langgraph-api 0.2.126__py3-none-any.whl → 0.2.129__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.126"
+__version__ = "0.2.129"

langgraph_api/models/run.py

@@ -275,6 +275,20 @@ async def create_valid_run(
     config = payload.get("config") or {}
     context = payload.get("context") or {}
     configurable = config.setdefault("configurable", {})
+
+    if configurable and context:
+        raise HTTPException(
+            status_code=400,
+            detail="Cannot specify both configurable and context. Prefer setting context alone. Context was introduced in LangGraph 0.6.0 and is the long term planned replacement for configurable.",
+        )
+
+    # Keep config and context in sync for user provided params
+    if context:
+        configurable = context.copy()
+        config["configurable"] = configurable
+    else:
+        context = configurable.copy()
+
     if checkpoint_id:
         configurable["checkpoint_id"] = str(checkpoint_id)
     if checkpoint := payload.get("checkpoint"):
@@ -300,6 +314,7 @@ async def create_valid_run(
     configurable["__after_seconds__"] = after_seconds
     put_time_start = time.time()
     if_not_exists = payload.get("if_not_exists", "reject")
+
     run_coro = Runs.put(
         conn,
         assistant_id,

langgraph_api/server.py

@@ -71,7 +71,11 @@ middleware.extend(
                 allow_credentials=True,
                 allow_methods=["*"],
                 allow_headers=["*"],
-                expose_headers=["x-pagination-total", "x-pagination-next"],
+                expose_headers=[
+                    "x-pagination-total",
+                    "x-pagination-next",
+                    "content-location",
+                ],
             )
             if config.CORS_CONFIG is None
             else Middleware(

langgraph_api/stream.py

@@ -40,6 +40,42 @@ from langgraph_runtime.ops import Runs
 
 logger = structlog.stdlib.get_logger(__name__)
 
+
+async def _filter_context_by_schema(
+    context: dict[str, Any], context_schema: dict | None
+) -> dict[str, Any]:
+    """Filter context parameters based on the context schema.
+
+    Args:
+        context: The context dictionary to filter
+        context_schema: The JSON schema for valid context parameters
+
+    Returns:
+        Filtered context dictionary containing only valid parameters
+    """
+    if not context_schema or not context:
+        return context
+
+    # Extract valid properties from the schema
+    properties = context_schema.get("properties", {})
+    if not properties:
+        return context
+
+    # Filter context to only include parameters defined in the schema
+    filtered_context = {}
+    for key, value in context.items():
+        if key in properties:
+            filtered_context[key] = value
+        else:
+            await logger.adebug(
+                f"Filtering out context parameter '{key}' not found in context schema",
+                context_key=key,
+                available_keys=list(properties.keys()),
+            )
+
+    return filtered_context
+
+
 AnyStream = AsyncIterator[tuple[str, Any]]
 
 
@@ -107,6 +143,17 @@ async def astream_state(
             checkpointer=None if temporary else Checkpointer(),
         )
     )
+
+    # Filter context parameters based on context schema if available
+    if context and USE_RUNTIME_CONTEXT_API and not isinstance(graph, BaseRemotePregel):
+        try:
+            context_schema = graph.get_context_jsonschema()
+            context = await _filter_context_by_schema(context, context_schema)
+        except Exception as e:
+            await logger.adebug(
+                f"Failed to get context schema for filtering: {e}", exc_info=e
+            )
+
     input = kwargs.pop("input")
     if cmd := kwargs.pop("command"):
         input = map_cmd(cmd)
@@ -348,7 +395,17 @@ async def astream_state(
         yield "feedback", feedback_urls
 
 
-async def consume(stream: AnyStream, run_id: str, resumable: bool = False) -> None:
+async def consume(
+    stream: AnyStream,
+    run_id: str,
+    resumable: bool = False,
+    stream_modes: set[StreamMode] | None = None,
+) -> None:
+    stream_modes = stream_modes or set()
+    if "messages-tuple" in stream_modes:
+        stream_modes.add("messages")
+    stream_modes.add("metadata")
+
     async with aclosing(stream):
         try:
             async for mode, payload in stream:
@@ -356,7 +413,7 @@ async def consume(stream: AnyStream, run_id: str, resumable: bool = False) -> No
                     run_id,
                     mode,
                     await run_in_executor(None, json_dumpb, payload),
-                    resumable=resumable,
+                    resumable=resumable and mode.split("|")[0] in stream_modes,
                 )
         except Exception as e:
             if isinstance(e, ExceptionGroup):
@@ -365,7 +422,6 @@ async def consume(stream: AnyStream, run_id: str, resumable: bool = False) -> No
                 run_id,
                 "error",
                 await run_in_executor(None, json_dumpb, e),
-                resumable=resumable,
             )
             raise e
 

langgraph_api/worker.py

@@ -20,7 +20,7 @@ from langgraph_api.config import (
 from langgraph_api.errors import UserInterrupt, UserRollback, UserTimeout
 from langgraph_api.js.errors import RemoteException
 from langgraph_api.metadata import incr_runs
-from langgraph_api.schema import Run
+from langgraph_api.schema import Run, StreamMode
 from langgraph_api.state import state_snapshot_to_thread_state
 from langgraph_api.stream import AnyStream, astream_state, consume
 from langgraph_api.utils import with_user
@@ -130,9 +130,11 @@ async def worker(
                     break
 
     # Wrap the graph execution to separate user errors from server errors
-    async def wrap_user_errors(stream: AnyStream, run_id: str, resumable: bool):
+    async def wrap_user_errors(
+        stream: AnyStream, run_id: str, resumable: bool, stream_modes: set[StreamMode]
+    ):
         try:
-            await consume(stream, run_id, resumable)
+            await consume(stream, run_id, resumable, stream_modes)
         except Exception as e:
             if not isinstance(e, UserRollback | UserInterrupt):
                 logger.error(
@@ -184,8 +186,11 @@ async def worker(
                         on_checkpoint=on_checkpoint,
                         on_task_result=on_task_result,
                     )
+                stream_modes: set[StreamMode] = set(
+                    run["kwargs"].get("stream_mode", [])
+                )
                 await asyncio.wait_for(
-                    wrap_user_errors(stream, run_id, resumable),
+                    wrap_user_errors(stream, run_id, resumable, stream_modes),
                     BG_JOB_TIMEOUT_SECS,
                 )
         except (Exception, asyncio.CancelledError) as ee:

{langgraph_api-0.2.126.dist-info → langgraph_api-0.2.129.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.2.126
+Version: 0.2.129
 Author-email: Nuno Campos <nuno@langchain.dev>, Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE

{langgraph_api-0.2.126.dist-info → langgraph_api-0.2.129.dist-info}/RECORD

@@ -1,4 +1,4 @@
-langgraph_api/__init__.py,sha256=MJXz1iIiYzSsVkIx-r66BL7rUK8vsMX6Q89086HsRSU,24
+langgraph_api/__init__.py,sha256=HccA10414fLnmUwuQYGEKdJkoD7ewxAva8kIUEgZE6c,24
 langgraph_api/asgi_transport.py,sha256=eqifhHxNnxvI7jJqrY1_8RjL4Fp9NdN4prEub2FWBt8,5091
 langgraph_api/asyncio.py,sha256=Wv4Rwm-a-Cf6JpfgJmVuVlXQ7SlwrjbTn0eq1ux8I2Q,9652
 langgraph_api/cli.py,sha256=xQojITwmmKSJw48Lr2regcnRPRq2FJqWlPpeyr5TgbU,16158
@@ -17,17 +17,17 @@ langgraph_api/queue_entrypoint.py,sha256=JzJCB3iYvep-GoAHQ_-H2ZxXVgmzYfvjQsmdBRx
 langgraph_api/route.py,sha256=4VBkJMeusfiZtLzyUaKm1HwLHTq0g15y2CRiRhM6xyA,4773
 langgraph_api/schema.py,sha256=1L7g4TUJjmsaBUCSThycH11-hrdPLxB6mtc3tIHmpbM,6371
 langgraph_api/serde.py,sha256=0ALETUn582vNF-m0l_WOZGF_scL1VPA39fDkwMJQPrg,5187
-langgraph_api/server.py,sha256=KBnMFt3f9RVLVu_NqyeRc13D_Lq62Rk_2czZKEUMU5E,6994
+langgraph_api/server.py,sha256=9Y3qPixq2MuTs2tTB3CCW4cT5ueFJK7yce3GwHyi_L0,7093
 langgraph_api/sse.py,sha256=SLdtZmTdh5D8fbWrQjuY9HYLd2dg8Rmi6ZMmFMVc2iE,4204
 langgraph_api/state.py,sha256=P2mCo-0bqPu2v9FSFGJtUCjPPNvv6wLUKQh8SdxAtc8,4387
 langgraph_api/store.py,sha256=srRI0fQXNFo_RSUs4apucr4BEp_KrIseJksZXs32MlQ,4635
-langgraph_api/stream.py,sha256=RvO0mYEzU7XTSQz2PDvj3KzMO_T2Hpmsbwff0GoRDmI,15741
+langgraph_api/stream.py,sha256=ABYlcmNLqElA71Mn8I3cejBeSJaf5ZhdA-ymfs0uZEw,17517
 langgraph_api/thread_ttl.py,sha256=7H3gFlWcUiODPoaEzcwB0LR61uvcuyjD0ew_4BztB7k,1902
 langgraph_api/traceblock.py,sha256=2aWS6TKGTcQ0G1fOtnjVrzkpeGvDsR0spDbfddEqgRU,594
 langgraph_api/utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/validation.py,sha256=zMuKmwUEBjBgFMwAaeLZmatwGVijKv2sOYtYg7gfRtc,4950
 langgraph_api/webhook.py,sha256=VCJp4dI5E1oSJ15XP34cnPiOi8Ya8Q1BnBwVGadOpLI,1636
-langgraph_api/worker.py,sha256=LVvjvigurlDgpNjFcbAvRH7744fE01Lirrg2ZlHtORE,14245
+langgraph_api/worker.py,sha256=0Yt3L4_D0vu0rL2l8ZeQIin-X12H6WPRQeLxYiKYMSo,14458
 langgraph_api/api/__init__.py,sha256=WHy6oNLWtH1K7AxmmsU9RD-Vm6WP-Ov16xS8Ey9YCmQ,6090
 langgraph_api/api/assistants.py,sha256=ecHaID71ReTAZF4qsJzDe5L-2T5iOL2v8p6kQVHLKFk,16009
 langgraph_api/api/mcp.py,sha256=qe10ZRMN3f-Hli-9TI8nbQyWvMeBb72YB1PZVbyqBQw,14418
@@ -74,7 +74,7 @@ langgraph_api/middleware/http_logger.py,sha256=tUdWuIKtDa2EkFtG_kCjw1Wkgv7gbGH3h
 langgraph_api/middleware/private_network.py,sha256=eYgdyU8AzU2XJu362i1L8aSFoQRiV7_aLBPw7_EgeqI,2111
 langgraph_api/middleware/request_id.py,sha256=SDj3Yi3WvTbFQ2ewrPQBjAV8sYReOJGeIiuoHeZpR9g,1242
 langgraph_api/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/models/run.py,sha256=Yn1VuAAzxLMb9akOw8mLMWgx2cxPhJiIZ4-lYe75guQ,14808
+langgraph_api/models/run.py,sha256=kfoNs-PdurTsaV9tpQzaR59aRXPFYMJNJbDz4uH7Dis,15323
 langgraph_api/tunneling/cloudflare.py,sha256=iKb6tj-VWPlDchHFjuQyep2Dpb-w2NGfJKt-WJG9LH0,3650
 langgraph_api/utils/__init__.py,sha256=EQu0PShwHhxUI_9mDFgqlAf5_y5bX8TEk723P5iby24,4161
 langgraph_api/utils/cache.py,sha256=SrtIWYibbrNeZzLXLUGBFhJPkMVNQnVxR5giiYGHEfI,1810
@@ -95,8 +95,8 @@ langgraph_runtime/store.py,sha256=7mowndlsIroGHv3NpTSOZDJR0lCuaYMBoTnTrewjslw,11
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=SPCrzYpta2xTl-WE2W6qwosYdQqLeB8qpzaYEbcK44k,150725
-langgraph_api-0.2.126.dist-info/METADATA,sha256=2nFffECkS7whKzQeFOf4N-BWWgyRgAbKq1VVLqDIIA4,3890
-langgraph_api-0.2.126.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langgraph_api-0.2.126.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
-langgraph_api-0.2.126.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.2.126.dist-info/RECORD,,
+langgraph_api-0.2.129.dist-info/METADATA,sha256=sSSIekJ2xlZNS6V6rXhkLiMouTrMysiZAyuV8SeMhpw,3890
+langgraph_api-0.2.129.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langgraph_api-0.2.129.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
+langgraph_api-0.2.129.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.2.129.dist-info/RECORD,,