langgraph-api 0.2.44__py3-none-any.whl → 0.2.45__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.44"
+__version__ = "0.2.45"

langgraph_api/js/remote.py

@@ -452,27 +452,27 @@ async def run_js_http_process(paths_str: str, http_config: dict, watch: bool = F
                 attempt += 1
 
 
-def _get_passthrough_checkpointer(conn: AsyncConnectionProto):
-    from langgraph_runtime.checkpoint import Checkpointer
+class PassthroughSerialiser(SerializerProtocol):
+    def dumps(self, obj: Any) -> bytes:
+        return json_dumpb(obj)
 
-    class PassthroughSerialiser(SerializerProtocol):
-        def dumps(self, obj: Any) -> bytes:
-            return json_dumpb(obj)
+    def dumps_typed(self, obj: Any) -> tuple[str, bytes]:
+        return "json", json_dumpb(obj)
 
-        def dumps_typed(self, obj: Any) -> tuple[str, bytes]:
-            return "json", json_dumpb(obj)
+    def loads(self, data: bytes) -> Any:
+        return orjson.loads(data)
 
-        def loads(self, data: bytes) -> Any:
-            return orjson.loads(data)
+    def loads_typed(self, data: tuple[str, bytes]) -> Any:
+        type, payload = data
+        if type != "json":
+            raise ValueError(f"Unsupported type {type}")
+        return orjson.loads(payload)
 
-        def loads_typed(self, data: tuple[str, bytes]) -> Any:
-            type, payload = data
-            if type != "json":
-                raise ValueError(f"Unsupported type {type}")
-            return orjson.loads(payload)
 
-    checkpointer = Checkpointer(conn)
+def _get_passthrough_checkpointer(conn: AsyncConnectionProto):
+    from langgraph_runtime.checkpoint import Checkpointer
 
+    checkpointer = Checkpointer(conn)
     # This checkpointer does not attempt to revive LC-objects.
     # Instead, it will pass through the JSON values as-is.
     checkpointer.serde = PassthroughSerialiser()
@@ -687,7 +687,11 @@ async def run_remote_checkpointer():
                 payload = orjson.loads(await request.body())
                 return ApiResponse(await cb(payload))
             except ValueError as exc:
+                await logger.error(exc)
                 return ApiResponse({"error": str(exc)}, status_code=400)
+            except Exception as exc:
+                await logger.error(exc)
+                return ApiResponse({"error": str(exc)}, status_code=500)
 
         return wrapped
 
@@ -800,15 +804,18 @@ async def js_healthcheck():
                 transport=httpx.AsyncHTTPTransport(verify=SSL),
             ) as checkpointer_client,
         ):
+            graph_passed = False
             try:
                 res = await graph_client.get("/ok")
                 res.raise_for_status()
+                graph_passed = True
                 res = await checkpointer_client.get("/ok")
                 res.raise_for_status()
                 return True
             except httpx.HTTPError as exc:
                 logger.warning(
                     "JS healthcheck failed. Either the JS server is not running or the event loop is blocked by a CPU-intensive task.",
+                    graph_passed=graph_passed,
                     error=exc,
                 )
                 raise HTTPException(

langgraph_api/logging.py

@@ -57,6 +57,16 @@ class AddPrefixedEnvVars:
         return event_dict
 
 
+class AddApiVersion:
+    def __call__(
+        self, logger: logging.Logger, method_name: str, event_dict: EventDict
+    ) -> EventDict:
+        from langgraph_api import __version__
+
+        event_dict["langgraph_api_version"] = __version__
+        return event_dict
+
+
 class AddLoggingContext:
     def __call__(
         self, logger: logging.Logger, method_name: str, event_dict: EventDict
@@ -90,11 +100,14 @@ shared_processors = [
     structlog.stdlib.PositionalArgumentsFormatter(),
     structlog.stdlib.ExtraAdder(),
     AddPrefixedEnvVars("LANGSMITH_LANGGRAPH_"),  # injected by docker build
+    AddApiVersion(),
     structlog.processors.TimeStamper(fmt="iso", utc=True),
     structlog.processors.StackInfoRenderer(),
-    structlog.processors.dict_tracebacks
-    if LOG_JSON
-    else structlog.processors.format_exc_info,
+    (
+        structlog.processors.dict_tracebacks
+        if LOG_JSON
+        else structlog.processors.format_exc_info
+    ),
     structlog.processors.UnicodeDecoder(),
     AddLoggingContext(),
 ]

langgraph_api/models/run.py

@@ -29,6 +29,13 @@ from langgraph_api.utils import AsyncConnectionProto, get_auth_ctx
 from langgraph_runtime.ops import Runs, logger
 
 
+class LangSmithTracer(TypedDict, total=False):
+    """Configuration for LangSmith tracing."""
+
+    example_id: str | None
+    project_name: str | None
+
+
 class RunCreateDict(TypedDict):
     """Payload for creating a run."""
 
@@ -87,6 +94,8 @@ class RunCreateDict(TypedDict):
     """Start the run after this many seconds. Defaults to 0."""
     if_not_exists: IfNotExists
     """Create the thread if it doesn't exist. If False, reply with 404."""
+    langsmith_tracer: LangSmithTracer | None
+    """Configuration for additional tracing with LangSmith."""
 
 
 def ensure_ids(
@@ -295,6 +304,9 @@ async def create_valid_run(
         user_id = None
     if not configurable.get("langgraph_request_id"):
         configurable["langgraph_request_id"] = request_id
+    if ls_tracing := payload.get("langsmith_tracer"):
+        configurable["__langsmith_project__"] = ls_tracing.get("project_name")
+        configurable["__langsmith_example_id__"] = ls_tracing.get("example_id")
     if request_start_time:
         configurable["__request_start_time_ms__"] = request_start_time
     after_seconds = payload.get("after_seconds", 0)

langgraph_api/stream.py

@@ -1,5 +1,5 @@
 from collections.abc import AsyncIterator, Callable
-from contextlib import AsyncExitStack, aclosing
+from contextlib import AsyncExitStack, aclosing, asynccontextmanager
 from functools import lru_cache
 from typing import Any, cast
 
@@ -20,6 +20,7 @@ from langgraph.errors import (
     InvalidUpdateError,
 )
 from langgraph.pregel.debug import CheckpointPayload, TaskResultPayload
+from langsmith.utils import get_tracer_project
 from pydantic import ValidationError
 from pydantic.v1 import ValidationError as ValidationErrorLegacy
 
@@ -71,6 +72,12 @@ def _preprocess_debug_checkpoint(payload: CheckpointPayload | None) -> dict[str,
     return payload
 
 
+@asynccontextmanager
+async def async_tracing_context(*args, **kwargs):
+    with langsmith.tracing_context(*args, **kwargs):
+        yield
+
+
 async def astream_state(
     stack: AsyncExitStack,
     conn: AsyncConnectionProto,
@@ -131,6 +138,25 @@ async def astream_state(
     use_astream_events = "events" in stream_mode or isinstance(graph, BaseRemotePregel)
     # yield metadata chunk
     yield "metadata", {"run_id": run_id, "attempt": attempt}
+
+    #  is a langsmith tracing project is specified, additionally pass that in to tracing context
+    if ls_project := config["configurable"].get("__langsmith_project__"):
+        updates = None
+        if example_id := config["configurable"].get("__langsmith_example_id__"):
+            updates = {"reference_example_id": example_id}
+
+        await stack.enter_async_context(
+            async_tracing_context(
+                replicas=[
+                    (
+                        ls_project,
+                        updates,
+                    ),
+                    (get_tracer_project(), None),
+                ]
+            )
+        )
+
     # stream run
     if use_astream_events:
         async with (
@@ -269,7 +295,7 @@ async def astream_state(
                         yield mode, chunk
                 # --- end shared logic with astream_events ---
     if is_remote_pregel:
-        # increament the remote runs
+        # increment the remote runs
         try:
             nodes_executed = await graph.fetch_nodes_executed()
             incr_nodes(None, incr=nodes_executed)

{langgraph_api-0.2.44.dist-info → langgraph_api-0.2.45.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.2.44
+Version: 0.2.45
 Author-email: Nuno Campos <nuno@langchain.dev>, Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE
@@ -9,12 +9,12 @@ Requires-Dist: cloudpickle>=3.0.0
 Requires-Dist: cryptography<45.0,>=42.0.0
 Requires-Dist: httpx>=0.25.0
 Requires-Dist: jsonschema-rs<0.30,>=0.20.0
-Requires-Dist: langchain-core>=0.2.38
+Requires-Dist: langchain-core>=0.3.64
 Requires-Dist: langgraph-checkpoint>=2.0.23
 Requires-Dist: langgraph-runtime-inmem<0.3,>=0.2.0
 Requires-Dist: langgraph-sdk>=0.1.66
 Requires-Dist: langgraph>=0.3.27
-Requires-Dist: langsmith>=0.1.112
+Requires-Dist: langsmith>=0.3.45
 Requires-Dist: orjson>=3.9.7
 Requires-Dist: pyjwt>=2.9.0
 Requires-Dist: sse-starlette<2.2.0,>=2.1.0

{langgraph_api-0.2.44.dist-info → langgraph_api-0.2.45.dist-info}/RECORD

@@ -1,4 +1,4 @@
-langgraph_api/__init__.py,sha256=U9IXTG0wwP5qkavLVbDEmAfwEJjO-1-R5e__z2cNpz0,23
+langgraph_api/__init__.py,sha256=48yajatRDmJidbzrkcvrcmL96an-Z7rF0VahutQn2wk,23
 langgraph_api/asgi_transport.py,sha256=eqifhHxNnxvI7jJqrY1_8RjL4Fp9NdN4prEub2FWBt8,5091
 langgraph_api/asyncio.py,sha256=nelZwKL7iOjM5GHj1rVjiPE7igUIKLNKtc-3urxmlfo,9250
 langgraph_api/cli.py,sha256=9Ou3tGDDY_VVLt5DFle8UviJdpI4ZigC5hElYvq2-To,14519
@@ -8,7 +8,7 @@ langgraph_api/cron_scheduler.py,sha256=i87j4pJrcsmsqMKeKUs69gaAjrGaSM3pM3jnXdN5J
 langgraph_api/errors.py,sha256=Bu_i5drgNTyJcLiyrwVE_6-XrSU50BHf9TDpttki9wQ,1690
 langgraph_api/graph.py,sha256=MPm8DvInBQsq2em45c2YD5bW6T_G1LlDkAuWq-19gCQ,23240
 langgraph_api/http.py,sha256=gYbxxjY8aLnsXeJymcJ7G7Nj_yToOGpPYQqmZ1_ggfA,5240
-langgraph_api/logging.py,sha256=bx6kxV7O2FUnu0ubrgeJuj5q2EjYC4ve1BkbrVfzy28,3953
+langgraph_api/logging.py,sha256=1BXELwUBY8gZeOWCYElbBu_GyHLM2jjlDVJznlekqvQ,4268
 langgraph_api/metadata.py,sha256=2sz9ECnbnQtgqN6ptDkRmymaVKfQPgaX-JuDJDJB47c,4254
 langgraph_api/patch.py,sha256=Dgs0PXHytekX4SUL6KsjjN0hHcOtGLvv1GRGbh6PswU,1408
 langgraph_api/queue_entrypoint.py,sha256=_41ZveMDdn9bapjA7Ik9FG3r4hyIwXESUM5F1PdlieE,2309
@@ -19,7 +19,7 @@ langgraph_api/server.py,sha256=Z_VL-kIphybTRDWBIqHMfRhgCmAFyTRqAGlgnHQF0Zg,6973
 langgraph_api/sse.py,sha256=F7swfjKBDrlUmXZ_dWuDVHtp-3o1Cpjq1lwp0bJD-nw,4223
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
 langgraph_api/store.py,sha256=_xGhdwEIMoY1_hIy_oWwxZp4Y7FH833BNJfgFIhT80E,4640
-langgraph_api/stream.py,sha256=0qLblAQ406bGZEM1zRQKVyU407c9PjcBcgPCKN78ywc,12847
+langgraph_api/stream.py,sha256=NfpOI5_SzHLcExgcjaF8ggeKCk9n1ocoQTWmjFwsrVQ,13700
 langgraph_api/thread_ttl.py,sha256=-Ox8NFHqUH3wGNdEKMIfAXUubY5WGifIgCaJ7npqLgw,1762
 langgraph_api/utils.py,sha256=92mSti9GfGdMRRWyESKQW5yV-75Z9icGHnIrBYvdypU,3619
 langgraph_api/validation.py,sha256=zMuKmwUEBjBgFMwAaeLZmatwGVijKv2sOYtYg7gfRtc,4950
@@ -52,7 +52,7 @@ langgraph_api/js/client.mts,sha256=N9CTH7mbXGSD-gpv-XyruYsHI-rgrObL8cQoAp5s3_U,3
 langgraph_api/js/errors.py,sha256=Cm1TKWlUCwZReDC5AQ6SgNIVGD27Qov2xcgHyf8-GXo,361
 langgraph_api/js/global.d.ts,sha256=j4GhgtQSZ5_cHzjSPcHgMJ8tfBThxrH-pUOrrJGteOU,196
 langgraph_api/js/package.json,sha256=7_qkj-b0_bpHFFyBDgGaZl3BeuSqkbCD7wNn-ZvQeGA,1333
-langgraph_api/js/remote.py,sha256=utq7tjSFUf0zPLDFgC9lnsGKrtX3EVEX6IcNCc9Q1yM,35934
+langgraph_api/js/remote.py,sha256=yNK8ajiRXxpzbUFWT1wPdK0XyRW1TejzBPAAeVUCtQY,36192
 langgraph_api/js/schema.py,sha256=7idnv7URlYUdSNMBXQcw7E4SxaPxCq_Oxwnlml8q5ik,408
 langgraph_api/js/sse.py,sha256=lsfp4nyJyA1COmlKG9e2gJnTttf_HGCB5wyH8OZBER8,4105
 langgraph_api/js/tsconfig.json,sha256=imCYqVnqFpaBoZPx8k1nO4slHIWBFsSlmCYhO73cpBs,341
@@ -70,16 +70,24 @@ langgraph_api/middleware/http_logger.py,sha256=aj4mdisRobFePkD3Iy6-w_Mujwx4TQRaE
 langgraph_api/middleware/private_network.py,sha256=eYgdyU8AzU2XJu362i1L8aSFoQRiV7_aLBPw7_EgeqI,2111
 langgraph_api/middleware/request_id.py,sha256=fmtp0jLqoSqXTD39d9PW7jVlIvGqGurqgteMCeexRvY,1106
 langgraph_api/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/models/run.py,sha256=orI907khq8SNOJAtGed-jgMThCMzqheIZcuFT9I1_VA,14040
+langgraph_api/models/run.py,sha256=j1s9KRfFXgjKUudB9z7IVJ34Klo85PPeaVFtmWHhEdo,14514
 langgraph_api/tunneling/cloudflare.py,sha256=iKb6tj-VWPlDchHFjuQyep2Dpb-w2NGfJKt-WJG9LH0,3650
 langgraph_license/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_license/validation.py,sha256=ZKraAVJArAABKqrmHN-EN18ncoNUmRm500Yt1Sc7tUA,537
 langgraph_runtime/__init__.py,sha256=O4GgSmu33c-Pr8Xzxj_brcK5vkm70iNTcyxEjICFZxA,1075
+langgraph_runtime/checkpoint.py,sha256=J2ePryEyKJWGgxjs27qEHrjj87uPMX3Rqm3hLvG63uk,119
+langgraph_runtime/database.py,sha256=ANEtfm4psr19FtpVcNs5CFWHw-JhfHvIMnkaORa4QSM,117
+langgraph_runtime/lifespan.py,sha256=-YIHyEEaP_F2tSdTP0tNjfAJXs7KfxaIsWdmQAUi2KM,117
+langgraph_runtime/metrics.py,sha256=CIBw3tjTg1c-o3_2InA-qV34028fQcYWBYkpN6zdEoI,116
+langgraph_runtime/ops.py,sha256=ht_U9LPbHWy0l95b_Q0Vvtd7kYxeZsaSKSf0WpwHUoo,112
+langgraph_runtime/queue.py,sha256=m7req6Ca9NOw1yp-zo30zGhldRWDFk4QVL_tgrVrhQg,114
+langgraph_runtime/retry.py,sha256=V0duD01fO7GUQ_btQkp1aoXcEOFhXooGVP6q4yMfuyY,114
+langgraph_runtime/store.py,sha256=7mowndlsIroGHv3NpTSOZDJR0lCuaYMBoTnTrewjslw,114
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=_4GFDqbq1X9vD4_FxwahuVODJMOHx-U76gkY4rdy3DA,138189
-langgraph_api-0.2.44.dist-info/METADATA,sha256=9OXLai4MqakDiva3wfVGk3vmX4akdCJ1zKuvNpl62t4,3892
-langgraph_api-0.2.44.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langgraph_api-0.2.44.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
-langgraph_api-0.2.44.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.2.44.dist-info/RECORD,,
+langgraph_api-0.2.45.dist-info/METADATA,sha256=C9WHgMJRJQpBF0i6FG8m4eqhgPTPLJ8V1JbBpRi2tAM,3891
+langgraph_api-0.2.45.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langgraph_api-0.2.45.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
+langgraph_api-0.2.45.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.2.45.dist-info/RECORD,,

langgraph_runtime/checkpoint.py

@@ -0,0 +1,4 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from langgraph_runtime_inmem.checkpoint import *  # noqa: F403

langgraph_runtime/database.py

@@ -0,0 +1,4 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from langgraph_runtime_inmem.database import *  # noqa: F403

langgraph_runtime/lifespan.py

@@ -0,0 +1,4 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from langgraph_runtime_inmem.lifespan import *  # noqa: F403

langgraph_runtime/metrics.py

@@ -0,0 +1,4 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from langgraph_runtime_inmem.metrics import *  # noqa: F403

langgraph_runtime/ops.py

@@ -0,0 +1,4 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from langgraph_runtime_inmem.ops import *  # noqa: F403

langgraph_runtime/queue.py

@@ -0,0 +1,4 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from langgraph_runtime_inmem.queue import *  # noqa: F403

langgraph_runtime/retry.py

@@ -0,0 +1,4 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from langgraph_runtime_inmem.retry import *  # noqa: F403

langgraph_runtime/store.py

@@ -0,0 +1,4 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from langgraph_runtime_inmem.store import *  # noqa: F403