langgraph-api 0.5.4__py3-none-any.whl → 0.7.3__py3-none-any.whl

langgraph_api/{grpc_ops → grpc}/generated/core_api_pb2_grpc.py

@@ -524,21 +524,26 @@ class ThreadsStub(object):
                 request_serializer=core__api__pb2.SetThreadStatusRequest.SerializeToString,
                 response_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
                 _registered_method=True)
+        self.Stream = channel.unary_stream(
+                '/coreApi.Threads/Stream',
+                request_serializer=core__api__pb2.StreamThreadRequest.SerializeToString,
+                response_deserializer=core__api__pb2.StreamEvent.FromString,
+                _registered_method=True)
         self.SetJointStatus = channel.unary_unary(
                 '/coreApi.Threads/SetJointStatus',
                 request_serializer=core__api__pb2.SetThreadJointStatusRequest.SerializeToString,
                 response_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
                 _registered_method=True)
-        self.JointRollback = channel.unary_unary(
-                '/coreApi.Threads/JointRollback',
-                request_serializer=core__api__pb2.JointRollbackRequest.SerializeToString,
-                response_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
-                _registered_method=True)
         self.SweepTTL = channel.unary_unary(
                 '/coreApi.Threads/SweepTTL',
                 request_serializer=core__api__pb2.SweepThreadsTTLRequest.SerializeToString,
                 response_deserializer=core__api__pb2.SweepThreadsTTLResponse.FromString,
                 _registered_method=True)
+        self.GetGraphID = channel.unary_unary(
+                '/coreApi.Threads/GetGraphID',
+                request_serializer=core__api__pb2.GetGraphIDRequest.SerializeToString,
+                response_deserializer=core__api__pb2.GetGraphIDResponse.FromString,
+                _registered_method=True)
 
 
 class ThreadsServicer(object):
@@ -592,15 +597,15 @@ class ThreadsServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
-    def SetJointStatus(self, request, context):
-        """Set the status of a thread and run atomically.
+    def Stream(self, request, context):
+        """Stream run events for all runs associated with the thread.
         """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
-    def JointRollback(self, request, context):
-        """Roll back a run and update a thread's status atomically.
+    def SetJointStatus(self, request, context):
+        """Set the status of a thread and run atomically.
         """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
         context.set_details('Method not implemented!')
@@ -613,6 +618,13 @@ class ThreadsServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def GetGraphID(self, request, context):
+        """Get graph ID for the latest run in a thread (internal method, no auth).
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_ThreadsServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -656,21 +668,26 @@ def add_ThreadsServicer_to_server(servicer, server):
                     request_deserializer=core__api__pb2.SetThreadStatusRequest.FromString,
                     response_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
             ),
+            'Stream': grpc.unary_stream_rpc_method_handler(
+                    servicer.Stream,
+                    request_deserializer=core__api__pb2.StreamThreadRequest.FromString,
+                    response_serializer=core__api__pb2.StreamEvent.SerializeToString,
+            ),
             'SetJointStatus': grpc.unary_unary_rpc_method_handler(
                     servicer.SetJointStatus,
                     request_deserializer=core__api__pb2.SetThreadJointStatusRequest.FromString,
                     response_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
             ),
-            'JointRollback': grpc.unary_unary_rpc_method_handler(
-                    servicer.JointRollback,
-                    request_deserializer=core__api__pb2.JointRollbackRequest.FromString,
-                    response_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
-            ),
             'SweepTTL': grpc.unary_unary_rpc_method_handler(
                     servicer.SweepTTL,
                     request_deserializer=core__api__pb2.SweepThreadsTTLRequest.FromString,
                     response_serializer=core__api__pb2.SweepThreadsTTLResponse.SerializeToString,
             ),
+            'GetGraphID': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetGraphID,
+                    request_deserializer=core__api__pb2.GetGraphIDRequest.FromString,
+                    response_serializer=core__api__pb2.GetGraphIDResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'coreApi.Threads', rpc_method_handlers)
@@ -899,7 +916,7 @@ class Threads(object):
             _registered_method=True)
 
     @staticmethod
-    def SetJointStatus(request,
+    def Stream(request,
             target,
             options=(),
             channel_credentials=None,
@@ -909,12 +926,12 @@ class Threads(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(
+        return grpc.experimental.unary_stream(
             request,
             target,
-            '/coreApi.Threads/SetJointStatus',
-            core__api__pb2.SetThreadJointStatusRequest.SerializeToString,
-            google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+            '/coreApi.Threads/Stream',
+            core__api__pb2.StreamThreadRequest.SerializeToString,
+            core__api__pb2.StreamEvent.FromString,
             options,
             channel_credentials,
             insecure,
@@ -926,7 +943,7 @@ class Threads(object):
             _registered_method=True)
 
     @staticmethod
-    def JointRollback(request,
+    def SetJointStatus(request,
             target,
             options=(),
             channel_credentials=None,
@@ -939,8 +956,8 @@ class Threads(object):
         return grpc.experimental.unary_unary(
             request,
             target,
-            '/coreApi.Threads/JointRollback',
-            core__api__pb2.JointRollbackRequest.SerializeToString,
+            '/coreApi.Threads/SetJointStatus',
+            core__api__pb2.SetThreadJointStatusRequest.SerializeToString,
             google_dot_protobuf_dot_empty__pb2.Empty.FromString,
             options,
             channel_credentials,
@@ -979,6 +996,33 @@ class Threads(object):
             metadata,
             _registered_method=True)
 
+    @staticmethod
+    def GetGraphID(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,
+            '/coreApi.Threads/GetGraphID',
+            core__api__pb2.GetGraphIDRequest.SerializeToString,
+            core__api__pb2.GetGraphIDResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
 
 class RunsStub(object):
     """Missing associated documentation comment in .proto file."""
@@ -992,7 +1036,7 @@ class RunsStub(object):
         self.Create = channel.unary_unary(
                 '/coreApi.Runs/Create',
                 request_serializer=core__api__pb2.CreateRunRequest.SerializeToString,
-                response_deserializer=core__api__pb2.Run.FromString,
+                response_deserializer=core__api__pb2.CreateRunResponse.FromString,
                 _registered_method=True)
         self.Get = channel.unary_unary(
                 '/coreApi.Runs/Get',
@@ -1019,10 +1063,30 @@ class RunsStub(object):
                 request_serializer=core__api__pb2.SetRunStatusRequest.SerializeToString,
                 response_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
                 _registered_method=True)
+        self.Stream = channel.unary_stream(
+                '/coreApi.Runs/Stream',
+                request_serializer=core__api__pb2.StreamRunRequest.SerializeToString,
+                response_deserializer=core__api__pb2.StreamEvent.FromString,
+                _registered_method=True)
+        self.Enter = channel.unary_stream(
+                '/coreApi.Runs/Enter',
+                request_serializer=core__api__pb2.EnterRunRequest.SerializeToString,
+                response_deserializer=core__api__pb2.ControlEvent.FromString,
+                _registered_method=True)
+        self.MarkDone = channel.unary_unary(
+                '/coreApi.Runs/MarkDone',
+                request_serializer=core__api__pb2.MarkRunDoneRequest.SerializeToString,
+                response_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+                _registered_method=True)
         self.Stats = channel.unary_unary(
                 '/coreApi.Runs/Stats',
                 request_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
-                response_deserializer=core__api__pb2.QueueStats.FromString,
+                response_deserializer=core__api__pb2.RunStats.FromString,
+                _registered_method=True)
+        self.Count = channel.unary_unary(
+                '/coreApi.Runs/Count',
+                request_serializer=core__api__pb2.CountRunsRequest.SerializeToString,
+                response_deserializer=core__api__pb2.CountResponse.FromString,
                 _registered_method=True)
         self.Next = channel.unary_unary(
                 '/coreApi.Runs/Next',
@@ -1075,15 +1139,44 @@ class RunsServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def Stream(self, request, context):
+        """Stream run events.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Enter(self, request, context):
+        """Enter a run - starts heartbeat and streams control signals (internal worker method)
+        Returns a stream of interrupt/rollback signals for the specified run
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def MarkDone(self, request, context):
+        """Mark a run as done (internal worker method)
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def Stats(self, request, context):
-        """Get run queue statistics (internal method)
+        """Get run statistics (internal method, no auth)
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Count(self, request, context):
+        """Count runs matching criteria (internal method, no auth)
         """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
     def Next(self, request, context):
-        """Get next run from queue (internal worker method)
+        """Get next run from queue (internal worker method, no auth)
         TODO: come back
         """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -1091,7 +1184,7 @@ class RunsServicer(object):
         raise NotImplementedError('Method not implemented!')
 
     def Sweep(self, request, context):
-        """Sweep abandoned runs (internal method)
+        """Sweep abandoned runs (internal method, no auth)
         """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
         context.set_details('Method not implemented!')
@@ -1103,7 +1196,7 @@ def add_RunsServicer_to_server(servicer, server):
             'Create': grpc.unary_unary_rpc_method_handler(
                     servicer.Create,
                     request_deserializer=core__api__pb2.CreateRunRequest.FromString,
-                    response_serializer=core__api__pb2.Run.SerializeToString,
+                    response_serializer=core__api__pb2.CreateRunResponse.SerializeToString,
             ),
             'Get': grpc.unary_unary_rpc_method_handler(
                     servicer.Get,
@@ -1130,10 +1223,30 @@ def add_RunsServicer_to_server(servicer, server):
                     request_deserializer=core__api__pb2.SetRunStatusRequest.FromString,
                     response_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
             ),
+            'Stream': grpc.unary_stream_rpc_method_handler(
+                    servicer.Stream,
+                    request_deserializer=core__api__pb2.StreamRunRequest.FromString,
+                    response_serializer=core__api__pb2.StreamEvent.SerializeToString,
+            ),
+            'Enter': grpc.unary_stream_rpc_method_handler(
+                    servicer.Enter,
+                    request_deserializer=core__api__pb2.EnterRunRequest.FromString,
+                    response_serializer=core__api__pb2.ControlEvent.SerializeToString,
+            ),
+            'MarkDone': grpc.unary_unary_rpc_method_handler(
+                    servicer.MarkDone,
+                    request_deserializer=core__api__pb2.MarkRunDoneRequest.FromString,
+                    response_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
+            ),
             'Stats': grpc.unary_unary_rpc_method_handler(
                     servicer.Stats,
                     request_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
-                    response_serializer=core__api__pb2.QueueStats.SerializeToString,
+                    response_serializer=core__api__pb2.RunStats.SerializeToString,
+            ),
+            'Count': grpc.unary_unary_rpc_method_handler(
+                    servicer.Count,
+                    request_deserializer=core__api__pb2.CountRunsRequest.FromString,
+                    response_serializer=core__api__pb2.CountResponse.SerializeToString,
             ),
             'Next': grpc.unary_unary_rpc_method_handler(
                     servicer.Next,
@@ -1172,7 +1285,7 @@ class Runs(object):
             target,
             '/coreApi.Runs/Create',
             core__api__pb2.CreateRunRequest.SerializeToString,
-            core__api__pb2.Run.FromString,
+            core__api__pb2.CreateRunResponse.FromString,
             options,
             channel_credentials,
             insecure,
@@ -1318,6 +1431,87 @@ class Runs(object):
             metadata,
             _registered_method=True)
 
+    @staticmethod
+    def Stream(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,
+            '/coreApi.Runs/Stream',
+            core__api__pb2.StreamRunRequest.SerializeToString,
+            core__api__pb2.StreamEvent.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def Enter(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,
+            '/coreApi.Runs/Enter',
+            core__api__pb2.EnterRunRequest.SerializeToString,
+            core__api__pb2.ControlEvent.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def MarkDone(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,
+            '/coreApi.Runs/MarkDone',
+            core__api__pb2.MarkRunDoneRequest.SerializeToString,
+            google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
     @staticmethod
     def Stats(request,
             target,
@@ -1334,7 +1528,34 @@ class Runs(object):
             target,
             '/coreApi.Runs/Stats',
             google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
-            core__api__pb2.QueueStats.FromString,
+            core__api__pb2.RunStats.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def Count(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,
+            '/coreApi.Runs/Count',
+            core__api__pb2.CountRunsRequest.SerializeToString,
+            core__api__pb2.CountResponse.FromString,
             options,
             channel_credentials,
             insecure,