langgraph-api 0.0.2__tar.gz → 0.0.4__tar.gz

{langgraph_api-0.0.2 → langgraph_api-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.2
+Version: 0.0.4
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -14,8 +14,8 @@ Requires-Dist: cryptography (>=43.0.3,<44.0.0)
 Requires-Dist: httpx (>=0.27.0)
 Requires-Dist: jsonschema-rs (>=0.25.0,<0.26.0)
 Requires-Dist: langchain-core (>=0.2.38,<0.4.0)
-Requires-Dist: langgraph (>=0.2.52)
-Requires-Dist: langgraph-checkpoint (>=2.0.5,<3.0)
+Requires-Dist: langgraph (>=0.2.52,<0.3.0)
+Requires-Dist: langgraph-checkpoint (>=2.0.6,<3.0)
 Requires-Dist: langsmith (>=0.1.63,<0.2.0)
 Requires-Dist: orjson (>=3.10.1)
 Requires-Dist: pyjwt (>=2.9.0,<3.0.0)

{langgraph_api-0.0.2 → langgraph_api-0.0.4}/langgraph_api/cli.py

@@ -4,12 +4,35 @@ import logging
 import os
 import pathlib
 import threading
-from collections.abc import Mapping
+from collections.abc import Mapping, Sequence
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 
+def _get_org_id() -> str | None:
+    from langsmith.client import Client
+    from langsmith.utils import tracing_is_enabled
+
+    # Yes, the organizationId is actually the workspace iD
+    # which is actually the tenantID which we actually get via
+    # the sessions endpoint
+
+    if not tracing_is_enabled():
+        return
+    client = Client()
+    try:
+        response = client.request_with_retries(
+            "GET", "/api/v1/sessions", params={"limit": 1}
+        )
+        result = response.json()
+        if result:
+            return result[0]["tenant_id"]
+    except Exception as e:
+        logger.debug("Failed to get organization ID: %s", str(e))
+        return None
+
+
 @contextlib.contextmanager
 def patch_environment(**kwargs):
     """Temporarily patch environment variables.
@@ -47,6 +70,8 @@ def run_server(
     open_browser: bool = False,
     debug_port: int | None = None,
     env: str | pathlib.Path | Mapping[str, str] | None = None,
+    reload_includes: Sequence[str] | None = None,
+    reload_excludes: Sequence[str] | None = None,
 ):
     """Run the LangGraph API server."""
     import uvicorn
@@ -90,14 +115,21 @@ def run_server(
     studio_url = f"https://smith.langchain.com/studio/?baseUrl={local_url}"
 
     def _open_browser():
+        nonlocal studio_url
         import time
         import urllib.request
         import webbrowser
 
+        org_id = _get_org_id()
+        if org_id:
+            studio_url = f"https://smith.langchain.com/studio/?baseUrl={local_url}&organizationId={org_id}"
+
         while True:
             try:
                 with urllib.request.urlopen(f"{local_url}/ok") as response:
                     if response.status == 200:
+                        logger.info("🎨 Opening Studio in your browser...")
+                        logger.info("URL: " + studio_url)
                         webbrowser.open(studio_url)
                         return
             except urllib.error.URLError:
@@ -127,7 +159,7 @@ For production use, please use LangGraph Cloud.
         REDIS_URI="fake",
         N_JOBS_PER_WORKER=str(n_jobs_per_worker if n_jobs_per_worker else 1),
         LANGSERVE_GRAPHS=json.dumps(graphs) if graphs else None,
-        LANGSMITH_LANGGRAPH_API_VARIANT="test",
+        LANGSMITH_LANGGRAPH_API_VARIANT="local_dev",
         **(env_vars or {}),
     ):
         if open_browser:
@@ -140,6 +172,8 @@ For production use, please use LangGraph Cloud.
             reload=reload,
             env_file=env_file,
             access_log=False,
+            reload_includes=reload_includes,
+            reload_excludes=reload_excludes,
             log_config={
                 "version": 1,
                 "incremental": False,

{langgraph_api-0.0.2 → langgraph_api-0.0.4}/langgraph_api/graph.py

@@ -157,18 +157,20 @@ async def collect_graphs_from_env(register: bool = False) -> None:
 
     if paths_str:
         specs = [
-            GraphSpec(
-                key,
-                module=value.split(":")[0],
-                variable=value.split(":")[1],
-                config=config_per_graph.get(key),
-            )
-            if "/" not in value
-            else GraphSpec(
-                key,
-                path=value.split(":")[0],
-                variable=value.split(":")[1],
-                config=config_per_graph.get(key),
+            (
+                GraphSpec(
+                    key,
+                    module=value.split(":")[0],
+                    variable=value.split(":")[1],
+                    config=config_per_graph.get(key),
+                )
+                if "/" not in value
+                else GraphSpec(
+                    key,
+                    path=value.split(":")[0],
+                    variable=value.split(":")[1],
+                    config=config_per_graph.get(key),
+                )
             )
             for key, value in json.loads(paths_str).items()
         ]
@@ -268,7 +270,15 @@ def _graph_from_spec(spec: GraphSpec) -> GraphValue:
             modspec.loader.exec_module(module)
         except ImportError as e:
             e.add_note(f"Could not import python module for graph: {spec}")
-            raise e
+            if os.environ.get("LANGSMITH_LANGGRAPH_API_VARIANT") == "local_dev":
+                e.add_note(
+                    "This error likely means you haven't installed your project and its dependencies yet. Before running the server, install your project:\n\n"
+                    "If you are using requirements.txt:\n"
+                    "python -m pip install -r requirements.txt\n\n"
+                    "If you are using pyproject.toml or setuptools:\n"
+                    "python -m pip install -e .\n\n"
+                    "Make sure to run this command from your project's root directory (where your setup.py or pyproject.toml is located)"
+                )
     else:
         raise ValueError("Graph specification must have a path or module")
 

{langgraph_api-0.0.2 → langgraph_api-0.0.4}/langgraph_api/js/src/hooks.mjs

@@ -5,7 +5,11 @@ export async function resolve(specifier, context, nextResolve) {
   const parentURL = new URL("./graph.mts", import.meta.url).toString();
 
   if (specifier.startsWith("@langchain/langgraph")) {
-    return nextResolve(specifier, { ...context, parentURL });
+    try {
+      return nextResolve(specifier, { ...context, parentURL });
+    } catch (error) {
+      return nextResolve(specifier, context);
+    }
   }
 
   return nextResolve(specifier, context);

{langgraph_api-0.0.2 → langgraph_api-0.0.4}/langgraph_api/queue.py

@@ -5,7 +5,7 @@ from random import random
 from typing import cast
 
 import structlog
-from langgraph.pregel.debug import CheckpointPayload
+from langgraph.pregel.debug import CheckpointPayload, TaskResultPayload
 
 from langgraph_api.config import BG_JOB_NO_DELAY, STATS_INTERVAL_SECS
 from langgraph_api.errors import (
@@ -129,6 +129,13 @@ async def worker(
             nonlocal checkpoint
             checkpoint = checkpoint_arg
 
+        def on_task_result(task_result: TaskResultPayload):
+            if checkpoint is not None:
+                for task in checkpoint["tasks"]:
+                    if task["id"] == task_result["id"]:
+                        task.update(task_result)
+                        break
+
         try:
             if attempt > MAX_RETRY_ATTEMPTS:
                 raise RuntimeError(f"Run {run['run_id']} exceeded max attempts")
@@ -144,6 +151,7 @@ async def worker(
                     attempt,
                     done,
                     on_checkpoint=on_checkpoint,
+                    on_task_result=on_task_result,
                 )
             await asyncio.wait_for(consume(stream, run_id), timeout)
             await logger.ainfo(

{langgraph_api-0.0.2 → langgraph_api-0.0.4}/langgraph_api/schema.py

@@ -98,6 +98,8 @@ class Thread(TypedDict):
     """The status of the thread. One of 'idle', 'busy', 'interrupted', "error"."""
     values: Fragment
     """The current state of the thread."""
+    interrupts: Fragment
+    """The current interrupts of the thread, a map of task_id to list of interrupts."""
 
 
 class ThreadTask(TypedDict):

{langgraph_api-0.0.2 → langgraph_api-0.0.4}/langgraph_api/stream.py

@@ -18,7 +18,7 @@ from langgraph.errors import (
     GraphRecursionError,
     InvalidUpdateError,
 )
-from langgraph.pregel.debug import CheckpointPayload
+from langgraph.pregel.debug import CheckpointPayload, TaskResultPayload
 from langgraph.types import Command, Send
 from pydantic import ValidationError
 from pydantic.v1 import ValidationError as ValidationErrorLegacy
@@ -90,10 +90,11 @@ async def astream_state(
     done: ValueEvent,
     *,
     on_checkpoint: Callable[[CheckpointPayload], None] = lambda _: None,
+    on_task_result: Callable[[TaskResultPayload], None] = lambda _: None,
 ) -> AnyStream:
     """Stream messages from the runnable."""
     run_id = str(run["run_id"])
-    await stack.enter_async_context(conn.pipeline())
+    pipe = await stack.enter_async_context(conn.pipeline())
     # extract args from run
     kwargs = run["kwargs"].copy()
     subgraphs = kwargs.get("subgraphs", False)
@@ -102,7 +103,7 @@ async def astream_state(
     graph = get_graph(
         config["configurable"]["graph_id"],
         config,
-        store=None if not conn else Store(conn),
+        store=None if not conn else Store(conn, pipe=pipe),
         checkpointer=None if temporary else Checkpointer(conn),
     )
     input = kwargs.pop("input")
@@ -174,6 +175,8 @@ async def astream_state(
                         if chunk["type"] == "checkpoint":
                             checkpoint = _preprocess_debug_checkpoint(chunk["payload"])
                             on_checkpoint(checkpoint)
+                        elif chunk["type"] == "task_result":
+                            on_task_result(chunk["payload"])
                     if mode == "messages":
                         if "messages-tuple" in stream_mode:
                             yield "messages", chunk
@@ -185,9 +188,11 @@ async def astream_state(
                                 messages[msg.id] = msg
                                 yield "messages/metadata", {msg.id: {"metadata": meta}}
                             yield (
-                                "messages/partial"
-                                if isinstance(msg, BaseMessageChunk)
-                                else "messages/complete",
+                                (
+                                    "messages/partial"
+                                    if isinstance(msg, BaseMessageChunk)
+                                    else "messages/complete"
+                                ),
                                 [message_chunk_to_message(messages[msg.id])],
                             )
                     elif mode in stream_mode:
@@ -221,6 +226,8 @@ async def astream_state(
                     if chunk["type"] == "checkpoint":
                         checkpoint = _preprocess_debug_checkpoint(chunk["payload"])
                         on_checkpoint(checkpoint)
+                    elif chunk["type"] == "task_result":
+                        on_task_result(chunk["payload"])
                 if mode == "messages":
                     if "messages-tuple" in stream_mode:
                         yield "messages", chunk
@@ -232,9 +239,11 @@ async def astream_state(
                             messages[msg.id] = msg
                             yield "messages/metadata", {msg.id: {"metadata": meta}}
                         yield (
-                            "messages/partial"
-                            if isinstance(msg, BaseMessageChunk)
-                            else "messages/complete",
+                            (
+                                "messages/partial"
+                                if isinstance(msg, BaseMessageChunk)
+                                else "messages/complete"
+                            ),
                             [message_chunk_to_message(messages[msg.id])],
                         )
                 elif mode in stream_mode:

{langgraph_api-0.0.2 → langgraph_api-0.0.4}/langgraph_license/validation.py

@@ -7,5 +7,5 @@ async def get_license_status() -> bool:
 
 
 def plus_features_enabled() -> bool:
-    """Always return true"""
+    """Always return false"""
     return False

{langgraph_api-0.0.2 → langgraph_api-0.0.4}/langgraph_storage/checkpoint.py

@@ -10,6 +10,8 @@ from langgraph.checkpoint.base import (
 )
 from langgraph.checkpoint.memory import MemorySaver, PersistentDict
 
+from langgraph_api.serde import Serializer
+
 _EXCLUDED_KEYS = {"checkpoint_ns", "checkpoint_id", "run_id", "thread_id"}
 
 
@@ -37,7 +39,7 @@ class InMemorySaver(MemorySaver):
             return d
 
         super().__init__(
-            serde=serde,
+            serde=serde if serde is not None else Serializer(),
             factory=factory,
         )
 

{langgraph_api-0.0.2 → langgraph_api-0.0.4}/langgraph_storage/ops.py

@@ -583,6 +583,15 @@ class Threads:
                 "updated_at": datetime.now(UTC),
                 "values": checkpoint["values"] if checkpoint else None,
                 "status": status,
+                "interrupts": (
+                    {
+                        t["id"]: t["interrupts"]
+                        for t in checkpoint["tasks"]
+                        if t["interrupts"]
+                    }
+                    if checkpoint
+                    else {}
+                ),
             }
         )
 

langgraph_api-0.0.4/langgraph_storage/store.py

@@ -0,0 +1,44 @@
+import os
+from typing import Any
+
+from langgraph.checkpoint.memory import PersistentDict
+from langgraph.store.memory import InMemoryStore
+
+
+class DiskBackedInMemStore(InMemoryStore):
+    def __init__(self, *args: Any, filename: str | None = None, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+        self.filename = filename
+        self._data = PersistentDict(dict, filename=self.filename)
+        self._load_data()
+
+    def _load_data(self) -> None:
+        if not self.filename:
+            return
+        try:
+            self._data.load()
+        except FileNotFoundError:
+            # It's okay if the file doesn't exist yet
+            pass
+        except (EOFError, ValueError) as e:
+            raise RuntimeError(
+                f"Failed to load store from {self.filename}. "
+                "This may be due to changes in the stored data structure. "
+                "Consider clearing the local store by running: rm -rf .langgraph_api"
+            ) from e
+        except Exception as e:
+            raise RuntimeError(
+                f"Unexpected error loading store from {self.filename}: {str(e)}"
+            ) from e
+
+    def close(self) -> None:
+        self._data.close()
+
+
+_STORE_FILE = os.path.join(".langgraph_api", "store.pckl")
+os.makedirs(".langgraph_api", exist_ok=True)
+STORE = DiskBackedInMemStore(filename=_STORE_FILE)
+
+
+def Store(*args: Any, **kwargs: Any) -> DiskBackedInMemStore:
+    return STORE

{langgraph_api-0.0.2 → langgraph_api-0.0.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.0.2"
+version = "0.0.4"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",
@@ -23,8 +23,8 @@ python = ">=3.11.0,<4.0"
 sse-starlette = "^2.1.0"
 starlette = ">=0.38.6"
 watchfiles = ">=0.13"
-langgraph = ">=0.2.52"
-langgraph-checkpoint = ">=2.0.5,<3.0"
+langgraph = ">=0.2.52,<0.3.0"
+langgraph-checkpoint = ">=2.0.6,<3.0"
 orjson = ">=3.10.1"
 uvicorn = ">=0.26.0"
 langsmith = "^0.1.63"

langgraph_api-0.0.2/langgraph_storage/store.py

@@ -1,28 +0,0 @@
-import os
-
-from langgraph.checkpoint.memory import PersistentDict
-from langgraph.store.memory import InMemoryStore
-
-
-class DiskBackedInMemStore(InMemoryStore):
-    def __init__(self, *args, filename=None, **kwargs):
-        super().__init__(*args, **kwargs)
-        self.filename = filename
-        self._data = PersistentDict(dict, filename=self.filename)
-        try:
-            self._data.load()
-        except FileNotFoundError:
-            pass
-
-    def close(self):
-        self._data.close()
-
-
-_STORE_FILE = os.path.join(".langgraph_api", "store.pckl")
-if not os.path.exists(".langgraph_api"):
-    os.mkdir(".langgraph_api")
-STORE = DiskBackedInMemStore(filename=_STORE_FILE)
-
-
-def Store(*args, **kwargs):
-    return STORE