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

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

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

langgraph_api-0.2.98/langgraph_api/__init__.py

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

{langgraph_api-0.2.96 → langgraph_api-0.2.98}/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.98}/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.98}/langgraph_api/logging.py

@@ -68,11 +68,31 @@ class AddApiVersion:
 
 
 class AddLoggingContext:
+    def __init__(self):
+        try:
+            from langchain_core.runnables.config import (
+                RunnableConfig,
+                var_child_runnable_config,
+            )
+
+            self.cvar: contextvars.ContextVar[RunnableConfig | None] = (
+                var_child_runnable_config
+            )
+        except Exception:
+            self.cvar = False
+
     def __call__(
         self, logger: logging.Logger, method_name: str, event_dict: EventDict
     ) -> EventDict:
         if (ctx := worker_config.get()) is not None:
             event_dict.update(ctx)
+        if (
+            self.cvar is not None
+            and (conf := self.cvar.get())
+            and (metadata := conf.get("metadata"))
+            and (lgnode := metadata.get("langgraph_node"))
+        ):
+            event_dict["langgraph_node"] = lgnode
         return event_dict
 
 
@@ -128,6 +148,7 @@ class Formatter(structlog.stdlib.ProcessorFormatter):
         super().__init__(
             processors=[
                 structlog.stdlib.ProcessorFormatter.remove_processors_meta,
+                AddLoggingContext(),
                 renderer,
             ],
             foreign_pre_chain=shared_processors,
@@ -136,7 +157,6 @@ class Formatter(structlog.stdlib.ProcessorFormatter):
 
 
 # configure structlog
-
 if not structlog.is_configured():
     structlog.configure(
         processors=[

{langgraph_api-0.2.96 → langgraph_api-0.2.98}/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"