langgraph-executor 0.0.1a0__py3-none-any.whl → 0.0.1a1__py3-none-any.whl

langgraph_executor/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.1a0"
+__version__ = "0.0.1a1"

langgraph_executor/execute_task.py

@@ -58,6 +58,7 @@ def reconstruct_task(
     *,
     store: BaseStore | None = None,
     config: RunnableConfig | None = None,
+    custom_stream_writer=None,
 ) -> PregelExecutableTask:
     pb_task = request.task
 
@@ -91,7 +92,9 @@ def reconstruct_task(
             val = pb_to_val(pb_task.input["PUSH_INPUT"])
 
         writes = deque()
-        runtime = ensure_runtime(configurable, store, graph)
+        runtime = ensure_runtime(
+            configurable, store, graph, custom_stream_writer=custom_stream_writer
+        )
 
         # Generate cache key if cache policy exists
         cache_policy = getattr(proc, "cache_policy", None)
@@ -184,15 +187,31 @@ def create_scratchpad(
     return scratchpad
 
 
-def ensure_runtime(configurable, store, graph):
+def ensure_runtime(configurable, store, graph, custom_stream_writer=None):
     runtime = configurable.get(CONFIG_KEY_RUNTIME)
+
+    # Prepare runtime overrides
+    overrides = {"store": store}
+    if custom_stream_writer is not None:
+        overrides["stream_writer"] = custom_stream_writer
+
     if runtime is None:
-        return DEFAULT_RUNTIME.override(store=store)
+        return DEFAULT_RUNTIME.override(**overrides)
     if isinstance(runtime, Runtime):
-        return runtime.override(store=store)
+        return runtime.override(**overrides)
     if isinstance(runtime, dict):
         context = _coerce_context(graph, runtime.get("context"))
-        return Runtime(**(runtime | {"store": store, "context": context}))
+        return Runtime(
+            **(
+                runtime
+                | {"store": store, "context": context}
+                | (
+                    {"stream_writer": custom_stream_writer}
+                    if custom_stream_writer
+                    else {}
+                )
+            )
+        )
     raise ValueError("Invalid runtime")
 
 

langgraph_executor/executor.py

@@ -113,18 +113,24 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
 
         # Right now, only handle task execution without interrupts, etc
         try:
-            # Get request
             request = get_init_request(request_iterator)
 
             # Reconstruct PregelExecutableTask
             graph = get_graph(request.graph_name, self.graphs)
-            task = reconstruct_task(request, graph)
-
-            # Check if streaming is requested (for messages mode)
             stream_messages = "messages" in request.stream_modes
+            stream_custom = "custom" in request.stream_modes
 
-            # Set up streaming callback if needed
             stream_chunks = []
+
+            custom_stream_writer = (
+                self._create_custom_stream_writer(stream_chunks)
+                if stream_custom
+                else None
+            )
+
+            task = reconstruct_task(
+                request, graph, custom_stream_writer=custom_stream_writer
+            )
             if stream_messages:
 
                 def stream_callback(message: BaseMessageChunk, metadata: dict):
@@ -201,7 +207,6 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
                     task,
                     retry_policy=None,
                 )
-
                 # Yield any accumulated stream chunks
                 yield from stream_chunks
 
@@ -285,6 +290,36 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
             self.logger.exception(f"ApplyWrites error: {e}")
             context.abort(grpc.StatusCode.INTERNAL, str(e))
 
+    def _create_custom_stream_writer(self, stream_chunks):
+        """Create a proper stream_writer function for custom mode (like langgraph does)."""
+        from google.protobuf.struct_pb2 import Struct  # type: ignore[unresolved-import]
+
+        def stream_writer(content):
+            """Custom stream writer that creates CustomStreamEvent messages."""
+            try:
+                # Create payload struct (like langgraph does)
+                payload = Struct()
+                if isinstance(content, str):
+                    payload.update({"content": content})
+                elif isinstance(content, dict):
+                    payload.update(content)
+                else:
+                    payload.update({"content": str(content)})
+
+                # Create CustomStreamEvent
+                custom_event = executor_pb2.CustomStreamEvent(payload=payload)
+                custom_event_response = executor_pb2.ExecuteTaskResponse(
+                    custom_stream_event=custom_event
+                )
+                stream_chunks.append(custom_event_response)
+
+            except Exception as e:
+                self.logger.warning(
+                    f"Failed to create custom stream event: {e}", exc_info=True
+                )
+
+        return stream_writer
+
 
 def extract_output_messages(writes: Sequence[Any]) -> list[types_pb2.Message]:  # type: ignore[name-defined]
     messages = []

langgraph_executor/extract_graph.py

@@ -80,11 +80,8 @@ def extract_reserved_configurable(
     )
 
 
-def extract_nodes(nodes: dict[str, PregelNode]) -> dict[str, graph_pb2.NodeDefinition]:
-    out = {}
-    for k, v in nodes.items():
-        out[k] = extract_node(k, v)
-    return out
+def extract_nodes(nodes: dict[str, PregelNode]) -> list[graph_pb2.NodeDefinition]:
+    return [extract_node(k, v) for k, v in nodes.items()]
 
 
 def extract_node(name: str, node: PregelNode) -> graph_pb2.NodeDefinition:

langgraph_executor/pb/executor_pb2.py

@@ -24,9 +24,10 @@ _sym_db = _symbol_database.Default()
 
 from . import types_pb2 as types__pb2
 from . import graph_pb2 as graph__pb2
+from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0e\x65xecutor.proto\x12\x08\x65xecutor\x1a\x0btypes.proto\x1a\x0bgraph.proto\"\x13\n\x11ListGraphsRequest\")\n\x12ListGraphsResponse\x12\x13\n\x0bgraph_names\x18\x01 \x03(\t\"%\n\x0fGetGraphRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\"D\n\x10GetGraphResponse\x12\x30\n\x10graph_definition\x18\x01 \x01(\x0b\x32\x16.graph.GraphDefinition\"\x8a\x02\n\x1d\x43hannelsFromCheckpointRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12\x1e\n\x05specs\x18\x02 \x01(\x0b\x32\x0f.types.Channels\x12g\n\x19\x63heckpoint_channel_values\x18\x03 \x03(\x0b\x32\x44.executor.ChannelsFromCheckpointRequest.CheckpointChannelValuesEntry\x1aL\n\x1c\x43heckpointChannelValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\"C\n\x1e\x43hannelsFromCheckpointResponse\x12!\n\x08\x63hannels\x18\x01 \x01(\x0b\x32\x0f.types.Channels\"\x88\x01\n\x12\x45xecuteTaskRequest\x12)\n\x04init\x18\x01 \x01(\x0b\x32\x19.executor.ExecuteTaskInitH\x00\x12<\n\x14\x63\x61\x63he_check_response\x18\x02 \x01(\x0b\x32\x1c.executor.CacheCheckResponseH\x00\x42\t\n\x07message\"\xd6\x01\n\x0f\x45xecuteTaskInit\x12\x19\n\x04task\x18\x01 \x01(\x0b\x32\x0b.types.Task\x12!\n\x08\x63hannels\x18\x02 \x01(\x0b\x32\x0f.types.Channels\x12\x14\n\x0cstream_modes\x18\x03 \x03(\t\x12\x0c\n\x04step\x18\x04 \x01(\x05\x12\x15\n\rcheckpoint_ns\x18\x05 \x03(\t\x12\x13\n\x0boutput_keys\x18\x06 \x03(\t\x12\x13\n\x0bstream_keys\x18\x07 \x03(\t\x12\x12\n\ngraph_name\x18\x08 \x01(\t\x12\x0c\n\x04stop\x18\t \x01(\x05\"\xbd\x01\n\x13\x45xecuteTaskResponse\x12+\n\x0btask_result\x18\x01 \x01(\x0b\x32\x14.executor.TaskResultH\x00\x12\x32\n\x18message_or_message_chunk\x18\x02 \x01(\x0b\x32\x0e.types.MessageH\x00\x12:\n\x13\x63\x61\x63he_check_request\x18\x03 \x01(\x0b\x32\x1b.executor.CacheCheckRequestH\x00\x42\t\n\x07message\"^\n\nTaskResult\x12(\n\x05\x65rror\x18\x01 \x01(\x0b\x32\x14.types.ExecutorErrorH\x00\x88\x01\x01\x12\x1c\n\x06writes\x18\x02 \x03(\x0b\x32\x0c.types.WriteB\x08\n\x06_error\"\xa4\x01\n\x12\x41pplyWritesRequest\x12%\n\ncheckpoint\x18\x01 \x01(\x0b\x32\x11.types.Checkpoint\x12!\n\x08\x63hannels\x18\x02 \x01(\x0b\x32\x0f.types.Channels\x12\x1a\n\x05tasks\x18\x03 \x03(\x0b\x32\x0b.types.Task\x12\x14\n\x0cnext_version\x18\x04 \x01(\t\x12\x12\n\ngraph_name\x18\x05 \x01(\t\"b\n\x13\x41pplyWritesResponse\x12!\n\x07updates\x18\x01 \x01(\x0b\x32\x0e.types.UpdatesH\x00\x12\x1d\n\x05\x65rror\x18\x02 \x01(\x0b\x32\x0c.types.ValueH\x00\x42\t\n\x07message\"\xc8\x01\n\x17GenerateCacheKeyRequest\x12\x11\n\tnode_name\x18\x01 \x01(\t\x12\x16\n\x0einput_channels\x18\x02 \x03(\t\x12L\n\x0e\x63hannel_values\x18\x03 \x03(\x0b\x32\x34.executor.GenerateCacheKeyRequest.ChannelValuesEntry\x1a\x34\n\x12\x43hannelValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x0c:\x02\x38\x01\"-\n\x18GenerateCacheKeyResponse\x12\x11\n\tcache_key\x18\x01 \x01(\t\"Y\n\x11\x43\x61\x63heCheckRequest\x12\x17\n\x0f\x63\x61\x63he_namespace\x18\x01 \x03(\t\x12\x11\n\tcache_key\x18\x02 \x01(\t\x12\x10\n\x03ttl\x18\x03 \x01(\x05H\x00\x88\x01\x01\x42\x06\n\x04_ttl\"\'\n\x12\x43\x61\x63heCheckResponse\x12\x11\n\tcache_hit\x18\x01 \x01(\x08\x32\x83\x04\n\x11LangGraphExecutor\x12G\n\nListGraphs\x12\x1b.executor.ListGraphsRequest\x1a\x1c.executor.ListGraphsResponse\x12\x41\n\x08GetGraph\x12\x19.executor.GetGraphRequest\x1a\x1a.executor.GetGraphResponse\x12k\n\x16\x43hannelsFromCheckpoint\x12\'.executor.ChannelsFromCheckpointRequest\x1a(.executor.ChannelsFromCheckpointResponse\x12N\n\x0b\x45xecuteTask\x12\x1c.executor.ExecuteTaskRequest\x1a\x1d.executor.ExecuteTaskResponse(\x01\x30\x01\x12J\n\x0b\x41pplyWrites\x12\x1c.executor.ApplyWritesRequest\x1a\x1d.executor.ApplyWritesResponse\x12Y\n\x10GenerateCacheKey\x12!.executor.GenerateCacheKeyRequest\x1a\".executor.GenerateCacheKeyResponseB1Z/github.com/langchain-ai/langgraph-go/runtime/pbb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0e\x65xecutor.proto\x12\x08\x65xecutor\x1a\x0btypes.proto\x1a\x0bgraph.proto\x1a\x1cgoogle/protobuf/struct.proto\"\x13\n\x11ListGraphsRequest\")\n\x12ListGraphsResponse\x12\x13\n\x0bgraph_names\x18\x01 \x03(\t\"%\n\x0fGetGraphRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\"D\n\x10GetGraphResponse\x12\x30\n\x10graph_definition\x18\x01 \x01(\x0b\x32\x16.graph.GraphDefinition\"\x8a\x02\n\x1d\x43hannelsFromCheckpointRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12\x1e\n\x05specs\x18\x02 \x01(\x0b\x32\x0f.types.Channels\x12g\n\x19\x63heckpoint_channel_values\x18\x03 \x03(\x0b\x32\x44.executor.ChannelsFromCheckpointRequest.CheckpointChannelValuesEntry\x1aL\n\x1c\x43heckpointChannelValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\"C\n\x1e\x43hannelsFromCheckpointResponse\x12!\n\x08\x63hannels\x18\x01 \x01(\x0b\x32\x0f.types.Channels\"\x88\x01\n\x12\x45xecuteTaskRequest\x12)\n\x04init\x18\x01 \x01(\x0b\x32\x19.executor.ExecuteTaskInitH\x00\x12<\n\x14\x63\x61\x63he_check_response\x18\x02 \x01(\x0b\x32\x1c.executor.CacheCheckResponseH\x00\x42\t\n\x07message\"\xd6\x01\n\x0f\x45xecuteTaskInit\x12\x19\n\x04task\x18\x01 \x01(\x0b\x32\x0b.types.Task\x12!\n\x08\x63hannels\x18\x02 \x01(\x0b\x32\x0f.types.Channels\x12\x14\n\x0cstream_modes\x18\x03 \x03(\t\x12\x0c\n\x04step\x18\x04 \x01(\x05\x12\x15\n\rcheckpoint_ns\x18\x05 \x03(\t\x12\x13\n\x0boutput_keys\x18\x06 \x03(\t\x12\x13\n\x0bstream_keys\x18\x07 \x03(\t\x12\x12\n\ngraph_name\x18\x08 \x01(\t\x12\x0c\n\x04stop\x18\t \x01(\x05\"\xf9\x01\n\x13\x45xecuteTaskResponse\x12+\n\x0btask_result\x18\x01 \x01(\x0b\x32\x14.executor.TaskResultH\x00\x12\x32\n\x18message_or_message_chunk\x18\x02 \x01(\x0b\x32\x0e.types.MessageH\x00\x12:\n\x13\x63\x61\x63he_check_request\x18\x03 \x01(\x0b\x32\x1b.executor.CacheCheckRequestH\x00\x12:\n\x13\x63ustom_stream_event\x18\x04 \x01(\x0b\x32\x1b.executor.CustomStreamEventH\x00\x42\t\n\x07message\"^\n\nTaskResult\x12(\n\x05\x65rror\x18\x01 \x01(\x0b\x32\x14.types.ExecutorErrorH\x00\x88\x01\x01\x12\x1c\n\x06writes\x18\x02 \x03(\x0b\x32\x0c.types.WriteB\x08\n\x06_error\"=\n\x11\x43ustomStreamEvent\x12(\n\x07payload\x18\x01 \x01(\x0b\x32\x17.google.protobuf.Struct\"\xa4\x01\n\x12\x41pplyWritesRequest\x12%\n\ncheckpoint\x18\x01 \x01(\x0b\x32\x11.types.Checkpoint\x12!\n\x08\x63hannels\x18\x02 \x01(\x0b\x32\x0f.types.Channels\x12\x1a\n\x05tasks\x18\x03 \x03(\x0b\x32\x0b.types.Task\x12\x14\n\x0cnext_version\x18\x04 \x01(\t\x12\x12\n\ngraph_name\x18\x05 \x01(\t\"b\n\x13\x41pplyWritesResponse\x12!\n\x07updates\x18\x01 \x01(\x0b\x32\x0e.types.UpdatesH\x00\x12\x1d\n\x05\x65rror\x18\x02 \x01(\x0b\x32\x0c.types.ValueH\x00\x42\t\n\x07message\"\xc8\x01\n\x17GenerateCacheKeyRequest\x12\x11\n\tnode_name\x18\x01 \x01(\t\x12\x16\n\x0einput_channels\x18\x02 \x03(\t\x12L\n\x0e\x63hannel_values\x18\x03 \x03(\x0b\x32\x34.executor.GenerateCacheKeyRequest.ChannelValuesEntry\x1a\x34\n\x12\x43hannelValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x0c:\x02\x38\x01\"-\n\x18GenerateCacheKeyResponse\x12\x11\n\tcache_key\x18\x01 \x01(\t\"Y\n\x11\x43\x61\x63heCheckRequest\x12\x17\n\x0f\x63\x61\x63he_namespace\x18\x01 \x03(\t\x12\x11\n\tcache_key\x18\x02 \x01(\t\x12\x10\n\x03ttl\x18\x03 \x01(\x05H\x00\x88\x01\x01\x42\x06\n\x04_ttl\"\'\n\x12\x43\x61\x63heCheckResponse\x12\x11\n\tcache_hit\x18\x01 \x01(\x08\x32\x83\x04\n\x11LangGraphExecutor\x12G\n\nListGraphs\x12\x1b.executor.ListGraphsRequest\x1a\x1c.executor.ListGraphsResponse\x12\x41\n\x08GetGraph\x12\x19.executor.GetGraphRequest\x1a\x1a.executor.GetGraphResponse\x12k\n\x16\x43hannelsFromCheckpoint\x12\'.executor.ChannelsFromCheckpointRequest\x1a(.executor.ChannelsFromCheckpointResponse\x12N\n\x0b\x45xecuteTask\x12\x1c.executor.ExecuteTaskRequest\x1a\x1d.executor.ExecuteTaskResponse(\x01\x30\x01\x12J\n\x0b\x41pplyWrites\x12\x1c.executor.ApplyWritesRequest\x1a\x1d.executor.ApplyWritesResponse\x12Y\n\x10GenerateCacheKey\x12!.executor.GenerateCacheKeyRequest\x1a\".executor.GenerateCacheKeyResponseB1Z/github.com/langchain-ai/langgraph-go/runtime/pbb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -38,42 +39,44 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_CHANNELSFROMCHECKPOINTREQUEST_CHECKPOINTCHANNELVALUESENTRY']._serialized_options = b'8\001'
   _globals['_GENERATECACHEKEYREQUEST_CHANNELVALUESENTRY']._loaded_options = None
   _globals['_GENERATECACHEKEYREQUEST_CHANNELVALUESENTRY']._serialized_options = b'8\001'
-  _globals['_LISTGRAPHSREQUEST']._serialized_start=54
-  _globals['_LISTGRAPHSREQUEST']._serialized_end=73
-  _globals['_LISTGRAPHSRESPONSE']._serialized_start=75
-  _globals['_LISTGRAPHSRESPONSE']._serialized_end=116
-  _globals['_GETGRAPHREQUEST']._serialized_start=118
-  _globals['_GETGRAPHREQUEST']._serialized_end=155
-  _globals['_GETGRAPHRESPONSE']._serialized_start=157
-  _globals['_GETGRAPHRESPONSE']._serialized_end=225
-  _globals['_CHANNELSFROMCHECKPOINTREQUEST']._serialized_start=228
-  _globals['_CHANNELSFROMCHECKPOINTREQUEST']._serialized_end=494
-  _globals['_CHANNELSFROMCHECKPOINTREQUEST_CHECKPOINTCHANNELVALUESENTRY']._serialized_start=418
-  _globals['_CHANNELSFROMCHECKPOINTREQUEST_CHECKPOINTCHANNELVALUESENTRY']._serialized_end=494
-  _globals['_CHANNELSFROMCHECKPOINTRESPONSE']._serialized_start=496
-  _globals['_CHANNELSFROMCHECKPOINTRESPONSE']._serialized_end=563
-  _globals['_EXECUTETASKREQUEST']._serialized_start=566
-  _globals['_EXECUTETASKREQUEST']._serialized_end=702
-  _globals['_EXECUTETASKINIT']._serialized_start=705
-  _globals['_EXECUTETASKINIT']._serialized_end=919
-  _globals['_EXECUTETASKRESPONSE']._serialized_start=922
-  _globals['_EXECUTETASKRESPONSE']._serialized_end=1111
-  _globals['_TASKRESULT']._serialized_start=1113
-  _globals['_TASKRESULT']._serialized_end=1207
-  _globals['_APPLYWRITESREQUEST']._serialized_start=1210
-  _globals['_APPLYWRITESREQUEST']._serialized_end=1374
-  _globals['_APPLYWRITESRESPONSE']._serialized_start=1376
-  _globals['_APPLYWRITESRESPONSE']._serialized_end=1474
-  _globals['_GENERATECACHEKEYREQUEST']._serialized_start=1477
-  _globals['_GENERATECACHEKEYREQUEST']._serialized_end=1677
-  _globals['_GENERATECACHEKEYREQUEST_CHANNELVALUESENTRY']._serialized_start=1625
-  _globals['_GENERATECACHEKEYREQUEST_CHANNELVALUESENTRY']._serialized_end=1677
-  _globals['_GENERATECACHEKEYRESPONSE']._serialized_start=1679
-  _globals['_GENERATECACHEKEYRESPONSE']._serialized_end=1724
-  _globals['_CACHECHECKREQUEST']._serialized_start=1726
-  _globals['_CACHECHECKREQUEST']._serialized_end=1815
-  _globals['_CACHECHECKRESPONSE']._serialized_start=1817
-  _globals['_CACHECHECKRESPONSE']._serialized_end=1856
-  _globals['_LANGGRAPHEXECUTOR']._serialized_start=1859
-  _globals['_LANGGRAPHEXECUTOR']._serialized_end=2374
+  _globals['_LISTGRAPHSREQUEST']._serialized_start=84
+  _globals['_LISTGRAPHSREQUEST']._serialized_end=103
+  _globals['_LISTGRAPHSRESPONSE']._serialized_start=105
+  _globals['_LISTGRAPHSRESPONSE']._serialized_end=146
+  _globals['_GETGRAPHREQUEST']._serialized_start=148
+  _globals['_GETGRAPHREQUEST']._serialized_end=185
+  _globals['_GETGRAPHRESPONSE']._serialized_start=187
+  _globals['_GETGRAPHRESPONSE']._serialized_end=255
+  _globals['_CHANNELSFROMCHECKPOINTREQUEST']._serialized_start=258
+  _globals['_CHANNELSFROMCHECKPOINTREQUEST']._serialized_end=524
+  _globals['_CHANNELSFROMCHECKPOINTREQUEST_CHECKPOINTCHANNELVALUESENTRY']._serialized_start=448
+  _globals['_CHANNELSFROMCHECKPOINTREQUEST_CHECKPOINTCHANNELVALUESENTRY']._serialized_end=524
+  _globals['_CHANNELSFROMCHECKPOINTRESPONSE']._serialized_start=526
+  _globals['_CHANNELSFROMCHECKPOINTRESPONSE']._serialized_end=593
+  _globals['_EXECUTETASKREQUEST']._serialized_start=596
+  _globals['_EXECUTETASKREQUEST']._serialized_end=732
+  _globals['_EXECUTETASKINIT']._serialized_start=735
+  _globals['_EXECUTETASKINIT']._serialized_end=949
+  _globals['_EXECUTETASKRESPONSE']._serialized_start=952
+  _globals['_EXECUTETASKRESPONSE']._serialized_end=1201
+  _globals['_TASKRESULT']._serialized_start=1203
+  _globals['_TASKRESULT']._serialized_end=1297
+  _globals['_CUSTOMSTREAMEVENT']._serialized_start=1299
+  _globals['_CUSTOMSTREAMEVENT']._serialized_end=1360
+  _globals['_APPLYWRITESREQUEST']._serialized_start=1363
+  _globals['_APPLYWRITESREQUEST']._serialized_end=1527
+  _globals['_APPLYWRITESRESPONSE']._serialized_start=1529
+  _globals['_APPLYWRITESRESPONSE']._serialized_end=1627
+  _globals['_GENERATECACHEKEYREQUEST']._serialized_start=1630
+  _globals['_GENERATECACHEKEYREQUEST']._serialized_end=1830
+  _globals['_GENERATECACHEKEYREQUEST_CHANNELVALUESENTRY']._serialized_start=1778
+  _globals['_GENERATECACHEKEYREQUEST_CHANNELVALUESENTRY']._serialized_end=1830
+  _globals['_GENERATECACHEKEYRESPONSE']._serialized_start=1832
+  _globals['_GENERATECACHEKEYRESPONSE']._serialized_end=1877
+  _globals['_CACHECHECKREQUEST']._serialized_start=1879
+  _globals['_CACHECHECKREQUEST']._serialized_end=1968
+  _globals['_CACHECHECKRESPONSE']._serialized_start=1970
+  _globals['_CACHECHECKRESPONSE']._serialized_end=2009
+  _globals['_LANGGRAPHEXECUTOR']._serialized_start=2012
+  _globals['_LANGGRAPHEXECUTOR']._serialized_end=2527
 # @@protoc_insertion_point(module_scope)

langgraph_executor/pb/executor_pb2.pyi

@@ -8,6 +8,7 @@ import collections.abc
 import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.message
+import google.protobuf.struct_pb2
 import graph_pb2
 import types_pb2
 import typing
@@ -214,22 +215,26 @@ class ExecuteTaskResponse(google.protobuf.message.Message):
     TASK_RESULT_FIELD_NUMBER: builtins.int
     MESSAGE_OR_MESSAGE_CHUNK_FIELD_NUMBER: builtins.int
     CACHE_CHECK_REQUEST_FIELD_NUMBER: builtins.int
+    CUSTOM_STREAM_EVENT_FIELD_NUMBER: builtins.int
     @property
     def task_result(self) -> global___TaskResult: ...
     @property
     def message_or_message_chunk(self) -> types_pb2.Message: ...
     @property
     def cache_check_request(self) -> global___CacheCheckRequest: ...
+    @property
+    def custom_stream_event(self) -> global___CustomStreamEvent: ...
     def __init__(
         self,
         *,
         task_result: global___TaskResult | None = ...,
         message_or_message_chunk: types_pb2.Message | None = ...,
         cache_check_request: global___CacheCheckRequest | None = ...,
+        custom_stream_event: global___CustomStreamEvent | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing.Literal["cache_check_request", b"cache_check_request", "message", b"message", "message_or_message_chunk", b"message_or_message_chunk", "task_result", b"task_result"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing.Literal["cache_check_request", b"cache_check_request", "message", b"message", "message_or_message_chunk", b"message_or_message_chunk", "task_result", b"task_result"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing.Literal["message", b"message"]) -> typing.Literal["task_result", "message_or_message_chunk", "cache_check_request"] | None: ...
+    def HasField(self, field_name: typing.Literal["cache_check_request", b"cache_check_request", "custom_stream_event", b"custom_stream_event", "message", b"message", "message_or_message_chunk", b"message_or_message_chunk", "task_result", b"task_result"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["cache_check_request", b"cache_check_request", "custom_stream_event", b"custom_stream_event", "message", b"message", "message_or_message_chunk", b"message_or_message_chunk", "task_result", b"task_result"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["message", b"message"]) -> typing.Literal["task_result", "message_or_message_chunk", "cache_check_request", "custom_stream_event"] | None: ...
 
 global___ExecuteTaskResponse = ExecuteTaskResponse
 
@@ -255,6 +260,23 @@ class TaskResult(google.protobuf.message.Message):
 
 global___TaskResult = TaskResult
 
+@typing.final
+class CustomStreamEvent(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PAYLOAD_FIELD_NUMBER: builtins.int
+    @property
+    def payload(self) -> google.protobuf.struct_pb2.Struct: ...
+    def __init__(
+        self,
+        *,
+        payload: google.protobuf.struct_pb2.Struct | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["payload", b"payload"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["payload", b"payload"]) -> None: ...
+
+global___CustomStreamEvent = CustomStreamEvent
+
 @typing.final
 class ApplyWritesRequest(google.protobuf.message.Message):
     """Apply Writes"""

langgraph_executor/pb/graph_pb2.py

@@ -26,7 +26,7 @@ from . import types_pb2 as types__pb2
 from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0bgraph.proto\x12\x05graph\x1a\x0btypes.proto\x1a\x1cgoogle/protobuf/struct.proto\"\xd5\x05\n\x0fGraphDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x30\n\x05nodes\x18\x02 \x03(\x0b\x32!.graph.GraphDefinition.NodesEntry\x12!\n\x08\x63hannels\x18\x03 \x01(\x0b\x32\x0f.types.Channels\x12\x16\n\x0einput_channels\x18\x04 \x03(\t\x12\x17\n\x0foutput_channels\x18\x05 \x03(\t\x12\x1e\n\x16input_channels_is_list\x18\x06 \x01(\x08\x12\x1f\n\x17output_channels_is_list\x18\x07 \x01(\x08\x12\x1e\n\x16interrupt_before_nodes\x18\x08 \x03(\t\x12\x1d\n\x15interrupt_after_nodes\x18\t \x03(\t\x12\x44\n\x10trigger_to_nodes\x18\n \x03(\x0b\x32*.graph.GraphDefinition.TriggerToNodesEntry\x12\x13\n\x0bstream_mode\x18\x0b \x03(\t\x12\x14\n\x0cstream_eager\x18\x0c \x01(\x08\x12\x17\n\x0fstream_channels\x18\r \x03(\t\x12\x14\n\x0cstep_timeout\x18\x0e \x01(\x02\x12\r\n\x05\x64\x65\x62ug\x18\x0f \x01(\x08\x12\x1b\n\x05\x63\x61\x63he\x18\x10 \x01(\x0b\x32\x0c.graph.Cache\x12(\n\x0cretry_policy\x18\x11 \x01(\x0b\x32\x12.graph.RetryPolicy\x12%\n\x06\x63onfig\x18\x12 \x01(\x0b\x32\x15.types.RunnableConfig\x1a\x43\n\nNodesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12$\n\x05value\x18\x02 \x01(\x0b\x32\x15.graph.NodeDefinition:\x02\x38\x01\x1aL\n\x13TriggerToNodesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12$\n\x05value\x18\x02 \x01(\x0b\x32\x15.graph.TriggerMapping:\x02\x38\x01\"\x8c\x01\n\x0eNodeDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x10\n\x08triggers\x18\x02 \x03(\t\x12\x0f\n\x07writers\x18\x03 \x03(\t\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12)\n\x08metadata\x18\x05 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x10\n\x08\x63hannels\x18\x06 \x03(\t\"{\n\x0bRetryPolicy\x12\x18\n\x10initial_interval\x18\x01 \x01(\x02\x12\x16\n\x0e\x62\x61\x63koff_factor\x18\x02 \x01(\x02\x12\x14\n\x0cmax_interval\x18\x03 \x01(\x02\x12\x14\n\x0cmax_attempts\x18\x04 \x01(\x05\x12\x0e\n\x06jitter\x18\x05 \x01(\x08\"\x1b\n\x05\x43\x61\x63he\x12\x12\n\ncache_type\x18\x01 \x01(\t\"\x1f\n\x0eTriggerMapping\x12\r\n\x05nodes\x18\x01 \x03(\tB1Z/github.com/langchain-ai/langgraph-go/runtime/pbb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0bgraph.proto\x12\x05graph\x1a\x0btypes.proto\x1a\x1cgoogle/protobuf/struct.proto\"\x84\x05\n\x0fGraphDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12$\n\x05nodes\x18\x02 \x03(\x0b\x32\x15.graph.NodeDefinition\x12!\n\x08\x63hannels\x18\x03 \x01(\x0b\x32\x0f.types.Channels\x12\x16\n\x0einput_channels\x18\x04 \x03(\t\x12\x17\n\x0foutput_channels\x18\x05 \x03(\t\x12\x1e\n\x16input_channels_is_list\x18\x06 \x01(\x08\x12\x1f\n\x17output_channels_is_list\x18\x07 \x01(\x08\x12\x1e\n\x16interrupt_before_nodes\x18\x08 \x03(\t\x12\x1d\n\x15interrupt_after_nodes\x18\t \x03(\t\x12\x44\n\x10trigger_to_nodes\x18\n \x03(\x0b\x32*.graph.GraphDefinition.TriggerToNodesEntry\x12\x13\n\x0bstream_mode\x18\x0b \x03(\t\x12\x14\n\x0cstream_eager\x18\x0c \x01(\x08\x12\x17\n\x0fstream_channels\x18\r \x03(\t\x12\x14\n\x0cstep_timeout\x18\x0e \x01(\x02\x12\r\n\x05\x64\x65\x62ug\x18\x0f \x01(\x08\x12\x1b\n\x05\x63\x61\x63he\x18\x10 \x01(\x0b\x32\x0c.graph.Cache\x12(\n\x0cretry_policy\x18\x11 \x01(\x0b\x32\x12.graph.RetryPolicy\x12%\n\x06\x63onfig\x18\x12 \x01(\x0b\x32\x15.types.RunnableConfig\x1aL\n\x13TriggerToNodesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12$\n\x05value\x18\x02 \x01(\x0b\x32\x15.graph.TriggerMapping:\x02\x38\x01\"\x8c\x01\n\x0eNodeDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x10\n\x08triggers\x18\x02 \x03(\t\x12\x0f\n\x07writers\x18\x03 \x03(\t\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12)\n\x08metadata\x18\x05 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x10\n\x08\x63hannels\x18\x06 \x03(\t\"{\n\x0bRetryPolicy\x12\x18\n\x10initial_interval\x18\x01 \x01(\x02\x12\x16\n\x0e\x62\x61\x63koff_factor\x18\x02 \x01(\x02\x12\x14\n\x0cmax_interval\x18\x03 \x01(\x02\x12\x14\n\x0cmax_attempts\x18\x04 \x01(\x05\x12\x0e\n\x06jitter\x18\x05 \x01(\x08\"\x1b\n\x05\x43\x61\x63he\x12\x12\n\ncache_type\x18\x01 \x01(\t\"\x1f\n\x0eTriggerMapping\x12\r\n\x05nodes\x18\x01 \x03(\tB1Z/github.com/langchain-ai/langgraph-go/runtime/pbb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -34,22 +34,18 @@ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'graph_pb2', _globals)
 if not _descriptor._USE_C_DESCRIPTORS:
   _globals['DESCRIPTOR']._loaded_options = None
   _globals['DESCRIPTOR']._serialized_options = b'Z/github.com/langchain-ai/langgraph-go/runtime/pb'
-  _globals['_GRAPHDEFINITION_NODESENTRY']._loaded_options = None
-  _globals['_GRAPHDEFINITION_NODESENTRY']._serialized_options = b'8\001'
   _globals['_GRAPHDEFINITION_TRIGGERTONODESENTRY']._loaded_options = None
   _globals['_GRAPHDEFINITION_TRIGGERTONODESENTRY']._serialized_options = b'8\001'
   _globals['_GRAPHDEFINITION']._serialized_start=66
-  _globals['_GRAPHDEFINITION']._serialized_end=791
-  _globals['_GRAPHDEFINITION_NODESENTRY']._serialized_start=646
-  _globals['_GRAPHDEFINITION_NODESENTRY']._serialized_end=713
-  _globals['_GRAPHDEFINITION_TRIGGERTONODESENTRY']._serialized_start=715
-  _globals['_GRAPHDEFINITION_TRIGGERTONODESENTRY']._serialized_end=791
-  _globals['_NODEDEFINITION']._serialized_start=794
-  _globals['_NODEDEFINITION']._serialized_end=934
-  _globals['_RETRYPOLICY']._serialized_start=936
-  _globals['_RETRYPOLICY']._serialized_end=1059
-  _globals['_CACHE']._serialized_start=1061
-  _globals['_CACHE']._serialized_end=1088
-  _globals['_TRIGGERMAPPING']._serialized_start=1090
-  _globals['_TRIGGERMAPPING']._serialized_end=1121
+  _globals['_GRAPHDEFINITION']._serialized_end=710
+  _globals['_GRAPHDEFINITION_TRIGGERTONODESENTRY']._serialized_start=634
+  _globals['_GRAPHDEFINITION_TRIGGERTONODESENTRY']._serialized_end=710
+  _globals['_NODEDEFINITION']._serialized_start=713
+  _globals['_NODEDEFINITION']._serialized_end=853
+  _globals['_RETRYPOLICY']._serialized_start=855
+  _globals['_RETRYPOLICY']._serialized_end=978
+  _globals['_CACHE']._serialized_start=980
+  _globals['_CACHE']._serialized_end=1007
+  _globals['_TRIGGERMAPPING']._serialized_start=1009
+  _globals['_TRIGGERMAPPING']._serialized_end=1040
 # @@protoc_insertion_point(module_scope)

langgraph_executor/pb/graph_pb2.pyi

@@ -18,24 +18,6 @@ DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 class GraphDefinition(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
-    @typing.final
-    class NodesEntry(google.protobuf.message.Message):
-        DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-        KEY_FIELD_NUMBER: builtins.int
-        VALUE_FIELD_NUMBER: builtins.int
-        key: builtins.str
-        @property
-        def value(self) -> global___NodeDefinition: ...
-        def __init__(
-            self,
-            *,
-            key: builtins.str = ...,
-            value: global___NodeDefinition | None = ...,
-        ) -> None: ...
-        def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
-        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
-
     @typing.final
     class TriggerToNodesEntry(google.protobuf.message.Message):
         DESCRIPTOR: google.protobuf.descriptor.Descriptor
@@ -79,7 +61,7 @@ class GraphDefinition(google.protobuf.message.Message):
     step_timeout: builtins.float
     debug: builtins.bool
     @property
-    def nodes(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___NodeDefinition]: ...
+    def nodes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___NodeDefinition]: ...
     @property
     def channels(self) -> types_pb2.Channels: ...
     @property
@@ -106,7 +88,7 @@ class GraphDefinition(google.protobuf.message.Message):
         self,
         *,
         name: builtins.str = ...,
-        nodes: collections.abc.Mapping[builtins.str, global___NodeDefinition] | None = ...,
+        nodes: collections.abc.Iterable[global___NodeDefinition] | None = ...,
         channels: types_pb2.Channels | None = ...,
         input_channels: collections.abc.Iterable[builtins.str] | None = ...,
         output_channels: collections.abc.Iterable[builtins.str] | None = ...,

langgraph_executor/pb/runtime_pb2.py

@@ -27,7 +27,7 @@ from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
 from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\rruntime.proto\x12\x07runtime\x1a\x0btypes.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x1cgoogle/protobuf/struct.proto\"%\n\x12\x41\x64\x64\x45xecutorRequest\x12\x0f\n\x07\x61\x64\x64ress\x18\x01 \x01(\t\"L\n\x0f\x41\x64\x64GraphRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12+\n\x0c\x63heckpointer\x18\x02 \x01(\x0b\x32\x15.runtime.Checkpointer\"}\n\x0c\x43heckpointer\x12/\n\x06memory\x18\x02 \x01(\x0b\x32\x1d.runtime.InMemoryCheckpointerH\x00\x12\x31\n\x08postgres\x18\x03 \x01(\x0b\x32\x1d.runtime.PostgresCheckpointerH\x00\x42\t\n\x07message\"\x16\n\x14InMemoryCheckpointer\"1\n\x14PostgresCheckpointer\x12\x19\n\x11\x63onnection_string\x18\x01 \x01(\t\"\xcf\x01\n\rInvokeRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12\x1b\n\x05input\x18\x02 \x01(\x0b\x32\x0c.types.Value\x12%\n\x06\x63onfig\x18\x03 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x08run_opts\x18\x04 \x01(\x0b\x32\x10.runtime.RunOptsH\x00\x88\x01\x01\x12$\n\x07\x63ontext\x18\x05 \x01(\x0b\x32\x0e.types.ContextH\x01\x88\x01\x01\x42\x0b\n\t_run_optsB\n\n\x08_context\"\xcf\x01\n\rStreamRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12\x1b\n\x05input\x18\x02 \x01(\x0b\x32\x0c.types.Value\x12%\n\x06\x63onfig\x18\x03 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x08run_opts\x18\x04 \x01(\x0b\x32\x10.runtime.RunOptsH\x00\x88\x01\x01\x12$\n\x07\x63ontext\x18\x05 \x01(\x0b\x32\x0e.types.ContextH\x01\x88\x01\x01\x42\x0b\n\t_run_optsB\n\n\x08_context\"\xd2\x01\n\x07RunOpts\x12\x13\n\x0boutput_keys\x18\x01 \x03(\t\x12\x18\n\x10interrupt_before\x18\x02 \x03(\t\x12\x17\n\x0finterrupt_after\x18\x03 \x03(\t\x12\x17\n\ndurability\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05\x64\x65\x62ug\x18\x05 \x01(\x08H\x01\x88\x01\x01\x12\x16\n\tsubgraphs\x18\x06 \x01(\x08H\x02\x88\x01\x01\x12\x13\n\x0bstream_mode\x18\x07 \x03(\tB\r\n\x0b_durabilityB\x08\n\x06_debugB\x0c\n\n_subgraphs\"1\n\tChunkList\x12$\n\x06\x63hunks\x18\x01 \x03(\x0b\x32\x14.runtime.OutputChunk\"\x86\x01\n\x0bOutputChunk\x12#\n\x05\x63hunk\x18\x01 \x01(\x0b\x32\x12.types.StreamChunkH\x00\x12\x1d\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x0c.types.ValueH\x00\x12(\n\nchunk_list\x18\x05 \x01(\x0b\x32\x12.runtime.ChunkListH\x00\x42\t\n\x07message\"_\n\x0fGetStateRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12%\n\x06\x63onfig\x18\x02 \x01(\x0b\x32\x15.types.RunnableConfig\x12\x11\n\tsubgraphs\x18\x03 \x01(\x08\"7\n\x10GetStateResponse\x12#\n\x05state\x18\x01 \x01(\x0b\x32\x14.types.StateSnapshot\"\xb2\x01\n\x16GetStateHistoryRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12%\n\x06\x63onfig\x18\x02 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x06\x66ilter\x18\x03 \x01(\x0b\x32\x17.google.protobuf.Struct\x12%\n\x06\x62\x65\x66ore\x18\x04 \x01(\x0b\x32\x15.types.RunnableConfig\x12\r\n\x05limit\x18\x05 \x01(\x03\"@\n\x17GetStateHistoryResponse\x12%\n\x07history\x18\x01 \x03(\x0b\x32\x14.types.StateSnapshot2\x9d\x03\n\x10LangGraphRuntime\x12\x42\n\x0b\x41\x64\x64\x45xecutor\x12\x1b.runtime.AddExecutorRequest\x1a\x16.google.protobuf.Empty\x12<\n\x08\x41\x64\x64Graph\x12\x18.runtime.AddGraphRequest\x1a\x16.google.protobuf.Empty\x12\x36\n\x06Invoke\x12\x16.runtime.InvokeRequest\x1a\x14.runtime.OutputChunk\x12\x38\n\x06Stream\x12\x16.runtime.StreamRequest\x1a\x14.runtime.OutputChunk0\x01\x12?\n\x08GetState\x12\x18.runtime.GetStateRequest\x1a\x19.runtime.GetStateResponse\x12T\n\x0fGetStateHistory\x12\x1f.runtime.GetStateHistoryRequest\x1a .runtime.GetStateHistoryResponseB1Z/github.com/langchain-ai/langgraph-go/runtime/pbb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\rruntime.proto\x12\x07runtime\x1a\x0btypes.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x1cgoogle/protobuf/struct.proto\"%\n\x12\x41\x64\x64\x45xecutorRequest\x12\x0f\n\x07\x61\x64\x64ress\x18\x01 \x01(\t\"L\n\x0f\x41\x64\x64GraphRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12+\n\x0c\x63heckpointer\x18\x02 \x01(\x0b\x32\x15.runtime.Checkpointer\"}\n\x0c\x43heckpointer\x12/\n\x06memory\x18\x02 \x01(\x0b\x32\x1d.runtime.InMemoryCheckpointerH\x00\x12\x31\n\x08postgres\x18\x03 \x01(\x0b\x32\x1d.runtime.PostgresCheckpointerH\x00\x42\t\n\x07message\"\x16\n\x14InMemoryCheckpointer\",\n\x14PostgresCheckpointer\x12\x14\n\x0c\x64\x61tabase_uri\x18\x01 \x01(\t\"\xcf\x01\n\rInvokeRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12\x1b\n\x05input\x18\x02 \x01(\x0b\x32\x0c.types.Value\x12%\n\x06\x63onfig\x18\x03 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x08run_opts\x18\x04 \x01(\x0b\x32\x10.runtime.RunOptsH\x00\x88\x01\x01\x12$\n\x07\x63ontext\x18\x05 \x01(\x0b\x32\x0e.types.ContextH\x01\x88\x01\x01\x42\x0b\n\t_run_optsB\n\n\x08_context\"\xcf\x01\n\rStreamRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12\x1b\n\x05input\x18\x02 \x01(\x0b\x32\x0c.types.Value\x12%\n\x06\x63onfig\x18\x03 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x08run_opts\x18\x04 \x01(\x0b\x32\x10.runtime.RunOptsH\x00\x88\x01\x01\x12$\n\x07\x63ontext\x18\x05 \x01(\x0b\x32\x0e.types.ContextH\x01\x88\x01\x01\x42\x0b\n\t_run_optsB\n\n\x08_context\"\xd2\x01\n\x07RunOpts\x12\x13\n\x0boutput_keys\x18\x01 \x03(\t\x12\x18\n\x10interrupt_before\x18\x02 \x03(\t\x12\x17\n\x0finterrupt_after\x18\x03 \x03(\t\x12\x17\n\ndurability\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05\x64\x65\x62ug\x18\x05 \x01(\x08H\x01\x88\x01\x01\x12\x16\n\tsubgraphs\x18\x06 \x01(\x08H\x02\x88\x01\x01\x12\x13\n\x0bstream_mode\x18\x07 \x03(\tB\r\n\x0b_durabilityB\x08\n\x06_debugB\x0c\n\n_subgraphs\"1\n\tChunkList\x12$\n\x06\x63hunks\x18\x01 \x03(\x0b\x32\x14.runtime.OutputChunk\"\x86\x01\n\x0bOutputChunk\x12#\n\x05\x63hunk\x18\x01 \x01(\x0b\x32\x12.types.StreamChunkH\x00\x12\x1d\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x0c.types.ValueH\x00\x12(\n\nchunk_list\x18\x05 \x01(\x0b\x32\x12.runtime.ChunkListH\x00\x42\t\n\x07message\"_\n\x0fGetStateRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12%\n\x06\x63onfig\x18\x02 \x01(\x0b\x32\x15.types.RunnableConfig\x12\x11\n\tsubgraphs\x18\x03 \x01(\x08\"7\n\x10GetStateResponse\x12#\n\x05state\x18\x01 \x01(\x0b\x32\x14.types.StateSnapshot\"\xb2\x01\n\x16GetStateHistoryRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12%\n\x06\x63onfig\x18\x02 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x06\x66ilter\x18\x03 \x01(\x0b\x32\x17.google.protobuf.Struct\x12%\n\x06\x62\x65\x66ore\x18\x04 \x01(\x0b\x32\x15.types.RunnableConfig\x12\r\n\x05limit\x18\x05 \x01(\x03\"@\n\x17GetStateHistoryResponse\x12%\n\x07history\x18\x01 \x03(\x0b\x32\x14.types.StateSnapshot2\x9d\x03\n\x10LangGraphRuntime\x12\x42\n\x0b\x41\x64\x64\x45xecutor\x12\x1b.runtime.AddExecutorRequest\x1a\x16.google.protobuf.Empty\x12<\n\x08\x41\x64\x64Graph\x12\x18.runtime.AddGraphRequest\x1a\x16.google.protobuf.Empty\x12\x36\n\x06Invoke\x12\x16.runtime.InvokeRequest\x1a\x14.runtime.OutputChunk\x12\x38\n\x06Stream\x12\x16.runtime.StreamRequest\x1a\x14.runtime.OutputChunk0\x01\x12?\n\x08GetState\x12\x18.runtime.GetStateRequest\x1a\x19.runtime.GetStateResponse\x12T\n\x0fGetStateHistory\x12\x1f.runtime.GetStateHistoryRequest\x1a .runtime.GetStateHistoryResponseB1Z/github.com/langchain-ai/langgraph-go/runtime/pbb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -44,25 +44,25 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_INMEMORYCHECKPOINTER']._serialized_start=342
   _globals['_INMEMORYCHECKPOINTER']._serialized_end=364
   _globals['_POSTGRESCHECKPOINTER']._serialized_start=366
-  _globals['_POSTGRESCHECKPOINTER']._serialized_end=415
-  _globals['_INVOKEREQUEST']._serialized_start=418
-  _globals['_INVOKEREQUEST']._serialized_end=625
-  _globals['_STREAMREQUEST']._serialized_start=628
-  _globals['_STREAMREQUEST']._serialized_end=835
-  _globals['_RUNOPTS']._serialized_start=838
-  _globals['_RUNOPTS']._serialized_end=1048
-  _globals['_CHUNKLIST']._serialized_start=1050
-  _globals['_CHUNKLIST']._serialized_end=1099
-  _globals['_OUTPUTCHUNK']._serialized_start=1102
-  _globals['_OUTPUTCHUNK']._serialized_end=1236
-  _globals['_GETSTATEREQUEST']._serialized_start=1238
-  _globals['_GETSTATEREQUEST']._serialized_end=1333
-  _globals['_GETSTATERESPONSE']._serialized_start=1335
-  _globals['_GETSTATERESPONSE']._serialized_end=1390
-  _globals['_GETSTATEHISTORYREQUEST']._serialized_start=1393
-  _globals['_GETSTATEHISTORYREQUEST']._serialized_end=1571
-  _globals['_GETSTATEHISTORYRESPONSE']._serialized_start=1573
-  _globals['_GETSTATEHISTORYRESPONSE']._serialized_end=1637
-  _globals['_LANGGRAPHRUNTIME']._serialized_start=1640
-  _globals['_LANGGRAPHRUNTIME']._serialized_end=2053
+  _globals['_POSTGRESCHECKPOINTER']._serialized_end=410
+  _globals['_INVOKEREQUEST']._serialized_start=413
+  _globals['_INVOKEREQUEST']._serialized_end=620
+  _globals['_STREAMREQUEST']._serialized_start=623
+  _globals['_STREAMREQUEST']._serialized_end=830
+  _globals['_RUNOPTS']._serialized_start=833
+  _globals['_RUNOPTS']._serialized_end=1043
+  _globals['_CHUNKLIST']._serialized_start=1045
+  _globals['_CHUNKLIST']._serialized_end=1094
+  _globals['_OUTPUTCHUNK']._serialized_start=1097
+  _globals['_OUTPUTCHUNK']._serialized_end=1231
+  _globals['_GETSTATEREQUEST']._serialized_start=1233
+  _globals['_GETSTATEREQUEST']._serialized_end=1328
+  _globals['_GETSTATERESPONSE']._serialized_start=1330
+  _globals['_GETSTATERESPONSE']._serialized_end=1385
+  _globals['_GETSTATEHISTORYREQUEST']._serialized_start=1388
+  _globals['_GETSTATEHISTORYREQUEST']._serialized_end=1566
+  _globals['_GETSTATEHISTORYRESPONSE']._serialized_start=1568
+  _globals['_GETSTATEHISTORYRESPONSE']._serialized_end=1632
+  _globals['_LANGGRAPHRUNTIME']._serialized_start=1635
+  _globals['_LANGGRAPHRUNTIME']._serialized_end=2048
 # @@protoc_insertion_point(module_scope)

langgraph_executor/pb/runtime_pb2.pyi

@@ -91,14 +91,14 @@ global___InMemoryCheckpointer = InMemoryCheckpointer
 class PostgresCheckpointer(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
-    CONNECTION_STRING_FIELD_NUMBER: builtins.int
-    connection_string: builtins.str
+    DATABASE_URI_FIELD_NUMBER: builtins.int
+    database_uri: builtins.str
     def __init__(
         self,
         *,
-        connection_string: builtins.str = ...,
+        database_uri: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing.Literal["connection_string", b"connection_string"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["database_uri", b"database_uri"]) -> None: ...
 
 global___PostgresCheckpointer = PostgresCheckpointer
 

{langgraph_executor-0.0.1a0.dist-info → langgraph_executor-0.0.1a1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-executor
-Version: 0.0.1a0
+Version: 0.0.1a1
 Summary: LangGraph python RPC server executable by the langgraph-go orchestrator.
 Requires-Python: >=3.11
 Requires-Dist: grpcio>=1.73.1

{langgraph_executor-0.0.1a0.dist-info → langgraph_executor-0.0.1a1.dist-info}/RECORD

@@ -1,31 +1,31 @@
-langgraph_executor/__init__.py,sha256=508eDu4nPCyqYyg_NQBRjRZmK_68ggYywDmqDMXRF1I,24
+langgraph_executor/__init__.py,sha256=qEdg2MhZc4iFQOS_8-O-er6N8n5kRoiTg9Mrx_ssBJ8,24
 langgraph_executor/common.py,sha256=w75Bqbrj5LOtiWoOOdIi45yVB05xYXDAnokwg7MgDQE,13688
 langgraph_executor/example.py,sha256=TcfxgC9VfpZFliWnuVdJMllCDa8ji7vrSCRWnmdsUA8,900
-langgraph_executor/execute_task.py,sha256=DmsLHhkfhBYhZeHFq2eWHCkJuK9WfcDSkUQ8a0nwCB0,6932
-langgraph_executor/executor.py,sha256=9YI_-meJHz9ei9azWQjZGLzhgHDpa7Xj6Sb9HO01s_w,13242
-langgraph_executor/extract_graph.py,sha256=VlbQ6KausNd7ZPtivItMP1g4bYDwhZtttOHtbuFy4Hc,6407
+langgraph_executor/execute_task.py,sha256=-QwlbWlE8uUNFkIVtvcaWUeu4HszDF5fxTixHnE7JZk,7459
+langgraph_executor/executor.py,sha256=vCvEpL1QhPXgz5J2fbyeJ39_98x78cnTp-DU_Js0AAQ,14688
+langgraph_executor/extract_graph.py,sha256=f7NNeZigRQlTAYg_-mV8eBgQ9q8kuThCDwLjRO6wReQ,6365
 langgraph_executor/info_logger.py,sha256=BUcrg_nui_dN4JBUVyO7jurmCnt8gQ0TOubUNbyvYBk,3095
 langgraph_executor/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_executor/server.py,sha256=avtZzStLiIDvNCKyytFU3E2Li1_TBgz5oyWUBRbPzFk,5200
 langgraph_executor/setup.sh,sha256=QI505EIya8sjEIohom6GDfFckFqOMF8bIEX-hSWcLUI,627
 langgraph_executor/stream_utils.py,sha256=snxlEoZe0d4ae3h_6Ct3zNzLV_ugvPANcr4VECYofwk,2974
 langgraph_executor/pb/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_executor/pb/executor_pb2.py,sha256=G9BI0TLB39yACBo2QX3eT0jDbeNFJyEY0IlGuOAeEH8,7783
-langgraph_executor/pb/executor_pb2.pyi,sha256=bW6TYrLG8SySVqx1ie-5igw2PEdvHiQLIvnyb59GATc,16575
+langgraph_executor/pb/executor_pb2.py,sha256=cFwtWNUw8jxoKpA8EE2XGHg2uBgZYRInM9VLTD9keuY,8196
+langgraph_executor/pb/executor_pb2.pyi,sha256=CjyrE4iJSReAWokBgD8L4TMj1dZv0PvW8jYMtIFVHWA,17535
 langgraph_executor/pb/executor_pb2_grpc.py,sha256=IvKDVBWxrWU7Rcntall4KAvpQ8CdLkbljKaCEloimQE,12224
 langgraph_executor/pb/executor_pb2_grpc.pyi,sha256=iGHLqEpl7FFrBvBmBfbacvl0YgtZf8dH5pBkAaB0d64,5270
-langgraph_executor/pb/graph_pb2.py,sha256=FyhQlHv1lIivUL_HoKVxB6x1oXxgdugV2bsKJd55A8o,4430
-langgraph_executor/pb/graph_pb2.pyi,sha256=inoFjgLD42hxkG1ORIU3Lrrk57ubdmZZtRw53JZUsGI,10161
+langgraph_executor/pb/graph_pb2.py,sha256=V4by2e0nEN-oF6AeX-SSzgN96d8BbE923p91Cp91UFQ,4021
+langgraph_executor/pb/graph_pb2.pyi,sha256=xmhlJdwZ6g-ckwxjLKEIs7pJPiGIbin9m2rqdKMOWyI,9470
 langgraph_executor/pb/graph_pb2_grpc.py,sha256=8j8j0GTUo21GL7RO-_UgVPN27DRnAfus0lhTWKI49no,886
 langgraph_executor/pb/graph_pb2_grpc.pyi,sha256=Dl8kkjhqb6F1Kt24mcFg7ppish4iKVfjRiiBxEjsMMA,413
-langgraph_executor/pb/runtime_pb2.py,sha256=tYVvki3PixSiXnrBh53izR9IynGJxtynKtkZs6Aoeu4,6445
-langgraph_executor/pb/runtime_pb2.pyi,sha256=QNkLw4OV3IJQjpX3j-t_s6_zkcxMyuYnJCwVPGaBcJ0,14398
+langgraph_executor/pb/runtime_pb2.py,sha256=o6g0hem5Izqslbgy4yQgVrxERaaUTrsOVZfLVD1W0zw,6443
+langgraph_executor/pb/runtime_pb2.pyi,sha256=mcej7UAR7Pxpkl_3Pw4ENg3oGjln6L5-_XHqkpzZFoU,14373
 langgraph_executor/pb/runtime_pb2_grpc.py,sha256=HKn8fgGT9zTDAS-ZGCk-1AO2QForvCZH_BVnufTS0sM,11791
 langgraph_executor/pb/runtime_pb2_grpc.pyi,sha256=TeEHQYvJDdypawBFA6JP4OiQ9NFa_RNVKzGwHBSlFNs,4614
 langgraph_executor/pb/types_pb2.py,sha256=rUboWBthlLM-iJOmZZta-HSekwTNFuXfWLaYHes1918,15570
 langgraph_executor/pb/types_pb2.pyi,sha256=tBX4LzmfWZ-IYscIeAEBTDC02GWWKEAQVB2JPNHChDc,40957
 langgraph_executor/pb/types_pb2_grpc.py,sha256=EPv87wCc-6BNJ2xTNcb9d3ictDerK5cBt7qhd7EmJiQ,886
 langgraph_executor/pb/types_pb2_grpc.pyi,sha256=Dl8kkjhqb6F1Kt24mcFg7ppish4iKVfjRiiBxEjsMMA,413
-langgraph_executor-0.0.1a0.dist-info/METADATA,sha256=0v94S42MrwYAAitdIL1UfZGCHb5BVwtcDeFev_I8Ptk,433
-langgraph_executor-0.0.1a0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langgraph_executor-0.0.1a0.dist-info/RECORD,,
+langgraph_executor-0.0.1a1.dist-info/METADATA,sha256=pZesnbhRr1FPbzudYlalRzmxb5sj9ytDzuc5k9PKjrU,433
+langgraph_executor-0.0.1a1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langgraph_executor-0.0.1a1.dist-info/RECORD,,