Ensure first parent uses cache

maxi297 · maxi297 · commit 3098764064cd · 2024-12-05T14:28:24.000-05:00
diff --git a/airbyte_cdk/sources/declarative/concurrent_declarative_source.py b/airbyte_cdk/sources/declarative/concurrent_declarative_source.py
@@ -201,7 +201,7 @@ def _group_streams(
         state_manager = ConnectorStateManager(state=self._state)  # type: ignore  # state is always in the form of List[AirbyteStateMessage]. The ConnectorStateManager should use generics, but this can be done later
 
         name_to_stream_mapping = {
-            stream["name"]: stream for stream in self.resolved_manifest["streams"]
+            stream["name"]: stream for stream in self._initialize_cache_for_parent_streams(self.resolved_manifest["streams"])
         }
 
         for declarative_stream in self.streams(config=config):

Original file line number	Diff line number	Diff line change
`@@ -201,7 +201,7 @@ def _group_streams(`
`201`	`201`	`state_manager = ConnectorStateManager(state=self._state) # type: ignore # state is always in the form of List[AirbyteStateMessage]. The ConnectorStateManager should use generics, but this can be done later`
`202`	`202`
`203`	`203`	`name_to_stream_mapping = {`
`204`		`- stream["name"]: stream for stream in self.resolved_manifest["streams"]`
	`204`	`+ stream["name"]: stream for stream in self._initialize_cache_for_parent_streams(self.resolved_manifest["streams"])`
`205`	`205`	`}`
`206`	`206`
`207`	`207`	`for declarative_stream in self.streams(config=config):`