langgraph-api 0.2.113__tar.gz → 0.2.114__tar.gz

{langgraph_api-0.2.113 → langgraph_api-0.2.114}/PKG-INFO

@@ -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.114/langgraph_api/__init__.py

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

{langgraph_api-0.2.113 → langgraph_api-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-0.2.113 → langgraph_api-0.2.114}/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-0.2.113 → langgraph_api-0.2.114}/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-0.2.113 → langgraph_api-0.2.114}/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-0.2.113 → langgraph_api-0.2.114}/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 → langgraph_api-0.2.114}/pyproject.toml

@@ -31,7 +31,7 @@ dependencies = [
     "cryptography>=42.0.0,<45.0",
     "langgraph-sdk>=0.2.0",
     "cloudpickle>=3.0.0",
-    "langgraph-runtime-inmem>=0.6.7,<0.7",
+    "langgraph-runtime-inmem>=0.6.8,<0.7",
     "truststore>=0.1",
 ]
 

langgraph_api-0.2.113/langgraph_api/__init__.py

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