langgraph-api 0.2.11__py3-none-any.whl → 0.2.12__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.11"
+__version__ = "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/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/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.11.dist-info → langgraph_api-0.2.12.dist-info}/METADATA

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

{langgraph_api-0.2.11.dist-info → langgraph_api-0.2.12.dist-info}/RECORD

@@ -1,11 +1,11 @@
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api/__init__.py,sha256=_MLx4ac1juJPWEEiC9kMQISX3x3jFBr507jM2P_hxMg,23
+langgraph_api/__init__.py,sha256=X4KG3FscE5AhbGbcdDDgdDC550CVpxNMwdNLcx6EQ7M,23
 langgraph_api/api/__init__.py,sha256=YVzpbn5IQotvuuLG9fhS9QMrxXfP4s4EpEMG0n4q3Nw,5625
 langgraph_api/api/assistants.py,sha256=mcKaVeNG8hQAV4IQDhaukS7FgVqTIVQNTyti3GfK2KI,14649
 langgraph_api/api/mcp.py,sha256=RvRYgANqRzNQzSmgjNkq4RlKTtoEJYil04ot9lsmEtE,14352
 langgraph_api/api/meta.py,sha256=sTgkhE-DaFWpERG6F7KelZfDsmJAiVc4j5dg50tDkSo,2950
 langgraph_api/api/openapi.py,sha256=WYpkjwsVV056qBkBVbGjhh-TfU9-IgZMY2au1s1PhZ4,15953
-langgraph_api/api/runs.py,sha256=dhHZ3xu7V0anftqzXaOYnhVEryJpVeqzu60MFiUw4u8,18010
+langgraph_api/api/runs.py,sha256=ys8X3g2EGbuF_OF1htM4jcLu4Mqztd8v7ttosmIbdsw,17972
 langgraph_api/api/store.py,sha256=G4Fm8hgFLlJUW5_ekLS_IOgCfpFy-TK9uq5r9QTOELQ,5447
 langgraph_api/api/threads.py,sha256=ogMKmEoiycuaV3fa5kpupDohJ7fwUOfVczt6-WSK4FE,9322
 langgraph_api/api/ui.py,sha256=2nlipYV2nUGR4T9pceaAbgN1lS3-T2zPBh7Nv3j9eZQ,2479
@@ -20,7 +20,7 @@ langgraph_api/auth/noop.py,sha256=Bk6Nf3p8D_iMVy_OyfPlyiJp_aEwzL-sHrbxoXpCbac,58
 langgraph_api/auth/studio_user.py,sha256=FzFQRROKDlA9JjtBuwyZvk6Mbwno5M9RVYjDO6FU3F8,186
 langgraph_api/cli.py,sha256=9Ou3tGDDY_VVLt5DFle8UviJdpI4ZigC5hElYvq2-To,14519
 langgraph_api/command.py,sha256=3O9v3i0OPa96ARyJ_oJbLXkfO8rPgDhLCswgO9koTFA,768
-langgraph_api/config.py,sha256=KXQ3W0LzGuEqc0kfB2B37tvqFBwl66rW7vXtw1gIZGc,10773
+langgraph_api/config.py,sha256=Jsv1Qw1esmO1hllo2_0UhDtcIxOAmF38-8xutpxDcw4,10862
 langgraph_api/cron_scheduler.py,sha256=i87j4pJrcsmsqMKeKUs69gaAjrGaSM3pM3jnXdN5JDQ,2630
 langgraph_api/errors.py,sha256=Bu_i5drgNTyJcLiyrwVE_6-XrSU50BHf9TDpttki9wQ,1690
 langgraph_api/graph.py,sha256=YyWCPtoI9VDV0knjCMUFoH4r9OFVsAiv5K8FzbziMqs,21488
@@ -78,7 +78,7 @@ langgraph_api/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 langgraph_api/models/run.py,sha256=hPiiP6jgJX-jsg4Aihwi0tpC-nkTabV9YE0WNlD4EjI,13166
 langgraph_api/patch.py,sha256=Dgs0PXHytekX4SUL6KsjjN0hHcOtGLvv1GRGbh6PswU,1408
 langgraph_api/queue_entrypoint.py,sha256=gjtajZfnDXhsi7JElfmkY-p0ENBiKBDJ4Ugiw-exapw,1839
-langgraph_api/route.py,sha256=fM4qYCGbmH0a3_cV8uKocb1sLklehxO6HhdRXqLK6OM,4421
+langgraph_api/route.py,sha256=uN311KjIugyNHG3rmVw_ms61QO1W1l16jJx03rf0R_s,4630
 langgraph_api/schema.py,sha256=2711t4PIBk5dky4gmMndrTRC9CVvAgH47C9FKDxhkBo,5444
 langgraph_api/serde.py,sha256=TVsx2QQtepf8Wsgsabcku1NV4Vbugu4Oujmdnq4qMS0,3964
 langgraph_api/server.py,sha256=lCv_ZHLbMnhKRhdaSJi1edk-O9We-MnQafkGzdWlngw,6599
@@ -96,8 +96,8 @@ langgraph_license/validation.py,sha256=ZKraAVJArAABKqrmHN-EN18ncoNUmRm500Yt1Sc7t
 langgraph_runtime/__init__.py,sha256=O4GgSmu33c-Pr8Xzxj_brcK5vkm70iNTcyxEjICFZxA,1075
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=GefWJwBrbrN_jNDAEFlwNEXx1ottRtvjOmKBi965KLU,133756
-langgraph_api-0.2.11.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.2.11.dist-info/METADATA,sha256=Viok-PPckUyL3Gq-DZo_zpO5yCDkdN8bC97j8ZGPvQg,4241
-langgraph_api-0.2.11.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-langgraph_api-0.2.11.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.2.11.dist-info/RECORD,,
+langgraph_api-0.2.12.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.2.12.dist-info/METADATA,sha256=Eu9KpsvHFh84m1dIz0C5-QjU5B_IdKHOgpqPV691TqM,4241
+langgraph_api-0.2.12.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+langgraph_api-0.2.12.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.2.12.dist-info/RECORD,,