langgraph-api 0.0.41__tar.gz → 0.0.42__tar.gz

{langgraph_api-0.0.41 → langgraph_api-0.0.42}/PKG-INFO

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

{langgraph_api-0.0.41 → langgraph_api-0.0.42}/langgraph_api/api/assistants.py

@@ -124,6 +124,7 @@ async def create_assistant(request: ApiRequest) -> ApiResponse:
             metadata=payload.get("metadata") or {},
             if_exists=payload.get("if_exists") or "raise",
             name=payload.get("name") or "Untitled",
+            description=payload.get("description"),
         )
     return ApiResponse(await fetchone(assistant, not_found_code=409))
 
@@ -312,6 +313,7 @@ async def patch_assistant(
             graph_id=payload.get("graph_id"),
             metadata=payload.get("metadata"),
             name=payload.get("name"),
+            description=payload.get("description"),
         )
     return ApiResponse(await fetchone(assistant))
 

{langgraph_api-0.0.41 → langgraph_api-0.0.42}/langgraph_api/api/runs.py

@@ -273,7 +273,7 @@ async def wait_run_stateless(request: ApiRequest):
             except TimeoutError:
                 yield b"\n"
             except asyncio.CancelledError:
-                stream.cancel()
+                stream.cancel("Run stream cancelled")
                 await stream
                 raise
 

{langgraph_api-0.0.41 → langgraph_api-0.0.42}/langgraph_api/asyncio.py

@@ -63,7 +63,9 @@ async def wait_if_not_done(coro: Coroutine[Any, Any, T], done: ValueEvent) -> T:
         async with asyncio.TaskGroup() as tg:
             coro_task = tg.create_task(coro)
             done_task = tg.create_task(done.wait())
-            coro_task.add_done_callback(lambda _: done_task.cancel())
+            coro_task.add_done_callback(
+                lambda _: done_task.cancel("Coro task completed")
+            )
             done_task.add_done_callback(lambda _: coro_task.cancel(done._value))
             try:
                 return await coro_task
@@ -131,6 +133,7 @@ class SimpleTaskGroup(AbstractAsyncContextManager["SimpleTaskGroup"]):
         cancel: bool = False,
         wait: bool = True,
         taskset: set[asyncio.Task] | None = None,
+        taskgroup_name: str | None = None,
     ) -> None:
         self.tasks = taskset if taskset is not None else set()
         self.cancel = cancel
@@ -140,6 +143,7 @@ class SimpleTaskGroup(AbstractAsyncContextManager["SimpleTaskGroup"]):
                 task.add_done_callback(partial(self._create_task_done_callback, ()))
         for coro in coros:
             self.create_task(coro)
+        self.taskgroup_name = f" {taskgroup_name} " if taskgroup_name else ""
 
     def _create_task_done_callback(
         self, ignore_exceptions: tuple[Exception, ...], task: asyncio.Task
@@ -175,7 +179,7 @@ class SimpleTaskGroup(AbstractAsyncContextManager["SimpleTaskGroup"]):
         # cancel all tasks
         if self.cancel:
             for task in tasks:
-                task.cancel()
+                task.cancel(f"Task group{self.taskgroup_name}cancelled.")
         # wait for all tasks
         if self.wait:
             await asyncio.gather(*tasks, return_exceptions=True)

{langgraph_api-0.0.41 → langgraph_api-0.0.42}/langgraph_api/config.py

@@ -105,6 +105,20 @@ class StoreConfig(TypedDict, total=False):
     ttl: TTLConfig
 
 
+class CheckpointerConfig(TypedDict, total=False):
+    """Configuration for the built-in checkpointer, which handles checkpointing of state.
+
+    If omitted, no checkpointer is set up (the object store will still be present, however).
+    """
+
+    ttl: ThreadTTLConfig | None
+    """Optional. Defines the TTL (time-to-live) behavior configuration.
+    
+    If provided, the checkpointer will apply TTL settings according to the configuration.
+    If omitted, no TTL behavior is configured.
+    """
+
+
 # env
 
 env = Config()
@@ -223,9 +237,14 @@ def _parse_thread_ttl(value: str | None) -> ThreadTTLConfig | None:
     }
 
 
+CHECKPOINTER_CONFIG: CheckpointerConfig | None = env(
+    "LANGGRAPH_CHECKPOINTER", cast=_parse_json, default=None
+)
 THREAD_TTL: ThreadTTLConfig | None = env(
     "LANGGRAPH_THREAD_TTL", cast=_parse_thread_ttl, default=None
 )
+if THREAD_TTL is None and CHECKPOINTER_CONFIG is not None:
+    THREAD_TTL = CHECKPOINTER_CONFIG.get("ttl")
 
 N_JOBS_PER_WORKER = env("N_JOBS_PER_WORKER", cast=int, default=10)
 BG_JOB_TIMEOUT_SECS = env("BG_JOB_TIMEOUT_SECS", cast=float, default=3600)

{langgraph_api-0.0.41 → langgraph_api-0.0.42}/langgraph_api/graph.py

@@ -291,7 +291,7 @@ def _handle_exception(task: asyncio.Task) -> None:
 async def stop_remote_graphs() -> None:
     logger.info("Cancelling remote graphs")
     for task in js_bg_tasks:
-        task.cancel()
+        task.cancel("Stopping remote graphs.")
 
 
 def verify_graphs() -> None:

{langgraph_api-0.0.41 → langgraph_api-0.0.42}/langgraph_api/lifespan.py

@@ -47,7 +47,9 @@ async def lifespan(
     await start_pool()
     await collect_graphs_from_env(True)
     try:
-        async with SimpleTaskGroup(cancel=True, taskset=taskset) as tg:
+        async with SimpleTaskGroup(
+            cancel=True, taskset=taskset, taskgroup_name="Lifespan"
+        ) as tg:
             tg.create_task(metadata_loop())
             if config.N_JOBS_PER_WORKER > 0:
                 tg.create_task(queue())

{langgraph_api-0.0.41 → langgraph_api-0.0.42}/langgraph_api/schema.py

@@ -73,6 +73,8 @@ class Assistant(TypedDict):
     """The ID of the graph."""
     name: str
     """The name of the assistant."""
+    description: str | None
+    """The description of the assistant."""
     config: Config
     """The assistant config."""
     created_at: datetime

{langgraph_api-0.0.41 → langgraph_api-0.0.42}/langgraph_storage/database.py

@@ -23,6 +23,7 @@ class Assistant(TypedDict):
     assistant_id: UUID
     graph_id: str
     name: str
+    description: str | None
     created_at: NotRequired[datetime]
     updated_at: NotRequired[datetime]
     config: dict[str, Any]

{langgraph_api-0.0.41 → langgraph_api-0.0.42}/langgraph_storage/ops.py

@@ -177,6 +177,7 @@ class Assistants(Authenticated):
         if_exists: OnConflictBehavior,
         name: str,
         ctx: Auth.types.BaseAuthContext | None = None,
+        description: str | None = None,
     ) -> AsyncIterator[Assistant]:
         """Insert an assistant."""
         assistant_id = _ensure_uuid(assistant_id)
@@ -224,6 +225,7 @@ class Assistants(Authenticated):
             "created_at": now,
             "updated_at": now,
             "version": 1,
+            "description": description,
         }
         new_version = {
             "assistant_id": assistant_id,
@@ -251,16 +253,20 @@ class Assistants(Authenticated):
         graph_id: str | None = None,
         metadata: MetadataInput | None = None,
         name: str | None = None,
+        description: str | None = None,
         ctx: Auth.types.BaseAuthContext | None = None,
     ) -> AsyncIterator[Assistant]:
         """Update an assistant.
 
         Args:
+            conn: The connection to the in-memory store.
             assistant_id: The assistant ID.
             graph_id: The graph ID.
             config: The assistant config.
             metadata: The assistant metadata.
             name: The assistant name.
+            description: The assistant description.
+            ctx: The auth context.
 
         Returns:
             return the updated assistant model.
@@ -317,6 +323,9 @@ class Assistants(Authenticated):
             "metadata": metadata if metadata is not None else assistant["metadata"],
             "created_at": now,
             "name": name if name is not None else assistant["name"],
+            "description": description
+            if description is not None
+            else assistant["description"],
         }
         conn.store["assistant_versions"].append(new_version_entry)
 
@@ -327,6 +336,9 @@ class Assistants(Authenticated):
                 "config": new_version_entry["config"],
                 "metadata": new_version_entry["metadata"],
                 "name": name if name is not None else assistant["name"],
+                "description": description
+                if description is not None
+                else assistant["description"],
                 "updated_at": now,
                 "version": new_version,
             }
@@ -1350,7 +1362,7 @@ class Runs(Authenticated):
         # Get queue for this run
         queue = await Runs.Stream.subscribe(run_id)
 
-        async with SimpleTaskGroup(cancel=True) as tg:
+        async with SimpleTaskGroup(cancel=True, taskgroup_name="Runs.enter") as tg:
             done = ValueEvent()
             tg.create_task(listen_for_cancellation(queue, run_id, done))
 

{langgraph_api-0.0.41 → langgraph_api-0.0.42}/langgraph_storage/queue.py

@@ -141,9 +141,9 @@ async def queue():
             bb.deactivate()
         logger.info("Shutting down background workers")
         for task in WORKERS:
-            task.cancel()
+            task.cancel("Shutting down background workers.")
         for task in WEBHOOKS:
-            task.cancel()
+            task.cancel("Shutting down webhooks for background workers.")
         await asyncio.wait_for(
             asyncio.gather(*WORKERS, *WEBHOOKS, return_exceptions=True),
             SHUTDOWN_GRACE_PERIOD_SECS,

{langgraph_api-0.0.41 → langgraph_api-0.0.42}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.0.41"
+version = "0.0.42"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",

langgraph_api-0.0.41/langgraph_storage/ttl_dict.py

@@ -1,54 +0,0 @@
-import asyncio
-
-
-# TODO: Make this class thread-safe
-class TTLDict:
-    def __init__(self, timeout=10):
-        self._store = {}
-        self._timeouts = {}
-        self._tasks = {}  # Track tasks for key expiry
-        self._timeout = timeout
-
-    async def _remove_key_after_timeout(self, key):
-        await asyncio.sleep(self._timeouts[key])
-        if key in self._store:
-            del self._store[key]
-            del self._timeouts[key]
-
-    def set(self, key, value, timeout=None):
-        self._store[key] = value
-        key_timeout = timeout if timeout is not None else self._timeout
-        self._timeouts[key] = key_timeout
-
-        # Cancel the previous task if it exists
-        if key in self._tasks:
-            self._tasks[key].cancel()
-
-        # Schedule the removal of the key
-        self._tasks[key] = asyncio.create_task(self._remove_key_after_timeout(key))
-
-    def get(self, key, default=None):
-        return self._store.get(key, default)
-
-    def __contains__(self, key):
-        return key in self._store
-
-    def remove(self, key):
-        if key in self._store:
-            del self._store[key]
-            del self._timeouts[key]
-            if key in self._tasks:
-                self._tasks[key].cancel()
-                del self._tasks[key]
-
-    def keys(self):
-        return self._store.keys()
-
-    def values(self):
-        return self._store.values()
-
-    def items(self):
-        return self._store.items()
-
-    def __repr__(self):
-        return f"{self.__class__.__name__}({self._store})"