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

{langgraph_executor-0.0.1a7 → langgraph_executor-0.0.1a9}/.gitignore

@@ -50,3 +50,6 @@ logs/
 langgraph_runtime_integration/bin/*
 *.egg-info/
 .gocache
+.cache
+
+.idea*

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

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: langgraph-executor
-Version: 0.0.1a7
+Version: 0.0.1a9
 Summary: LangGraph python RPC server executable by the langgraph-go orchestrator.
 Requires-Python: >=3.11
 Requires-Dist: grpcio>=1.73.1
 Requires-Dist: langgraph>=0.6.5
-Requires-Dist: protobuf>=4.25.0
+Requires-Dist: protobuf<7,>=6.31.1
 Requires-Dist: psutil>=7.0.0
 Description-Content-Type: text/markdown
 

langgraph_executor-0.0.1a9/langgraph_executor/__init__.py

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

{langgraph_executor-0.0.1a7 → langgraph_executor-0.0.1a9}/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-0.0.1a9}/pyproject.toml

@@ -11,7 +11,7 @@ readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
     "grpcio>=1.73.1",
-    "protobuf>=4.25.0",
+    "protobuf>=6.31.1,<7",
     "langgraph>=0.6.5",
     "psutil>=7.0.0",
 ]
@@ -99,4 +99,4 @@ patterns = ["*.py", "../api/**/*.py"]
 # https://docs.pytest.org/en/7.1.x/reference/reference.html
 # --strict-config       any warnings encountered while parsing the `pytest`
 #                       section of the configuration file raise errors.
-addopts = "--strict-markers --strict-config --durations=5 -vv"
+addopts = "--strict-markers --strict-config --durations=5 -vv"

langgraph_executor-0.0.1a7/langgraph_executor/__init__.py

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

langgraph_executor-0.0.1a7/langgraph_executor/setup.sh

@@ -1,29 +0,0 @@
-#!/bin/bash
-
-echo "Setting up Python environment..."
-
-# Create virtual environment if it doesn't exist
-if [ ! -d ".venv" ]; then
-    echo "Creating virtual environment..."
-    python -m venv .venv
-fi
-
-# Activate virtual environment
-source .venv/bin/activate
-
-# Install dependencies
-echo "Installing dependencies..."
-python -m pip install -r requirements.txt
-
-# Deactivate environment 
-deactivate
-
-# Generate protobuf files
-# echo "Generating Python protobuf files..."
-# source .venv/bin/activate
-# ./generate_proto.sh
-
-echo "Python setup complete!"
-echo ""
-echo "To activate the environment:"
-echo "  source .venv/bin/activate"