langgraph-api 0.18.0.dev14__tar.gz → 0.18.0rc2__tar.gz

{langgraph_api-0.18.0.dev14 → langgraph_api-0.18.0rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.18.0.dev14
+Version: 0.18.0rc2
 Author-email: Will Fu-Hinthorn <will@langchain.dev>, Josh Rogers <josh@langchain.dev>, Parker Rule <parker@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE

langgraph_api-0.18.0rc2/langgraph_api/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.18.0rc2"

{langgraph_api-0.18.0.dev14 → langgraph_api-0.18.0rc2}/langgraph_api/executor/__main__.py

@@ -16,8 +16,10 @@ import grpc.aio as grpc_aio
 from langgraph_grpc_common.proto import orchestrator_pb2, orchestrator_pb2_grpc
 
 from langgraph_api.asyncio import ValueEvent
+from langgraph_api.encryption.context import set_encryption_context
 from langgraph_api.errors import UserInterrupt, UserRollback
 from langgraph_api.graph import collect_graphs_from_env
+from langgraph_api.grpc.ops import encryption_context_from_proto
 from langgraph_api.grpc.ops.runs import proto_to_run
 from langgraph_api.serde import json_dumpb
 from langgraph_api.worker_runtime import (
@@ -112,7 +114,12 @@ async def _run_graph(
     attempt: int,
     failed_attempts: int,
     cancel: ValueEvent,
+    encryption_context: dict[str, Any] | None = None,
 ) -> AsyncIterator[orchestrator_pb2.ExecutorMessage]:
+    # Claimed run context for custom encryption key derivation. Always set:
+    # this task is reused across dispatches, unlike worker.py. Do not log;
+    # metadata is tenant/user key material.
+    set_encryption_context(encryption_context or {})
     tracker = RunCheckpointTracker()
     temporary = run["kwargs"].get("temporary", False)
     try:
@@ -235,7 +242,16 @@ async def _dispatch_once(stub: orchestrator_pb2_grpc.OrchestratorStub) -> bool:
             return True
 
         async with aclosing(
-            _run_graph(run, execute.attempt, execute.failed_attempts, cancel)
+            _run_graph(
+                run,
+                execute.attempt,
+                execute.failed_attempts,
+                cancel,
+                encryption_context=encryption_context_from_proto(
+                    execute.encryption_context
+                )
+                or None,
+            )
         ) as results:
             async for result in results:
                 if not await _try_write(call, result):
@@ -257,6 +273,14 @@ async def _dispatch_once(stub: orchestrator_pb2_grpc.OrchestratorStub) -> bool:
 async def _bootstrap() -> None:
     from langgraph_api import _checkpointer as api_checkpointer  # noqa: PLC0415
     from langgraph_api import store as api_store  # noqa: PLC0415
+    from langgraph_api.asyncio import set_event_loop  # noqa: PLC0415
+
+    # ASGITransport hops loopback onto _MAIN_LOOP (same as runtime lifespan).
+    # The executor does not enter API lifespan, so set the pointer here on
+    # the loop that starts the pools and later runs stream_run.
+    # get_client() with no URL imports langgraph_api.server itself.
+    set_event_loop(asyncio.get_running_loop())
+
     from langgraph_runtime import database  # noqa: PLC0415
 
     await database.start_pool()

{langgraph_api-0.18.0.dev14 → langgraph_api-0.18.0rc2}/langgraph_api/grpc/ops/__init__.py

@@ -158,24 +158,24 @@ def build_encryption_context(model: str) -> enc_pb.EncryptionContext | None:
     return enc_ctx
 
 
+def encryption_context_from_proto(
+    enc_ctx: enc_pb.EncryptionContext | None,
+) -> dict[str, Any]:
+    """Decode EncryptionContext proto metadata into the dict set_encryption_context expects."""
+    encryption_context: dict[str, Any] = {}
+    if not enc_ctx or not enc_ctx.metadata:
+        return encryption_context
+    for key, value_bytes in enc_ctx.metadata.items():
+        try:
+            encryption_context[key] = orjson.loads(value_bytes)
+        except Exception:
+            encryption_context[key] = value_bytes
+    return encryption_context
+
+
 def extract_encryption_context(run_with_attempt: pb.RunWithAttempt) -> dict[str, Any]:
     """Extract the encryption context from a gRPC EncryptionContext proto message."""
-    # Extract encryption context if present
-    encryption_context = {}
-    if (
-        run_with_attempt.encryption_context
-        and run_with_attempt.encryption_context.metadata
-    ):
-        for (
-            key,
-            value_bytes,
-        ) in run_with_attempt.encryption_context.metadata.items():
-            try:
-                encryption_context[key] = orjson.loads(value_bytes)
-            except Exception:
-                # If we can't decode, keep as bytes
-                encryption_context[key] = value_bytes
-    return encryption_context
+    return encryption_context_from_proto(run_with_attempt.encryption_context)
 
 
 def _handle_grpc_error(error: AioRpcError) -> None:

{langgraph_api-0.18.0.dev14 → langgraph_api-0.18.0rc2}/langgraph_api/priority_settings.py

@@ -138,7 +138,8 @@ def _int_in_range(
 
 def validate_priority_settings(result: Any) -> ThreadPrioritySettings:
     """The settings to store for the callback's result, with defaults for what it left out.
-    An empty fairness_group is the default group, like a missing one.
+    An empty fairness_group is the default group, like a missing one, and a fairness_weight
+    is dropped with a warning.
 
     Raises:
         HTTPException(500): the result isn't a dict or None, has an unknown key, or has a
@@ -148,6 +149,16 @@ def validate_priority_settings(result: Any) -> ThreadPrioritySettings:
         result = {}
     if not isinstance(result, dict):
         raise _invalid(f"a {type(result).__name__}, not a dict or None")
+    if "fairness_weight" in result:
+        # langgraph-sdk 0.4.9 to 0.4.11 type the result with a fairness_weight, which the
+        # server never stores, so callbacks written against them may still return it.
+        result = {
+            key: value for key, value in result.items() if key != "fairness_weight"
+        }
+        logger.warning(
+            "Priority callback returned fairness_weight, which has no effect and is"
+            " ignored; remove it from the callback's result"
+        )
     unknown = sorted(set(result) - set(ThreadPrioritySettings.__annotations__))
     if unknown:
         raise _invalid(f"unknown keys {unknown}")

{langgraph_api-0.18.0.dev14 → langgraph_api-0.18.0rc2}/langgraph_api/server.py

@@ -23,7 +23,7 @@ from starlette.applications import Starlette
 from starlette.middleware import Middleware
 from starlette.middleware.cors import CORSMiddleware
 from starlette.routing import BaseRoute, Mount
-from starlette.types import Receive, Scope, Send
+from starlette.types import ASGIApp, Receive, Scope, Send
 
 import langgraph_api.config as config
 from langgraph_api.api import (
@@ -97,23 +97,42 @@ def update_openapi_spec(app):
         set_custom_spec(spec)
 
 
+def _with_leaf_path_params(route: BaseRoute, app: ASGIApp) -> ASGIApp:
+    """Resolves FastAPI 0.137+ included-router path params before middleware runs."""
+
+    async def resolved(scope: Scope, receive: Receive, send: Send) -> None:
+        # Uses FastAPI's private `_IncludedRouter._match`; CI exercises this path on latest FastAPI.
+        branch, child_scope = route, {}
+        while isinstance(branch, type(route)):
+            _, child_scope, branch, _ = branch._match(scope)
+        if "path_params" in child_scope:
+            scope["path_params"] = child_scope["path_params"]
+        await app(scope, receive, send)
+
+    return resolved
+
+
 def apply_middleware(
     routes: list[BaseRoute], middleware: list[Middleware]
 ) -> list[BaseRoute]:
-    """Applies middleware to a list of routes.
-
-    Routes are modified in place (only the `app` attribute is modified);
-    the modified routes are returned for convenience.
-    """
-    middleware_routes = []
+    """Applies middleware to routes in place, preserving route metadata."""
     for route in routes:
+        # Routes with an `app` are wrapped after matching and the method check;
+        # app-less FastAPI 0.137+ included routers are wrapped at `handle`.
+        attr = "app" if hasattr(route, "app") else "handle"
+        wrapped = getattr(route, attr)
         for cls, args, kwargs in reversed(middleware):
-            if hasattr(route, "app"):
-                route.app = cls(route.app, *args, **kwargs)
+            wrapped = cls(wrapped, *args, **kwargs)
+        if attr == "handle" and middleware:
+            if hasattr(route, "_match"):
+                wrapped = _with_leaf_path_params(route, wrapped)
             else:
-                raise ValueError(f"Cannot apply middleware: route {route} has no app")
-        middleware_routes.append(route)
-    return middleware_routes
+                logger.warning(
+                    "Middleware cannot resolve path params for route",
+                    route_type=type(route).__name__,
+                )
+        setattr(route, attr, wrapped)
+    return routes
 
 
 def validate_router_lifespan_hooks(router: typing.Any) -> None:

langgraph_api-0.18.0rc2/langgraph_grpc_common/proto/orchestrator_pb2.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: orchestrator.proto
+# Protobuf Python Version: 6.33.5
+"""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,
+    33,
+    5,
+    '',
+    'orchestrator.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from . import core_api_pb2 as core__api__pb2
+from . import encryption_pb2 as encryption__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x12orchestrator.proto\x12\x0corchestrator\x1a\x0e\x63ore-api.proto\x1a\x10\x65ncryption.proto\"\xa6\x01\n\x13OrchestratorMessage\x12/\n\x07\x65xecute\x18\x01 \x01(\x0b\x32\x1c.orchestrator.ExecuteRequestH\x00\x12-\n\x06\x63\x61ncel\x18\x02 \x01(\x0b\x32\x1b.orchestrator.CancelRequestH\x00\x12(\n\x07release\x18\x03 \x01(\x0b\x32\x15.orchestrator.ReleaseH\x00\x42\x05\n\x03msg\"\t\n\x07Release\"\x90\x01\n\x0e\x45xecuteRequest\x12\x19\n\x03run\x18\x01 \x01(\x0b\x32\x0c.coreApi.Run\x12\x0f\n\x07\x61ttempt\x18\x02 \x01(\x05\x12\x17\n\x0f\x66\x61iled_attempts\x18\x03 \x01(\x05\x12\x39\n\x12\x65ncryption_context\x18\x04 \x01(\x0b\x32\x1d.encryption.EncryptionContext\"\x1f\n\rCancelRequest\x12\x0e\n\x06\x61\x63tion\x18\x01 \x01(\t\"b\n\x0f\x45xecutorMessage\x12$\n\x05\x63hunk\x18\x01 \x01(\x0b\x32\x13.orchestrator.ChunkH\x00\x12\"\n\x04\x64one\x18\x02 \x01(\x0b\x32\x12.orchestrator.DoneH\x00\x42\x05\n\x03msg\"%\n\x05\x43hunk\x12\x0c\n\x04mode\x18\x01 \x01(\t\x12\x0e\n\x06packet\x18\x02 \x01(\x0c\"X\n\x04\x44one\x12\x0e\n\x06status\x18\x01 \x01(\t\x12\x0f\n\x07message\x18\x02 \x01(\t\x12\x17\n\x0f\x63heckpoint_json\x18\x03 \x01(\x0c\x12\x16\n\x0e\x65xception_json\x18\x04 \x01(\x0c\x32`\n\x0cOrchestrator\x12P\n\x08\x44ispatch\x12\x1d.orchestrator.ExecutorMessage\x1a!.orchestrator.OrchestratorMessage(\x01\x30\x01\x42\x42Z@github.com/langchain-ai/langgraph-api/core/internal/executors/pbb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'orchestrator_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z@github.com/langchain-ai/langgraph-api/core/internal/executors/pb'
+  _globals['_ORCHESTRATORMESSAGE']._serialized_start=71
+  _globals['_ORCHESTRATORMESSAGE']._serialized_end=237
+  _globals['_RELEASE']._serialized_start=239
+  _globals['_RELEASE']._serialized_end=248
+  _globals['_EXECUTEREQUEST']._serialized_start=251
+  _globals['_EXECUTEREQUEST']._serialized_end=395
+  _globals['_CANCELREQUEST']._serialized_start=397
+  _globals['_CANCELREQUEST']._serialized_end=428
+  _globals['_EXECUTORMESSAGE']._serialized_start=430
+  _globals['_EXECUTORMESSAGE']._serialized_end=528
+  _globals['_CHUNK']._serialized_start=530
+  _globals['_CHUNK']._serialized_end=567
+  _globals['_DONE']._serialized_start=569
+  _globals['_DONE']._serialized_end=657
+  _globals['_ORCHESTRATOR']._serialized_start=659
+  _globals['_ORCHESTRATOR']._serialized_end=755
+# @@protoc_insertion_point(module_scope)

{langgraph_api-0.18.0.dev14 → langgraph_api-0.18.0rc2}/langgraph_grpc_common/proto/orchestrator_pb2.pyi

@@ -7,6 +7,7 @@ from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
 import builtins as _builtins
 from . import core_api_pb2 as _core_api_pb2
+from . import encryption_pb2 as _encryption_pb2
 import sys
 import typing as _typing
 
@@ -71,6 +72,7 @@ class ExecuteRequest(_message.Message):
     RUN_FIELD_NUMBER: _builtins.int
     ATTEMPT_FIELD_NUMBER: _builtins.int
     FAILED_ATTEMPTS_FIELD_NUMBER: _builtins.int
+    ENCRYPTION_CONTEXT_FIELD_NUMBER: _builtins.int
     attempt: _builtins.int
     failed_attempts: _builtins.int
     """Claims that were not cooperative drains. fetch_checkpoint_after_failure
@@ -81,16 +83,24 @@ class ExecuteRequest(_message.Message):
     def run(self) -> _core_api_pb2.Run:
         """Same coreApi.Run the HTTP/API path gets via proto_to_run."""
 
+    @_builtins.property
+    def encryption_context(self) -> _encryption_pb2.EncryptionContext:
+        """Request-scoped custom-encryption metadata claimed with the run. Same
+        payload RunWithAttempt.encryption_context carries to the in-process
+        worker. Absent when the run has no context (including AES-only).
+        """
+
     def __init__(
         self,
         *,
         run: _core_api_pb2.Run | None = ...,
         attempt: _builtins.int = ...,
         failed_attempts: _builtins.int = ...,
+        encryption_context: _encryption_pb2.EncryptionContext | None = ...,
     ) -> None: ...
-    _HasFieldArgType: _TypeAlias = _typing.Literal["run", b"run"]  # noqa: Y015
+    _HasFieldArgType: _TypeAlias = _typing.Literal["encryption_context", b"encryption_context", "run", b"run"]  # noqa: Y015
     def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
-    _ClearFieldArgType: _TypeAlias = _typing.Literal["attempt", b"attempt", "failed_attempts", b"failed_attempts", "run", b"run"]  # noqa: Y015
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["attempt", b"attempt", "encryption_context", b"encryption_context", "failed_attempts", b"failed_attempts", "run", b"run"]  # noqa: Y015
     def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
     def WhichOneof(self, oneof_group: _Never) -> None: ...
 

langgraph_api-0.18.0.dev14/langgraph_api/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.18.0.dev14"

langgraph_api-0.18.0.dev14/langgraph_grpc_common/proto/orchestrator_pb2.py

@@ -1,52 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# NO CHECKED-IN PROTOBUF GENCODE
-# source: orchestrator.proto
-# Protobuf Python Version: 6.33.5
-"""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,
-    33,
-    5,
-    '',
-    'orchestrator.proto'
-)
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-from . import core_api_pb2 as core__api__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x12orchestrator.proto\x12\x0corchestrator\x1a\x0e\x63ore-api.proto\"\xa6\x01\n\x13OrchestratorMessage\x12/\n\x07\x65xecute\x18\x01 \x01(\x0b\x32\x1c.orchestrator.ExecuteRequestH\x00\x12-\n\x06\x63\x61ncel\x18\x02 \x01(\x0b\x32\x1b.orchestrator.CancelRequestH\x00\x12(\n\x07release\x18\x03 \x01(\x0b\x32\x15.orchestrator.ReleaseH\x00\x42\x05\n\x03msg\"\t\n\x07Release\"U\n\x0e\x45xecuteRequest\x12\x19\n\x03run\x18\x01 \x01(\x0b\x32\x0c.coreApi.Run\x12\x0f\n\x07\x61ttempt\x18\x02 \x01(\x05\x12\x17\n\x0f\x66\x61iled_attempts\x18\x03 \x01(\x05\"\x1f\n\rCancelRequest\x12\x0e\n\x06\x61\x63tion\x18\x01 \x01(\t\"b\n\x0f\x45xecutorMessage\x12$\n\x05\x63hunk\x18\x01 \x01(\x0b\x32\x13.orchestrator.ChunkH\x00\x12\"\n\x04\x64one\x18\x02 \x01(\x0b\x32\x12.orchestrator.DoneH\x00\x42\x05\n\x03msg\"%\n\x05\x43hunk\x12\x0c\n\x04mode\x18\x01 \x01(\t\x12\x0e\n\x06packet\x18\x02 \x01(\x0c\"X\n\x04\x44one\x12\x0e\n\x06status\x18\x01 \x01(\t\x12\x0f\n\x07message\x18\x02 \x01(\t\x12\x17\n\x0f\x63heckpoint_json\x18\x03 \x01(\x0c\x12\x16\n\x0e\x65xception_json\x18\x04 \x01(\x0c\x32`\n\x0cOrchestrator\x12P\n\x08\x44ispatch\x12\x1d.orchestrator.ExecutorMessage\x1a!.orchestrator.OrchestratorMessage(\x01\x30\x01\x42\x42Z@github.com/langchain-ai/langgraph-api/core/internal/executors/pbb\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'orchestrator_pb2', _globals)
-if not _descriptor._USE_C_DESCRIPTORS:
-  _globals['DESCRIPTOR']._loaded_options = None
-  _globals['DESCRIPTOR']._serialized_options = b'Z@github.com/langchain-ai/langgraph-api/core/internal/executors/pb'
-  _globals['_ORCHESTRATORMESSAGE']._serialized_start=53
-  _globals['_ORCHESTRATORMESSAGE']._serialized_end=219
-  _globals['_RELEASE']._serialized_start=221
-  _globals['_RELEASE']._serialized_end=230
-  _globals['_EXECUTEREQUEST']._serialized_start=232
-  _globals['_EXECUTEREQUEST']._serialized_end=317
-  _globals['_CANCELREQUEST']._serialized_start=319
-  _globals['_CANCELREQUEST']._serialized_end=350
-  _globals['_EXECUTORMESSAGE']._serialized_start=352
-  _globals['_EXECUTORMESSAGE']._serialized_end=450
-  _globals['_CHUNK']._serialized_start=452
-  _globals['_CHUNK']._serialized_end=489
-  _globals['_DONE']._serialized_start=491
-  _globals['_DONE']._serialized_end=579
-  _globals['_ORCHESTRATOR']._serialized_start=581
-  _globals['_ORCHESTRATOR']._serialized_end=677
-# @@protoc_insertion_point(module_scope)