langgraph-executor 0.0.1a0__py3-none-any.whl

langgraph_executor/pb/executor_pb2_grpc.py

@@ -0,0 +1,321 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+from . import executor_pb2 as executor__pb2
+
+GRPC_GENERATED_VERSION = '1.74.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in executor_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class LangGraphExecutorStub(object):
+    """Main service for LangGraph runtime operations
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.ListGraphs = channel.unary_unary(
+                '/executor.LangGraphExecutor/ListGraphs',
+                request_serializer=executor__pb2.ListGraphsRequest.SerializeToString,
+                response_deserializer=executor__pb2.ListGraphsResponse.FromString,
+                _registered_method=True)
+        self.GetGraph = channel.unary_unary(
+                '/executor.LangGraphExecutor/GetGraph',
+                request_serializer=executor__pb2.GetGraphRequest.SerializeToString,
+                response_deserializer=executor__pb2.GetGraphResponse.FromString,
+                _registered_method=True)
+        self.ChannelsFromCheckpoint = channel.unary_unary(
+                '/executor.LangGraphExecutor/ChannelsFromCheckpoint',
+                request_serializer=executor__pb2.ChannelsFromCheckpointRequest.SerializeToString,
+                response_deserializer=executor__pb2.ChannelsFromCheckpointResponse.FromString,
+                _registered_method=True)
+        self.ExecuteTask = channel.stream_stream(
+                '/executor.LangGraphExecutor/ExecuteTask',
+                request_serializer=executor__pb2.ExecuteTaskRequest.SerializeToString,
+                response_deserializer=executor__pb2.ExecuteTaskResponse.FromString,
+                _registered_method=True)
+        self.ApplyWrites = channel.unary_unary(
+                '/executor.LangGraphExecutor/ApplyWrites',
+                request_serializer=executor__pb2.ApplyWritesRequest.SerializeToString,
+                response_deserializer=executor__pb2.ApplyWritesResponse.FromString,
+                _registered_method=True)
+        self.GenerateCacheKey = channel.unary_unary(
+                '/executor.LangGraphExecutor/GenerateCacheKey',
+                request_serializer=executor__pb2.GenerateCacheKeyRequest.SerializeToString,
+                response_deserializer=executor__pb2.GenerateCacheKeyResponse.FromString,
+                _registered_method=True)
+
+
+class LangGraphExecutorServicer(object):
+    """Main service for LangGraph runtime operations
+    """
+
+    def ListGraphs(self, request, context):
+        """List available graphs
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetGraph(self, request, context):
+        """Get graph definition
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ChannelsFromCheckpoint(self, request, context):
+        """Get channels initialized from checkpoint
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ExecuteTask(self, request_iterator, context):
+        """Execute a node - bidirectional streaming for interactive execution
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ApplyWrites(self, request, context):
+        """Update channel values with writes
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GenerateCacheKey(self, request, context):
+        """Generate cache key for a node execution
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_LangGraphExecutorServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'ListGraphs': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListGraphs,
+                    request_deserializer=executor__pb2.ListGraphsRequest.FromString,
+                    response_serializer=executor__pb2.ListGraphsResponse.SerializeToString,
+            ),
+            'GetGraph': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetGraph,
+                    request_deserializer=executor__pb2.GetGraphRequest.FromString,
+                    response_serializer=executor__pb2.GetGraphResponse.SerializeToString,
+            ),
+            'ChannelsFromCheckpoint': grpc.unary_unary_rpc_method_handler(
+                    servicer.ChannelsFromCheckpoint,
+                    request_deserializer=executor__pb2.ChannelsFromCheckpointRequest.FromString,
+                    response_serializer=executor__pb2.ChannelsFromCheckpointResponse.SerializeToString,
+            ),
+            'ExecuteTask': grpc.stream_stream_rpc_method_handler(
+                    servicer.ExecuteTask,
+                    request_deserializer=executor__pb2.ExecuteTaskRequest.FromString,
+                    response_serializer=executor__pb2.ExecuteTaskResponse.SerializeToString,
+            ),
+            'ApplyWrites': grpc.unary_unary_rpc_method_handler(
+                    servicer.ApplyWrites,
+                    request_deserializer=executor__pb2.ApplyWritesRequest.FromString,
+                    response_serializer=executor__pb2.ApplyWritesResponse.SerializeToString,
+            ),
+            'GenerateCacheKey': grpc.unary_unary_rpc_method_handler(
+                    servicer.GenerateCacheKey,
+                    request_deserializer=executor__pb2.GenerateCacheKeyRequest.FromString,
+                    response_serializer=executor__pb2.GenerateCacheKeyResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'executor.LangGraphExecutor', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('executor.LangGraphExecutor', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class LangGraphExecutor(object):
+    """Main service for LangGraph runtime operations
+    """
+
+    @staticmethod
+    def ListGraphs(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/executor.LangGraphExecutor/ListGraphs',
+            executor__pb2.ListGraphsRequest.SerializeToString,
+            executor__pb2.ListGraphsResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def GetGraph(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/executor.LangGraphExecutor/GetGraph',
+            executor__pb2.GetGraphRequest.SerializeToString,
+            executor__pb2.GetGraphResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def ChannelsFromCheckpoint(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/executor.LangGraphExecutor/ChannelsFromCheckpoint',
+            executor__pb2.ChannelsFromCheckpointRequest.SerializeToString,
+            executor__pb2.ChannelsFromCheckpointResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def ExecuteTask(request_iterator,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.stream_stream(
+            request_iterator,
+            target,
+            '/executor.LangGraphExecutor/ExecuteTask',
+            executor__pb2.ExecuteTaskRequest.SerializeToString,
+            executor__pb2.ExecuteTaskResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def ApplyWrites(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/executor.LangGraphExecutor/ApplyWrites',
+            executor__pb2.ApplyWritesRequest.SerializeToString,
+            executor__pb2.ApplyWritesResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def GenerateCacheKey(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/executor.LangGraphExecutor/GenerateCacheKey',
+            executor__pb2.GenerateCacheKeyRequest.SerializeToString,
+            executor__pb2.GenerateCacheKeyResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

langgraph_executor/pb/executor_pb2_grpc.pyi

@@ -0,0 +1,150 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import abc
+import collections.abc
+import executor_pb2
+import grpc
+import grpc.aio
+import typing
+
+_T = typing.TypeVar("_T")
+
+class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta): ...
+
+class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext):  # type: ignore[misc, type-arg]
+    ...
+
+class LangGraphExecutorStub:
+    """Main service for LangGraph runtime operations"""
+
+    def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
+    ListGraphs: grpc.UnaryUnaryMultiCallable[
+        executor_pb2.ListGraphsRequest,
+        executor_pb2.ListGraphsResponse,
+    ]
+    """List available graphs"""
+
+    GetGraph: grpc.UnaryUnaryMultiCallable[
+        executor_pb2.GetGraphRequest,
+        executor_pb2.GetGraphResponse,
+    ]
+    """Get graph definition"""
+
+    ChannelsFromCheckpoint: grpc.UnaryUnaryMultiCallable[
+        executor_pb2.ChannelsFromCheckpointRequest,
+        executor_pb2.ChannelsFromCheckpointResponse,
+    ]
+    """Get channels initialized from checkpoint"""
+
+    ExecuteTask: grpc.StreamStreamMultiCallable[
+        executor_pb2.ExecuteTaskRequest,
+        executor_pb2.ExecuteTaskResponse,
+    ]
+    """Execute a node - bidirectional streaming for interactive execution"""
+
+    ApplyWrites: grpc.UnaryUnaryMultiCallable[
+        executor_pb2.ApplyWritesRequest,
+        executor_pb2.ApplyWritesResponse,
+    ]
+    """Update channel values with writes"""
+
+    GenerateCacheKey: grpc.UnaryUnaryMultiCallable[
+        executor_pb2.GenerateCacheKeyRequest,
+        executor_pb2.GenerateCacheKeyResponse,
+    ]
+    """Generate cache key for a node execution"""
+
+class LangGraphExecutorAsyncStub:
+    """Main service for LangGraph runtime operations"""
+
+    ListGraphs: grpc.aio.UnaryUnaryMultiCallable[
+        executor_pb2.ListGraphsRequest,
+        executor_pb2.ListGraphsResponse,
+    ]
+    """List available graphs"""
+
+    GetGraph: grpc.aio.UnaryUnaryMultiCallable[
+        executor_pb2.GetGraphRequest,
+        executor_pb2.GetGraphResponse,
+    ]
+    """Get graph definition"""
+
+    ChannelsFromCheckpoint: grpc.aio.UnaryUnaryMultiCallable[
+        executor_pb2.ChannelsFromCheckpointRequest,
+        executor_pb2.ChannelsFromCheckpointResponse,
+    ]
+    """Get channels initialized from checkpoint"""
+
+    ExecuteTask: grpc.aio.StreamStreamMultiCallable[
+        executor_pb2.ExecuteTaskRequest,
+        executor_pb2.ExecuteTaskResponse,
+    ]
+    """Execute a node - bidirectional streaming for interactive execution"""
+
+    ApplyWrites: grpc.aio.UnaryUnaryMultiCallable[
+        executor_pb2.ApplyWritesRequest,
+        executor_pb2.ApplyWritesResponse,
+    ]
+    """Update channel values with writes"""
+
+    GenerateCacheKey: grpc.aio.UnaryUnaryMultiCallable[
+        executor_pb2.GenerateCacheKeyRequest,
+        executor_pb2.GenerateCacheKeyResponse,
+    ]
+    """Generate cache key for a node execution"""
+
+class LangGraphExecutorServicer(metaclass=abc.ABCMeta):
+    """Main service for LangGraph runtime operations"""
+
+    @abc.abstractmethod
+    def ListGraphs(
+        self,
+        request: executor_pb2.ListGraphsRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[executor_pb2.ListGraphsResponse, collections.abc.Awaitable[executor_pb2.ListGraphsResponse]]:
+        """List available graphs"""
+
+    @abc.abstractmethod
+    def GetGraph(
+        self,
+        request: executor_pb2.GetGraphRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[executor_pb2.GetGraphResponse, collections.abc.Awaitable[executor_pb2.GetGraphResponse]]:
+        """Get graph definition"""
+
+    @abc.abstractmethod
+    def ChannelsFromCheckpoint(
+        self,
+        request: executor_pb2.ChannelsFromCheckpointRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[executor_pb2.ChannelsFromCheckpointResponse, collections.abc.Awaitable[executor_pb2.ChannelsFromCheckpointResponse]]:
+        """Get channels initialized from checkpoint"""
+
+    @abc.abstractmethod
+    def ExecuteTask(
+        self,
+        request_iterator: _MaybeAsyncIterator[executor_pb2.ExecuteTaskRequest],
+        context: _ServicerContext,
+    ) -> typing.Union[collections.abc.Iterator[executor_pb2.ExecuteTaskResponse], collections.abc.AsyncIterator[executor_pb2.ExecuteTaskResponse]]:
+        """Execute a node - bidirectional streaming for interactive execution"""
+
+    @abc.abstractmethod
+    def ApplyWrites(
+        self,
+        request: executor_pb2.ApplyWritesRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[executor_pb2.ApplyWritesResponse, collections.abc.Awaitable[executor_pb2.ApplyWritesResponse]]:
+        """Update channel values with writes"""
+
+    @abc.abstractmethod
+    def GenerateCacheKey(
+        self,
+        request: executor_pb2.GenerateCacheKeyRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[executor_pb2.GenerateCacheKeyResponse, collections.abc.Awaitable[executor_pb2.GenerateCacheKeyResponse]]:
+        """Generate cache key for a node execution"""
+
+def add_LangGraphExecutorServicer_to_server(servicer: LangGraphExecutorServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...

langgraph_executor/pb/graph_pb2.py

@@ -0,0 +1,55 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: graph.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'graph.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+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')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_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
+# @@protoc_insertion_point(module_scope)