langgraph-api 0.2.128__tar.gz → 0.2.129__tar.gz

{langgraph_api-0.2.128 → langgraph_api-0.2.129}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.2.128
+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.129/langgraph_api/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.2.129"

{langgraph_api-0.2.128 → langgraph_api-0.2.129}/langgraph_api/models/run.py

@@ -282,6 +282,13 @@ async def create_valid_run(
             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"):
@@ -308,11 +315,6 @@ async def create_valid_run(
     put_time_start = time.time()
     if_not_exists = payload.get("if_not_exists", "reject")
 
-    # Keep config and context in sync
-    # Configurable is either A) just internal config or B) internal config + user config (and context is empty). Either way, configurable is the default.
-    context = {**context, **configurable}
-    config["configurable"] = context
-
     run_coro = Runs.put(
         conn,
         assistant_id,

{langgraph_api-0.2.128 → langgraph_api-0.2.129}/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-0.2.128 → langgraph_api-0.2.129}/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)

{langgraph_api-0.2.128 → langgraph_api-0.2.129}/langgraph_api/worker.py

@@ -54,7 +54,6 @@ async def set_auth_ctx_for_run(
         user = normalize_user(user)
         # Reapply normalization to the kwargs
         run_kwargs["config"]["configurable"]["langgraph_auth_user"] = user
-        run_kwargs["context"]["langgraph_auth_user"] = user
     except Exception:
         user = SimpleUser(user_id) if user_id is not None else None
         permissions = None

langgraph_api-0.2.128/langgraph_api/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.2.128"