langgraph-api 0.2.10__tar.gz → 0.2.12__tar.gz

{langgraph_api-0.2.10 → langgraph_api-0.2.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langgraph-api
-Version: 0.2.10
+Version: 0.2.12
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos

langgraph_api-0.2.12/langgraph_api/__init__.py

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

{langgraph_api-0.2.10 → langgraph_api-0.2.12}/langgraph_api/api/runs.py

@@ -285,7 +285,7 @@ async def wait_run_stateless(request: ApiRequest):
 
 
 @retry_db
-async def list_runs_http(
+async def list_runs(
     request: ApiRequest,
 ):
     """List all runs for a thread."""
@@ -312,7 +312,7 @@ async def list_runs_http(
 
 
 @retry_db
-async def get_run_http(request: ApiRequest):
+async def get_run(request: ApiRequest):
     """Get a run by ID."""
     thread_id = request.path_params["thread_id"]
     run_id = request.path_params["run_id"]
@@ -349,7 +349,7 @@ async def join_run(request: ApiRequest):
 
 
 @retry_db
-async def join_run_stream_endpoint(request: ApiRequest):
+async def join_run_stream(request: ApiRequest):
     """Wait for a run to finish."""
     thread_id = request.path_params["thread_id"]
     run_id = request.path_params["run_id"]
@@ -541,11 +541,11 @@ runs_routes = [
     ApiRoute("/threads/{thread_id}/runs/{run_id}/join", join_run, methods=["GET"]),
     ApiRoute(
         "/threads/{thread_id}/runs/{run_id}/stream",
-        join_run_stream_endpoint,
+        join_run_stream,
         methods=["GET"],
     ),
     ApiRoute("/threads/{thread_id}/runs/{run_id}/cancel", cancel_run, methods=["POST"]),
-    ApiRoute("/threads/{thread_id}/runs/{run_id}", get_run_http, methods=["GET"]),
+    ApiRoute("/threads/{thread_id}/runs/{run_id}", get_run, methods=["GET"]),
     ApiRoute("/threads/{thread_id}/runs/{run_id}", delete_run, methods=["DELETE"]),
     ApiRoute("/threads/{thread_id}/runs/stream", stream_run, methods=["POST"]),
     ApiRoute("/threads/{thread_id}/runs/wait", wait_run, methods=["POST"]),
@@ -557,7 +557,7 @@ runs_routes = [
         if config.FF_CRONS_ENABLED and plus_features_enabled()
         else None
     ),
-    ApiRoute("/threads/{thread_id}/runs", list_runs_http, methods=["GET"]),
+    ApiRoute("/threads/{thread_id}/runs", list_runs, methods=["GET"]),
     (
         ApiRoute("/runs/crons/{cron_id}", delete_cron, methods=["DELETE"])
         if config.FF_CRONS_ENABLED and plus_features_enabled()

{langgraph_api-0.2.10 → langgraph_api-0.2.12}/langgraph_api/config.py

@@ -148,6 +148,8 @@ STATS_INTERVAL_SECS = env("STATS_INTERVAL_SECS", cast=int, default=60)
 DATABASE_URI = env("DATABASE_URI", cast=str, default=getenv("POSTGRES_URI", undefined))
 MIGRATIONS_PATH = env("MIGRATIONS_PATH", cast=str, default="/storage/migrations")
 
+POSTGRES_POOL_MAX_SIZE = env("LANGGRAPH_POSTGRES_POOL_MAX_SIZE", cast=int, default=150)
+
 
 def _get_encryption_key(key_str: str | None):
     if not key_str:

{langgraph_api-0.2.10 → langgraph_api-0.2.12}/langgraph_api/route.py

@@ -9,16 +9,25 @@ from starlette._utils import is_async_callable
 from starlette.concurrency import run_in_threadpool
 from starlette.middleware import Middleware
 from starlette.requests import Request
-from starlette.responses import JSONResponse, Response
+from starlette.responses import JSONResponse
 from starlette.routing import Route, compile_path, get_name
 from starlette.types import ASGIApp, Receive, Scope, Send
 
 from langgraph_api.serde import json_dumpb
 from langgraph_api.utils import get_auth_ctx, with_user
 
+SchemaType = (
+    jsonschema_rs.Draft4Validator
+    | jsonschema_rs.Draft6Validator
+    | jsonschema_rs.Draft7Validator
+    | jsonschema_rs.Draft201909Validator
+    | jsonschema_rs.Draft202012Validator
+    | None
+)
+
 
 def api_request_response(
-    func: typing.Callable[[Request], typing.Awaitable[Response] | Response],
+    func: typing.Callable[[Request], typing.Awaitable[ASGIApp] | ASGIApp],
 ) -> ASGIApp:
     """
     Takes a function or coroutine `func(request) -> response`,
@@ -30,9 +39,11 @@ def api_request_response(
 
         async def app(scope: Scope, receive: Receive, send: Send) -> None:
             if is_async_callable(func):
-                response = await func(request)
+                response: ASGIApp = await func(request)
             else:
-                response = await run_in_threadpool(func, request)
+                response = await run_in_threadpool(
+                    typing.cast(typing.Callable[[Request], ASGIApp], func), request
+                )
             await response(scope, receive, send)
 
         await wrap_app_handling_exceptions(app, request)(scope, receive, send)
@@ -45,9 +56,7 @@ class ApiResponse(JSONResponse):
         return json_dumpb(content)
 
 
-def _json_loads(
-    content: bytearray, schema: jsonschema_rs.Draft4Validator | None
-) -> typing.Any:
+def _json_loads(content: bytearray, schema: SchemaType) -> typing.Any:
     json = orjson.loads(content)
     if schema is not None:
         schema.validate(json)
@@ -55,7 +64,7 @@ def _json_loads(
 
 
 class ApiRequest(Request):
-    async def body(self) -> typing.Coroutine[typing.Any, typing.Any, bytearray]:
+    async def body(self) -> bytearray:
         if not hasattr(self, "_body"):
             chunks = bytearray()
             async for chunk in self.stream():
@@ -63,7 +72,7 @@ class ApiRequest(Request):
             self._body = chunks
         return self._body
 
-    async def json(self, schema: jsonschema_rs.Draft4Validator | None) -> typing.Any:
+    async def json(self, schema: SchemaType = None) -> typing.Any:
         if not hasattr(self, "_json"):
             body = await self.body()
             self._json = await run_in_threadpool(_json_loads, body, schema)

{langgraph_api-0.2.10 → langgraph_api-0.2.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.2.10"
+version = "0.2.12"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",

langgraph_api-0.2.10/langgraph_api/__init__.py

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