langgraph-executor 0.0.1a3__tar.gz → 0.0.1a5__tar.gz

{langgraph_executor-0.0.1a3 → langgraph_executor-0.0.1a5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-executor
-Version: 0.0.1a3
+Version: 0.0.1a5
 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.1a5/langgraph_executor/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.0.1a5"

{langgraph_executor-0.0.1a3 → langgraph_executor-0.0.1a5}/langgraph_executor/executor_base.py

@@ -3,7 +3,14 @@ import contextlib
 import functools
 import logging
 import uuid
-from collections.abc import AsyncIterator, Callable, Collection, Iterator, Sequence
+from collections.abc import (
+    AsyncIterator,
+    Awaitable,
+    Callable,
+    Collection,
+    Iterator,
+    Sequence,
+)
 from typing import Any, Protocol, cast
 
 import grpc
@@ -17,6 +24,7 @@ from langgraph.pregel import Pregel
 from langgraph.pregel._algo import apply_writes
 from langgraph.pregel._checkpoint import channels_from_checkpoint
 from langgraph.pregel._retry import arun_with_retry
+from langgraph.store.base import BaseStore
 from langgraph.types import PregelExecutableTask
 
 from langgraph_executor.common import (
@@ -74,6 +82,7 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
         ]
         | None = None,
         on_custom: Callable[[Any], None] | None = None,
+        get_store: Callable[[], Awaitable[BaseStore]] | None = None,
     ):
         """Initialize the servicer with compiled graphs.
 
@@ -93,6 +102,7 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
         self._graph_definition_cache: dict[str, executor_pb2.GetGraphResponse] = {}
         self.on_message = on_message
         self.on_custom = on_custom
+        self.get_store = get_store
 
     async def ListGraphs(
         self, request: Any, context: grpc.aio.ServicerContext
@@ -119,7 +129,7 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
         if (resp := self._graph_definition_cache.get(name)) is not None:
             return resp
         async with self.get_graph(name, RunnableConfig()) as graph:
-            graph_definition = extract_graph(graph)
+            graph_definition = extract_graph(graph, name=name)
 
             resp = executor_pb2.GetGraphResponse(
                 graph_definition=graph_definition,
@@ -185,6 +195,7 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
         try:
             request = await _get_init_request(request_iterator)
             config = reconstruct_config(request.task.config)
+            store = await self.get_store() if self.get_store is not None else None
             async with self.get_graph(request.graph_name, config) as graph:
                 stream_messages = "messages" in request.stream_modes
                 stream_custom = "custom" in request.stream_modes
@@ -200,7 +211,10 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
                 )
 
                 task = reconstruct_task(
-                    request, graph, custom_stream_writer=custom_stream_writer
+                    request,
+                    graph,
+                    custom_stream_writer=custom_stream_writer,
+                    store=store,
                 )
                 if stream_messages:
                     # Create and inject callback handler

{langgraph_executor-0.0.1a3 → langgraph_executor-0.0.1a5}/langgraph_executor/extract_graph.py

@@ -117,14 +117,14 @@ def extract_trigger_to_nodes(
     return trigger_map
 
 
-def extract_graph(graph: Pregel) -> graph_pb2.GraphDefinition:
+def extract_graph(graph: Pregel, name: str | None = None) -> graph_pb2.GraphDefinition:
     """Extract graph information from a compiled LangGraph graph.
 
     Returns a protobuf message that contains all relevant orchestration information about the graph
     """
     # Handle input_channels and output_channels oneof
     graph_def = graph_pb2.GraphDefinition(
-        name=str(graph.name),
+        name=name or str(graph.name),
         channels=extract_channels(graph.channels),
         interrupt_before_nodes=list(graph.interrupt_before_nodes),
         interrupt_after_nodes=list(graph.interrupt_after_nodes),

langgraph_executor-0.0.1a3/langgraph_executor/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.0.1a3"