langgraph-sdk 0.3.2__tar.gz → 0.3.4__tar.gz

{langgraph_sdk-0.3.2 → langgraph_sdk-0.3.4}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: langgraph-sdk
-Version: 0.3.2
+Version: 0.3.4
 Summary: SDK for interacting with LangGraph API
 Project-URL: Source, https://github.com/langchain-ai/langgraph/tree/main/libs/sdk-py
-Project-URL: Twitter, https://x.com/LangChainAI
+Project-URL: Twitter, https://x.com/LangChain
 Project-URL: Slack, https://www.langchain.com/join-community
 Project-URL: Reddit, https://www.reddit.com/r/LangChain/
 License-Expression: MIT

{langgraph_sdk-0.3.2 → langgraph_sdk-0.3.4}/langgraph_sdk/__init__.py

@@ -3,6 +3,6 @@ from langgraph_sdk.client import get_client, get_sync_client
 from langgraph_sdk.encryption import Encryption
 from langgraph_sdk.encryption.types import EncryptionContext
 
-__version__ = "0.3.2"
+__version__ = "0.3.4"
 
 __all__ = ["Auth", "Encryption", "EncryptionContext", "get_client", "get_sync_client"]

{langgraph_sdk-0.3.2 → langgraph_sdk-0.3.4}/langgraph_sdk/client.py

@@ -17,6 +17,7 @@ import re
 import sys
 import warnings
 from collections.abc import AsyncIterator, Callable, Iterator, Mapping, Sequence
+from datetime import datetime
 from types import TracebackType
 from typing import (
     Any,
@@ -2982,6 +2983,8 @@ class CronClient:
         interrupt_after: All | list[str] | None = None,
         webhook: str | None = None,
         multitask_strategy: str | None = None,
+        end_time: datetime | None = None,
+        enabled: bool | None = None,
         headers: Mapping[str, str] | None = None,
         params: QueryParamTypes | None = None,
     ) -> Run:
@@ -2992,6 +2995,7 @@ class CronClient:
             assistant_id: The assistant ID or graph name to use for the cron job.
                 If using graph name, will default to first assistant created from that graph.
             schedule: The cron schedule to execute this job on.
+                Schedules are interpreted in UTC.
             input: The input to the graph.
             metadata: Metadata to assign to the cron job runs.
             config: The configuration for the assistant.
@@ -3005,6 +3009,8 @@ class CronClient:
             webhook: Webhook to call after LangGraph API call is done.
             multitask_strategy: Multitask strategy to use.
                 Must be one of 'reject', 'interrupt', 'rollback', or 'enqueue'.
+            end_time: The time to stop running the cron job. If not provided, the cron job will run indefinitely.
+            enabled: Whether the cron job is enabled or not.
             headers: Optional custom headers to include with the request.
             params: Optional query parameters to include with the request.
 
@@ -3025,7 +3031,8 @@ class CronClient:
                 interrupt_before=["node_to_stop_before_1","node_to_stop_before_2"],
                 interrupt_after=["node_to_stop_after_1","node_to_stop_after_2"],
                 webhook="https://my.fake.webhook.com",
-                multitask_strategy="interrupt"
+                multitask_strategy="interrupt",
+                enabled=True,
             )
             ```
         """
@@ -3040,6 +3047,8 @@ class CronClient:
             "interrupt_before": interrupt_before,
             "interrupt_after": interrupt_after,
             "webhook": webhook,
+            "end_time": end_time.isoformat() if end_time else None,
+            "enabled": enabled,
         }
         if multitask_strategy:
             payload["multitask_strategy"] = multitask_strategy
@@ -3066,6 +3075,8 @@ class CronClient:
         webhook: str | None = None,
         on_run_completed: OnCompletionBehavior | None = None,
         multitask_strategy: str | None = None,
+        end_time: datetime | None = None,
+        enabled: bool | None = None,
         headers: Mapping[str, str] | None = None,
         params: QueryParamTypes | None = None,
     ) -> Run:
@@ -3075,6 +3086,7 @@ class CronClient:
             assistant_id: The assistant ID or graph name to use for the cron job.
                 If using graph name, will default to first assistant created from that graph.
             schedule: The cron schedule to execute this job on.
+                Schedules are interpreted in UTC.
             input: The input to the graph.
             metadata: Metadata to assign to the cron job runs.
             config: The configuration for the assistant.
@@ -3090,6 +3102,8 @@ class CronClient:
                 clean them up. Clients are responsible for cleaning up kept threads.
             multitask_strategy: Multitask strategy to use.
                 Must be one of 'reject', 'interrupt', 'rollback', or 'enqueue'.
+            end_time: The time to stop running the cron job. If not provided, the cron job will run indefinitely.
+            enabled: Whether the cron job is enabled or not.
             headers: Optional custom headers to include with the request.
             params: Optional query parameters to include with the request.
 
@@ -3109,7 +3123,8 @@ class CronClient:
                 interrupt_before=["node_to_stop_before_1","node_to_stop_before_2"],
                 interrupt_after=["node_to_stop_after_1","node_to_stop_after_2"],
                 webhook="https://my.fake.webhook.com",
-                multitask_strategy="interrupt"
+                multitask_strategy="interrupt",
+                enabled=True,
             )
             ```
 
@@ -3126,6 +3141,8 @@ class CronClient:
             "interrupt_after": interrupt_after,
             "webhook": webhook,
             "on_run_completed": on_run_completed,
+            "end_time": end_time.isoformat() if end_time else None,
+            "enabled": enabled,
         }
         if multitask_strategy:
             payload["multitask_strategy"] = multitask_strategy
@@ -3163,11 +3180,88 @@ class CronClient:
         """
         await self.http.delete(f"/runs/crons/{cron_id}", headers=headers, params=params)
 
+    async def update(
+        self,
+        cron_id: str,
+        *,
+        schedule: str | None = None,
+        end_time: datetime | None = None,
+        input: Input | None = None,
+        metadata: Mapping[str, Any] | None = None,
+        config: Config | None = None,
+        context: Context | None = None,
+        webhook: str | None = None,
+        interrupt_before: All | list[str] | None = None,
+        interrupt_after: All | list[str] | None = None,
+        on_run_completed: OnCompletionBehavior | None = None,
+        enabled: bool | None = None,
+        headers: Mapping[str, str] | None = None,
+        params: QueryParamTypes | None = None,
+    ) -> Cron:
+        """Update a cron job by ID.
+
+        Args:
+            cron_id: The cron ID to update.
+            schedule: The cron schedule to execute this job on.
+                Schedules are interpreted in UTC.
+            end_time: The end date to stop running the cron.
+            input: The input to the graph.
+            metadata: Metadata to assign to the cron job runs.
+            config: The configuration for the assistant.
+            context: Static context added to the assistant.
+            webhook: Webhook to call after LangGraph API call is done.
+            interrupt_before: Nodes to interrupt immediately before they get executed.
+            interrupt_after: Nodes to interrupt immediately after they get executed.
+            on_run_completed: What to do with the thread after the run completes.
+                Must be one of 'delete' or 'keep'. 'delete' removes the thread
+                after execution. 'keep' creates a new thread for each execution but does not
+                clean them up.
+            enabled: Enable or disable the cron job.
+            headers: Optional custom headers to include with the request.
+            params: Optional query parameters to include with the request.
+
+        Returns:
+            The updated cron job.
+
+        ???+ example "Example Usage"
+
+            ```python
+            client = get_client(url="http://localhost:2024")
+            updated_cron = await client.crons.update(
+                cron_id="1ef3cefa-4c09-6926-96d0-3dc97fd5e39b",
+                schedule="0 10 * * *",
+                enabled=False,
+            )
+            ```
+
+        """
+        payload = {
+            "schedule": schedule,
+            "end_time": end_time.isoformat() if end_time else None,
+            "input": input,
+            "metadata": metadata,
+            "config": config,
+            "context": context,
+            "webhook": webhook,
+            "interrupt_before": interrupt_before,
+            "interrupt_after": interrupt_after,
+            "on_run_completed": on_run_completed,
+            "enabled": enabled,
+        }
+        payload = {k: v for k, v in payload.items() if v is not None}
+        return await self.http.patch(
+            f"/runs/crons/{cron_id}",
+            json=payload,
+            headers=headers,
+            params=params,
+        )
+
     async def search(
         self,
         *,
         assistant_id: str | None = None,
         thread_id: str | None = None,
+        enabled: bool | None = None,
         limit: int = 10,
         offset: int = 0,
         sort_by: CronSortBy | None = None,
@@ -3181,6 +3275,7 @@ class CronClient:
         Args:
             assistant_id: The assistant ID or graph name to search for.
             thread_id: the thread ID to search for.
+            enabled: The enabled status to search for.
             limit: The maximum number of results to return.
             offset: The number of results to skip.
             headers: Optional custom headers to include with the request.
@@ -3196,6 +3291,7 @@ class CronClient:
             cron_jobs = await client.crons.search(
                 assistant_id="my_assistant_id",
                 thread_id="my_thread_id",
+                enabled=True,
                 limit=5,
                 offset=5,
             )
@@ -3230,6 +3326,7 @@ class CronClient:
         payload = {
             "assistant_id": assistant_id,
             "thread_id": thread_id,
+            "enabled": enabled,
             "limit": limit,
             "offset": offset,
         }
@@ -6284,6 +6381,8 @@ class SyncCronClient:
         interrupt_after: All | list[str] | None = None,
         webhook: str | None = None,
         multitask_strategy: str | None = None,
+        end_time: datetime | None = None,
+        enabled: bool | None = None,
         headers: Mapping[str, str] | None = None,
         params: QueryParamTypes | None = None,
     ) -> Run:
@@ -6294,6 +6393,7 @@ class SyncCronClient:
             assistant_id: The assistant ID or graph name to use for the cron job.
                 If using graph name, will default to first assistant created from that graph.
             schedule: The cron schedule to execute this job on.
+                Schedules are interpreted in UTC.
             input: The input to the graph.
             metadata: Metadata to assign to the cron job runs.
             config: The configuration for the assistant.
@@ -6305,6 +6405,8 @@ class SyncCronClient:
             webhook: Webhook to call after LangGraph API call is done.
             multitask_strategy: Multitask strategy to use.
                 Must be one of 'reject', 'interrupt', 'rollback', or 'enqueue'.
+            end_time: The time to stop running the cron job. If not provided, the cron job will run indefinitely.
+            enabled: Whether the cron job is enabled. By default, it is considered enabled.
             headers: Optional custom headers to include with the request.
 
         Returns:
@@ -6324,7 +6426,8 @@ class SyncCronClient:
                 interrupt_before=["node_to_stop_before_1","node_to_stop_before_2"],
                 interrupt_after=["node_to_stop_after_1","node_to_stop_after_2"],
                 webhook="https://my.fake.webhook.com",
-                multitask_strategy="interrupt"
+                multitask_strategy="interrupt",
+                enabled=True
             )
             ```
         """
@@ -6340,6 +6443,8 @@ class SyncCronClient:
             "checkpoint_during": checkpoint_during,
             "webhook": webhook,
             "multitask_strategy": multitask_strategy,
+            "end_time": end_time.isoformat() if end_time else None,
+            "enabled": enabled,
         }
         payload = {k: v for k, v in payload.items() if v is not None}
         return self.http.post(
@@ -6364,6 +6469,8 @@ class SyncCronClient:
         webhook: str | None = None,
         on_run_completed: OnCompletionBehavior | None = None,
         multitask_strategy: str | None = None,
+        end_time: datetime | None = None,
+        enabled: bool | None = None,
         headers: Mapping[str, str] | None = None,
         params: QueryParamTypes | None = None,
     ) -> Run:
@@ -6373,6 +6480,7 @@ class SyncCronClient:
             assistant_id: The assistant ID or graph name to use for the cron job.
                 If using graph name, will default to first assistant created from that graph.
             schedule: The cron schedule to execute this job on.
+                Schedules are interpreted in UTC.
             input: The input to the graph.
             metadata: Metadata to assign to the cron job runs.
             config: The configuration for the assistant.
@@ -6388,6 +6496,8 @@ class SyncCronClient:
                 clean them up. Clients are responsible for cleaning up kept threads.
             multitask_strategy: Multitask strategy to use.
                 Must be one of 'reject', 'interrupt', 'rollback', or 'enqueue'.
+            end_time: The time to stop running the cron job. If not provided, the cron job will run indefinitely.
+            enabled: Whether the cron job is enabled. By default, it is considered enabled.
             headers: Optional custom headers to include with the request.
 
         Returns:
@@ -6407,7 +6517,8 @@ class SyncCronClient:
                 interrupt_before=["node_to_stop_before_1","node_to_stop_before_2"],
                 interrupt_after=["node_to_stop_after_1","node_to_stop_after_2"],
                 webhook="https://my.fake.webhook.com",
-                multitask_strategy="interrupt"
+                multitask_strategy="interrupt",
+                enabled=True
             )
             ```
 
@@ -6425,6 +6536,8 @@ class SyncCronClient:
             "checkpoint_during": checkpoint_during,
             "on_run_completed": on_run_completed,
             "multitask_strategy": multitask_strategy,
+            "end_time": end_time.isoformat() if end_time else None,
+            "enabled": enabled,
         }
         payload = {k: v for k, v in payload.items() if v is not None}
         return self.http.post(
@@ -6460,11 +6573,88 @@ class SyncCronClient:
         """
         self.http.delete(f"/runs/crons/{cron_id}", headers=headers, params=params)
 
+    def update(
+        self,
+        cron_id: str,
+        *,
+        schedule: str | None = None,
+        end_time: datetime | None = None,
+        input: Input | None = None,
+        metadata: Mapping[str, Any] | None = None,
+        config: Config | None = None,
+        context: Context | None = None,
+        webhook: str | None = None,
+        interrupt_before: All | list[str] | None = None,
+        interrupt_after: All | list[str] | None = None,
+        on_run_completed: OnCompletionBehavior | None = None,
+        enabled: bool | None = None,
+        headers: Mapping[str, str] | None = None,
+        params: QueryParamTypes | None = None,
+    ) -> Cron:
+        """Update a cron job by ID.
+
+        Args:
+            cron_id: The cron ID to update.
+            schedule: The cron schedule to execute this job on.
+                Schedules are interpreted in UTC.
+            end_time: The end date to stop running the cron.
+            input: The input to the graph.
+            metadata: Metadata to assign to the cron job runs.
+            config: The configuration for the assistant.
+            context: Static context added to the assistant.
+            webhook: Webhook to call after LangGraph API call is done.
+            interrupt_before: Nodes to interrupt immediately before they get executed.
+            interrupt_after: Nodes to interrupt immediately after they get executed.
+            on_run_completed: What to do with the thread after the run completes.
+                Must be one of 'delete' or 'keep'. 'delete' removes the thread
+                after execution. 'keep' creates a new thread for each execution but does not
+                clean them up.
+            enabled: Enable or disable the cron job.
+            headers: Optional custom headers to include with the request.
+            params: Optional query parameters to include with the request.
+
+        Returns:
+            The updated cron job.
+
+        ???+ example "Example Usage"
+
+            ```python
+            client = get_sync_client(url="http://localhost:8123")
+            updated_cron = client.crons.update(
+                cron_id="1ef3cefa-4c09-6926-96d0-3dc97fd5e39b",
+                schedule="0 10 * * *",
+                enabled=False,
+            )
+            ```
+
+        """
+        payload = {
+            "schedule": schedule,
+            "end_time": end_time.isoformat() if end_time else None,
+            "input": input,
+            "metadata": metadata,
+            "config": config,
+            "context": context,
+            "webhook": webhook,
+            "interrupt_before": interrupt_before,
+            "interrupt_after": interrupt_after,
+            "on_run_completed": on_run_completed,
+            "enabled": enabled,
+        }
+        payload = {k: v for k, v in payload.items() if v is not None}
+        return self.http.patch(
+            f"/runs/crons/{cron_id}",
+            json=payload,
+            headers=headers,
+            params=params,
+        )
+
     def search(
         self,
         *,
         assistant_id: str | None = None,
         thread_id: str | None = None,
+        enabled: bool | None = None,
         limit: int = 10,
         offset: int = 0,
         sort_by: CronSortBy | None = None,
@@ -6478,6 +6668,7 @@ class SyncCronClient:
         Args:
             assistant_id: The assistant ID or graph name to search for.
             thread_id: the thread ID to search for.
+            enabled: Whether the cron job is enabled.
             limit: The maximum number of results to return.
             offset: The number of results to skip.
             headers: Optional custom headers to include with the request.
@@ -6492,6 +6683,7 @@ class SyncCronClient:
             cron_jobs = client.crons.search(
                 assistant_id="my_assistant_id",
                 thread_id="my_thread_id",
+                enabled=True,
                 limit=5,
                 offset=5,
             )
@@ -6525,6 +6717,7 @@ class SyncCronClient:
         payload = {
             "assistant_id": assistant_id,
             "thread_id": thread_id,
+            "enabled": enabled,
             "limit": limit,
             "offset": offset,
         }

{langgraph_sdk-0.3.2 → langgraph_sdk-0.3.4}/langgraph_sdk/errors.py

@@ -38,7 +38,7 @@ class APIError(httpx.HTTPStatusError, LangGraphError):
             response = None
 
         httpx.HTTPStatusError.__init__(self, message, request=req, response=response)  # type: ignore[arg-type]
-        LangGraphError.__init__(self)
+        LangGraphError.__init__(self, message)
 
         self.request = req
         self.message = message

{langgraph_sdk-0.3.2 → langgraph_sdk-0.3.4}/langgraph_sdk/schema.py

@@ -372,6 +372,35 @@ class Cron(TypedDict):
     """The next run date of the cron."""
     metadata: dict
     """The metadata of the cron."""
+    enabled: bool
+    """Whether the cron is enabled."""
+
+
+class CronUpdate(TypedDict, total=False):
+    """Payload for updating a cron job. All fields are optional."""
+
+    schedule: str
+    """The cron schedule to execute this job on."""
+    end_time: datetime
+    """The end date to stop running the cron."""
+    input: Input
+    """The input to the graph."""
+    metadata: dict[str, Any]
+    """Metadata to assign to the cron job runs."""
+    config: Config
+    """The configuration for the assistant."""
+    context: Context
+    """Static context added to the assistant."""
+    webhook: str
+    """Webhook to call after LangGraph API call is done."""
+    interrupt_before: All | list[str]
+    """Nodes to interrupt immediately before they get executed."""
+    interrupt_after: All | list[str]
+    """Nodes to interrupt immediately after they get executed."""
+    on_run_completed: OnCompletionBehavior
+    """What to do with the thread after the run completes."""
+    enabled: bool
+    """Enable or disable the cron job."""
 
 
 # Select field aliases for client-side typing of `select` parameters.
@@ -428,6 +457,7 @@ CronSelectField = Literal[
     "metadata",
     "now",
     "on_run_completed",
+    "enabled",
 ]
 
 PrimitiveData = str | int | float | bool | None

{langgraph_sdk-0.3.2 → langgraph_sdk-0.3.4}/pyproject.toml

@@ -18,7 +18,7 @@ path = "langgraph_sdk/__init__.py"
 
 [project.urls]
 Source = "https://github.com/langchain-ai/langgraph/tree/main/libs/sdk-py"
-Twitter = "https://x.com/LangChainAI"
+Twitter = "https://x.com/LangChain"
 Slack = "https://www.langchain.com/join-community"
 Reddit = "https://www.reddit.com/r/LangChain/"
 
@@ -30,7 +30,7 @@ test = [
     "pytest-watch",
 ]
 lint = [
-    "ruff==0.14.7",
+    "ruff==0.14.11",
     "codespell",
     "mypy==1.19.0",
     "ty==0.0.1a27",