langgraph-api 0.2.68__py3-none-any.whl → 0.2.70__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.68"
+__version__ = "0.2.70"

langgraph_api/api/runs.py

@@ -548,14 +548,19 @@ async def search_crons(request: ApiRequest):
         validate_uuid(thread_id, "Invalid thread ID: must be a UUID")
 
     async with connect() as conn:
-        crons_iter = await Crons.search(
+        crons_iter, total = await Crons.search(
             conn,
             assistant_id=assistant_id,
             thread_id=thread_id,
             limit=int(payload.get("limit", 10)),
             offset=int(payload.get("offset", 0)),
+            sort_by=payload.get("sort_by"),
+            sort_order=payload.get("sort_order"),
         )
-    return ApiResponse([cron async for cron in crons_iter])
+    return ApiResponse(
+        [cron async for cron in crons_iter],
+        headers={"X-Pagination-Total": str(total)},
+    )
 
 
 runs_routes = [

langgraph_api/errors.py

@@ -50,3 +50,9 @@ class UserInterrupt(Exception):
 class UserRollback(UserInterrupt):
     def __init__(self):
         super().__init__("User requested rollback of the run")
+
+
+class UserTimeout(Exception):
+    def __init__(self, timeout_error: TimeoutError):
+        super().__init__(timeout_error)
+        self.timeout_error = timeout_error

langgraph_api/schema.py

@@ -170,6 +170,8 @@ class Cron(TypedDict):
 
     cron_id: UUID
     """The ID of the cron."""
+    assistant_id: UUID
+    """The ID of the assistant."""
     thread_id: UUID | None
     """The ID of the thread."""
     end_time: datetime | None
@@ -180,8 +182,14 @@ class Cron(TypedDict):
     """The time the cron was created."""
     updated_at: datetime
     """The last time the cron was updated."""
+    user_id: UUID | None
+    """The ID of the user."""
     payload: Fragment
     """The run payload to use for creating new run."""
+    next_run_date: datetime
+    """The next run date of the cron."""
+    metadata: Fragment
+    """The cron metadata."""
 
 
 class ThreadUpdateResponse(TypedDict):

langgraph_api/worker.py

@@ -16,12 +16,12 @@ from langgraph_api.config import (
     BG_JOB_MAX_RETRIES,
     BG_JOB_TIMEOUT_SECS,
 )
-from langgraph_api.errors import UserInterrupt, UserRollback
+from langgraph_api.errors import UserInterrupt, UserRollback, UserTimeout
 from langgraph_api.js.errors import RemoteException
 from langgraph_api.metadata import incr_runs
 from langgraph_api.schema import Run
 from langgraph_api.state import state_snapshot_to_thread_state
-from langgraph_api.stream import astream_state, consume
+from langgraph_api.stream import AnyStream, astream_state, consume
 from langgraph_api.utils import with_user
 from langgraph_runtime.database import connect
 from langgraph_runtime.ops import Runs, Threads
@@ -79,8 +79,11 @@ async def worker(
         "__request_start_time_ms__"
     )
     after_seconds = run["kwargs"]["config"]["configurable"].get("__after_seconds__", 0)
+    run_started_at_dt = datetime.now(UTC)
+    run_started_at = run_started_at_dt.isoformat()
+    run_ended_at_dt: datetime | None = None
     run_ended_at: str | None = None
-    run_started_at = datetime.now(UTC)
+
     # Note that "created_at" is inclusive of the `after_seconds`
     run_creation_ms = (
         int(
@@ -103,13 +106,13 @@ async def worker(
             "request_id": _get_request_id(run),
         }
     )
-    run_stream_started_at = datetime.now(UTC)
+    run_stream_started_at_dt = datetime.now(UTC)
     await logger.ainfo(
         "Starting background run",
-        run_started_at=run_started_at.isoformat(),
+        run_started_at=run_started_at,
         run_creation_ms=run_creation_ms,
-        run_queue_ms=ms(run_started_at, run["created_at"]),
-        run_stream_start_ms=ms(run_stream_started_at, run_started_at),
+        run_queue_ms=ms(run_started_at_dt, run["created_at"]),
+        run_stream_start_ms=ms(run_stream_started_at_dt, run_started_at_dt),
     )
 
     def on_checkpoint(checkpoint_arg: CheckpointPayload):
@@ -123,6 +126,21 @@ async def worker(
                     task.update(task_result)
                     break
 
+    # Wrap the graph execution to separate user errors from server errors
+    async def wrap_user_errors(stream: AnyStream, run_id: str, resumable: bool):
+        try:
+            await consume(stream, run_id, resumable)
+        except Exception as e:
+            logger.info(
+                f"Run encountered an error in graph: {type(e)}({e})",
+                exc_info=e,
+            )
+            # TimeoutError is a special case where we rely on asyncio.wait_for to timeout runs
+            # Convert user TimeoutErrors to a custom class so we can distinguish and later convert back
+            if isinstance(e, TimeoutError):
+                raise UserTimeout(e) from e
+            raise
+
     async with Runs.enter(run_id, main_loop) as done:
         # attempt the run
         try:
@@ -163,25 +181,9 @@ async def worker(
                         on_task_result=on_task_result,
                     )
                 await asyncio.wait_for(
-                    consume(stream, run_id, resumable),
+                    wrap_user_errors(stream, run_id, resumable),
                     BG_JOB_TIMEOUT_SECS,
                 )
-                run_ended_at_dt = datetime.now(UTC)
-                run_ended_at = run_ended_at_dt.isoformat()
-                await logger.ainfo(
-                    "Background run succeeded",
-                    run_id=str(run_id),
-                    run_attempt=attempt,
-                    run_created_at=run_created_at,
-                    run_started_at=run_started_at.isoformat(),
-                    run_ended_at=run_ended_at,
-                    run_exec_ms=ms(run_ended_at_dt, run_started_at),
-                    run_completed_in_ms=(
-                        int((run_ended_at_dt.timestamp() * 1_000) - request_created_at)
-                        if request_created_at is not None
-                        else None
-                    ),
-                )
         except (Exception, asyncio.CancelledError) as ee:
             exception = ee
         except BaseException as eee:
@@ -192,11 +194,33 @@ async def worker(
                 exception=str(eee),
             )
             raise
+        finally:
+            run_ended_at_dt = datetime.now(UTC)
+            run_ended_at = run_ended_at_dt.isoformat()
 
         # handle exceptions and set status
         async with connect() as conn:
+            log_info = {
+                "run_id": str(run_id),
+                "run_attempt": attempt,
+                "run_created_at": run_created_at,
+                "run_started_at": run_started_at,
+                "run_ended_at": run_ended_at,
+                "run_exec_ms": ms(run_ended_at_dt, run_started_at_dt),
+                "run_completed_in_ms": (
+                    int((run_ended_at_dt.timestamp() * 1_000) - request_created_at)
+                    if request_created_at is not None
+                    else None
+                ),
+            }
+
             if exception is None:
                 status = "success"
+
+                await logger.ainfo(
+                    "Background run succeeded",
+                    **log_info,
+                )
                 # If a stateful run succeeded but no checkpoint was returned, likely
                 # there was a retriable exception that resumed right at the end
                 if checkpoint is None and not temporary:
@@ -222,56 +246,28 @@ async def worker(
                 )
             elif isinstance(exception, TimeoutError):
                 status = "timeout"
-                run_ended_at = datetime.now(UTC).isoformat()
                 await logger.awarning(
                     "Background run timed out",
-                    run_id=str(run_id),
-                    run_attempt=attempt,
-                    run_created_at=run_created_at,
-                    run_started_at=run_started_at.isoformat(),
-                    run_ended_at=run_ended_at,
-                    run_exec_ms=ms(datetime.now(UTC), run_started_at),
-                    run_completed_in_ms=(
-                        int((run_ended_at_dt.timestamp() * 1_000) - request_created_at)
-                        if request_created_at is not None
-                        else None
-                    ),
+                    **log_info,
                 )
                 await Threads.set_joint_status(
                     conn, run["thread_id"], run_id, status, checkpoint=checkpoint
                 )
             elif isinstance(exception, UserRollback):
                 status = "rollback"
-                run_ended_at_dt = datetime.now(UTC)
-                run_ended_at = run_ended_at_dt.isoformat()
                 try:
                     await Threads.set_joint_status(
                         conn, run["thread_id"], run_id, status, checkpoint=checkpoint
                     )
                     await logger.ainfo(
                         "Background run rolled back",
-                        run_id=str(run_id),
-                        run_attempt=attempt,
-                        run_created_at=run_created_at,
-                        run_started_at=run_started_at.isoformat(),
-                        run_ended_at=run_ended_at,
-                        run_exec_ms=ms(run_ended_at_dt, run_started_at),
-                        run_completed_in_ms=(
-                            int(
-                                (run_ended_at_dt.timestamp() * 1_000)
-                                - request_created_at
-                            )
-                            if request_created_at is not None
-                            else None
-                        ),
+                        **log_info,
                     )
                 except HTTPException as e:
                     if e.status_code == 404:
                         await logger.ainfo(
                             "Ignoring rollback error for missing run",
-                            run_id=str(run_id),
-                            run_attempt=attempt,
-                            run_created_at=run_created_at,
+                            **log_info,
                         )
                     else:
                         raise
@@ -279,54 +275,32 @@ async def worker(
                 checkpoint = None  # reset the checkpoint
             elif isinstance(exception, UserInterrupt):
                 status = "interrupted"
-                run_ended_at_dt = datetime.now(UTC)
-                run_ended_at = run_ended_at_dt.isoformat()
                 await logger.ainfo(
                     "Background run interrupted",
-                    run_id=str(run_id),
-                    run_attempt=attempt,
-                    run_created_at=run_created_at,
-                    run_started_at=run_started_at.isoformat(),
-                    run_ended_at=run_ended_at,
-                    run_exec_ms=ms(run_ended_at_dt, run_started_at),
-                    run_completed_in_ms=(
-                        int((run_ended_at_dt.timestamp() * 1_000) - request_created_at)
-                        if request_created_at is not None
-                        else None
-                    ),
+                    **log_info,
                 )
                 await Threads.set_joint_status(
                     conn, run["thread_id"], run_id, status, checkpoint, exception
                 )
             elif isinstance(exception, ALL_RETRIABLE_EXCEPTIONS):
                 status = "retry"
-                run_ended_at_dt = datetime.now(UTC)
-                run_ended_at = run_ended_at_dt.isoformat()
                 await logger.awarning(
                     f"Background run failed, will retry. Exception: {type(exception)}({exception})",
-                    exc_info=True,
-                    run_id=str(run_id),
-                    run_attempt=attempt,
-                    run_created_at=run_created_at,
-                    run_started_at=run_started_at.isoformat(),
-                    run_ended_at=run_ended_at,
-                    run_exec_ms=ms(run_ended_at_dt, run_started_at),
+                    **log_info,
                 )
                 # Don't update thread status yet.
                 await Runs.set_status(conn, run_id, "pending")
             else:
                 status = "error"
-                run_ended_at_dt = datetime.now(UTC)
-                run_ended_at = run_ended_at_dt.isoformat()
+
+                # Convert UserTimeout to TimeoutError for customers
+                if isinstance(exception, UserTimeout):
+                    exception = exception.timeout_error
+
                 await logger.aexception(
                     f"Background run failed. Exception: {type(exception)}({exception})",
                     exc_info=not isinstance(exception, RemoteException),
-                    run_id=str(run_id),
-                    run_attempt=attempt,
-                    run_created_at=run_created_at,
-                    run_started_at=run_started_at.isoformat(),
-                    run_ended_at=run_ended_at,
-                    run_exec_ms=ms(run_ended_at_dt, run_started_at),
+                    **log_info,
                 )
                 await Threads.set_joint_status(
                     conn, run["thread_id"], run_id, status, checkpoint, exception
@@ -363,7 +337,7 @@ async def worker(
         exception=exception,
         run=run,
         webhook=webhook,
-        run_started_at=run_started_at.isoformat(),
+        run_started_at=run_started_at,
         run_ended_at=run_ended_at,
     )
 

{langgraph_api-0.2.68.dist-info → langgraph_api-0.2.70.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.2.68
+Version: 0.2.70
 Author-email: Nuno Campos <nuno@langchain.dev>, Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE
@@ -12,7 +12,7 @@ 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.4,>=0.3.0
-Requires-Dist: langgraph-sdk>=0.1.66
+Requires-Dist: langgraph-sdk>=0.1.71
 Requires-Dist: langgraph>=0.3.27
 Requires-Dist: langsmith>=0.3.45
 Requires-Dist: orjson>=3.9.7

{langgraph_api-0.2.68.dist-info → langgraph_api-0.2.70.dist-info}/RECORD

@@ -1,11 +1,11 @@
-langgraph_api/__init__.py,sha256=VhcIWUc8CMbglTSx2PRXJQXLxGPIWajEYdegulJBOPk,23
+langgraph_api/__init__.py,sha256=nvBmUT-8UzEzR76L42ZDv_0mtdkVtoC42_s4j3hsMkk,23
 langgraph_api/asgi_transport.py,sha256=eqifhHxNnxvI7jJqrY1_8RjL4Fp9NdN4prEub2FWBt8,5091
 langgraph_api/asyncio.py,sha256=Odnc6mAJIGF3eFWT8Xcrg2Zam7FwzXkfCWEHaXfrzQQ,9371
 langgraph_api/cli.py,sha256=13mKb-WT7fGx_yqcbWITPB9ICEHCrPzIP1ddZ5RbXbY,16015
 langgraph_api/command.py,sha256=3O9v3i0OPa96ARyJ_oJbLXkfO8rPgDhLCswgO9koTFA,768
 langgraph_api/config.py,sha256=jmtO5LXubet2Hl7S2tF2BRyB_q2yqCzFqCwN7n6za4Y,11727
 langgraph_api/cron_scheduler.py,sha256=i87j4pJrcsmsqMKeKUs69gaAjrGaSM3pM3jnXdN5JDQ,2630
-langgraph_api/errors.py,sha256=Bu_i5drgNTyJcLiyrwVE_6-XrSU50BHf9TDpttki9wQ,1690
+langgraph_api/errors.py,sha256=zlnl3xXIwVG0oGNKKpXf1an9Rn_SBDHSyhe53hU6aLw,1858
 langgraph_api/graph.py,sha256=t18rq0NHHRILUI5cONhKPzKhbqLGn6_BRytiEh0gdcg,23561
 langgraph_api/http.py,sha256=gYbxxjY8aLnsXeJymcJ7G7Nj_yToOGpPYQqmZ1_ggfA,5240
 langgraph_api/logging.py,sha256=Do62kNJtY7Ns7q7XrP2VIxho3c85jt8_l3yfRtqM7xM,4370
@@ -13,7 +13,7 @@ langgraph_api/metadata.py,sha256=Gx0b6YszLRjdWLDVN8OcVgC_YYQG_nQitPfUfgQx1w8,464
 langgraph_api/patch.py,sha256=Dgs0PXHytekX4SUL6KsjjN0hHcOtGLvv1GRGbh6PswU,1408
 langgraph_api/queue_entrypoint.py,sha256=hC8j-A4cUxibusiiPJBlK0mkmChNZxNcXn5GVwL0yic,4889
 langgraph_api/route.py,sha256=4VBkJMeusfiZtLzyUaKm1HwLHTq0g15y2CRiRhM6xyA,4773
-langgraph_api/schema.py,sha256=2711t4PIBk5dky4gmMndrTRC9CVvAgH47C9FKDxhkBo,5444
+langgraph_api/schema.py,sha256=a6it0h9ku4jrTXiW9MhnGok_wignyQ4cXBra67FiryM,5678
 langgraph_api/serde.py,sha256=8fQXg7T7RVUqj_jgOoSOJrWVpQDW0qJKjAjSsEhPHo4,4803
 langgraph_api/server.py,sha256=Z_VL-kIphybTRDWBIqHMfRhgCmAFyTRqAGlgnHQF0Zg,6973
 langgraph_api/sse.py,sha256=F7swfjKBDrlUmXZ_dWuDVHtp-3o1Cpjq1lwp0bJD-nw,4223
@@ -24,13 +24,13 @@ langgraph_api/thread_ttl.py,sha256=-Ox8NFHqUH3wGNdEKMIfAXUubY5WGifIgCaJ7npqLgw,1
 langgraph_api/utils.py,sha256=92mSti9GfGdMRRWyESKQW5yV-75Z9icGHnIrBYvdypU,3619
 langgraph_api/validation.py,sha256=zMuKmwUEBjBgFMwAaeLZmatwGVijKv2sOYtYg7gfRtc,4950
 langgraph_api/webhook.py,sha256=1ncwO0rIZcj-Df9sxSnFEzd1gP1bfS4okeZQS8NSRoE,1382
-langgraph_api/worker.py,sha256=2LZUzcZGyK8_-ajmGHdoKhU7HT0SdBPLDLEJlBDyz2w,16021
+langgraph_api/worker.py,sha256=Cl1sS63HYngQ92vkPFef6qYp-LLIJQyExqNzBlpVl5E,14145
 langgraph_api/api/__init__.py,sha256=YVzpbn5IQotvuuLG9fhS9QMrxXfP4s4EpEMG0n4q3Nw,5625
 langgraph_api/api/assistants.py,sha256=6IPVKQBlI95-Z4nYdqBY9st9oynGJAocL67cwnDaZCk,15744
 langgraph_api/api/mcp.py,sha256=RvRYgANqRzNQzSmgjNkq4RlKTtoEJYil04ot9lsmEtE,14352
 langgraph_api/api/meta.py,sha256=MU9Ehdo2M8oaxGVBXVQFNRP6qSTXyrsGXFcndRlnvIE,3924
 langgraph_api/api/openapi.py,sha256=362m6Ny8wOwZ6HrDK9JAVUzPkyLYWKeV1E71hPOaA0U,11278
-langgraph_api/api/runs.py,sha256=9jU9C4myBhgZXyvR9MzNn9KrpNo7DvWbg8uNeWzSmKE,19524
+langgraph_api/api/runs.py,sha256=1rNUbFleakTYAqG20cGtpWyY2TS5EoVgT4Pxe1AQ68E,19692
 langgraph_api/api/store.py,sha256=TSeMiuMfrifmEnEbL0aObC2DPeseLlmZvAMaMzPgG3Y,5535
 langgraph_api/api/threads.py,sha256=ogMKmEoiycuaV3fa5kpupDohJ7fwUOfVczt6-WSK4FE,9322
 langgraph_api/api/ui.py,sha256=2nlipYV2nUGR4T9pceaAbgN1lS3-T2zPBh7Nv3j9eZQ,2479
@@ -85,9 +85,9 @@ langgraph_runtime/retry.py,sha256=V0duD01fO7GUQ_btQkp1aoXcEOFhXooGVP6q4yMfuyY,11
 langgraph_runtime/store.py,sha256=7mowndlsIroGHv3NpTSOZDJR0lCuaYMBoTnTrewjslw,114
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
-openapi.json,sha256=J4ICw3-W3LdxQOfsTuzcZVYt3e9UFeLb6oPr_a-5iHU,141823
-langgraph_api-0.2.68.dist-info/METADATA,sha256=tMf14kWcDmBYEZlH7COkzGx7xq6WD4g1-9m3c7d0AH8,3891
-langgraph_api-0.2.68.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langgraph_api-0.2.68.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
-langgraph_api-0.2.68.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.2.68.dist-info/RECORD,,
+openapi.json,sha256=NVgY0hdCu6v5anNus9I-CVYKKcrDLKd1qhILy-kQpq8,142590
+langgraph_api-0.2.70.dist-info/METADATA,sha256=nbh82X9tCCJpITWyy2Ex8uGZ4WhQHoaF0rly6mvjQkU,3891
+langgraph_api-0.2.70.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langgraph_api-0.2.70.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
+langgraph_api-0.2.70.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.2.70.dist-info/RECORD,,

openapi.json

@@ -3243,6 +3243,12 @@
             "title": "Schedule",
             "description": "The cron schedule to execute this job on."
           },
+          "end_time": {
+            "type": "string",
+            "format": "date-time",
+            "title": "End Time",
+            "description": "The end date to stop running the cron."
+          },
           "assistant_id": {
             "anyOf": [
               {
@@ -3392,6 +3398,20 @@
             "description": "The number of results to skip.",
             "default": 0,
             "minimum": 0
+          },
+          "sort_by": {
+            "type": "string",
+            "title": "Sort By",
+            "description": "The field to sort by.",
+            "default": "created_at",
+            "enum": ["cron_id", "assistant_id", "thread_id", "next_run_date", "end_time", "created_at", "updated_at"]
+          },
+          "sort_order": {
+            "type": "string",
+            "title": "Sort Order",
+            "description": "The order to sort by.",
+            "default": "desc",
+            "enum": ["asc", "desc"]
           }
         },
         "type": "object",
@@ -3513,13 +3533,14 @@
             "type": "string",
             "enum": [
               "pending",
+              "running",
               "error",
               "success",
               "timeout",
               "interrupted"
             ],
             "title": "Status",
-            "description": "The status of the run. One of 'pending', 'error', 'success', 'timeout', 'interrupted'."
+            "description": "The status of the run. One of 'pending', 'running', 'error', 'success', 'timeout', 'interrupted'."
           },
           "metadata": {
             "type": "object",