langgraph-api 0.2.113__py3-none-any.whl → 0.2.114__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.113"
+__version__ = "0.2.114"

langgraph_api/api/assistants.py

@@ -17,7 +17,7 @@ from langgraph_api.graph import get_assistant_id, get_graph
 from langgraph_api.js.base import BaseRemotePregel
 from langgraph_api.route import ApiRequest, ApiResponse, ApiRoute
 from langgraph_api.serde import ajson_loads
-from langgraph_api.utils import fetchone, validate_uuid
+from langgraph_api.utils import fetchone, get_pagination_headers, validate_uuid
 from langgraph_api.validation import (
     AssistantCreate,
     AssistantPatch,
@@ -172,20 +172,21 @@ async def search_assistants(
 ) -> ApiResponse:
     """List assistants."""
     payload = await request.json(AssistantSearchRequest)
+    offset = int(payload.get("offset") or 0)
     async with connect() as conn:
-        assistants_iter, total = await Assistants.search(
+        assistants_iter, next_offset = await Assistants.search(
             conn,
             graph_id=payload.get("graph_id"),
             metadata=payload.get("metadata"),
             limit=int(payload.get("limit") or 10),
-            offset=int(payload.get("offset") or 0),
+            offset=offset,
             sort_by=payload.get("sort_by"),
             sort_order=payload.get("sort_order"),
         )
-    return ApiResponse(
-        [assistant async for assistant in assistants_iter],
-        headers={"X-Pagination-Total": str(total)},
+    assistants, response_headers = await get_pagination_headers(
+        assistants_iter, next_offset, offset
     )
+    return ApiResponse(assistants, headers=response_headers)
 
 
 @retry_db

langgraph_api/api/runs.py

@@ -12,7 +12,7 @@ from langgraph_api.asyncio import ValueEvent, aclosing
 from langgraph_api.models.run import create_valid_run
 from langgraph_api.route import ApiRequest, ApiResponse, ApiRoute
 from langgraph_api.sse import EventSourceResponse
-from langgraph_api.utils import fetchone, validate_uuid
+from langgraph_api.utils import fetchone, get_pagination_headers, validate_uuid
 from langgraph_api.validation import (
     CronCreate,
     CronSearch,
@@ -568,20 +568,21 @@ async def search_crons(request: ApiRequest):
     if thread_id := payload.get("thread_id"):
         validate_uuid(thread_id, "Invalid thread ID: must be a UUID")
 
+    offset = int(payload.get("offset", 0))
     async with connect() as conn:
-        crons_iter, total = await Crons.search(
+        crons_iter, next_offset = await Crons.search(
             conn,
             assistant_id=assistant_id,
             thread_id=thread_id,
             limit=int(payload.get("limit", 10)),
-            offset=int(payload.get("offset", 0)),
+            offset=offset,
             sort_by=payload.get("sort_by"),
             sort_order=payload.get("sort_order"),
         )
-    return ApiResponse(
-        [cron async for cron in crons_iter],
-        headers={"X-Pagination-Total": str(total)},
+    crons, response_headers = await get_pagination_headers(
+        crons_iter, next_offset, offset
     )
+    return ApiResponse(crons, headers=response_headers)
 
 
 runs_routes = [

langgraph_api/api/threads.py

@@ -6,7 +6,7 @@ from starlette.routing import BaseRoute
 
 from langgraph_api.route import ApiRequest, ApiResponse, ApiRoute
 from langgraph_api.state import state_snapshot_to_thread_state
-from langgraph_api.utils import fetchone, validate_uuid
+from langgraph_api.utils import fetchone, get_pagination_headers, validate_uuid
 from langgraph_api.validation import (
     ThreadCreate,
     ThreadPatch,
@@ -61,7 +61,7 @@ async def search_threads(
     limit = int(payload.get("limit") or 10)
     offset = int(payload.get("offset") or 0)
     async with connect() as conn:
-        iter, next_offset = await Threads.search(
+        threads_iter, next_offset = await Threads.search(
             conn,
             status=payload.get("status"),
             values=payload.get("values"),
@@ -71,15 +71,9 @@ async def search_threads(
             sort_by=payload.get("sort_by"),
             sort_order=payload.get("sort_order"),
         )
-    threads = [thread async for thread in iter]
-    if next_offset is None:
-        response_headers = {"X-Pagination-Total": str(len(threads) + offset)}
-    else:
-        cur = str(next_offset)
-        response_headers = {
-            "X-Pagination-Total": cur,
-            "X-Pagination-Next": cur,
-        }
+    threads, response_headers = await get_pagination_headers(
+        threads_iter, next_offset, offset
+    )
     return ApiResponse(threads, headers=response_headers)
 
 

langgraph_api/server.py

@@ -71,7 +71,7 @@ middleware.extend(
                 allow_credentials=True,
                 allow_methods=["*"],
                 allow_headers=["*"],
-                expose_headers=["x-pagination-total"],
+                expose_headers=["x-pagination-total", "x-pagination-next"],
             )
             if config.CORS_CONFIG is None
             else Middleware(

langgraph_api/utils/__init__.py

@@ -1,6 +1,6 @@
 import contextvars
 import uuid
-from collections.abc import AsyncIterator
+from collections.abc import AsyncGenerator, AsyncIterator
 from contextlib import asynccontextmanager
 from datetime import datetime
 from typing import Any, Protocol, TypeAlias, TypeVar
@@ -127,3 +127,19 @@ class SchemaGenerator(BaseSchemaGenerator):
             schema["paths"][endpoint.path][endpoint.http_method] = parsed
 
         return schema
+
+
+async def get_pagination_headers(
+    resource: AsyncGenerator[T],
+    next_offset: int | None,
+    offset: int,
+) -> tuple[list[T], dict[str, str]]:
+    resources = [r async for r in resource]
+    if next_offset is None:
+        response_headers = {"X-Pagination-Total": str(len(resources) + offset)}
+    else:
+        response_headers = {
+            "X-Pagination-Total": str(next_offset + 1),  # Next offset will be "n"
+            "X-Pagination-Next": str(next_offset),
+        }
+    return resources, response_headers

{langgraph_api-0.2.113.dist-info → langgraph_api-0.2.114.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.2.113
+Version: 0.2.114
 Author-email: Nuno Campos <nuno@langchain.dev>, Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE
@@ -11,7 +11,7 @@ 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>=2.0.23
-Requires-Dist: langgraph-runtime-inmem<0.7,>=0.6.7
+Requires-Dist: langgraph-runtime-inmem<0.7,>=0.6.8
 Requires-Dist: langgraph-sdk>=0.2.0
 Requires-Dist: langgraph>=0.4.0
 Requires-Dist: langsmith>=0.3.45

{langgraph_api-0.2.113.dist-info → langgraph_api-0.2.114.dist-info}/RECORD

@@ -1,4 +1,4 @@
-langgraph_api/__init__.py,sha256=YuoNAr2ZnhYxr-DSsTTZheKFrr_w-5RjlSY_cdQOZ2A,24
+langgraph_api/__init__.py,sha256=FvhaSOAw0OQ1NvIUlXa-kO_z2gwtORe4nLr3ouqI-m8,24
 langgraph_api/asgi_transport.py,sha256=eqifhHxNnxvI7jJqrY1_8RjL4Fp9NdN4prEub2FWBt8,5091
 langgraph_api/asyncio.py,sha256=Wv4Rwm-a-Cf6JpfgJmVuVlXQ7SlwrjbTn0eq1ux8I2Q,9652
 langgraph_api/cli.py,sha256=xQojITwmmKSJw48Lr2regcnRPRq2FJqWlPpeyr5TgbU,16158
@@ -17,7 +17,7 @@ langgraph_api/queue_entrypoint.py,sha256=hC8j-A4cUxibusiiPJBlK0mkmChNZxNcXn5GVwL
 langgraph_api/route.py,sha256=4VBkJMeusfiZtLzyUaKm1HwLHTq0g15y2CRiRhM6xyA,4773
 langgraph_api/schema.py,sha256=WoA7uu1VA_ZRVSR4gpAnm1n7sKDfWUEojx7CFDiOH7Q,6341
 langgraph_api/serde.py,sha256=0ALETUn582vNF-m0l_WOZGF_scL1VPA39fDkwMJQPrg,5187
-langgraph_api/server.py,sha256=Z_VL-kIphybTRDWBIqHMfRhgCmAFyTRqAGlgnHQF0Zg,6973
+langgraph_api/server.py,sha256=KBnMFt3f9RVLVu_NqyeRc13D_Lq62Rk_2czZKEUMU5E,6994
 langgraph_api/sse.py,sha256=SLdtZmTdh5D8fbWrQjuY9HYLd2dg8Rmi6ZMmFMVc2iE,4204
 langgraph_api/state.py,sha256=P2mCo-0bqPu2v9FSFGJtUCjPPNvv6wLUKQh8SdxAtc8,4387
 langgraph_api/store.py,sha256=srRI0fQXNFo_RSUs4apucr4BEp_KrIseJksZXs32MlQ,4635
@@ -29,13 +29,13 @@ langgraph_api/validation.py,sha256=zMuKmwUEBjBgFMwAaeLZmatwGVijKv2sOYtYg7gfRtc,4
 langgraph_api/webhook.py,sha256=VCJp4dI5E1oSJ15XP34cnPiOi8Ya8Q1BnBwVGadOpLI,1636
 langgraph_api/worker.py,sha256=LVvjvigurlDgpNjFcbAvRH7744fE01Lirrg2ZlHtORE,14245
 langgraph_api/api/__init__.py,sha256=WHy6oNLWtH1K7AxmmsU9RD-Vm6WP-Ov16xS8Ey9YCmQ,6090
-langgraph_api/api/assistants.py,sha256=e6da5aYMVAxj2rZG9RxZbBKFM0bP_f8Xbx0P5UhK8Y8,15924
+langgraph_api/api/assistants.py,sha256=ecHaID71ReTAZF4qsJzDe5L-2T5iOL2v8p6kQVHLKFk,16009
 langgraph_api/api/mcp.py,sha256=qe10ZRMN3f-Hli-9TI8nbQyWvMeBb72YB1PZVbyqBQw,14418
 langgraph_api/api/meta.py,sha256=fmc7btbtl5KVlU_vQ3Bj4J861IjlqmjBKNtnxSV-S-Q,4198
 langgraph_api/api/openapi.py,sha256=KToI2glOEsvrhDpwdScdBnL9xoLOqkTxx5zKq2pMuKQ,11957
-langgraph_api/api/runs.py,sha256=whjpK5Kn3nQy9g9qQK94F_rTlX4oG65WODUyj5TSOwM,20877
+langgraph_api/api/runs.py,sha256=iPzPen-_UMikGuRtg4xiageZC6eKwsmsMnqpnGdbucY,20962
 langgraph_api/api/store.py,sha256=TSeMiuMfrifmEnEbL0aObC2DPeseLlmZvAMaMzPgG3Y,5535
-langgraph_api/api/threads.py,sha256=NvWQ7j9XMJjOnist2XpNg3rheUhHCiqacC7MsSXFpEQ,9895
+langgraph_api/api/threads.py,sha256=haV1VkROLhWz4NZpO28ncFog-tbv7_iWqE1I8ZrQiDs,9726
 langgraph_api/api/ui.py,sha256=17QrRy2XVzP7x_0RdRw7pmSv-n1lmnb54byHCGGeNhM,2490
 langgraph_api/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/auth/custom.py,sha256=ZtNSQ4hIldbd2HvRsilwKzN_hjCWIiIOHClmYyPi8FM,22206
@@ -76,7 +76,7 @@ langgraph_api/middleware/request_id.py,sha256=SDj3Yi3WvTbFQ2ewrPQBjAV8sYReOJGeIi
 langgraph_api/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/models/run.py,sha256=RX2LaE1kmTruX8o8HgvqeEt5YpPGHILWByChjMZVJ58,15035
 langgraph_api/tunneling/cloudflare.py,sha256=iKb6tj-VWPlDchHFjuQyep2Dpb-w2NGfJKt-WJG9LH0,3650
-langgraph_api/utils/__init__.py,sha256=92mSti9GfGdMRRWyESKQW5yV-75Z9icGHnIrBYvdypU,3619
+langgraph_api/utils/__init__.py,sha256=EQu0PShwHhxUI_9mDFgqlAf5_y5bX8TEk723P5iby24,4161
 langgraph_api/utils/cache.py,sha256=SrtIWYibbrNeZzLXLUGBFhJPkMVNQnVxR5giiYGHEfI,1810
 langgraph_api/utils/config.py,sha256=gONI0UsoSpuR72D9lSGAmpr-_iSMDFdD4M_tiXXjmNk,3936
 langgraph_api/utils/future.py,sha256=CGhUb_Ht4_CnTuXc2kI8evEn1gnMKYN0ce9ZyUkW5G4,7251
@@ -94,8 +94,8 @@ langgraph_runtime/store.py,sha256=7mowndlsIroGHv3NpTSOZDJR0lCuaYMBoTnTrewjslw,11
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=jyQZW5U4V15zWciiIvaDPasYZd3k1iMiQ2vkPxf3zb4,145614
-langgraph_api-0.2.113.dist-info/METADATA,sha256=KYzCdutNJiyluoAZUY-ff1_tokxpTsMK-fK_KzDdkHA,3890
-langgraph_api-0.2.113.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langgraph_api-0.2.113.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
-langgraph_api-0.2.113.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.2.113.dist-info/RECORD,,
+langgraph_api-0.2.114.dist-info/METADATA,sha256=vUC4EP4w8Mv9FFd6hsoODoI0ofHfjNmQ677f6sYRYpI,3890
+langgraph_api-0.2.114.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langgraph_api-0.2.114.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
+langgraph_api-0.2.114.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.2.114.dist-info/RECORD,,