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

{langgraph_api-0.2.126 → langgraph_api-0.2.128}/Makefile

@@ -54,7 +54,7 @@ start:
 	LANGGRAPH_RUNTIME_EDITION=inmem \
 	LANGGRAPH_AES_KEY='$(LANGGRAPH_AES_KEY)' \
 	N_JOBS_PER_WORKER=2 \
-	LANGSERVE_GRAPHS='{"agent": "./tests/graphs/agent.py:graph", "custom_lifespan": "./tests/graphs/my_router.py:graph", "single_node": "./tests/graphs/single_node.py:graph", "benchmark": "./tests/graphs/benchmark.py:graph", "other": "./tests/graphs/other.py:make_graph", "weather": "./tests/graphs/weather.py:mk_weather_graph", "searchy": "./tests/graphs/searchy.py:graph", "agent_simple": "./tests/graphs/agent_simple.py:graph", "simple_runtime": "./tests/graphs/simple_runtime.py:graph"}' \
+	LANGSERVE_GRAPHS='{"agent": "./tests/graphs/agent.py:graph", "custom_lifespan": "./tests/graphs/my_router.py:graph", "single_node": "./tests/graphs/single_node.py:graph", "benchmark": "./tests/graphs/benchmark.py:graph", "config_graph": "./tests/graphs/config_graph.py:graph", "other": "./tests/graphs/other.py:make_graph", "weather": "./tests/graphs/weather.py:mk_weather_graph", "searchy": "./tests/graphs/searchy.py:graph", "agent_simple": "./tests/graphs/agent_simple.py:graph", "simple_runtime": "./tests/graphs/simple_runtime.py:graph"}' \
 	LANGGRAPH_STORE='$(STORE_CONFIG)' \
 	LANGGRAPH_CONFIG='{"agent": {"configurable": {"model_name": "openai"}}}' \
 	LANGSMITH_LANGGRAPH_API_VARIANT=test \
@@ -77,7 +77,7 @@ start-auth-jwt:
 	LANGGRAPH_RUNTIME_EDITION=inmem LANGGRAPH_HTTP='$(HTTP_CONFIG)' \
 	LANGGRAPH_AES_KEY='$(LANGGRAPH_AES_KEY)' \
 	N_JOBS_PER_WORKER=2 \
-	LANGSERVE_GRAPHS='{"agent": "./tests/graphs/agent.py:graph", "other": "./tests/graphs/other.py:make_graph", "weather": "./tests/graphs/weather.py:mk_weather_graph", "searchy": "./tests/graphs/searchy.py:graph", "agent_simple": "./tests/graphs/agent_simple.py:graph", "simple_runtime": "./tests/graphs/simple_runtime.py:graph"}' \
+	LANGSERVE_GRAPHS='{"agent": "./tests/graphs/agent.py:graph", "config_graph": "./tests/graphs/config_graph.py:graph", "other": "./tests/graphs/other.py:make_graph", "weather": "./tests/graphs/weather.py:mk_weather_graph", "searchy": "./tests/graphs/searchy.py:graph", "agent_simple": "./tests/graphs/agent_simple.py:graph", "simple_runtime": "./tests/graphs/simple_runtime.py:graph"}' \
 	LANGGRAPH_STORE='$(STORE_CONFIG)' \
 	LANGGRAPH_AUTH='{"path": "tests/graphs/jwt_auth.py:auth"}' \
 	LANGSMITH_LANGGRAPH_API_VARIANT=test \
@@ -95,7 +95,7 @@ start-auth-jwt:
 start-auth-fastapi-jwt:
 	LANGGRAPH_RUNTIME_EDITION=inmem LANGGRAPH_HTTP='$(HTTP_CONFIG)' \
 	N_JOBS_PER_WORKER=2 \
-	LANGSERVE_GRAPHS='{"agent": "./tests/graphs/agent.py:graph", "other": "./tests/graphs/other.py:make_graph", "weather": "./tests/graphs/weather.py:mk_weather_graph", "searchy": "./tests/graphs/searchy.py:graph", "agent_simple": "./tests/graphs/agent_simple.py:graph", "simple_runtime": "./tests/graphs/simple_runtime.py:graph"}' \
+	LANGSERVE_GRAPHS='{"agent": "./tests/graphs/agent.py:graph", "config_graph": "./tests/graphs/config_graph.py:graph", "other": "./tests/graphs/other.py:make_graph", "weather": "./tests/graphs/weather.py:mk_weather_graph", "searchy": "./tests/graphs/searchy.py:graph", "agent_simple": "./tests/graphs/agent_simple.py:graph", "simple_runtime": "./tests/graphs/simple_runtime.py:graph"}' \
 	LANGGRAPH_STORE='$(STORE_CONFIG)' \
 	LANGGRAPH_AUTH='{"path": "./tests/graphs/fastapi_jwt_auth.py:auth"}' \
 	LANGSMITH_LANGGRAPH_API_VARIANT=test \

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

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

langgraph_api-0.2.128/langgraph_api/__init__.py

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

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

@@ -275,6 +275,13 @@ 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.",
+        )
+
     if checkpoint_id:
         configurable["checkpoint_id"] = str(checkpoint_id)
     if checkpoint := payload.get("checkpoint"):
@@ -300,6 +307,12 @@ async def create_valid_run(
     configurable["__after_seconds__"] = after_seconds
     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.126 → langgraph_api-0.2.128}/langgraph_api/stream.py

@@ -348,7 +348,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 +366,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 +375,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-0.2.126 → langgraph_api-0.2.128}/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
@@ -54,6 +54,7 @@ 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
@@ -130,9 +131,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 +187,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/langgraph_api/__init__.py

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