langgraph-executor 0.0.1a7__tar.gz → 0.0.1a8__tar.gz

{langgraph_executor-0.0.1a7 → langgraph_executor-0.0.1a8}/PKG-INFO

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

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

{langgraph_executor-0.0.1a7 → langgraph_executor-0.0.1a8}/langgraph_executor/executor_base.py

@@ -91,14 +91,16 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
         get_graph: GetGraph,
         runtime_channel: grpc.Channel | None = None,
         logger: Logger | None = None,
-        on_message: Callable[
-            [
-                BaseMessageChunk,
-                dict[str, Any],
-            ],
-            None,
-        ]
-        | None = None,
+        on_message: (
+            Callable[
+                [
+                    BaseMessageChunk,
+                    dict[str, Any],
+                ],
+                None,
+            ]
+            | None
+        ) = None,
         on_custom: Callable[[Any], None] | None = None,
         get_store: Callable[[], Awaitable[BaseStore]] | None = None,
     ):
@@ -143,7 +145,9 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
     ) -> executor_pb2.GetGraphResponse:  # type: ignore[name-defined]
         """Get graph definition."""
         try:
-            self.logger.debug("GetGraph called")
+            self.logger.debug(
+                "GetGraph called", extra={"graph_name": request.graph_name}
+            )
             graph_name: str = request.graph_name
             return await self._get_graph_definition(graph_name)
 
@@ -183,7 +187,14 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
         self, request: Any, context: grpc.aio.ServicerContext
     ) -> executor_pb2.ChannelsFromCheckpointResponse:  # type: ignore[name-defined]
         try:
-            self.logger.debug("ChannelsFromCheckpoint called")
+            self.logger.debug(
+                "ChannelsFromCheckpoint called",
+                extra={
+                    "graph_name": request.graph_name,
+                    "specs": request.specs,
+                    "checkpoint_channel_values": request.checkpoint_channel_values,
+                },
+            )
             async with self.get_graph(request.graph_name, RunnableConfig()) as graph:
                 # reconstruct specs
                 specs, _ = reconstruct_channels(
@@ -215,11 +226,16 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
         request_iterator: Iterator[executor_pb2.ExecuteTaskRequest],  # type: ignore[name-defined]
         context: grpc.aio.ServicerContext,
     ) -> AsyncIterator[executor_pb2.ExecuteTaskResponse]:  # type: ignore[name-defined]
-        self.logger.debug("ExecuteTask called")
-
-        # Right now, only handle task execution without interrupts, etc
         try:
             request = await _get_init_request(request_iterator)
+            self.logger.debug(
+                "ExecuteTask called",
+                extra={
+                    "graph_name": request.graph_name,
+                    "task": request.task,
+                    "stream_modes": request.stream_modes,
+                },
+            )
             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:
@@ -338,7 +354,15 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
         self, request: Any, context: grpc.aio.ServicerContext
     ) -> executor_pb2.ApplyWritesResponse:  # type: ignore[name-defined]
         # get graph
-        self.logger.debug("ApplyWrites called")
+        self.logger.debug(
+            "ApplyWrites called",
+            extra={
+                "graph_name": request.graph_name,
+                "tasks": request.tasks,
+                "channels": request.channels,
+                "checkpoint": request.checkpoint,
+            },
+        )
         try:
             async with self.get_graph(request.graph_name, RunnableConfig()) as graph:
                 channels, _ = reconstruct_channels(
@@ -382,7 +406,15 @@ class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
         request: executor_pb2.StateUpdateRequest,
         context: grpc.aio.ServicerContext,
     ) -> executor_pb2.TaskResult | None:
-        self.logger.debug("StateUpdate called")
+        self.logger.debug(
+            "StateUpdate called",
+            extra={
+                "graph_name": request.graph_name,
+                "node_name": request.node_name,
+                "task_id": request.task_id,
+                "values": request.values,
+            },
+        )
 
         try:
             async with self.get_graph(request.graph_name, RunnableConfig()) as graph:

langgraph_executor-0.0.1a7/langgraph_executor/__init__.py

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