langgraph-api 0.2.86__tar.gz → 0.2.89__tar.gz

{langgraph_api-0.2.86 → langgraph_api-0.2.89}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.2.86
+Version: 0.2.89
 Author-email: Nuno Campos <nuno@langchain.dev>, Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE
@@ -11,7 +11,7 @@ Requires-Dist: httpx>=0.25.0
 Requires-Dist: jsonschema-rs<0.30,>=0.20.0
 Requires-Dist: langchain-core>=0.3.64
 Requires-Dist: langgraph-checkpoint>=2.0.23
-Requires-Dist: langgraph-runtime-inmem<0.4,>=0.3.0
+Requires-Dist: langgraph-runtime-inmem<0.5,>=0.4.0
 Requires-Dist: langgraph-sdk>=0.1.71
 Requires-Dist: langgraph>=0.3.27
 Requires-Dist: langsmith>=0.3.45

langgraph_api-0.2.89/langgraph_api/__init__.py

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

{langgraph_api-0.2.86 → langgraph_api-0.2.89}/langgraph_api/auth/langsmith/backend.py

@@ -14,7 +14,6 @@ from langgraph_api.config import (
     LANGSMITH_AUTH_VERIFY_TENANT_ID,
     LANGSMITH_TENANT_ID,
 )
-from langgraph_api.utils.cache import LRUCache
 
 
 class AuthDict(TypedDict):
@@ -31,6 +30,8 @@ class AuthCacheEntry(TypedDict):
 
 class LangsmithAuthBackend(AuthenticationBackend):
     def __init__(self):
+        from langgraph_api.utils.cache import LRUCache
+
         self._cache = LRUCache[AuthCacheEntry](max_size=1000, ttl=60)
 
     def _get_cache_key(self, headers):

{langgraph_api-0.2.86 → langgraph_api-0.2.89}/langgraph_api/config.py

@@ -6,6 +6,8 @@ import orjson
 from starlette.config import Config, undefined
 from starlette.datastructures import CommaSeparatedStrings
 
+from langgraph_api import traceblock
+
 # types
 
 
@@ -372,3 +374,5 @@ if not os.getenv("LANGCHAIN_REVISION_ID") and (
     # This is respected by the langsmith SDK env inference
     # https://github.com/langchain-ai/langsmith-sdk/blob/1b93e4c13b8369d92db891ae3babc3e2254f0e56/python/langsmith/env/_runtime_env.py#L190
     os.environ["LANGCHAIN_REVISION_ID"] = ref_sha
+
+traceblock.patch_requests()

{langgraph_api-0.2.86 → langgraph_api-0.2.89}/langgraph_api/js/client.mts

@@ -56,6 +56,7 @@ import {
   getStaticGraphSchema,
 } from "@langchain/langgraph-api/schema";
 import { filterValidExportPath } from "./src/utils/files.mts";
+import { patchFetch } from "./traceblock.mts";
 
 const logger = createLogger({
   level: "debug",
@@ -1114,6 +1115,6 @@ async function getNodesExecutedRequest(
   nodesExecuted = 0;
   return { nodesExecuted: value };
 }
-
+patchFetch();
 asyncExitHook(() => awaitAllCallbacks(), { wait: 3_000 });
 main();

langgraph_api-0.2.89/langgraph_api/js/traceblock.mts

@@ -0,0 +1,25 @@
+import { overrideFetchImplementation } from "langsmith";
+
+const RUNS_RE = /^https:\/\/api\.smith\.langchain\.com\/.*runs(\/|$)/i;
+
+export function patchFetch() {
+  const shouldBlock =
+    typeof process !== "undefined" &&
+    !!(process.env && process.env.LANGSMITH_DISABLE_SAAS_RUNS === "true");
+
+  if (shouldBlock) {
+    overrideFetchImplementation(
+      async (input: RequestInfo, init?: RequestInit) => {
+        const req = input instanceof Request ? input : new Request(input, init);
+
+        if (req.method.toUpperCase() === "POST" && RUNS_RE.test(req.url)) {
+          throw new Error(
+            `Policy-blocked POST to ${new URL(req.url).pathname} — run tracking disabled`,
+          );
+        }
+
+        return fetch(req);
+      },
+    );
+  }
+}

{langgraph_api-0.2.86 → langgraph_api-0.2.89}/langgraph_api/models/run.py

@@ -312,6 +312,7 @@ async def create_valid_run(
     after_seconds = payload.get("after_seconds", 0)
     configurable["__after_seconds__"] = after_seconds
     put_time_start = time.time()
+    if_not_exists = payload.get("if_not_exists", "reject")
     run_coro = Runs.put(
         conn,
         assistant_id,
@@ -337,7 +338,7 @@ async def create_valid_run(
         multitask_strategy=multitask_strategy,
         prevent_insert_if_inflight=prevent_insert_if_inflight,
         after_seconds=after_seconds,
-        if_not_exists=payload.get("if_not_exists", "reject"),
+        if_not_exists=if_not_exists,
     )
     run_ = await run_coro
 
@@ -364,7 +365,7 @@ async def create_valid_run(
             stream_mode=stream_mode,
             temporary=temporary,
             after_seconds=after_seconds,
-            if_not_exists=payload.get("if_not_exists", "reject"),
+            if_not_exists=if_not_exists,
             run_create_ms=(
                 int(time.time() * 1_000) - request_start_time
                 if request_start_time

langgraph_api-0.2.89/langgraph_api/traceblock.py

@@ -0,0 +1,22 @@
+import os
+from urllib.parse import urlparse
+
+from requests.sessions import Session
+
+_HOST = "api.smith.langchain.com"
+_PATH_PREFIX = "/runs"
+
+
+def patch_requests():
+    if os.getenv("LANGSMITH_DISABLE_SAAS_RUNS") != "true":
+        return
+    _orig = Session.request
+
+    def _guard(self, method, url, *a, **kw):
+        if method.upper() == "POST":
+            u = urlparse(url)
+            if u.hostname == _HOST and _PATH_PREFIX in u.path:
+                raise RuntimeError(f"POST to {url} blocked by policy")
+        return _orig(self, method, url, *a, **kw)
+
+    Session.request = _guard

{langgraph_api-0.2.86 → langgraph_api-0.2.89}/langgraph_api/worker.py

@@ -113,6 +113,8 @@ async def worker(
         run_creation_ms=run_creation_ms,
         run_queue_ms=ms(run_started_at_dt, run["created_at"]),
         run_stream_start_ms=ms(run_stream_started_at_dt, run_started_at_dt),
+        temporary=temporary,
+        resumable=resumable,
     )
 
     def on_checkpoint(checkpoint_arg: CheckpointPayload):
@@ -242,47 +244,55 @@ async def worker(
                             run_id=str(run_id),
                             run_attempt=attempt,
                         )
-                await Threads.set_joint_status(
-                    conn, run["thread_id"], run_id, status, checkpoint=checkpoint
-                )
+                if not temporary:
+                    await Threads.set_joint_status(
+                        conn, run["thread_id"], run_id, status, checkpoint=checkpoint
+                    )
             elif isinstance(exception, TimeoutError):
                 status = "timeout"
                 await logger.awarning(
                     "Background run timed out",
                     **log_info,
                 )
-                await Threads.set_joint_status(
-                    conn, run["thread_id"], run_id, status, checkpoint=checkpoint
-                )
-            elif isinstance(exception, UserRollback):
-                status = "rollback"
-                try:
+                if not temporary:
                     await Threads.set_joint_status(
                         conn, run["thread_id"], run_id, status, checkpoint=checkpoint
                     )
-                    await logger.ainfo(
-                        "Background run rolled back",
-                        **log_info,
-                    )
-                except HTTPException as e:
-                    if e.status_code == 404:
+            elif isinstance(exception, UserRollback):
+                status = "rollback"
+                if not temporary:
+                    try:
+                        await Threads.set_joint_status(
+                            conn,
+                            run["thread_id"],
+                            run_id,
+                            status,
+                            checkpoint=checkpoint,
+                        )
                         await logger.ainfo(
-                            "Ignoring rollback error for missing run",
+                            "Background run rolled back",
                             **log_info,
                         )
-                    else:
-                        raise
-
-                checkpoint = None  # reset the checkpoint
+                    except HTTPException as e:
+                        if e.status_code == 404:
+                            await logger.ainfo(
+                                "Ignoring rollback error for missing run",
+                                **log_info,
+                            )
+                        else:
+                            raise
+
+                    checkpoint = None  # reset the checkpoint
             elif isinstance(exception, UserInterrupt):
                 status = "interrupted"
                 await logger.ainfo(
                     "Background run interrupted",
                     **log_info,
                 )
-                await Threads.set_joint_status(
-                    conn, run["thread_id"], run_id, status, checkpoint, exception
-                )
+                if not temporary:
+                    await Threads.set_joint_status(
+                        conn, run["thread_id"], run_id, status, checkpoint, exception
+                    )
             elif isinstance(exception, ALL_RETRIABLE_EXCEPTIONS):
                 status = "retry"
                 await logger.awarning(
@@ -290,6 +300,7 @@ async def worker(
                     **log_info,
                 )
                 # Don't update thread status yet.
+                # Apply this even for temporary runs, so we retry
                 await Runs.set_status(conn, run_id, "pending")
             else:
                 status = "error"
@@ -303,13 +314,14 @@ async def worker(
                     exc_info=not isinstance(exception, RemoteException),
                     **log_info,
                 )
-                await Threads.set_joint_status(
-                    conn, run["thread_id"], run_id, status, checkpoint, exception
-                )
+                if not temporary:
+                    await Threads.set_joint_status(
+                        conn, run["thread_id"], run_id, status, checkpoint, exception
+                    )
 
             # delete thread if it's temporary and we don't want to retry
             if temporary and not isinstance(exception, ALL_RETRIABLE_EXCEPTIONS):
-                await Threads.delete(conn, run["thread_id"])
+                await Threads._delete_with_run(conn, run["thread_id"], run_id)
 
         if isinstance(exception, ALL_RETRIABLE_EXCEPTIONS):
             await logger.awarning("RETRYING", exc_info=exception)

{langgraph_api-0.2.86 → langgraph_api-0.2.89}/pyproject.toml

@@ -31,7 +31,7 @@ dependencies = [
     "cryptography>=42.0.0,<45.0",
     "langgraph-sdk>=0.1.71",
     "cloudpickle>=3.0.0",
-    "langgraph-runtime-inmem>=0.3.0,<0.4",
+    "langgraph-runtime-inmem>=0.4.0,<0.5",
     "truststore>=0.1",
 ]
 

langgraph_api-0.2.86/langgraph_api/__init__.py

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