langgraph-api 0.2.96__tar.gz → 0.2.97__tar.gz

{langgraph_api-0.2.96 → langgraph_api-0.2.97}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.2.96
+Version: 0.2.97
 Author-email: Nuno Campos <nuno@langchain.dev>, Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE

langgraph_api-0.2.97/langgraph_api/__init__.py

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

{langgraph_api-0.2.96 → langgraph_api-0.2.97}/langgraph_api/api/threads.py

@@ -39,11 +39,15 @@ async def create_thread(
         )
 
         if supersteps := payload.get("supersteps"):
-            await Threads.State.bulk(
-                conn,
-                config={"configurable": {"thread_id": thread_id}},
-                supersteps=supersteps,
-            )
+            try:
+                await Threads.State.bulk(
+                    conn,
+                    config={"configurable": {"thread_id": thread_id}},
+                    supersteps=supersteps,
+                )
+            except HTTPException as e:
+                detail = f"Thread {thread_id} was created, but there were problems updating the state: {e.detail}"
+                raise HTTPException(status_code=201, detail=detail) from e
 
     return ApiResponse(await fetchone(iter, not_found_code=409))
 

{langgraph_api-0.2.96 → langgraph_api-0.2.97}/langgraph_api/asyncio.py

@@ -141,6 +141,13 @@ def call_soon_in_main_loop(coro: Coroutine[Any, Any, T]) -> asyncio.Future[T]:
     return this_loop_fut
 
 
+def call_soon_threadsafe(callback, *args) -> asyncio.Handle:
+    """Run a coroutine in the main event loop."""
+    if _MAIN_LOOP is None:
+        raise RuntimeError("No event loop set")
+    return _MAIN_LOOP.call_soon_threadsafe(callback, *args)
+
+
 class SimpleTaskGroup(AbstractAsyncContextManager["SimpleTaskGroup"]):
     """An async task group that can be configured to wait and/or cancel tasks on exit.
 

{langgraph_api-0.2.96 → langgraph_api-0.2.97}/langgraph_api/state.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import typing
 
-from langgraph.types import StateSnapshot
+from langgraph.types import Interrupt, StateSnapshot
 
 from langgraph_api.schema import Checkpoint, ThreadState
 
@@ -38,6 +38,23 @@ def runnable_config_to_checkpoint(
     return checkpoint
 
 
+def state_interrupt_to_thread_interrupt(interrupt: Interrupt | dict) -> dict:
+    # Sometimes the interrupt received in StateSnapshot is a dict.
+    # Let's convert it toan Interrupt object to gain access to dynamic `interrupt_id`.
+    if isinstance(interrupt, dict):
+        interrupt = Interrupt(**interrupt)
+
+    return {
+        "interrupt_id": interrupt.interrupt_id
+        if hasattr(interrupt, "interrupt_id")
+        else None,
+        "value": interrupt.value,
+        "resumable": interrupt.resumable,
+        "ns": interrupt.ns,
+        "when": interrupt.when,
+    }
+
+
 def state_snapshot_to_thread_state(state: StateSnapshot) -> ThreadState:
     return {
         "values": state.values,
@@ -48,7 +65,9 @@ def state_snapshot_to_thread_state(state: StateSnapshot) -> ThreadState:
                 "name": t.name,
                 "path": t.path,
                 "error": t.error,
-                "interrupts": t.interrupts,
+                "interrupts": [
+                    state_interrupt_to_thread_interrupt(i) for i in t.interrupts
+                ],
                 "checkpoint": t.state["configurable"]
                 if t.state is not None and not isinstance(t.state, StateSnapshot)
                 else None,

langgraph_api-0.2.96/langgraph_api/__init__.py

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