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

langgraph_executor/executor_base.py

@@ -0,0 +1,497 @@
+import asyncio
+import contextlib
+import functools
+import logging
+import uuid
+from collections.abc import AsyncIterator, Callable, Collection, Iterator, Sequence
+from typing import Any, Protocol, cast
+
+import grpc
+import grpc.aio
+from google.protobuf.struct_pb2 import Struct  # type: ignore[import-untyped]
+from langchain_core.messages import BaseMessage, BaseMessageChunk
+from langchain_core.runnables import RunnableConfig
+from langgraph.checkpoint.base import Checkpoint
+from langgraph.errors import GraphBubbleUp, GraphInterrupt
+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.types import PregelExecutableTask
+
+from langgraph_executor.common import (
+    checkpoint_to_proto,
+    exception_to_pb,
+    extract_channels,
+    pb_to_val,
+    reconstruct_channels,
+    reconstruct_checkpoint,
+    reconstruct_config,
+    reconstruct_task_writes,
+    updates_to_proto,
+)
+from langgraph_executor.execute_task import (
+    extract_writes,
+    reconstruct_task,
+)
+from langgraph_executor.extract_graph import extract_graph
+from langgraph_executor.pb import executor_pb2, executor_pb2_grpc, types_pb2
+from langgraph_executor.stream_utils import ExecutorStreamHandler
+
+
+class Logger(Protocol):
+    def debug(self, msg: str, *args: Any, **kwargs: Any) -> None: ...
+    def info(self, msg: str, *args: Any, **kwargs: Any) -> None: ...
+    def warning(self, msg: str, *args: Any, **kwargs: Any) -> None: ...
+    def error(self, msg: str, *args: Any, **kwargs: Any) -> None: ...
+    def exception(self, msg: str, *args: Any, **kwargs: Any) -> None: ...
+    def critical(self, msg: str, *args: Any, **kwargs: Any) -> None: ...
+
+
+LOGGER = logging.getLogger(__name__)
+SENTINEL = cast(executor_pb2.ExecuteTaskResponse, object())
+GetGraph = Callable[
+    [str, RunnableConfig], contextlib.AbstractAsyncContextManager[Pregel]
+]
+
+
+class LangGraphExecutorServicer(executor_pb2_grpc.LangGraphExecutorServicer):
+    """gRPC servicer for LangGraph runtime execution operations."""
+
+    def __init__(
+        self,
+        graphs: Collection[str],
+        *,
+        subgraph_map: dict[str, str],
+        get_graph: GetGraph,
+        logger: Logger | None = None,
+        on_message: Callable[
+            [
+                BaseMessageChunk,
+                dict[str, Any],
+            ],
+            None,
+        ]
+        | None = None,
+        on_custom: Callable[[Any], None] | None = None,
+    ):
+        """Initialize the servicer with compiled graphs.
+
+        Args:
+            graphs: Dictionary mapping graph names to compiled graphs
+            subgraph_map: Dictionary mapping subgraph names to parent graph names
+            get_graph: Function to get a graph by name
+            logger: Optional logger
+
+        """
+        self.logger = logger or LOGGER
+        self.graphs = set(graphs)
+        self.graph_names = sorted(self.graphs)
+        self.subgraph_map = subgraph_map
+        self.get_graph = get_graph
+        _patch_base_message_with_ids()
+        self._graph_definition_cache: dict[str, executor_pb2.GetGraphResponse] = {}
+        self.on_message = on_message
+        self.on_custom = on_custom
+
+    async def ListGraphs(
+        self, request: Any, context: grpc.aio.ServicerContext
+    ) -> executor_pb2.ListGraphsResponse:  # type: ignore[name-defined]
+        """List available graphs."""
+        return executor_pb2.ListGraphsResponse(
+            graph_names=self.graph_names,
+        )
+
+    async def GetGraph(
+        self, request: Any, context: grpc.aio.ServicerContext
+    ) -> executor_pb2.GetGraphResponse:  # type: ignore[name-defined]
+        """Get graph definition."""
+        try:
+            self.logger.debug("GetGraph called")
+            graph_name: str = request.graph_name
+            return await self._get_graph_definition(graph_name)
+
+        except Exception as e:
+            self.logger.error(f"GetGraph Error: {e}", exc_info=True)
+            await context.abort(grpc.StatusCode.INTERNAL, str(e))
+
+    async def _get_graph_definition(self, name: str) -> executor_pb2.GetGraphResponse:
+        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)
+
+            resp = executor_pb2.GetGraphResponse(
+                graph_definition=graph_definition,
+                parent_name=self.subgraph_map.get(name, None),
+                checkpointer=graph.checkpointer is not None,
+            )
+            self._graph_definition_cache[name] = resp
+            return resp
+
+    async def GetAllGraphs(
+        self,
+        request: executor_pb2.GetAllGraphsRequest,
+        context: grpc.aio.ServicerContext,
+    ) -> AsyncIterator[executor_pb2.GetGraphResponse]:
+        try:
+            self.logger.debug("GetAllGraphs called")
+            for name in self.graph_names:
+                yield await self._get_graph_definition(name)
+
+        except Exception as e:
+            self.logger.error(f"GetAllGraphs Error: {e}", exc_info=True)
+            await context.abort(grpc.StatusCode.INTERNAL, str(e))
+
+    async def ChannelsFromCheckpoint(
+        self, request: Any, context: grpc.aio.ServicerContext
+    ) -> executor_pb2.ChannelsFromCheckpointResponse:  # type: ignore[name-defined]
+        try:
+            self.logger.debug("ChannelsFromCheckpoint called")
+            async with self.get_graph(request.graph_name, RunnableConfig()) as graph:
+                # reconstruct specs
+                specs, _ = reconstruct_channels(
+                    request.specs.channels,
+                    graph,
+                    scratchpad=None,  # type: ignore[invalid-arg-type]
+                )
+
+                # initialize channels from specs and checkpoint channel values
+                checkpoint_dummy = Checkpoint(  # type: ignore[typeddict-item]
+                    channel_values={
+                        k: pb_to_val(v)
+                        for k, v in request.checkpoint_channel_values.items()
+                    },
+                )
+                channels, _ = channels_from_checkpoint(specs, checkpoint_dummy)
+
+                # channels to pb
+                channels = extract_channels(channels)
+
+                return executor_pb2.ChannelsFromCheckpointResponse(channels=channels)
+
+        except Exception as e:
+            self.logger.error(f"ChannelsFromCheckpoint Error: {e}", exc_info=True)
+            await context.abort(grpc.StatusCode.INTERNAL, str(e))
+
+    async def ExecuteTask(
+        self,
+        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)
+            config = reconstruct_config(request.task.config)
+            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
+
+                stream_queue = asyncio.Queue()
+
+                custom_stream_writer = (
+                    _create_custom_stream_writer(
+                        stream_queue, self.logger, on_custom=self.on_custom
+                    )
+                    if stream_custom
+                    else None
+                )
+
+                task = reconstruct_task(
+                    request, graph, custom_stream_writer=custom_stream_writer
+                )
+                if stream_messages:
+                    # Create and inject callback handler
+                    stream_handler = ExecutorStreamHandler(
+                        functools.partial(
+                            stream_callback,
+                            logger=self.logger,
+                            stream_queue=stream_queue,
+                            on_message=self.on_message,
+                        ),
+                        task.id,
+                    )
+
+                    # Add handler to task config callbacks
+                    if "callbacks" not in task.config:
+                        task.config["callbacks"] = []
+                    task.config["callbacks"].append(stream_handler)  # type: ignore[union-attr]
+
+                # Execute task, catching interrupts
+                # Check cache if task has cache key - send request to Go orchestrator
+                should_execute = True
+                if task.cache_key:
+                    self.logger.debug(
+                        f"Task {task.id} has cache key, sending cache check request to Go",
+                    )
+
+                    # Send cache check request to Go runtime
+                    cache_check_request = executor_pb2.CacheCheckRequest(
+                        cache_namespace=list(task.cache_key.ns),
+                        cache_key=task.cache_key.key,
+                        ttl=task.cache_key.ttl,
+                    )
+
+                    yield executor_pb2.ExecuteTaskResponse(
+                        cache_check_request=cache_check_request,
+                    )
+
+                    # Wait for Go's response via the bidirectional stream
+                    try:
+                        cache_response_request = next(request_iterator)
+                        if hasattr(cache_response_request, "cache_check_response"):
+                            cache_response = cache_response_request.cache_check_response
+                            should_execute = not cache_response.cache_hit
+                            self.logger.debug(
+                                f"Received cache response for task {task.id}: cache_hit={cache_response.cache_hit}",
+                            )
+                        else:
+                            self.logger.warning(
+                                f"Expected cache_check_response for task {task.id}, got unexpected message type",
+                            )
+                            should_execute = (
+                                True  # Default to execution if unexpected response
+                            )
+                    except StopIteration:
+                        self.logger.warning(
+                            f"No cache response received for task {task.id}, defaulting to execution",
+                        )
+                        should_execute = True  # Default to execution if no response
+
+                # TODO patch retry policy
+                # TODO configurable to deal with _call and the functional api
+                exception_pb = None
+                if should_execute:
+                    runner_task = asyncio.create_task(
+                        _run_task(task, logger=self.logger, stream_queue=stream_queue)
+                    )
+                    # Drain the queue and stream responses to client
+                    while True:
+                        item = await stream_queue.get()
+                        if item is SENTINEL:
+                            break
+                        yield item
+                    exception_pb = await runner_task
+
+                # Ensure the final chat messages are emitted (if any)
+                final_messages = _extract_output_messages(task.writes)
+                if final_messages:
+                    for message in final_messages:
+                        yield executor_pb2.ExecuteTaskResponse(
+                            message_or_message_chunk=message
+                        )
+
+                # Final task result
+                yield executor_pb2.ExecuteTaskResponse(
+                    task_result=executor_pb2.TaskResult(
+                        error=exception_pb, writes=extract_writes(task.writes)
+                    )
+                )
+
+        except Exception as e:
+            self.logger.exception(f"ExecuteTask error: {e}")
+            await context.abort(grpc.StatusCode.INTERNAL, str(e))
+
+    async def ApplyWrites(
+        self, request: Any, context: grpc.aio.ServicerContext
+    ) -> executor_pb2.ApplyWritesResponse:  # type: ignore[name-defined]
+        # get graph
+        self.logger.debug("ApplyWrites called")
+        try:
+            async with self.get_graph(request.graph_name, RunnableConfig()) as graph:
+                channels, _ = reconstruct_channels(
+                    request.channels.channels,
+                    graph,
+                    # TODO: figure this out
+                    scratchpad=None,  # type: ignore[invalid-arg-type]
+                )
+                checkpoint = reconstruct_checkpoint(request.checkpoint)
+                tasks = reconstruct_task_writes(request.tasks)
+
+                # apply writes
+                updated_channel_names_set = apply_writes(
+                    checkpoint,
+                    channels,
+                    tasks,
+                    lambda *args: request.next_version,
+                    graph.trigger_to_nodes,
+                )
+                updated_channel_names = list(updated_channel_names_set)
+
+                # Reconstruct protos
+                updated_channels = extract_channels(channels)
+                checkpoint_proto = checkpoint_to_proto(checkpoint)
+
+                # Respond with updates
+                return executor_pb2.ApplyWritesResponse(
+                    updates=updates_to_proto(
+                        checkpoint_proto,
+                        updated_channel_names,
+                        updated_channels,
+                    ),
+                )
+
+        except Exception as e:
+            self.logger.exception(f"ApplyWrites error: {e}")
+            await context.abort(grpc.StatusCode.INTERNAL, str(e))
+
+    async def GenerateCacheKey(
+        self,
+        request: executor_pb2.GenerateCacheKeyRequest,
+        context: grpc.aio.ServicerContext,
+    ) -> executor_pb2.GenerateCacheKeyResponse:
+        """Generate cache key for a node execution"""
+        raise NotImplementedError("GenerateCacheKey not implemented")
+
+
+# Helpers
+
+
+async def _run_task(
+    task: PregelExecutableTask,
+    *,
+    logger: Logger,
+    stream_queue: asyncio.Queue[executor_pb2.ExecuteTaskResponse],
+) -> types_pb2.ExecutorError | None:
+    try:
+        await arun_with_retry(
+            task,
+            retry_policy=None,
+        )
+
+    except Exception as e:
+        if isinstance(e, GraphBubbleUp | GraphInterrupt):
+            logger.info(f"Interrupt in task {task.id}: {e}")
+        else:
+            logger.exception(
+                f"Exception running task {task.id}: {e}\nTask: {task}\n\n",
+                exc_info=True,
+            )
+        return exception_to_pb(e)
+    finally:
+        await stream_queue.put(SENTINEL)
+
+
+def stream_callback(
+    message: BaseMessageChunk,
+    metadata: dict[str, Any],
+    *,
+    logger: Logger,
+    stream_queue: asyncio.Queue[executor_pb2.ExecuteTaskResponse],
+    on_message: Callable[[BaseMessageChunk, dict[str, Any]], None] | None = None,
+):
+    """Callback to capture stream chunks and queue them."""
+    try:
+        if on_message is not None:
+            on_message(message, metadata)
+        stream_queue.put_nowait(
+            executor_pb2.ExecuteTaskResponse(
+                message_or_message_chunk=_extract_output_message(message)
+            )
+        )
+    except Exception as e:
+        logger.warning(f"Failed to create stream chunk: {e}", exc_info=True)
+
+
+def _create_custom_stream_writer(
+    stream_queue: asyncio.Queue[Any],
+    logger: Logger,
+    *,
+    on_custom: Callable[[Any], None] | None = None,
+):
+    """Create a proper stream_writer function for custom mode (like langgraph does)."""
+
+    def stream_writer(content):
+        """Custom stream writer that creates CustomStreamEvent messages."""
+        try:
+            if on_custom is not None:
+                on_custom(content)
+            # 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_queue.put_nowait(custom_event_response)
+
+        except Exception as e:
+            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 = []
+    for write in writes:
+        # Not sure this check is right
+        if isinstance(write[1], BaseMessage):
+            messages.append(_extract_output_message(write[1]))
+        elif isinstance(write[1], Sequence):
+            messages.extend(
+                [
+                    _extract_output_message(w)
+                    for w in write[1]
+                    if isinstance(w, BaseMessage)
+                ]
+            )
+
+    return messages
+
+
+def _extract_output_message(write: Any) -> types_pb2.Message:  # type: ignore[name-defined]
+    message = Struct()
+    message.update(
+        {
+            "is_streaming_chunk": False,
+            "message": {
+                "id": getattr(write, "id", None) or uuid.uuid4().hex,
+                "type": getattr(write, "type", None),
+                "content": str(getattr(write, "content", "") or ""),
+                "additional_kwargs": getattr(write, "additional_kwargs", {}),
+                "usage_metadata": getattr(write, "usage_metadata", {}),
+                "tool_calls": getattr(write, "tool_calls", []),
+                "tool_call_id": getattr(write, "tool_call_id", ""),
+                "tool_call_chunks": getattr(write, "tool_call_chunks", []),
+                "response_metadata": getattr(write, "response_metadata", {}),
+            },
+            "metadata": {},
+        }
+    )
+    return types_pb2.Message(payload=message)
+
+
+async def _get_init_request(request_iterator):
+    request = await anext(request_iterator)
+
+    if not hasattr(request, "init"):
+        raise ValueError("First message must be init")
+
+    return request.init
+
+
+@functools.lru_cache(maxsize=1)
+def _patch_base_message_with_ids() -> None:
+    """Patch the specific BaseMessage class used in your system."""
+    try:
+        from langchain_core.messages import BaseMessage
+
+        original_init = BaseMessage.__init__
+
+        def patched_init(self, content: Any, **kwargs: Any) -> None:
+            original_init(self, content, **kwargs)
+            if self.id is None:
+                self.id = str(uuid.uuid4())
+
+        BaseMessage.__init__ = patched_init  # type: ignore[method-assign]
+    except Exception as e:
+        LOGGER.warning("Failed to patch BaseMessage with IDs: %s", e)

langgraph_executor/info_logger.py

@@ -8,7 +8,7 @@ from pathlib import Path
 class ExecutorInfo:
     id: str
     pid: int
-    port: int
+    address: str
     status: str
     start_time: float
     end_time: float | None = None
@@ -33,7 +33,7 @@ class ExecutorInfoLogger:
         data = {
             "id": executor_info.id,
             "pid": executor_info.pid,
-            "port": executor_info.port,
+            "address": executor_info.address,
             "status": executor_info.status,
             "start_time": executor_info.start_time,
             "end_time": executor_info.end_time,
@@ -61,7 +61,7 @@ class ExecutorInfoLogger:
             return ExecutorInfo(
                 id=data["id"],
                 pid=data["pid"],
-                port=data["port"],
+                address=data["address"],
                 status=data["status"],
                 start_time=data["start_time"],
                 end_time=data.get("end_time"),

langgraph_executor/pb/executor_pb2.py

@@ -27,7 +27,7 @@ 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\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')
+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\"\x9a\x01\n\x10GetGraphResponse\x12\x30\n\x10graph_definition\x18\x01 \x01(\x0b\x32\x16.graph.GraphDefinition\x12\x18\n\x0bparent_name\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x19\n\x0c\x63heckpointer\x18\x03 \x01(\x08H\x01\x88\x01\x01\x42\x0e\n\x0c_parent_nameB\x0f\n\r_checkpointer\"\x15\n\x13GetAllGraphsRequest\"\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\xd0\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\x0cGetAllGraphs\x12\x1d.executor.GetAllGraphsRequest\x1a\x1a.executor.GetGraphResponse0\x01\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)
@@ -45,38 +45,40 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _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
+  _globals['_GETGRAPHRESPONSE']._serialized_start=188
+  _globals['_GETGRAPHRESPONSE']._serialized_end=342
+  _globals['_GETALLGRAPHSREQUEST']._serialized_start=344
+  _globals['_GETALLGRAPHSREQUEST']._serialized_end=365
+  _globals['_CHANNELSFROMCHECKPOINTREQUEST']._serialized_start=368
+  _globals['_CHANNELSFROMCHECKPOINTREQUEST']._serialized_end=634
+  _globals['_CHANNELSFROMCHECKPOINTREQUEST_CHECKPOINTCHANNELVALUESENTRY']._serialized_start=558
+  _globals['_CHANNELSFROMCHECKPOINTREQUEST_CHECKPOINTCHANNELVALUESENTRY']._serialized_end=634
+  _globals['_CHANNELSFROMCHECKPOINTRESPONSE']._serialized_start=636
+  _globals['_CHANNELSFROMCHECKPOINTRESPONSE']._serialized_end=703
+  _globals['_EXECUTETASKREQUEST']._serialized_start=706
+  _globals['_EXECUTETASKREQUEST']._serialized_end=842
+  _globals['_EXECUTETASKINIT']._serialized_start=845
+  _globals['_EXECUTETASKINIT']._serialized_end=1059
+  _globals['_EXECUTETASKRESPONSE']._serialized_start=1062
+  _globals['_EXECUTETASKRESPONSE']._serialized_end=1311
+  _globals['_TASKRESULT']._serialized_start=1313
+  _globals['_TASKRESULT']._serialized_end=1407
+  _globals['_CUSTOMSTREAMEVENT']._serialized_start=1409
+  _globals['_CUSTOMSTREAMEVENT']._serialized_end=1470
+  _globals['_APPLYWRITESREQUEST']._serialized_start=1473
+  _globals['_APPLYWRITESREQUEST']._serialized_end=1637
+  _globals['_APPLYWRITESRESPONSE']._serialized_start=1639
+  _globals['_APPLYWRITESRESPONSE']._serialized_end=1737
+  _globals['_GENERATECACHEKEYREQUEST']._serialized_start=1740
+  _globals['_GENERATECACHEKEYREQUEST']._serialized_end=1940
+  _globals['_GENERATECACHEKEYREQUEST_CHANNELVALUESENTRY']._serialized_start=1888
+  _globals['_GENERATECACHEKEYREQUEST_CHANNELVALUESENTRY']._serialized_end=1940
+  _globals['_GENERATECACHEKEYRESPONSE']._serialized_start=1942
+  _globals['_GENERATECACHEKEYRESPONSE']._serialized_end=1987
+  _globals['_CACHECHECKREQUEST']._serialized_start=1989
+  _globals['_CACHECHECKREQUEST']._serialized_end=2078
+  _globals['_CACHECHECKRESPONSE']._serialized_start=2080
+  _globals['_CACHECHECKRESPONSE']._serialized_end=2119
+  _globals['_LANGGRAPHEXECUTOR']._serialized_start=2122
+  _globals['_LANGGRAPHEXECUTOR']._serialized_end=2714
 # @@protoc_insertion_point(module_scope)

langgraph_executor/pb/executor_pb2.pyi

@@ -65,18 +65,40 @@ class GetGraphResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     GRAPH_DEFINITION_FIELD_NUMBER: builtins.int
+    PARENT_NAME_FIELD_NUMBER: builtins.int
+    CHECKPOINTER_FIELD_NUMBER: builtins.int
+    parent_name: builtins.str
+    checkpointer: builtins.bool
     @property
     def graph_definition(self) -> graph_pb2.GraphDefinition: ...
     def __init__(
         self,
         *,
         graph_definition: graph_pb2.GraphDefinition | None = ...,
+        parent_name: builtins.str | None = ...,
+        checkpointer: builtins.bool | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing.Literal["graph_definition", b"graph_definition"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing.Literal["graph_definition", b"graph_definition"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["_checkpointer", b"_checkpointer", "_parent_name", b"_parent_name", "checkpointer", b"checkpointer", "graph_definition", b"graph_definition", "parent_name", b"parent_name"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["_checkpointer", b"_checkpointer", "_parent_name", b"_parent_name", "checkpointer", b"checkpointer", "graph_definition", b"graph_definition", "parent_name", b"parent_name"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["_checkpointer", b"_checkpointer"]) -> typing.Literal["checkpointer"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["_parent_name", b"_parent_name"]) -> typing.Literal["parent_name"] | None: ...
 
 global___GetGraphResponse = GetGraphResponse
 
+@typing.final
+class GetAllGraphsRequest(google.protobuf.message.Message):
+    """GetAllGraphs"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___GetAllGraphsRequest = GetAllGraphsRequest
+
 @typing.final
 class ChannelsFromCheckpointRequest(google.protobuf.message.Message):
     """GetChannelsFromCheckpoint"""