langgraph-api 0.2.43__tar.gz → 0.2.45__tar.gz

{langgraph_api-0.2.43 → langgraph_api-0.2.45}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.2.43
+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.43 → langgraph_api-0.2.45}/constraints.txt

@@ -4,8 +4,8 @@
 langgraph>=0.3.27
 langgraph-sdk>=0.1.66
 langgraph-checkpoint>=2.0.23
-langchain-core>=0.2.43
-langsmith>=0.1.112
+langchain-core>=0.3.64
+langsmith>=0.3.45
 orjson>=3.9.7,<3.10.17
 httpx>=0.25.0
 tenacity>=8.0.0

langgraph_api-0.2.45/langgraph_api/__init__.py

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

{langgraph_api-0.2.43 → langgraph_api-0.2.45}/langgraph_api/config.py

@@ -241,6 +241,9 @@ BG_JOB_ISOLATED_LOOPS = env("BG_JOB_ISOLATED_LOOPS", cast=bool, default=False)
 BG_JOB_SHUTDOWN_GRACE_PERIOD_SECS = env(
     "BG_JOB_SHUTDOWN_GRACE_PERIOD_SECS", cast=int, default=3600
 )
+MAX_STREAM_CHUNK_SIZE_BYTES = env(
+    "MAX_STREAM_CHUNK_SIZE_BYTES", cast=int, default=1024 * 1024 * 128
+)
 
 
 def _parse_thread_ttl(value: str | None) -> ThreadTTLConfig | None:

{langgraph_api-0.2.43 → langgraph_api-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-0.2.43 → langgraph_api-0.2.45}/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-0.2.43 → langgraph_api-0.2.45}/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-0.2.43 → langgraph_api-0.2.45}/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.43 → langgraph_api-0.2.45}/langgraph_api/worker.py

@@ -90,11 +90,11 @@ async def worker(
         if request_created_at is not None
         else None
     )
-
     async with (
         connect() as conn,
         set_auth_ctx_for_run(run["kwargs"]),
         Runs.enter(run_id, main_loop) as done,
+        AsyncExitStack() as exit_stack,
     ):
         temporary = run["kwargs"].get("temporary", False)
         resumable = run["kwargs"].get("resumable", False)
@@ -155,7 +155,6 @@ async def worker(
                     )
 
                 raise RuntimeError(error_message)
-            exit_stack = AsyncExitStack()
             if temporary:
                 stream = astream_state(exit_stack, conn, cast(Run, run), attempt, done)
             else:

langgraph_api-0.2.45/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_api-0.2.45/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_api-0.2.45/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_api-0.2.45/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_api-0.2.45/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_api-0.2.45/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_api-0.2.45/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_api-0.2.45/langgraph_runtime/store.py

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

{langgraph_api-0.2.43 → langgraph_api-0.2.45}/pyproject.toml

@@ -21,9 +21,9 @@ dependencies = [
     "langgraph-checkpoint>=2.0.23",
     "orjson>=3.9.7",
     "uvicorn>=0.26.0",
-    "langsmith>=0.1.112",
+    "langsmith>=0.3.45",
     "httpx>=0.25.0",
-    "langchain-core>=0.2.38",
+    "langchain-core>=0.3.64",
     "tenacity>=8.0.0",
     "jsonschema-rs>=0.20.0,<0.30",
     "structlog>=24.1.0,<26",