langgraph-api 0.5.3__tar.gz → 0.5.9__tar.gz

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/Makefile

@@ -2,8 +2,8 @@
 
 # Environment variables
 FF_USE_CORE_API ?= false
-LANGSERVE_GRAPHS_ALL = '{"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", "agent_interrupt": "./tests/graphs/agent_interrupt.py:graph", "message_type_test": "./tests/graphs/message_type_test.py:graph", "remote_subgraph_parent": "./tests/graphs/remote_subgraph_parent.py:graph", "simple_remote": "./tests/graphs/simple_remote.py:graph"}'
-LANGSERVE_GRAPHS_AUTH = '{"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"}'
+LANGSERVE_GRAPHS_ALL = '{"agent": {"path": "./tests/graphs/agent.py:graph", "description": "agent"}, "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", "agent_interrupt": "./tests/graphs/agent_interrupt.py:graph", "message_type_test": "./tests/graphs/message_type_test.py:graph", "remote_subgraph_parent": "./tests/graphs/remote_subgraph_parent.py:graph", "simple_remote": "./tests/graphs/simple_remote.py:graph", "nested_subgraphs": "./tests/graphs/nested_subgraphs.py:graph", "functional_fibonacci": "./tests/graphs/functional_fibonacci.py:fibonacci", "state_graph_fibonacci": "./tests/graphs/state_graph_fibonacci.py:fibonacci"}'
+LANGSERVE_GRAPHS_AUTH = '{"agent": {"path": "./tests/graphs/agent.py:graph", "description": "agent"}, "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", "functional_fibonacci": "./tests/graphs/functional_fibonacci.py:fibonacci", "state_graph_fibonacci": "./tests/graphs/state_graph_fibonacci.py:fibonacci"}'
 
 # Go server management
 build-go-server:
@@ -217,6 +217,7 @@ start-js-server:
 	cd ../public-api-server-js && FF_USE_CORE_API=true \
 	LANGSERVE_GRAPHS=$(LANGSERVE_GRAPHS_ALL) \
 	LANGGRAPH_CONFIG='{"agent": {"configurable": {"model_name": "openai"}}}' \
+	FF_USE_JS_API=true \
 	PORT=9123 yarn start
 
 VERSION_KIND ?= patch

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.5.3
+Version: 0.5.9
 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
@@ -12,8 +12,8 @@ Requires-Dist: grpcio<2.0.0,>=1.75.0
 Requires-Dist: httpx>=0.25.0
 Requires-Dist: jsonschema-rs<0.30,>=0.20.0
 Requires-Dist: langchain-core>=0.3.64
-Requires-Dist: langgraph-checkpoint<4,>=3
-Requires-Dist: langgraph-runtime-inmem<0.17.0,>=0.16.0
+Requires-Dist: langgraph-checkpoint<4,>=3.0.1
+Requires-Dist: langgraph-runtime-inmem<0.18.0,>=0.17.0
 Requires-Dist: langgraph-sdk>=0.2.0
 Requires-Dist: langgraph<2,>=0.4.10
 Requires-Dist: langsmith>=0.3.45

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/benchmark/benchmark-runners/assistant.js

@@ -59,7 +59,7 @@ export class Assistant extends BenchmarkRunner {
                 'Patch response contains the correct assistant': (r) => r.patchResponse.json().assistant_id === result.assistantId,
                 'Get response 2 contains the correct assistant': (r) => r.getResponse2.json().assistant_id === result.assistantId,
                 'Get response 2 contains the new description': (r) => r.getResponse2.json().metadata.description != result.getResponse.json().metadata.description && result.getResponse2.json().metadata.description === result.patchResponse.json().metadata.description,
-                'Get response 2 contains the correct created_by': (r) => r.getResponse2.json().metadata.created_by === 'josh',
+                'Get response 2 contains the correct created_by': (r) => r.getResponse2.json().metadata.created_by === 'benchmark',
                 'Count response contains the correct number of assistants': (r) => parseInt(r.countResponse.json()) === 1,
                 'Delete response is successful': (r) => r.deleteResponse.status === 204,
             });
@@ -68,7 +68,7 @@ export class Assistant extends BenchmarkRunner {
         }
 
         if (!success) {
-            if (result.searchResponse.status >= 500 || result.getResponse.status >= 500 || result.patchResponse.status >= 500 || result.getResponse2.status >= 500 || result.countResponse.status >= 500 || result.deleteResponse.status >= 500) {
+            if (result.searchResponse.status == 502 || result.getResponse.status == 502 || result.patchResponse.status == 502 || result.getResponse2.status == 502 || result.countResponse.status == 502 || result.deleteResponse.status == 502) {
                 errorMetrics.server_errors.add(1);
                 console.log(`Server error: ${result.searchResponse.status}, ${result.getResponse.status}, ${result.patchResponse.status}, ${result.getResponse2.status}, ${result.countResponse.status}, ${result.deleteResponse.status}`);
             } else if (result.searchResponse.status === 408 || result.getResponse.status === 408 || result.patchResponse.status === 408 || result.getResponse2.status === 408 || result.countResponse.status === 408 || result.deleteResponse.status === 408) {

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/benchmark/benchmark-runners/benchmark-runner.js

@@ -4,7 +4,7 @@
 export class BenchmarkRunner {
     /**
      * Run the benchmark type.
-     * @param {string} baseUrl - The base URL of the LangGraph server.
+     * @param {string} baseUrl - The base URL of the Agent Server.
      * @param {any} requestParams - The parameters to use for the request. Includes headers and other config like timeout.
      * @param {any} benchmarkGraphOptions - The options for the benchmark graph.
      * @returns {any} - The result of the benchmark type. This format will vary by benchmark type.

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/benchmark/burst.js

@@ -13,7 +13,7 @@ const serverErrors = new Counter('server_errors');
 const otherErrors = new Counter('other_errors');
 const burstSuccessRate = new Rate('burst_success_rate');
 
-// URL of your LangGraph server
+// URL of your Agent Server
 const BASE_URL = __ENV.BASE_URL || 'http://localhost:9123';
 // LangSmith API key only needed with a custom server endpoint
 const LANGSMITH_API_KEY = __ENV.LANGSMITH_API_KEY;

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/benchmark/clean.js

@@ -3,7 +3,7 @@
  * The default benchmark server has a thread TTL of one hour that should clean things up too so this doesn't run too long.
  */
 
-// URL of your LangGraph server
+// URL of your Agent Server
 const BASE_URL = process.env.BASE_URL || 'http://localhost:9123';
 // LangSmith API key only needed with a custom server endpoint
 const LANGSMITH_API_KEY = process.env.LANGSMITH_API_KEY;

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/benchmark/ramp.js

@@ -16,12 +16,12 @@ const otherErrors = new Counter('other_errors');
 const errorMetrics = {
   timeout_errors: timeoutErrors,
   connection_errors: connectionErrors,
-  server_errors: serverErrors,
   missing_message_errors: missingMessageErrors,
   other_errors: otherErrors,
+  server_errors: serverErrors,
 }
 
-// URL of your LangGraph server
+// URL of your Agent Server
 const BASE_URL = __ENV.BASE_URL || 'http://localhost:9123';
 // LangSmith API key only needed with a custom server endpoint
 const LANGSMITH_API_KEY = __ENV.LANGSMITH_API_KEY;

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/benchmark/update-revision.js

@@ -21,6 +21,11 @@ const REVISION_CONFIG = {
             name: "FF_LOG_DROPPED_EVENTS",
             value: "true",
             type: "secret"
+        },
+        {
+            name: "FF_USE_CORE_API",
+            value: "true",
+            type: "secret"
         }
     ],
     shareable: false

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/constraints.txt

@@ -3,7 +3,7 @@
 # https://github.com/langchain-ai/langgraph/blob/main/docs/docs/cloud/deployment/setup_pyproject.md
 langgraph>=0.4.10,<2
 langgraph-sdk>=0.2.0
-langgraph-checkpoint>=3,<4
+langgraph-checkpoint>=3.0.1,<4
 langchain-core>=0.3.64
 langsmith>=0.3.45
 orjson>=3.9.7,<3.10.17

langgraph_api-0.5.9/langgraph_api/__init__.py

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

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/langgraph_api/api/a2a.py

@@ -1035,7 +1035,6 @@ async def handle_tasks_get(
         assistant_id = run_info.get("assistant_id")
         if assistant_id:
             try:
-                # Verify that the assistant exists
                 await _get_assistant(assistant_id, request.headers)
             except ValueError as e:
                 return {
@@ -1053,7 +1052,9 @@ async def handle_tasks_get(
             a2a_state = "working"
         elif lg_status == "success":
             a2a_state = "completed"
-        elif lg_status in ["error", "timeout", "interrupted"]:
+        elif lg_status == "interrupted":
+            a2a_state = "input-required"
+        elif lg_status in ["error", "timeout"]:
             a2a_state = "failed"
         else:
             a2a_state = "submitted"

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/langgraph_api/api/assistants.py

@@ -16,7 +16,7 @@ from starlette.routing import BaseRoute
 from langgraph_api import store as api_store
 from langgraph_api.feature_flags import FF_USE_CORE_API, USE_RUNTIME_CONTEXT_API
 from langgraph_api.graph import get_assistant_id, get_graph
-from langgraph_api.grpc_ops.ops import Assistants as GrpcAssistants
+from langgraph_api.grpc.ops import Assistants as GrpcAssistants
 from langgraph_api.js.base import BaseRemotePregel
 from langgraph_api.route import ApiRequest, ApiResponse, ApiRoute
 from langgraph_api.schema import ASSISTANT_FIELDS

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/langgraph_api/api/meta.py

@@ -86,6 +86,12 @@ async def meta_metrics(request: ApiRequest):
                         "# HELP lg_api_num_running_runs The number of runs currently running.",
                         "# TYPE lg_api_num_running_runs gauge",
                         f'lg_api_num_running_runs{{project_id="{metadata.PROJECT_ID}", revision_id="{metadata.HOST_REVISION_ID}"}} {queue_stats["n_running"]}',
+                        "# HELP lg_api_pending_runs_wait_time_max The maximum time a run has been pending, in seconds.",
+                        "# TYPE lg_api_pending_runs_wait_time_max gauge",
+                        f'lg_api_pending_runs_wait_time_max{{project_id="{metadata.PROJECT_ID}", revision_id="{metadata.HOST_REVISION_ID}"}} {queue_stats.get("pending_runs_wait_time_max_secs") or 0}',
+                        "# HELP lg_api_pending_runs_wait_time_med The median pending wait time across runs, in seconds.",
+                        "# TYPE lg_api_pending_runs_wait_time_med gauge",
+                        f'lg_api_pending_runs_wait_time_med{{project_id="{metadata.PROJECT_ID}", revision_id="{metadata.HOST_REVISION_ID}"}} {queue_stats.get("pending_runs_wait_time_med_secs") or 0}',
                     ]
                 )
         except Exception as e:

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/langgraph_api/api/threads.py

@@ -6,7 +6,7 @@ from starlette.responses import Response
 from starlette.routing import BaseRoute
 
 from langgraph_api.feature_flags import FF_USE_CORE_API
-from langgraph_api.grpc_ops.ops import Threads as GrpcThreads
+from langgraph_api.grpc.ops import Threads as GrpcThreads
 from langgraph_api.route import ApiRequest, ApiResponse, ApiRoute
 from langgraph_api.schema import THREAD_FIELDS, ThreadStreamMode
 from langgraph_api.sse import EventSourceResponse

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/langgraph_api/asgi_transport.py

@@ -152,7 +152,7 @@ class ASGITransport(ASGITransportBase):
 
         try:
             await call_soon_in_main_loop(self.app(scope, receive, send))
-        except Exception:  # noqa: PIE-786
+        except Exception:  # noqa: BLE001
             if self.raise_app_exceptions:
                 raise
 

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/langgraph_api/auth/custom.py

@@ -355,34 +355,39 @@ def _solve_fastapi_dependencies(
     }
 
     async def decorator(scope: dict, request: Request):
-        async with AsyncExitStack() as stack:
-            all_solved = await asyncio.gather(
-                *(
-                    solve_dependencies(
-                        request=request,
-                        dependant=dependent,
-                        async_exit_stack=stack,
-                        embed_body_fields=False,
+        async with AsyncExitStack() as request_stack:
+            scope["fastapi_inner_astack"] = request_stack
+            async with AsyncExitStack() as stack:
+                scope["fastapi_function_astack"] = stack
+                all_solved = await asyncio.gather(
+                    *(
+                        solve_dependencies(
+                            request=request,
+                            dependant=dependent,
+                            async_exit_stack=stack,
+                            embed_body_fields=False,
+                        )
+                        for dependent in dependents.values()
                     )
-                    for dependent in dependents.values()
                 )
-            )
-            all_injected = await asyncio.gather(
-                *(
-                    _run_async(dependent.call, solved.values, is_async)
-                    for dependent, solved in zip(
-                        dependents.values(), all_solved, strict=False
+                all_injected = await asyncio.gather(
+                    *(
+                        _run_async(dependent.call, solved.values, is_async)
+                        for dependent, solved in zip(
+                            dependents.values(), all_solved, strict=False
+                        )
                     )
                 )
-            )
-            kwargs = {
-                name: value
-                for name, value in zip(dependents.keys(), all_injected, strict=False)
-            }
-            other_params = _extract_arguments_from_scope(
-                scope, _param_names, request=request
-            )
-            return await fn(**(kwargs | other_params))
+                kwargs = {
+                    name: value
+                    for name, value in zip(
+                        dependents.keys(), all_injected, strict=False
+                    )
+                }
+                other_params = _extract_arguments_from_scope(
+                    scope, _param_names, request=request
+                )
+                return await fn(**(kwargs | other_params))
 
     return decorator
 

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/langgraph_api/config.py

@@ -281,6 +281,9 @@ REDIS_URI = env("REDIS_URI", cast=str)
 REDIS_CLUSTER = env("REDIS_CLUSTER", cast=bool, default=False)
 REDIS_MAX_CONNECTIONS = env("REDIS_MAX_CONNECTIONS", cast=int, default=2000)
 REDIS_CONNECT_TIMEOUT = env("REDIS_CONNECT_TIMEOUT", cast=float, default=10.0)
+REDIS_HEALTH_CHECK_INTERVAL = env(
+    "REDIS_HEALTH_CHECK_INTERVAL", cast=float, default=10.0
+)
 REDIS_KEY_PREFIX = env("REDIS_KEY_PREFIX", cast=str, default="")
 RUN_STATS_CACHE_SECONDS = env("RUN_STATS_CACHE_SECONDS", cast=int, default=60)
 

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/langgraph_api/feature_flags.py

@@ -16,3 +16,9 @@ FF_USE_CORE_API = os.getenv("FF_USE_CORE_API", "false").lower() in (
     "1",
     "yes",
 )
+# Feature flag for using the JS native API
+FF_USE_JS_API = os.getenv("FF_USE_JS_API", "false").lower() in (
+    "true",
+    "1",
+    "yes",
+)

{langgraph_api-0.5.3 → langgraph_api-0.5.9}/langgraph_api/graph.py

@@ -51,7 +51,7 @@ async def register_graph(
     description: str | None = None,
 ) -> None:
     """Register a graph."""
-    from langgraph_api.grpc_ops.ops import Assistants as AssistantsGrpc
+    from langgraph_api.grpc.ops import Assistants as AssistantsGrpc
     from langgraph_runtime.database import connect
     from langgraph_runtime.ops import Assistants as AssistantsRuntime
 

{langgraph_api-0.5.3/langgraph_api/grpc_ops → langgraph_api-0.5.9/langgraph_api/grpc}/config_conversion.py

@@ -8,7 +8,10 @@ from typing import Any, cast
 import orjson
 from langchain_core.runnables.config import RunnableConfig
 
-from langgraph_api.grpc_ops.generated import engine_common_pb2
+from langgraph_api.grpc.generated import (
+    engine_common_pb2,
+    enum_durability_pb2,
+)
 
 CONFIG_KEY_SEND = "__pregel_send"
 CONFIG_KEY_READ = "__pregel_read"
@@ -25,27 +28,27 @@ CONFIG_KEY_GRAPH_ID = "graph_id"
 
 def _durability_to_proto(
     durability: str,
-) -> engine_common_pb2.Durability:
+) -> enum_durability_pb2.Durability:
     match durability:
         case "async":
-            return engine_common_pb2.Durability.ASYNC
+            return enum_durability_pb2.Durability.ASYNC
         case "sync":
-            return engine_common_pb2.Durability.SYNC
+            return enum_durability_pb2.Durability.SYNC
         case "exit":
-            return engine_common_pb2.Durability.EXIT
+            return enum_durability_pb2.Durability.EXIT
         case _:
             raise ValueError(f"invalid durability: {durability}")
 
 
 def _durability_from_proto(
-    durability: engine_common_pb2.Durability,
+    durability: enum_durability_pb2.Durability,
 ) -> str:
     match durability:
-        case engine_common_pb2.Durability.ASYNC:
+        case enum_durability_pb2.Durability.ASYNC:
             return "async"
-        case engine_common_pb2.Durability.SYNC:
+        case enum_durability_pb2.Durability.SYNC:
             return "sync"
-        case engine_common_pb2.Durability.EXIT:
+        case enum_durability_pb2.Durability.EXIT:
             return "exit"
         case _:
             raise ValueError(f"invalid durability: {durability}")