chalkruntime 3.40.2__tar.gz → 3.40.4__tar.gz

{chalkruntime-3.40.2 → chalkruntime-3.40.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkruntime
-Version: 3.40.2
+Version: 3.40.4
 Summary: Runtime support library for Chalk AI
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{chalkruntime-3.40.2 → chalkruntime-3.40.4}/chalkruntime/graph/protograph_deserializer.py

@@ -1880,7 +1880,7 @@ class FromProtoConverter:
         r: pb.StreamResolver,
         graph: ResolvedGraph,
         stream_sources: Dict[Tuple[str, str | None], StreamSource],
-        captured_global_values: Mapping[str, pb.FunctionReferenceCapturedGlobal] | None = None,
+        captured_global_values: Mapping[str, pb.FunctionReferenceCapturedGlobal] | None,
     ) -> StreamResolverParsed:
         # Resolver outputs
         outputs: list[TResolverFeatureType] = []
@@ -1923,11 +1923,22 @@ class FromProtoConverter:
                 window_index = i
                 continue
 
-        source = stream_sources.get((r.source_v2.source_type, r.source_v2.name or None))
-        if source is None:
-            raise ValueError(
-                f"Stream resolver '{r.fqn}' refers to unknown streaming source with type '{r.source_v2.source_type}' and name '{r.source_v2.name}'"
+        # Prefer the embedded full source. A (source_type, name) reference cannot uniquely identify a
+        # source -- every source field is customer-overridable, so two sources sharing (type, name)
+        # (e.g. one Kafka integration with many topics) are distinct. Fall back to the legacy reference
+        # lookup (last-writer-wins, with a duplicate warning) only for graphs serialized before
+        # source_full existed.
+        if r.HasField("source_full"):
+            source = cls.convert_stream_source(r.source_full)
+        else:
+            _logger.warning(
+                f"Stream resolver '{r.fqn}' bound its source via the legacy (source_type, name) reference ('{r.source_v2.source_type}', '{r.source_v2.name}')."
             )
+            source = stream_sources.get((r.source_v2.source_type, r.source_v2.name or None))
+            if source is None:
+                raise ValueError(
+                    f"Stream resolver '{r.fqn}' refers to unknown streaming source with type '{r.source_v2.source_type}' and name '{r.source_v2.name}'"
+                )
 
         parse_info: ParseInfoParsed | None = None
         if r.HasField("parse_info"):
@@ -2022,9 +2033,12 @@ class FromProtoConverter:
                     feature_expressions[feat_parsed] = underscore_parsed
         sink_config: StreamSinkConfigParsed | None = None
         if r.HasField("message_producer"):
-            # Parse send_to if present
+            # Parse send_to if present. Prefer the embedded full destination source; fall back to the
+            # (type, name) reference lookup for graphs from older clients (see source_full above).
             send_to: StreamSource | None = None
-            if r.message_producer.HasField("send_to"):
+            if r.message_producer.HasField("send_to_full"):
+                send_to = cls.convert_stream_source(r.message_producer.send_to_full)
+            elif r.message_producer.HasField("send_to"):
                 send_to_ref = r.message_producer.send_to
                 send_to = stream_sources.get((send_to_ref.source_type, send_to_ref.name or None))
                 if send_to is None:
@@ -2603,7 +2617,8 @@ class FromProtoConverter:
         )
 
         stream_sources: List[StreamSource] = [cls.convert_stream_source(ss) for ss in pb_graph.stream_sources_v2]
-        # (type x name) --> StreamSource
+        # (type x name) --> StreamSource. Only consulted for graphs that predate source_full; resolvers
+        # carrying source_full bind from it directly and never hit this map.
         stream_sources_dict: Dict[Tuple[str, str | None], StreamSource] = {}
         for ss in stream_sources:
             if (ss.streaming_type, ss.name) in stream_sources_dict:

{chalkruntime-3.40.2 → chalkruntime-3.40.4}/chalkruntime.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkruntime
-Version: 3.40.2
+Version: 3.40.4
 Summary: Runtime support library for Chalk AI
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{chalkruntime-3.40.2 → chalkruntime-3.40.4}/pyproject.toml

@@ -42,7 +42,7 @@ name = "chalkruntime"
 description = "Runtime support library for Chalk AI"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "3.40.2"
+version = "3.40.4"
 
 
 [tool.deptry]