langgraph-api 0.18.0.dev14__py3-none-any.whl → 0.19.0.dev1__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.18.0.dev14"
+__version__ = "0.19.0.dev1"

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/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/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.0.dev14.dist-info → langgraph_api-0.19.0.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.18.0.dev14
+Version: 0.19.0.dev1
 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
@@ -17,7 +17,7 @@ Requires-Dist: jsonschema-rs<0.45,>=0.35.0
 Requires-Dist: langchain-core>=0.3.64
 Requires-Dist: langchain-protocol<0.1,>=0.0.18
 Requires-Dist: langgraph-checkpoint<5,>=3.0.1
-Requires-Dist: langgraph-runtime-inmem<0.39.0.dev0,>=0.38.0.dev0
+Requires-Dist: langgraph-runtime-inmem<0.40.0.dev0,>=0.39.0.dev0
 Requires-Dist: langgraph-sdk>=0.4.4
 Requires-Dist: langgraph<2,>=0.4.10
 Requires-Dist: langsmith[otel]>=0.6.3

{langgraph_api-0.18.0.dev14.dist-info → langgraph_api-0.19.0.dev1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-langgraph_api/__init__.py,sha256=zv1xCdlBnNXrWdY6uZZEHPQ6QTC8EPhBvpamvZNDoYo,29
+langgraph_api/__init__.py,sha256=M3VEOvigObbwk6P59vFF1BinDCBC8ZKVvjJQebbGyB4,28
 langgraph_api/_factory_utils.py,sha256=_FqW0olMu2PGLsnRq0ylt3a4jDCe12YIFHExMXI6Nyo,5976
 langgraph_api/asgi_transport.py,sha256=XApY3lIWBZTMbbsl8dDJzl0cLGirmAGE0SifqZUnXvs,11896
 langgraph_api/asyncio.py,sha256=uC6et0GgUd3YHGMOw31bldbIvo0_8BTeiLLM74TZDlU,12742
@@ -29,7 +29,7 @@ langgraph_api/schema.py,sha256=bbKrHHHxJasf_DT44nPfyslbtfOClCoFWAizpohcFJ4,12528
 langgraph_api/self_hosted_logs.py,sha256=CSWJH-Qrt99AFQw1-8u78wgKDkbyByQ-0tXfP2PDzGs,6096
 langgraph_api/self_hosted_metrics.py,sha256=Ez1RkraEH_23VW19SpXnKTvSBmLxZvE6pJEw6ON6PwU,20043
 langgraph_api/serde.py,sha256=sMjrORY2WCNmZEXDaE3lydi1un686qR1E_Wxw4mFgjU,9144
-langgraph_api/server.py,sha256=_DG7dCL2wtlaXdppTce2e0k8PunCcSRj79ncQfqmK_Q,10305
+langgraph_api/server.py,sha256=9WfBir5cgkaO-mUwIuv-XsteR4aTuSiKevhW8vXYt_Y,11262
 langgraph_api/sse.py,sha256=wF80kUI6KOPX7SATUvRxIAwpoJFN5rrien9FY1dRSgE,6477
 langgraph_api/state.py,sha256=Py8HG8k7u0FRAj4Ok89_uR9GerkyeFm90gSMTbWWQX0,8356
 langgraph_api/store.py,sha256=OXcdQSVCDKOzfS8S2aJHXbfuI5iYRoQBiW9WYKnCyx8,7122
@@ -105,12 +105,12 @@ langgraph_api/event_streaming/session.py,sha256=Lop4AD1CQTQGwg_yGpi4OLvLvTVaGmq5
 langgraph_api/event_streaming/state_normalizers.py,sha256=vgT4O4tJPr9VDBMn1EP994ieDGDYP43sROnOyLjkEAE,13659
 langgraph_api/event_streaming/types.py,sha256=RyZqfqgH-jmmmmAFQj5f6nH9M1rGK93zVG7nlmvqZgc,3647
 langgraph_api/executor/__init__.py,sha256=mZBT8HSqe9pMlBpE7sw6DMI4ULB4xwNPWGkN9zp1B7k,302
-langgraph_api/executor/__main__.py,sha256=SVFlKw4DVO8Raj_kNg7RkG_d3-tPt3qmEKgRDyrN80k,12239
+langgraph_api/executor/__main__.py,sha256=hBFU_zq-7yRahnFviAlYmEJE-jlfHw649IXETSyqgBY,13313
 langgraph_api/grpc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/grpc/client.py,sha256=L-FM_NrfO-pJHIk71g-rbIcR10tzCzJPv9VOszZoIkc,16100
 langgraph_api/grpc/server.py,sha256=Fknr6BJbdSK38wF7B1vPDplK25KB_UfMB9Fp1er6DPU,8366
 langgraph_api/grpc/generated/core_api_pb2.pyi,sha256=0PEzVh8s6rqHq52ekhqPaMI_TOsP4_xuUUv9_XS7r1w,54758
-langgraph_api/grpc/ops/__init__.py,sha256=KQL4w0xjvkCVz4j770QQVuP7MQh6qPaTZnxlO3KmoiA,18754
+langgraph_api/grpc/ops/__init__.py,sha256=6L07YhOGORfLu4fnVHEsXiQXiFA6xrLESGGBFb1biLE,18828
 langgraph_api/grpc/ops/assistants.py,sha256=T6ZIDfCkeiXvoHJyq-bUnl0j57-xLDrGwk0n-vFspz0,14199
 langgraph_api/grpc/ops/cache.py,sha256=MoPAmSuOwbt2Cud1YZJm_YJeaDHPFy1IiV4m0PDImgg,1399
 langgraph_api/grpc/ops/crons.py,sha256=aDlvNVYIfSP8gtr5NptLiAj-bfMtJOuMWYIA2NifWm0,19713
@@ -259,16 +259,16 @@ langgraph_grpc_common/proto/errors_pb2.py,sha256=b9RXbUmBpxuUSrBF6tSFAThRlKan3PB
 langgraph_grpc_common/proto/errors_pb2.pyi,sha256=rd3-BYUH8V-aO66taL7OOblaLgdrDtf1Vcd38GUoVVM,2181
 langgraph_grpc_common/proto/errors_pb2_grpc.py,sha256=ipvQ8CSnoewtfA_Y7uT14d6nrr-Q86f71vF-TfSrJl0,886
 langgraph_grpc_common/proto/errors_pb2_grpc.pyi,sha256=uC9Wnq6uyg488QiONpJ0ba1s_iouQCOYsjd_FDd1XUM,495
-langgraph_grpc_common/proto/orchestrator_pb2.py,sha256=E9UrdpeDtEwVK5tbX8ZUthb75bdJhREA7lkaY2YhxEQ,3314
-langgraph_grpc_common/proto/orchestrator_pb2.pyi,sha256=pfapHIUrcoiVR0p2eZyAh1OzcGE09NWmu31mzacQc2c,7752
+langgraph_grpc_common/proto/orchestrator_pb2.py,sha256=76bHsFbgRxd-LvfQuyY6poDWhoWXVBLUr8y-uaYTX4M,3486
+langgraph_grpc_common/proto/orchestrator_pb2.pyi,sha256=6WVGP0Lkf9qDwyV2OwNqeEMI1AqIezlCPhgbnj_DnBY,8354
 langgraph_grpc_common/proto/orchestrator_pb2_grpc.py,sha256=CefJAsnqkagiKfnLldQplJkrosUSlmmTmrlxwBfiE6Y,4112
 langgraph_grpc_common/proto/orchestrator_pb2_grpc.pyi,sha256=W_X9rYNVF8dwhWoM-AbMOq8q3f1fCuXrBYQPSggw628,3827
 langgraph_grpc_common/proto/store_pb2.py,sha256=VWJ7wtzL9pVmZZKbxMzOzCoiFn58bB2g8L6F7aT63Ow,8260
 langgraph_grpc_common/proto/store_pb2.pyi,sha256=atXejijuYEb9tE0stGQdZcfA9ucMFj-ytTMwvkxwviM,29322
 langgraph_grpc_common/proto/store_pb2_grpc.py,sha256=Kyt8eV2NXkjh3be_BYytGdD19kNf74961cruonMvCrw,3300
 langgraph_grpc_common/proto/store_pb2_grpc.pyi,sha256=oS1h2cDoq2OjyM8xazcmc8LwRDC5oCAD0zto2QUmPQw,2027
-langgraph_api-0.18.0.dev14.dist-info/METADATA,sha256=avJ8es63JRVpXdozERxdkyGGtUaLz-r7hHAbOj5w_xU,5940
-langgraph_api-0.18.0.dev14.dist-info/WHEEL,sha256=2iZrJkDIkeDh0k9SikKMqFuRP-ua6mb4Ir1AsER4BEw,87
-langgraph_api-0.18.0.dev14.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
-langgraph_api-0.18.0.dev14.dist-info/licenses/LICENSE,sha256=SCVQGLQfwOllsRFa9-Z3m8IYu4pnR9Vh2oANUCJiKqI,3860
-langgraph_api-0.18.0.dev14.dist-info/RECORD,,
+langgraph_api-0.19.0.dev1.dist-info/METADATA,sha256=6E3a5A5326VS22Ix2iEEVhWS8kWBeEfSD7_kqWSv6BU,5939
+langgraph_api-0.19.0.dev1.dist-info/WHEEL,sha256=2iZrJkDIkeDh0k9SikKMqFuRP-ua6mb4Ir1AsER4BEw,87
+langgraph_api-0.19.0.dev1.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
+langgraph_api-0.19.0.dev1.dist-info/licenses/LICENSE,sha256=SCVQGLQfwOllsRFa9-Z3m8IYu4pnR9Vh2oANUCJiKqI,3860
+langgraph_api-0.19.0.dev1.dist-info/RECORD,,

langgraph_grpc_common/proto/orchestrator_pb2.py

@@ -23,9 +23,10 @@ _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\"\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')
+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)
@@ -33,20 +34,20 @@ _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
+  _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_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: ...