langgraph-api 0.0.2__py3-none-any.whl → 0.0.3__py3-none-any.whl

langgraph_api/cli.py

@@ -4,7 +4,7 @@ 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__)
@@ -47,6 +47,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
@@ -127,7 +129,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 +142,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/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/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/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/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/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,6 +90,7 @@ 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"])
@@ -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
@@ -221,6 +224,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

{langgraph_api-0.0.2.dist-info → langgraph_api-0.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.2
+Version: 0.0.3
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -14,7 +14,7 @@ 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 (>=0.2.52,<0.3.0)
 Requires-Dist: langgraph-checkpoint (>=2.0.5,<3.0)
 Requires-Dist: langsmith (>=0.1.63,<0.2.0)
 Requires-Dist: orjson (>=3.10.1)

{langgraph_api-0.0.2.dist-info → langgraph_api-0.0.3.dist-info}/RECORD

@@ -14,11 +14,11 @@ langgraph_api/auth/langsmith/backend.py,sha256=uHeb5-h13NIjrX_LDAvfWYr3zpbJvlvbd
 langgraph_api/auth/langsmith/client.py,sha256=eKchvAom7hdkUXauD8vHNceBDDUijrFgdTV8bKd7x4Q,3998
 langgraph_api/auth/middleware.py,sha256=_gJTOskEaln4RUT2rVYdQGPJVAyAiq-orsL_eQ3CynE,1369
 langgraph_api/auth/noop.py,sha256=vDJmzG2vArJxVzdHePvrJWahEa0dvGnhc2LEMMeiFz0,391
-langgraph_api/cli.py,sha256=LTETSHr2j5pBwKJkidKxzXfCohEScMnXjjIpfANBJpY,6458
+langgraph_api/cli.py,sha256=NNtETXM9meeGVT9nsArpuxE0Vumk1el1z3x804Esz0Q,6663
 langgraph_api/config.py,sha256=cG6eO4P_SZ2pKedb2b4n4vnBHRQr0aiECvGvOA8ZlJA,2259
 langgraph_api/cron_scheduler.py,sha256=DAzY2DsADzEpPVbG2BOSLTIufI93yeRswd71Aby_lV0,2257
 langgraph_api/errors.py,sha256=Bu_i5drgNTyJcLiyrwVE_6-XrSU50BHf9TDpttki9wQ,1690
-langgraph_api/graph.py,sha256=A9-l7cUvesyC3ymY_XFXxT02VG2kkoqbSYUDvjga5S0,9771
+langgraph_api/graph.py,sha256=YM8XbgN8Q6FdqTO2LO_g_nV-cpbiJyOBMBtObnWXp18,10507
 langgraph_api/http.py,sha256=XrbyxpjtfSvnaWWh5ZLGpgZmY83WoDCrP_1GPguNiXI,4712
 langgraph_api/http_logger.py,sha256=Sxo_q-65tElauRvkzVLt9lJojgNdgtcHGBYD0IRyX7M,3146
 langgraph_api/js/.gitignore,sha256=qAah3Fq0HWAlfRj5ktZyC6QRQIsAolGLRGcRukA1XJI,33
@@ -29,7 +29,7 @@ langgraph_api/js/package.json,sha256=glTUiod6sTKOXoGX7C7XcUXLeB_6Xa0hIw7MmJq2EY8
 langgraph_api/js/remote.py,sha256=n34B4X1sZW70vEF1hMQ4vn28BroZJt7_7BGaJ0TNeBU,23306
 langgraph_api/js/server_sent_events.py,sha256=DLgXOHauemt7706vnfDUCG1GI3TidKycSizccdz9KgA,3702
 langgraph_api/js/src/graph.mts,sha256=Gh2-TcCepeRhqMk8Bm1tWm5sv53y2FxgSclUTamnLNw,2636
-langgraph_api/js/src/hooks.mjs,sha256=IfPwixktR6W5FA8yNmmBVDVAaqBA549WZKuj1uQedVY,511
+langgraph_api/js/src/hooks.mjs,sha256=rdAEcBmqtIwzTPzqxJ2e7TJq6FWnhh5Edpy6os8hmlk,597
 langgraph_api/js/src/parser/parser.mts,sha256=wXre7teh8N8RYmGcyhZp4vMJd0kNnnFgoSGEyMVPzpQ,13007
 langgraph_api/js/src/parser/parser.worker.mjs,sha256=2K6D0GlUmkk7LE39I8mryB8VZVE3-N9Cblji-ArPhFo,386
 langgraph_api/js/src/schema/types.mts,sha256=SUj0vpvWVbID1mnGr2SUtumDBQkJ9zjfvJdoFP7DIzk,66536
@@ -56,31 +56,31 @@ langgraph_api/metadata.py,sha256=mih2G7ScQxiqyUlbksVXkqR3Oo-pM1b6lXtzOsgR1sw,304
 langgraph_api/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/models/run.py,sha256=nNhlG-l2P-DThFWYmCaar0UuPctadB9sH3CCWJfLqVc,8178
 langgraph_api/patch.py,sha256=94ddcTSZJe22JcpjxiSNjFZdYVnmeoWjk4IX4iBSoyk,1249
-langgraph_api/queue.py,sha256=m_LcZ_KZO1PVROQxlHXZDud7CYBroZv9M3oLJ4n_qYA,9456
+langgraph_api/queue.py,sha256=7tsbDgv4GlUYieJsrvIJDMQUEok4Eu-n_PIQ93rwKjk,9810
 langgraph_api/route.py,sha256=Dzje_dSigJramglqkt4ERT9-cb2xFli7dx25ZV6B6mI,4147
-langgraph_api/schema.py,sha256=kEhwg9MUvlHKUDCvomRUts3ja53IlXHgz70H1_AXMvk,5138
+langgraph_api/schema.py,sha256=yHHQS_4KQjiENF21JexzdJ_RF_HSxMEQX2aP6yPVTo4,5251
 langgraph_api/serde.py,sha256=VoJ7Z1IuqrQGXFzEP1qijAITtWCrmjtVqlCRuScjXJI,3533
 langgraph_api/server.py,sha256=cCD2lVv0SZdgf0o797UfxUyjFwmoazJVCjl_j-8Ae7A,1523
 langgraph_api/sse.py,sha256=2wNodCOP2eg7a9mpSu0S3FQ0CHk2BBV_vv0UtIgJIcc,4034
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
-langgraph_api/stream.py,sha256=8VaGGUbh4BC_NJrAHeuPZgeLmRY0E8wNC0UkobLViqQ,11022
+langgraph_api/stream.py,sha256=N3qneAy09Wftm8Mr-7vJCA-jGXBEp2TXFi2p-QLvVzQ,11351
 langgraph_api/utils.py,sha256=FI50tOFMVidV4-1TefouL1N-OJX41qD_fSEoWigTtf0,1575
 langgraph_api/validation.py,sha256=McizHlz-Ez8Jhdbc79mbPSde7GIuf2Jlbjx2yv_l6dA,4475
 langgraph_license/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_license/middleware.py,sha256=_ODIYzQkymr6W9_Fp9wtf1kAQspnpsmr53xuzyF2GA0,612
-langgraph_license/validation.py,sha256=th7OP0JZ8R8uf7vKRpfl0K3LOI0dyCKBPBxwdmgcDOk,202
+langgraph_license/validation.py,sha256=Uu_G8UGO_WTlLsBEY0gTVWjRR4czYGfw5YAD3HLZoj0,203
 langgraph_storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_storage/checkpoint.py,sha256=9mJTKXo4LIatr0nD7AX2mU8p9q18XaRnSSi25kNQcGc,2691
+langgraph_storage/checkpoint.py,sha256=55J7W0s5Z7svqqan9wphubSgCphWUItq98j-iWVwbH8,2774
 langgraph_storage/database.py,sha256=0bB4y2rWBYiT4Jsmvbbsams4Q8G0CgEwPVwclz6iQrM,4924
-langgraph_storage/ops.py,sha256=0btWlZnl7hl3KC7C9IIKzlsmUPilF-MQe4WwLVLymfs,51884
+langgraph_storage/ops.py,sha256=mehNrHleaPAjZ2sFISNuD7iwi_ewyX_IMmSPwreHhR0,52186
 langgraph_storage/queue.py,sha256=6cTZ0ubHu3S1T43yxHMVOwsQsDaJupByiU0sTUFFls8,3261
 langgraph_storage/retry.py,sha256=uvYFuXJ-T6S1QY1ZwkZHyZQbsvS-Ab68LSbzbUUSI2E,696
 langgraph_storage/store.py,sha256=Z033CojJb6jMZbMu3VPtwR0bFdfyfdUS8dQSaXUASYU,731
 langgraph_storage/ttl_dict.py,sha256=FlpEY8EANeXWKo_G5nmIotPquABZGyIJyk6HD9u6vqY,1533
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=JaieC_zSdQ9bzqJYdHUfCOnNt0ALBWcdj7uVjRLh9M8,122950
-langgraph_api-0.0.2.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.0.2.dist-info/METADATA,sha256=JlZkWP9lx3w6asP1m_U2NsCBkftoHp44PlEYWFmLezM,3986
-langgraph_api-0.0.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-langgraph_api-0.0.2.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.0.2.dist-info/RECORD,,
+langgraph_api-0.0.3.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.0.3.dist-info/METADATA,sha256=h9q2RmDz5Z1hCr1Tqxbd390gZzyzfUEjoFhXXW9Xz3k,3993
+langgraph_api-0.0.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+langgraph_api-0.0.3.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.0.3.dist-info/RECORD,,

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_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_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 {}
+                ),
             }
         )