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

langgraph_executor/pb/executor_pb2_grpc.py

@@ -45,6 +45,11 @@ class LangGraphExecutorStub(object):
                 request_serializer=executor__pb2.GetGraphRequest.SerializeToString,
                 response_deserializer=executor__pb2.GetGraphResponse.FromString,
                 _registered_method=True)
+        self.GetAllGraphs = channel.unary_stream(
+                '/executor.LangGraphExecutor/GetAllGraphs',
+                request_serializer=executor__pb2.GetAllGraphsRequest.SerializeToString,
+                response_deserializer=executor__pb2.GetGraphResponse.FromString,
+                _registered_method=True)
         self.ChannelsFromCheckpoint = channel.unary_unary(
                 '/executor.LangGraphExecutor/ChannelsFromCheckpoint',
                 request_serializer=executor__pb2.ChannelsFromCheckpointRequest.SerializeToString,
@@ -85,6 +90,13 @@ class LangGraphExecutorServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def GetAllGraphs(self, request, context):
+        """Get all supported graph definitions 
+        """
+        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
         """
@@ -126,6 +138,11 @@ def add_LangGraphExecutorServicer_to_server(servicer, server):
                     request_deserializer=executor__pb2.GetGraphRequest.FromString,
                     response_serializer=executor__pb2.GetGraphResponse.SerializeToString,
             ),
+            'GetAllGraphs': grpc.unary_stream_rpc_method_handler(
+                    servicer.GetAllGraphs,
+                    request_deserializer=executor__pb2.GetAllGraphsRequest.FromString,
+                    response_serializer=executor__pb2.GetGraphResponse.SerializeToString,
+            ),
             'ChannelsFromCheckpoint': grpc.unary_unary_rpc_method_handler(
                     servicer.ChannelsFromCheckpoint,
                     request_deserializer=executor__pb2.ChannelsFromCheckpointRequest.FromString,
@@ -212,6 +229,33 @@ class LangGraphExecutor(object):
             metadata,
             _registered_method=True)
 
+    @staticmethod
+    def GetAllGraphs(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_stream(
+            request,
+            target,
+            '/executor.LangGraphExecutor/GetAllGraphs',
+            executor__pb2.GetAllGraphsRequest.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,

langgraph_executor/pb/executor_pb2_grpc.pyi

@@ -33,6 +33,12 @@ class LangGraphExecutorStub:
     ]
     """Get graph definition"""
 
+    GetAllGraphs: grpc.UnaryStreamMultiCallable[
+        executor_pb2.GetAllGraphsRequest,
+        executor_pb2.GetGraphResponse,
+    ]
+    """Get all supported graph definitions"""
+
     ChannelsFromCheckpoint: grpc.UnaryUnaryMultiCallable[
         executor_pb2.ChannelsFromCheckpointRequest,
         executor_pb2.ChannelsFromCheckpointResponse,
@@ -72,6 +78,12 @@ class LangGraphExecutorAsyncStub:
     ]
     """Get graph definition"""
 
+    GetAllGraphs: grpc.aio.UnaryStreamMultiCallable[
+        executor_pb2.GetAllGraphsRequest,
+        executor_pb2.GetGraphResponse,
+    ]
+    """Get all supported graph definitions"""
+
     ChannelsFromCheckpoint: grpc.aio.UnaryUnaryMultiCallable[
         executor_pb2.ChannelsFromCheckpointRequest,
         executor_pb2.ChannelsFromCheckpointResponse,
@@ -115,6 +127,14 @@ class LangGraphExecutorServicer(metaclass=abc.ABCMeta):
     ) -> typing.Union[executor_pb2.GetGraphResponse, collections.abc.Awaitable[executor_pb2.GetGraphResponse]]:
         """Get graph definition"""
 
+    @abc.abstractmethod
+    def GetAllGraphs(
+        self,
+        request: executor_pb2.GetAllGraphsRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[collections.abc.Iterator[executor_pb2.GetGraphResponse], collections.abc.AsyncIterator[executor_pb2.GetGraphResponse]]:
+        """Get all supported graph definitions"""
+
     @abc.abstractmethod
     def ChannelsFromCheckpoint(
         self,

langgraph_executor/server.py

@@ -1,4 +1,5 @@
 import argparse
+import asyncio
 import logging
 import os
 import pathlib
@@ -6,18 +7,13 @@ import signal
 import sys
 import time
 import uuid
-from concurrent import futures
 
-import grpc
 from langgraph.pregel import Pregel
 
-from langgraph_executor.executor import LangGraphExecutorServicer
+from langgraph_executor.executor import create_server
 from langgraph_executor.info_logger import ExecutorInfo, ExecutorInfoLogger
-from langgraph_executor.pb.executor_pb2_grpc import (
-    add_LangGraphExecutorServicer_to_server,
-)
 
-EXECUTOR_DEFAULT_PORT = 50052
+DEFAULT_EXECUTOR_ADDRESS = "127.0.0.1:50052"
 
 EXECUTOR_INFO_FILE_NAME = "info.json"
 
@@ -43,7 +39,6 @@ def setup_server_logging(component_id: str, debug=False, color=YELLOW):
     root_logger = logging.getLogger()
     for handler in root_logger.handlers[:]:
         root_logger.removeHandler(handler)
-
     # Configure logging
     logging.basicConfig(
         level=level,
@@ -63,11 +58,11 @@ def setup_server_logging(component_id: str, debug=False, color=YELLOW):
         )
 
 
-def create_executor_info(port: int, id: str):
+def create_executor_info(address: str, id: str):
     return ExecutorInfo(
         id=id,
         pid=os.getpid(),
-        port=port,
+        address=address,
         start_time=time.time(),
         status="starting",
         error_message=None,
@@ -82,10 +77,11 @@ def signum_to_name(signum):
         return f"UNKNOWN_SIGNAL_{signum}"
 
 
-def serve(
+async def serve(
     graphs: dict[str, Pregel],
-    port: int,
-    debug: bool,
+    *,
+    address: str = DEFAULT_EXECUTOR_ADDRESS,
+    debug: bool = False,
     id: str | None = None,
     log_dir: pathlib.Path | None = None,
 ):
@@ -107,16 +103,17 @@ def serve(
         log_dir or pathlib.Path(__file__).resolve().parent.parent / "logs",
     )
 
-    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
-    add_LangGraphExecutorServicer_to_server(LangGraphExecutorServicer(graphs), server)
-    port = server.add_insecure_port(f"[::]:{port}")
+    server = create_server(graphs, address)
 
-    server.start()
+    await server.start()
+    loop = asyncio.get_event_loop()
 
     # Signal handler for graceful shutdown
     def signal_handler(signum, frame):
         logger.info(f"Received signal {signum_to_name(signum)}. Shutting down...")
-        server.stop(5)  # Give 5 seconds for graceful shutdown
+        asyncio.run_coroutine_threadsafe(
+            server.stop(5), loop
+        )  # Give 5 seconds for graceful shutdown
         info_logger.update_executor_info(
             executor_info.id,
             status="stopped",
@@ -130,16 +127,16 @@ def serve(
     signal.signal(signal.SIGINT, signal_handler)
     signal.signal(signal.SIGTERM, signal_handler)
 
-    executor_info = create_executor_info(port, id_)
+    executor_info = create_executor_info(address, id_)
     info_logger.write_executor_info(executor_info)
 
     try:
         info_logger.update_executor_info(executor_info.id, status="running")
-        logger.info("Listening...")
-        server.wait_for_termination()
+        logger.info(f"Listening at address {address}...")
+        await server.wait_for_termination()
     except Exception as e:
         logger.exception("Unexpected error in executor")
-        server.stop(0)
+        await server.stop(0)
         info_logger.update_executor_info(
             executor_info.id,
             status="error",
@@ -152,8 +149,8 @@ def serve(
 def main():
     """Start a LangGraph executor server."""
     parser = argparse.ArgumentParser()
-    parser.add_argument("--port", type=int, default=EXECUTOR_DEFAULT_PORT)
-    parser.add_argument("--debug", action="store_true")
+    parser.add_argument("--address", type=int, default=DEFAULT_EXECUTOR_ADDRESS)
+    parser.add_argument("--debug", action="store_false")
     parser.add_argument(
         "--log-dir",
         type=pathlib.Path,
@@ -179,7 +176,7 @@ def main():
     graphs = GRAPHS
 
     # serve
-    serve(graphs, args.port, args.debug, log_dir=args.log_dir)
+    serve(graphs, address=args.address, debug=args.debug, log_dir=args.log_dir)
 
 
 if __name__ == "__main__":

{langgraph_executor-0.0.1a1.dist-info → langgraph_executor-0.0.1a2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-executor
-Version: 0.0.1a1
+Version: 0.0.1a2
 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.1a1.dist-info → langgraph_executor-0.0.1a2.dist-info}/RECORD

@@ -1,19 +1,20 @@
-langgraph_executor/__init__.py,sha256=qEdg2MhZc4iFQOS_8-O-er6N8n5kRoiTg9Mrx_ssBJ8,24
+langgraph_executor/__init__.py,sha256=3EXaFnXv1z-8AVCGrDM9MpCrpg8I15YhcHCsHzX3iGI,24
 langgraph_executor/common.py,sha256=w75Bqbrj5LOtiWoOOdIi45yVB05xYXDAnokwg7MgDQE,13688
 langgraph_executor/example.py,sha256=TcfxgC9VfpZFliWnuVdJMllCDa8ji7vrSCRWnmdsUA8,900
-langgraph_executor/execute_task.py,sha256=-QwlbWlE8uUNFkIVtvcaWUeu4HszDF5fxTixHnE7JZk,7459
-langgraph_executor/executor.py,sha256=vCvEpL1QhPXgz5J2fbyeJ39_98x78cnTp-DU_Js0AAQ,14688
+langgraph_executor/execute_task.py,sha256=_m-WyVODhShI29qzkK0DXSPoTY4E_izbuaT7F_fVZVw,7262
+langgraph_executor/executor.py,sha256=oUg6q8WWjAIwuh2ik76IbCPMKfmegT5ePZsWFnvR4WE,5296
+langgraph_executor/executor_base.py,sha256=OpFhZNd-FehaTkSlnmD4bw9cAQ3FF8nWO9mim5YL0Cg,18509
 langgraph_executor/extract_graph.py,sha256=f7NNeZigRQlTAYg_-mV8eBgQ9q8kuThCDwLjRO6wReQ,6365
-langgraph_executor/info_logger.py,sha256=BUcrg_nui_dN4JBUVyO7jurmCnt8gQ0TOubUNbyvYBk,3095
+langgraph_executor/info_logger.py,sha256=kgoVnMxm_aloHal89NHfH_Qb9eixrqD0dHmgnrm_vL4,3110
 langgraph_executor/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_executor/server.py,sha256=avtZzStLiIDvNCKyytFU3E2Li1_TBgz5oyWUBRbPzFk,5200
+langgraph_executor/server.py,sha256=EVwZjsO2S0qhhU9C_4e85gfrxyOCo3H72bBi28vSFIw,5134
 langgraph_executor/setup.sh,sha256=QI505EIya8sjEIohom6GDfFckFqOMF8bIEX-hSWcLUI,627
 langgraph_executor/stream_utils.py,sha256=snxlEoZe0d4ae3h_6Ct3zNzLV_ugvPANcr4VECYofwk,2974
 langgraph_executor/pb/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_executor/pb/executor_pb2.py,sha256=cFwtWNUw8jxoKpA8EE2XGHg2uBgZYRInM9VLTD9keuY,8196
-langgraph_executor/pb/executor_pb2.pyi,sha256=CjyrE4iJSReAWokBgD8L4TMj1dZv0PvW8jYMtIFVHWA,17535
-langgraph_executor/pb/executor_pb2_grpc.py,sha256=IvKDVBWxrWU7Rcntall4KAvpQ8CdLkbljKaCEloimQE,12224
-langgraph_executor/pb/executor_pb2_grpc.pyi,sha256=iGHLqEpl7FFrBvBmBfbacvl0YgtZf8dH5pBkAaB0d64,5270
+langgraph_executor/pb/executor_pb2.py,sha256=ny5SLOq4VqbZ-1XxPdytfi3Y-H5FuQwL4CK3fgMh_EI,8617
+langgraph_executor/pb/executor_pb2.pyi,sha256=iVySzUzD9REkvJzet00YOL9Be3YCRizCqnsvGU_oYD4,18612
+langgraph_executor/pb/executor_pb2_grpc.py,sha256=tsYWd0X0hmGPOO_Vf-4Pv0-vgMutzaTOTM_UnDjUeIc,13938
+langgraph_executor/pb/executor_pb2_grpc.pyi,sha256=5ipWaunudj0rGP1yoRRf3V-2kjKiMkr4HHIuGqUa24s,5979
 langgraph_executor/pb/graph_pb2.py,sha256=V4by2e0nEN-oF6AeX-SSzgN96d8BbE923p91Cp91UFQ,4021
 langgraph_executor/pb/graph_pb2.pyi,sha256=xmhlJdwZ6g-ckwxjLKEIs7pJPiGIbin9m2rqdKMOWyI,9470
 langgraph_executor/pb/graph_pb2_grpc.py,sha256=8j8j0GTUo21GL7RO-_UgVPN27DRnAfus0lhTWKI49no,886
@@ -26,6 +27,6 @@ langgraph_executor/pb/types_pb2.py,sha256=rUboWBthlLM-iJOmZZta-HSekwTNFuXfWLaYHe
 langgraph_executor/pb/types_pb2.pyi,sha256=tBX4LzmfWZ-IYscIeAEBTDC02GWWKEAQVB2JPNHChDc,40957
 langgraph_executor/pb/types_pb2_grpc.py,sha256=EPv87wCc-6BNJ2xTNcb9d3ictDerK5cBt7qhd7EmJiQ,886
 langgraph_executor/pb/types_pb2_grpc.pyi,sha256=Dl8kkjhqb6F1Kt24mcFg7ppish4iKVfjRiiBxEjsMMA,413
-langgraph_executor-0.0.1a1.dist-info/METADATA,sha256=pZesnbhRr1FPbzudYlalRzmxb5sj9ytDzuc5k9PKjrU,433
-langgraph_executor-0.0.1a1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langgraph_executor-0.0.1a1.dist-info/RECORD,,
+langgraph_executor-0.0.1a2.dist-info/METADATA,sha256=-xr_G-Z1Huy6YMjYSJHYZZkPD_tu7Q-ZwDgLluOmkTo,433
+langgraph_executor-0.0.1a2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langgraph_executor-0.0.1a2.dist-info/RECORD,,