langgraph-api 0.0.16__py3-none-any.whl → 0.0.18__py3-none-any.whl

langgraph_api/api/assistants.py

@@ -1,6 +1,7 @@
 from typing import Any
 from uuid import uuid4
 
+import structlog
 from langchain_core.runnables.utils import create_model
 from langgraph.pregel import Pregel
 from starlette.exceptions import HTTPException
@@ -23,6 +24,8 @@ from langgraph_storage.database import connect
 from langgraph_storage.ops import Assistants
 from langgraph_storage.retry import retry_db
 
+logger = structlog.stdlib.get_logger(__name__)
+
 
 def _state_jsonschema(graph: Pregel) -> dict | None:
     fields: dict = {}
@@ -39,11 +42,17 @@ def _state_jsonschema(graph: Pregel) -> dict | None:
 def _graph_schemas(graph: Pregel) -> dict:
     try:
         input_schema = graph.get_input_jsonschema()
-    except Exception:
+    except Exception as e:
+        logger.warning(
+            f"Failed to get input schema for graph {graph.name} with error: `{str(e)}`"
+        )
         input_schema = None
     try:
         output_schema = graph.get_output_jsonschema()
-    except Exception:
+    except Exception as e:
+        logger.warning(
+            f"Failed to get output schema for graph {graph.name} with error: `{str(e)}`"
+        )
         output_schema = None
     state_schema = _state_jsonschema(graph)
     try:
@@ -52,7 +61,10 @@ def _graph_schemas(graph: Pregel) -> dict:
             if "configurable" in graph.config_schema().__fields__
             else {}
         )
-    except Exception:
+    except Exception as e:
+        logger.warning(
+            f"Failed to get config schema for graph {graph.name} with error: `{str(e)}`"
+        )
         config_schema = None
     return {
         "input_schema": input_schema,
@@ -141,7 +153,13 @@ async def get_assistant_graph(
         if isinstance(graph, BaseRemotePregel):
             drawable_graph = await graph.fetch_graph(xray=xray)
             return ApiResponse(drawable_graph.to_json())
-        return ApiResponse(graph.get_graph(xray=xray).to_json())
+
+        try:
+            return ApiResponse(graph.get_graph(xray=xray).to_json())
+        except NotImplementedError:
+            raise HTTPException(
+                422, detail="The graph does not support visualization"
+            ) from None
 
 
 @retry_db
@@ -167,16 +185,21 @@ async def get_assistant_subgraphs(
                 )
             )
 
-        return ApiResponse(
-            {
-                ns: _graph_schemas(subgraph)
-                async for ns, subgraph in graph.aget_subgraphs(
-                    namespace=namespace,
-                    recurse=request.query_params.get("recurse", "False")
-                    in ("true", "True"),
-                )
-            }
-        )
+        try:
+            return ApiResponse(
+                {
+                    ns: _graph_schemas(subgraph)
+                    async for ns, subgraph in graph.aget_subgraphs(
+                        namespace=namespace,
+                        recurse=request.query_params.get("recurse", "False")
+                        in ("true", "True"),
+                    )
+                }
+            )
+        except NotImplementedError:
+            raise HTTPException(
+                422, detail="The graph does not support visualization"
+            ) from None
 
 
 @retry_db
@@ -205,11 +228,17 @@ async def get_assistant_schemas(
 
         try:
             input_schema = graph.get_input_schema().schema()
-        except Exception:
+        except Exception as e:
+            logger.warning(
+                f"Failed to get input schema for graph {graph.name} with error: `{str(e)}`"
+            )
             input_schema = None
         try:
             output_schema = graph.get_output_schema().schema()
-        except Exception:
+        except Exception as e:
+            logger.warning(
+                f"Failed to get output schema for graph {graph.name} with error: `{str(e)}`"
+            )
             output_schema = None
 
         state_schema = _state_jsonschema(graph)
@@ -219,8 +248,11 @@ async def get_assistant_schemas(
                 if "configurable" in graph.config_schema().__fields__
                 else {}
             )
-        except Exception:
+        except Exception as e:
             config_schema = None
+            logger.warning(
+                f"Failed to get config schema for graph {graph.name} with error: `{str(e)}`"
+            )
         return ApiResponse(
             {
                 "graph_id": assistant["graph_id"],

langgraph_api/auth/custom.py

@@ -16,7 +16,6 @@ from starlette.authentication import (
     AuthenticationBackend,
     AuthenticationError,
     BaseUser,
-    SimpleUser,
 )
 from starlette.concurrency import run_in_threadpool
 from starlette.exceptions import HTTPException
@@ -478,6 +477,11 @@ class ProxyUser(BaseUser):
         return getattr(self._user, name)
 
 
+class SimpleUser(ProxyUser):
+    def __init__(self, username: str):
+        super().__init__(DotDict({"identity": username}))
+
+
 def _normalize_auth_response(
     response: Any,
 ) -> tuple[AuthCredentials, BaseUser]:

langgraph_api/auth/noop.py

@@ -2,11 +2,19 @@ from starlette.authentication import (
     AuthCredentials,
     AuthenticationBackend,
     BaseUser,
-    UnauthenticatedUser,
+)
+from starlette.authentication import (
+    UnauthenticatedUser as StarletteUnauthenticatedUser,
 )
 from starlette.requests import HTTPConnection
 
 
+class UnauthenticatedUser(StarletteUnauthenticatedUser):
+    @property
+    def identity(self) -> str:
+        return ""
+
+
 class NoopAuthBackend(AuthenticationBackend):
     async def authenticate(
         self, conn: HTTPConnection

langgraph_api/config.py

@@ -6,6 +6,16 @@ from starlette.datastructures import CommaSeparatedStrings
 
 env = Config()
 
+
+def _parse_json(json: str | None) -> dict | None:
+    if not json:
+        return None
+    parsed = orjson.loads(json)
+    if not parsed:
+        return None
+    return parsed
+
+
 STATS_INTERVAL_SECS = env("STATS_INTERVAL_SECS", cast=int, default=60)
 HTTP_CONCURRENCY = env("HTTP_CONCURRENCY", cast=int, default=10)
 
@@ -21,10 +31,57 @@ REDIS_MAX_CONNECTIONS = env("REDIS_MAX_CONNECTIONS", cast=int, default=500)
 # server
 
 CORS_ALLOW_ORIGINS = env("CORS_ALLOW_ORIGINS", cast=CommaSeparatedStrings, default="*")
+CORS_CONFIG = env("CORS_CONFIG", cast=_parse_json, default=None)
+"""
+{
+    "type": "object",
+    "properties": {
+        "allow_origins": {
+            "type": "array",
+            "items": {"type": "string"},
+            "default": []
+        },
+        "allow_methods": {
+            "type": "array", 
+            "items": {"type": "string"},
+            "default": ["GET"]
+        },
+        "allow_headers": {
+            "type": "array",
+            "items": {"type": "string"},
+            "default": []
+        },
+        "allow_credentials": {
+            "type": "boolean",
+            "default": false
+        },
+        "allow_origin_regex": {
+            "type": ["string", "null"],
+            "default": null
+        },
+        "expose_headers": {
+            "type": "array",
+            "items": {"type": "string"},
+            "default": []
+        },
+        "max_age": {
+            "type": "integer",
+            "default": 600
+        }
+    }
+}
+"""
+if CORS_CONFIG is not None and CORS_ALLOW_ORIGINS != "*":
+    raise ValueError("CORS_CONFIG and CORS_ALLOW_ORIGINS cannot be set together")
 
 # queue
 
-BG_JOB_NO_DELAY = env("BG_JOB_NO_DELAY", cast=bool, default=False)
+BG_JOB_NO_DELAY = env("BG_JOB_NO_DELAY", cast=bool, default=None)
+BG_JOB_DELAY = env("BG_JOB_DELAY", cast=float, default=0.5)
+if BG_JOB_NO_DELAY is True:
+    BG_JOB_DELAY = 0
+
+
 N_JOBS_PER_WORKER = env("N_JOBS_PER_WORKER", cast=int, default=10)
 BG_JOB_TIMEOUT_SECS = env("BG_JOB_TIMEOUT_SECS", cast=float, default=3600)
 FF_CRONS_ENABLED = env("FF_CRONS_ENABLED", cast=bool, default=True)
@@ -33,18 +90,7 @@ FF_JS_ZEROMQ_ENABLED = env("FF_JS_ZEROMQ_ENABLED", cast=bool, default=False)
 # auth
 
 LANGGRAPH_AUTH_TYPE = env("LANGGRAPH_AUTH_TYPE", cast=str, default="noop")
-
-
-def _parse_auth(auth: str | None) -> dict | None:
-    if not auth:
-        return None
-    parsed = orjson.loads(auth)
-    if not parsed:
-        return None
-    return parsed
-
-
-LANGGRAPH_AUTH = env("LANGGRAPH_AUTH", cast=_parse_auth, default=None)
+LANGGRAPH_AUTH = env("LANGGRAPH_AUTH", cast=_parse_json, default=None)
 LANGSMITH_TENANT_ID = env("LANGSMITH_TENANT_ID", cast=str, default=None)
 LANGSMITH_AUTH_VERIFY_TENANT_ID = env(
     "LANGSMITH_AUTH_VERIFY_TENANT_ID",
@@ -52,7 +98,6 @@ LANGSMITH_AUTH_VERIFY_TENANT_ID = env(
     default=LANGSMITH_TENANT_ID is not None,
 )
 
-
 if LANGGRAPH_AUTH_TYPE == "langsmith":
     LANGSMITH_AUTH_ENDPOINT = env("LANGSMITH_AUTH_ENDPOINT", cast=str)
     LANGSMITH_TENANT_ID = env("LANGSMITH_TENANT_ID", cast=str)

langgraph_api/cron_scheduler.py

@@ -3,10 +3,10 @@ from random import random
 
 import structlog
 from langchain_core.runnables.config import run_in_executor
-from starlette.authentication import SimpleUser
 
 from langgraph_api.models.run import create_valid_run
-from langgraph_api.utils import next_cron_date, with_user
+from langgraph_api.queue import set_auth_ctx_for_run
+from langgraph_api.utils import next_cron_date
 from langgraph_storage.database import connect
 from langgraph_storage.ops import Crons
 from langgraph_storage.retry import retry_db
@@ -23,10 +23,13 @@ async def cron_scheduler():
         try:
             async with connect() as conn:
                 async for cron in Crons.next(conn):
-                    async with with_user(SimpleUser(str(cron["user_id"])), None):
+                    run_payload = cron["payload"]
+
+                    async with set_auth_ctx_for_run(
+                        run_payload, user_id=cron["user_id"]
+                    ):
                         logger.debug(f"Scheduling cron run {cron}")
                         try:
-                            run_payload = cron["payload"]
                             run = await create_valid_run(
                                 conn,
                                 thread_id=(

langgraph_api/js/client.mts

@@ -446,6 +446,7 @@ const StreamModeSchema = z.union([
 const ExtraStreamModeSchema = z.union([
   StreamModeSchema,
   z.literal("messages"),
+  z.literal("messages-tuple"),
 ]);
 
 const StreamEventsPayload = z.object({
@@ -476,11 +477,14 @@ async function* streamEventsRequest(
         ? payload.stream_mode
         : [payload.stream_mode];
 
-  const graphStreamMode: Set<"updates" | "debug" | "values"> = new Set();
+  const graphStreamMode: Set<"updates" | "debug" | "values" | "messages"> =
+    new Set();
   if (payload.stream_mode) {
     for (const mode of userStreamMode) {
       if (mode === "messages") {
         graphStreamMode.add("values");
+      } else if (mode === "messages-tuple") {
+        graphStreamMode.add("messages");
       } else {
         graphStreamMode.add(mode);
       }

langgraph_api/js/client.new.mts

@@ -504,6 +504,7 @@ const StreamModeSchema = z.union([
 const ExtraStreamModeSchema = z.union([
   StreamModeSchema,
   z.literal("messages"),
+  z.literal("messages-tuple"),
 ]);
 
 const StreamEventsPayload = z.object({
@@ -534,11 +535,14 @@ async function* streamEventsRequest(
         ? payload.stream_mode
         : [payload.stream_mode];
 
-  const graphStreamMode: Set<"updates" | "debug" | "values"> = new Set();
+  const graphStreamMode: Set<"updates" | "debug" | "values" | "messages"> =
+    new Set();
   if (payload.stream_mode) {
     for (const mode of userStreamMode) {
       if (mode === "messages") {
         graphStreamMode.add("values");
+      } else if (mode === "messages-tuple") {
+        graphStreamMode.add("messages");
       } else {
         graphStreamMode.add(mode);
       }

langgraph_api/js/package.json

@@ -22,7 +22,7 @@
     "undici": "^6.19.7",
     "uuid": "^10.0.0",
     "winston": "^3.15.0",
-    "zeromq": "^6.1.1",
+    "zeromq": "^6.3.0",
     "zod": "^3.23.8"
   },
   "devDependencies": {

langgraph_api/js/tests/api.test.mts

@@ -686,6 +686,40 @@ describe("runs", () => {
     expect(run.status).toBe("success");
   });
 
+  it.concurrent("stream messages tuple", async () => {
+    const assistant = await client.assistants.create({ graphId: "agent" });
+    const thread = await client.threads.create();
+    const input = {
+      messages: [{ type: "human", content: "foo", id: "initial-message" }],
+    };
+    const stream = await client.runs.stream(
+      thread.thread_id,
+      assistant.assistant_id,
+      { input, streamMode: "messages-tuple", config: globalConfig }
+    );
+
+    const chunks = await gatherIterator(stream);
+    const runId = findLast(chunks, (i) => i.event === "metadata")?.data.run_id;
+    expect(runId).not.toBeNull();
+
+    const messages = chunks
+      .filter((i) => i.event === "messages")
+      .map((i) => i.data[0]);
+
+    expect(messages).toHaveLength("begin".length + "end".length + 1);
+    expect(messages).toMatchObject([
+      ..."begin".split("").map((c) => ({ content: c })),
+      { content: "tool_call__begin" },
+      ..."end".split("").map((c) => ({ content: c })),
+    ]);
+
+    const seenEventTypes = new Set(chunks.map((i) => i.event));
+    expect(seenEventTypes).toEqual(new Set(["metadata", "messages"]));
+
+    const run = await client.runs.get(thread.thread_id, runId as string);
+    expect(run.status).toBe("success");
+  });
+
   it.concurrent("stream mixed modes", async () => {
     const assistant = await client.assistants.create({ graphId: "agent" });
     const thread = await client.threads.create();

langgraph_api/js/tests/graphs/agent.mts

@@ -6,11 +6,13 @@ import {
   END,
   messagesStateReducer,
   SharedValue,
-  LangGraphRunnableConfig,
   interrupt,
+  type LangGraphRunnableConfig,
 } from "@langchain/langgraph";
 import { FakeListChatModel } from "@langchain/core/utils/testing";
-
+import { ChatGenerationChunk } from "@langchain/core/outputs";
+import { v4 as uuidv4 } from "uuid";
+import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
 const GraphAnnotationOutput = Annotation.Root({
   messages: Annotation<BaseMessage[]>({
     reducer: messagesStateReducer,
@@ -20,6 +22,7 @@ const GraphAnnotationOutput = Annotation.Root({
   interrupt: Annotation<boolean>(),
   keyOne: Annotation<string | null>(),
   keyTwo: Annotation<string | null>(),
+  sleep: Annotation<number | null>(),
 });
 
 const GraphAnnotationInput = Annotation.Root({
@@ -28,13 +31,58 @@ const GraphAnnotationInput = Annotation.Root({
   sharedStateFromStoreConfig: Annotation<Record<string, any> | null>,
 });
 
+class StableFakeListChatModel extends FakeListChatModel {
+  streamMessageId: string = uuidv4();
+
+  async *_streamResponseChunks(
+    _messages: BaseMessage[],
+    options: this["ParsedCallOptions"],
+    runManager?: CallbackManagerForLLMRun
+  ): AsyncGenerator<ChatGenerationChunk> {
+    const response = this._currentResponse();
+    this._incrementResponse();
+    this.streamMessageId = uuidv4();
+
+    if (this.emitCustomEvent) {
+      await runManager?.handleCustomEvent("some_test_event", {
+        someval: true,
+      });
+    }
+
+    for await (const text of response) {
+      await this._sleepIfRequested();
+      if (options?.thrownErrorString) {
+        throw new Error(options.thrownErrorString);
+      }
+      const chunk = this._createResponseChunk(text);
+
+      // ensure stable ID
+      chunk.message.id = this.streamMessageId;
+      chunk.message.lc_kwargs.id = this.streamMessageId;
+
+      yield chunk;
+
+      void runManager?.handleLLMNewToken(
+        text,
+        undefined,
+        undefined,
+        undefined,
+        undefined,
+        { chunk }
+      );
+    }
+  }
+}
+
 // For shared state
 const namespace = ["sharedState", "data"];
 const key = "user_id";
 
 const modelMap: Record<string, FakeListChatModel> = {};
 const getModel = (threadId: string) => {
-  modelMap[threadId] ??= new FakeListChatModel({ responses: ["begin", "end"] });
+  modelMap[threadId] ??= new StableFakeListChatModel({
+    responses: ["begin", "end"],
+  });
   return modelMap[threadId];
 };
 
@@ -44,6 +92,11 @@ const agentNode = async (
 ) => {
   if (state.interrupt) interrupt("i want to interrupt");
 
+  if (state.sleep != null && state.messages.at(-1)?.getType() === "human") {
+    const sleep = state.sleep;
+    await new Promise((resolve) => setTimeout(resolve, sleep * 1000));
+  }
+
   const model = getModel(config.configurable?.thread_id ?? "$");
   const response = await model.invoke(state.messages);
   const sharedStateValue = state.sharedState?.data?.user_id ?? null;

langgraph_api/js/yarn.lock

@@ -1486,10 +1486,10 @@ nanoid@^3.3.7:
   resolved "https://registry.yarnpkg.com/nanoid/-/nanoid-3.3.7.tgz#d0c301a691bc8d54efa0a2226ccf3fe2fd656bd8"
   integrity sha512-eSRppjcPIatRIMC1U6UngP8XFcz8MQWGQdt1MTBQ7NaAmvXDfvNxbvWV3x2y6CdEUciCSsDHDQZbhYaB8QEo2g==
 
-node-addon-api@^8.2.1:
-  version "8.2.1"
-  resolved "https://registry.yarnpkg.com/node-addon-api/-/node-addon-api-8.2.1.tgz#43a993f110b88e22ba48bcd65e16b92165a6b002"
-  integrity sha512-vmEOvxwiH8tlOcv4SyE8RH34rI5/nWVaigUeAUPawC6f0+HoDthwI0vkMu4tbtsZrXq6QXFfrkhjofzKEs5tpA==
+node-addon-api@^8.3.0:
+  version "8.3.0"
+  resolved "https://registry.yarnpkg.com/node-addon-api/-/node-addon-api-8.3.0.tgz#ec3763f18befc1cdf66d11e157ce44d5eddc0603"
+  integrity sha512-8VOpLHFrOQlAH+qA0ZzuGRlALRA6/LVh8QJldbrC4DY0hXoMP0l4Acq8TzFC018HztWiRqyCEj2aTWY2UvnJUg==
 
 node-int64@^0.4.0:
   version "0.4.0"
@@ -2173,13 +2173,13 @@ yallist@^4.0.0:
   resolved "https://registry.yarnpkg.com/yallist/-/yallist-4.0.0.tgz#9bb92790d9c0effec63be73519e11a35019a3a72"
   integrity sha512-3wdGidZyq5PB084XLES5TpOSRA3wjXAlIWMhum2kRcv/41Sn2emQ0dycQW4uZXLejwKvg6EsvbdlVL+FYEct7A==
 
-zeromq@^6.1.1:
-  version "6.1.2"
-  resolved "https://registry.yarnpkg.com/zeromq/-/zeromq-6.1.2.tgz#89f1cae83f1a4615dab02dbc7881b10ea1a33c73"
-  integrity sha512-5lqsW2UXnKhFhBbIm0wzrRnZmWaGI2b2bm4E03rYMK1c1jBMN0wbPnuyqqhGg7QkzBUmpqf+LyjLhg2TRJmvow==
+zeromq@^6.3.0:
+  version "6.3.0"
+  resolved "https://registry.yarnpkg.com/zeromq/-/zeromq-6.3.0.tgz#19ffa286d128cc317847b5d615c079a67e3bae48"
+  integrity sha512-PG61AT4Y37NGJHSrp5SG2m4cKtRirso/5mm4Yf7l+upgNZZAugrGdhENsM5IcLS+WVVzbWVjEsLKpgLj/31SWQ==
   dependencies:
     "@aminya/cmake-ts" "^0.3.0-aminya.7"
-    node-addon-api "^8.2.1"
+    node-addon-api "^8.3.0"
 
 zod-to-json-schema@^3.22.3:
   version "3.23.5"

langgraph_api/models/run.py

@@ -8,7 +8,7 @@ from langgraph.checkpoint.base.id import uuid6
 from starlette.authentication import BaseUser
 from starlette.exceptions import HTTPException
 
-from langgraph_api.graph import get_assistant_id
+from langgraph_api.graph import GRAPHS, get_assistant_id
 from langgraph_api.schema import (
     All,
     Config,
@@ -92,7 +92,13 @@ def ensure_ids(
             assistant_id if isinstance(assistant_id, UUID) else UUID(assistant_id)
         ]
     except ValueError:
-        raise HTTPException(status_code=422, detail="Invalid assistant ID") from None
+        keys = ", ".join(GRAPHS.keys())
+        raise HTTPException(
+            status_code=422,
+            detail=f"Invalid assistant: '{assistant_id}'. Must be either:\n"
+            f"- A valid assistant UUID, or\n"
+            f"- One of the registered graphs: {keys}",
+        ) from None
     if thread_id:
         try:
             results.append(

langgraph_api/queue.py

@@ -2,14 +2,13 @@ import asyncio
 from collections.abc import AsyncGenerator
 from contextlib import AsyncExitStack, asynccontextmanager
 from datetime import UTC, datetime
-from random import random
 from typing import TypedDict, cast
 
 import structlog
 from langgraph.pregel.debug import CheckpointPayload, TaskResultPayload
 
-from langgraph_api.auth.custom import normalize_user
-from langgraph_api.config import BG_JOB_NO_DELAY, STATS_INTERVAL_SECS
+from langgraph_api.auth.custom import SimpleUser, normalize_user
+from langgraph_api.config import BG_JOB_DELAY, STATS_INTERVAL_SECS
 from langgraph_api.errors import (
     UserInterrupt,
     UserRollback,
@@ -127,13 +126,13 @@ async def queue(concurrency: int, timeout: float):
                 else:
                     semaphore.release()
                     await exit.aclose()
-                    await asyncio.sleep(0 if BG_JOB_NO_DELAY else random())
+                    await asyncio.sleep(BG_JOB_DELAY)
             except Exception as exc:
                 # keep trying to run the scheduler indefinitely
                 logger.exception("Background worker scheduler failed", exc_info=exc)
                 semaphore.release()
                 await exit.aclose()
-                await asyncio.sleep(0 if BG_JOB_NO_DELAY else random())
+                await asyncio.sleep(BG_JOB_DELAY)
     finally:
         logger.info("Shutting down background workers")
         for task in WORKERS:
@@ -157,12 +156,13 @@ class WorkerResult(TypedDict):
 
 
 @asynccontextmanager
-async def set_auth_ctx_for_run(run: Run) -> AsyncGenerator[None, None]:
+async def set_auth_ctx_for_run(
+    run_kwargs: dict, user_id: str | None = None
+) -> AsyncGenerator[None, None]:
+    # user_id is a fallback.
     try:
-        user = run["kwargs"]["config"]["configurable"]["langgraph_auth_user"]
-        permissions = run["kwargs"]["config"]["configurable"][
-            "langgraph_auth_permissions"
-        ]
+        user = run_kwargs["config"]["configurable"]["langgraph_auth_user"]
+        permissions = run_kwargs["config"]["configurable"]["langgraph_auth_permissions"]
         if user is not None:
             user = normalize_user(user)
             async with with_user(user, permissions):
@@ -171,6 +171,16 @@ async def set_auth_ctx_for_run(run: Run) -> AsyncGenerator[None, None]:
             yield None
 
     except KeyError:
+        if user_id is not None:
+            await logger.ainfo(
+                "Setting auth to backup user_id",
+                user_id=user_id,
+            )
+            async with with_user(SimpleUser(user_id)):
+                yield None
+        else:
+            yield None
+    except Exception:
         pass
 
 
@@ -191,7 +201,7 @@ async def worker(
     run_started_at = datetime.now(UTC)
     run_ended_at: str | None = None
 
-    async with set_auth_ctx_for_run(run), Runs.enter(run_id) as done, exit:
+    async with set_auth_ctx_for_run(run["kwargs"]), Runs.enter(run_id) as done, exit:
         temporary = run["kwargs"].get("temporary", False)
         run_created_at = run["created_at"].isoformat()
         await logger.ainfo(

langgraph_api/server.py

@@ -33,6 +33,11 @@ app = Starlette(
             allow_credentials=True,
             allow_methods=["*"],
             allow_headers=["*"],
+        )
+        if config.CORS_CONFIG is None
+        else Middleware(
+            CORSMiddleware,
+            **config.CORS_CONFIG,
         ),
         Middleware(LicenseValidationMiddleware),
         Middleware(AccessLoggerMiddleware, logger=logger),

langgraph_api/stream.py

@@ -130,7 +130,7 @@ async def astream_state(
     stream_modes_set: set[StreamMode] = set(stream_mode) - {"events"}
     if "debug" not in stream_modes_set:
         stream_modes_set.add("debug")
-    if "messages-tuple" in stream_modes_set:
+    if "messages-tuple" in stream_modes_set and not isinstance(graph, BaseRemotePregel):
         stream_modes_set.remove("messages-tuple")
         stream_modes_set.add("messages")
     # attach attempt metadata

langgraph_api/utils.py

@@ -6,9 +6,11 @@ from datetime import datetime
 from typing import Any, Protocol, TypeAlias, TypeVar
 
 from langgraph_sdk import Auth
-from starlette.authentication import AuthCredentials, BaseUser, SimpleUser
+from starlette.authentication import AuthCredentials, BaseUser
 from starlette.exceptions import HTTPException
 
+from langgraph_api.auth.custom import SimpleUser
+
 T = TypeVar("T")
 Row: TypeAlias = dict[str, Any]
 AuthContext = contextvars.ContextVar[Auth.types.BaseAuthContext | None](

{langgraph_api-0.0.16.dist-info → langgraph_api-0.0.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.16
+Version: 0.0.18
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos

{langgraph_api-0.0.16.dist-info → langgraph_api-0.0.18.dist-info}/RECORD

@@ -1,7 +1,7 @@
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 langgraph_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/api/__init__.py,sha256=zAdS_0jgjmCJK6E0VwEuNcNRkaknMXfQ2C_OES5tEI4,2066
-langgraph_api/api/assistants.py,sha256=nn-0Q5FTaEbdPq-oesrpVzqu223PDSzeejFy9fd5Xjw,11599
+langgraph_api/api/assistants.py,sha256=9wngelDC9vnSs2vYGTHDSCLf5KNds-6mgP1BWnfoY2M,12865
 langgraph_api/api/meta.py,sha256=hueasWpTDQ6xYLo9Bzt2jhNH8XQRzreH8FTeFfnRoxQ,2700
 langgraph_api/api/openapi.py,sha256=AUxfnD5hlRp7s-0g2hBC5dNSNk3HTwOLeJiF489DT44,2762
 langgraph_api/api/runs.py,sha256=wAzPXi_kcYB9BcLBL4FXgkBohWwCPIpe4XERnsnWnsA,16042
@@ -9,16 +9,16 @@ langgraph_api/api/store.py,sha256=VzAJVOwO0IxosBB7km5TTf2rhlWGyPkVz_LpvbxetVY,54
 langgraph_api/api/threads.py,sha256=taU61XPcCEhBPCYPZcMDsgVDwwWUWJs8p-PrXFXWY48,8661
 langgraph_api/asyncio.py,sha256=2fOlx-cZvuj1gQ867Kw1R_wsBsl9jdHYHcUtK2a-x-U,6264
 langgraph_api/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/auth/custom.py,sha256=g_u7FdKm1Qj8eu_MZdnJeMsI4DlWyU5Mg0rPJzdOTSE,20913
+langgraph_api/auth/custom.py,sha256=juCwlVxjWnCzcWI6Amae_4NfIOFtY39kaJ_zPF5AJaw,21025
 langgraph_api/auth/langsmith/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/auth/langsmith/backend.py,sha256=InScaL-HYCnxYEauhxU198gRZV9pJn9SzzBoR9Edn7g,2654
 langgraph_api/auth/langsmith/client.py,sha256=eKchvAom7hdkUXauD8vHNceBDDUijrFgdTV8bKd7x4Q,3998
 langgraph_api/auth/middleware.py,sha256=qc7SbaFoeWaqxS1wbjZ2PPQ4iI2p9T0shWL7c6g0ed4,1636
-langgraph_api/auth/noop.py,sha256=vDJmzG2vArJxVzdHePvrJWahEa0dvGnhc2LEMMeiFz0,391
+langgraph_api/auth/noop.py,sha256=Bk6Nf3p8D_iMVy_OyfPlyiJp_aEwzL-sHrbxoXpCbac,586
 langgraph_api/auth/studio_user.py,sha256=FzFQRROKDlA9JjtBuwyZvk6Mbwno5M9RVYjDO6FU3F8,186
 langgraph_api/cli.py,sha256=7vQQiD3F50r-8KkbuFjwIz8LLbdKUTd4xZGUJPiO3yQ,11688
-langgraph_api/config.py,sha256=HPNSCb-owhu_sU8KBLbTbKLdGIPGVLcSgdvKFKXg7uE,2816
-langgraph_api/cron_scheduler.py,sha256=CybK-9Jwopi_scObTHRyB7gyb0KjC4gqaT2GLe-WOFg,2587
+langgraph_api/config.py,sha256=YDGkyZQRdG2QRYwSwPKC9J3UOhq7Diy_ZdypAL-vbrw,4034
+langgraph_api/cron_scheduler.py,sha256=MW41-TSGUe5OuXycFTy7Ax7ypxHVAv-0ImLonRT8h8o,2629
 langgraph_api/errors.py,sha256=Bu_i5drgNTyJcLiyrwVE_6-XrSU50BHf9TDpttki9wQ,1690
 langgraph_api/graph.py,sha256=FombjYQkqj8jrXJFEVkl3m2UyFcq5nSVNswR2HoRsQY,16385
 langgraph_api/http.py,sha256=XrbyxpjtfSvnaWWh5ZLGpgZmY83WoDCrP_1GPguNiXI,4712
@@ -26,11 +26,11 @@ langgraph_api/http_logger.py,sha256=Sxo_q-65tElauRvkzVLt9lJojgNdgtcHGBYD0IRyX7M,
 langgraph_api/js/.gitignore,sha256=qAah3Fq0HWAlfRj5ktZyC6QRQIsAolGLRGcRukA1XJI,33
 langgraph_api/js/base.py,sha256=BpE8-xkUp8HFPRjSKx1tfUQubvoV4jYl6OwZdre3veI,209
 langgraph_api/js/build.mts,sha256=sAZXB3xVSoG8exJ1ZEytFyikRmEsXJ_0OlLjDDgD79o,1342
-langgraph_api/js/client.mts,sha256=ksiytm222KTUWj92ZnajqFku_y2AkRmfENmKie5LSPw,22519
-langgraph_api/js/client.new.mts,sha256=9FrArkM20IeD176Q7u5lJNruaQXqAfAdDcqJkF0TPPA,23493
+langgraph_api/js/client.mts,sha256=yFc4aeV5wm7N_z_K8gIukVXo6pHUBnp2Mlryl9B-yrA,22654
+langgraph_api/js/client.new.mts,sha256=54n0xc9SSDnrJDPsN_JcUKcRJJyQVIYuUrkKw3XDcio,23628
 langgraph_api/js/errors.py,sha256=Cm1TKWlUCwZReDC5AQ6SgNIVGD27Qov2xcgHyf8-GXo,361
 langgraph_api/js/global.d.ts,sha256=zR_zLYfpzyPfxpEFth5RgZoyfGulIXyZYPRf7cU0K0Y,106
-langgraph_api/js/package.json,sha256=BGnWhiMMvxocEuPciTq14fEHp5NFmHo6fYV8q62n3t4,840
+langgraph_api/js/package.json,sha256=i2wjOE0_u1DdUvlc7Ulgj-tg06kjKCsFL99nQKF7vQk,840
 langgraph_api/js/remote.py,sha256=D9cqcEgXau-fm_trpNwCHMra5BXntgUa469lgs_a9JQ,622
 langgraph_api/js/remote_new.py,sha256=T_Vr8459bax1C9xxqz_ZYmGivq5Vhspg2Iu9TL0Qc-Q,22707
 langgraph_api/js/remote_old.py,sha256=2a-3ooAYUZs8aPsfnXafbBd4pP7lRmokoU7TiO7P9Js,22546
@@ -45,10 +45,10 @@ langgraph_api/js/src/schema/types.template.mts,sha256=c-FA0Ykzp4KvPyYA6a-hDf60Kd
 langgraph_api/js/src/utils/importMap.mts,sha256=pX4TGOyUpuuWF82kXcxcv3-8mgusRezOGe6Uklm2O5A,1644
 langgraph_api/js/src/utils/pythonSchemas.mts,sha256=98IW7Z_VP7L_CHNRMb3_MsiV3BgLE2JsWQY_PQcRR3o,685
 langgraph_api/js/src/utils/serde.mts,sha256=OuyyO9btvwWd55rU_H4x91dFEJiaPxL-lL9O6Zgo908,742
-langgraph_api/js/tests/api.test.mts,sha256=2EpDEs888pJGdZnsyc76GdMp7uuRxM_SNlHBwITU-5I,55668
+langgraph_api/js/tests/api.test.mts,sha256=qJdISmuKo7AWZmR4thY9eq8h4EL5bWKlhI5x7LCJ8dQ,56919
 langgraph_api/js/tests/compose-postgres.yml,sha256=pbNfeqVUqhWILBuUdwAgQOYsVU_fgkCVm0YlTgU8og8,1721
 langgraph_api/js/tests/graphs/.gitignore,sha256=26J8MarZNXh7snXD5eTpV3CPFTht5Znv8dtHYCLNfkw,12
-langgraph_api/js/tests/graphs/agent.mts,sha256=fFHm9vW04UN_2mGcHVHqtFIvPhjyFurBg62DAW-GJd4,3889
+langgraph_api/js/tests/graphs/agent.mts,sha256=E9WMv0alMv0njUEECqEsqoRk9NXJUgXW7SyQJ3GOZ8k,5396
 langgraph_api/js/tests/graphs/delay.mts,sha256=CFneKxqI4bGGK0lYjSbe80QirowPQlsRSuhDUKfydhk,703
 langgraph_api/js/tests/graphs/error.mts,sha256=l4tk89449dj1BnEF_0ZcfPt0Ikk1gl8L1RaSnRfr3xo,487
 langgraph_api/js/tests/graphs/langgraph.json,sha256=frxd7ZWILdeMYSZgUBH6UO-IR7I2YJSOfOlx2mnO1sI,189
@@ -58,22 +58,22 @@ langgraph_api/js/tests/graphs/weather.mts,sha256=A7mLK3xW8h5B-ZyJNAyX2M2fJJwzPJz
 langgraph_api/js/tests/graphs/yarn.lock,sha256=q-1S--E5VWLYtkSv03shqtNzeDDv-N_J-N26FszLsjs,7903
 langgraph_api/js/tests/parser.test.mts,sha256=3zAbboUNhI-cY3hj4Ssr7J-sQXCBTeeI1ItrkG0Ftuk,26257
 langgraph_api/js/tests/utils.mts,sha256=2kTybJ3O7Yfe1q3ehDouqV54ibXkNzsPZ_wBZLJvY-4,421
-langgraph_api/js/yarn.lock,sha256=JtRgt5AXlsD4qBm1Nrtzxg-TLtZkFWtRKCvyS-V8CLg,103690
+langgraph_api/js/yarn.lock,sha256=4cFZXkAYtsdCO7nPBDNHehQ1kv3hYMRXwwXutupMHs8,103690
 langgraph_api/lifespan.py,sha256=Uj7NV-NqxxD1fgx_umM9pVqclcy-VlqrIxDljyj2he0,1820
 langgraph_api/logging.py,sha256=tiDNrEFwqaIdL5ywZv908OXlzzfXsPCws9GXeoFtBV8,3367
 langgraph_api/metadata.py,sha256=mih2G7ScQxiqyUlbksVXkqR3Oo-pM1b6lXtzOsgR1sw,3044
 langgraph_api/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/models/run.py,sha256=qBN_w8mUpSbAZNVL1683_DfZHQ0zvYMNyqGg-g1sz6Y,9625
+langgraph_api/models/run.py,sha256=1TzTmagDXFQD_LhIMRdguZHmrPSzztq1wiMjF63d2fc,9843
 langgraph_api/patch.py,sha256=94ddcTSZJe22JcpjxiSNjFZdYVnmeoWjk4IX4iBSoyk,1249
-langgraph_api/queue.py,sha256=nYtcjbqjwvELK5OXxD2aw5BWAlSJ-VPyCXSODMMXIj0,13353
+langgraph_api/queue.py,sha256=qRuM09mz8o7CkrBIwg-9lV3SW0TehdVGgeXvHc_adYk,13647
 langgraph_api/route.py,sha256=fM4qYCGbmH0a3_cV8uKocb1sLklehxO6HhdRXqLK6OM,4421
 langgraph_api/schema.py,sha256=4aZCFY-dxd_nTot71bdcd9S8QCIgKajuRyj0p2QfgJ4,5291
 langgraph_api/serde.py,sha256=VoJ7Z1IuqrQGXFzEP1qijAITtWCrmjtVqlCRuScjXJI,3533
-langgraph_api/server.py,sha256=afHDnL6b_fAIu_q4icnK60a74lHTTZOMIe1egdhRXIk,1522
+langgraph_api/server.py,sha256=OrFCbFMIL7L7K8Id4lNEI6soFBwCdcIN4maSwSfsj1o,1657
 langgraph_api/sse.py,sha256=2wNodCOP2eg7a9mpSu0S3FQ0CHk2BBV_vv0UtIgJIcc,4034
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
-langgraph_api/stream.py,sha256=Uygx6zcY5Wi9lBhRjtxqVDyLZSF1bsaqxg6mYoYVYcY,11900
-langgraph_api/utils.py,sha256=fMl3DHOQEwAqkFtrnP0Alfbrqw1UvwZ_JVLm-WTSQJk,2654
+langgraph_api/stream.py,sha256=MUYYNgwtLs1Mhq1dm12zda7j8uFYir49umigK6CnuXU,11944
+langgraph_api/utils.py,sha256=aIHPp_yu-NFUzs0jzNRm6mcqoZmtbLMZl1ugFmwWTss,2692
 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
@@ -87,9 +87,9 @@ langgraph_storage/retry.py,sha256=uvYFuXJ-T6S1QY1ZwkZHyZQbsvS-Ab68LSbzbUUSI2E,69
 langgraph_storage/store.py,sha256=D-p3cWc_umamkKp-6Cz3cAriSACpvM5nxUIvND6PuxE,2710
 langgraph_storage/ttl_dict.py,sha256=FlpEY8EANeXWKo_G5nmIotPquABZGyIJyk6HD9u6vqY,1533
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
-openapi.json,sha256=qf2Rw3ieawlAcSJu4mGonh9mNOb6solBD70CGL3w24A,124699
-langgraph_api-0.0.16.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.0.16.dist-info/METADATA,sha256=HLMkdfTmxuTMN59uOzDQ4VCRILZvgslETZShl2Mdruw,4041
-langgraph_api-0.0.16.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-langgraph_api-0.0.16.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.0.16.dist-info/RECORD,,
+openapi.json,sha256=L1Ap1o6oAg2vKqjWohq4gl1-GpxezjlddS_oZHjS-xE,125227
+langgraph_api-0.0.18.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.0.18.dist-info/METADATA,sha256=k6oxmbw-_QEHL7VOaAdecVtt7tkw7ccF6A4ywpGYYUk,4041
+langgraph_api-0.0.18.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+langgraph_api-0.0.18.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.0.18.dist-info/RECORD,,

openapi.json

@@ -3317,6 +3317,18 @@
               {
                 "type": "object"
               },
+              {
+                "type": "array"
+              },
+              {
+                "type": "string"
+              },
+              {
+                "type": "number"
+              },
+              {
+                "type": "boolean"
+              },
               {
                 "type": "null"
               }
@@ -3532,6 +3544,18 @@
               {
                 "type": "object"
               },
+              {
+                "type": "array"
+              },
+              {
+                "type": "string"
+              },
+              {
+                "type": "number"
+              },
+              {
+                "type": "boolean"
+              },
               {
                 "type": "null"
               }