langgraph-api 0.1.2__tar.gz → 0.1.3__tar.gz

{langgraph_api-0.1.2 → langgraph_api-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langgraph-api
-Version: 0.1.2
+Version: 0.1.3
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos

langgraph_api-0.1.3/langgraph_api/__init__.py

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

{langgraph_api-0.1.2 → langgraph_api-0.1.3}/langgraph_api/auth/middleware.py

@@ -36,7 +36,7 @@ def on_error(conn: HTTPConnection, exc: AuthenticationError):
 
 class ConditionalAuthenticationMiddleware(AuthenticationMiddleware):
     async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
-        if scope["root_path"] == "/noauth":
+        if (root_path := scope.get("root_path")) and root_path.startswith("/noauth"):
             # disable auth for requests originating from SDK ASGI transport
             # root_path cannot be set from a request, so safe to use as auth bypass
             await self.app(scope, receive, send)
@@ -46,7 +46,6 @@ class ConditionalAuthenticationMiddleware(AuthenticationMiddleware):
             # disable auth for UI asset requests
             await self.app(scope, receive, send)
             return
-
         return await super().__call__(scope, receive, send)
 
 

{langgraph_api-0.1.2 → langgraph_api-0.1.3}/langgraph_api/config.py

@@ -37,6 +37,8 @@ class HttpConfig(TypedDict, total=False):
     """Disable /ui routes"""
     disable_mcp: bool
     """Disable /mcp routes"""
+    mount_prefix: str
+    """Prefix for mounted routes. E.g., "/my-deployment/api"."""
 
 
 class ThreadTTLConfig(TypedDict, total=False):
@@ -169,11 +171,15 @@ HTTP_CONFIG: HttpConfig | None = env("LANGGRAPH_HTTP", cast=_parse_json, default
 STORE_CONFIG: StoreConfig | None = env(
     "LANGGRAPH_STORE", cast=_parse_json, default=None
 )
+
+MOUNT_PREFIX: str | None = env("MOUNT_PREFIX", cast=str, default=None) or (
+    HTTP_CONFIG.get("mount_prefix") if HTTP_CONFIG else None
+)
+
 CORS_ALLOW_ORIGINS = env("CORS_ALLOW_ORIGINS", cast=CommaSeparatedStrings, default="*")
-if HTTP_CONFIG and HTTP_CONFIG.get("cors"):
-    CORS_CONFIG = HTTP_CONFIG["cors"]
-else:
-    CORS_CONFIG: CorsConfig | None = env("CORS_CONFIG", cast=_parse_json, default=None)
+CORS_CONFIG: CorsConfig | None = env("CORS_CONFIG", cast=_parse_json, default=None) or (
+    HTTP_CONFIG.get("cors") if HTTP_CONFIG else None
+)
 """
 {
     "type": "object",

{langgraph_api-0.1.2 → langgraph_api-0.1.3}/langgraph_api/js/tests/api.test.mts

@@ -821,7 +821,7 @@ describe("runs", () => {
     },
   );
 
-  it.concurrent("human in the loop - modification", async () => {
+  it.concurrent("human in the loop - modification", {retry: 3}, async () => {
     // (2) interrupt, modify the message and then continue running
     const assistant = await client.assistants.create({ graphId: "agent" });
     const thread = await client.threads.create();
@@ -1204,7 +1204,7 @@ describe("subgraphs", () => {
   });
 
   // (1) interrupt and then continue running, no modification
-  it.concurrent("human in the loop - no modification", async () => {
+  it.concurrent("human in the loop - no modification", {retry: 3}, async () => {
     const assistant = await client.assistants.create({ graphId: "weather" });
     const thread = await client.threads.create();
 
@@ -1491,7 +1491,7 @@ describe("subgraphs", () => {
   });
 
   // (2) interrupt, modify the message and then continue running
-  it.concurrent("human in the loop - modification", async () => {
+  it.concurrent("human in the loop - modification", {retry: 3}, async () => {
     const assistant = await client.assistants.create({ graphId: "weather" });
     const thread = await client.threads.create();
     const input = {

{langgraph_api-0.1.2 → langgraph_api-0.1.3}/langgraph_api/metadata.py

@@ -70,6 +70,12 @@ async def metadata_loop() -> None:
     if not LANGGRAPH_CLOUD_LICENSE_KEY and not LANGSMITH_API_KEY:
         return
 
+    if LANGGRAPH_CLOUD_LICENSE_KEY and not LANGGRAPH_CLOUD_LICENSE_KEY.startswith(
+        "lcl_"
+    ):
+        logger.info("Running in air-gapped mode, skipping metadata loop")
+        return
+
     logger.info("Starting metadata loop")
 
     global RUN_COUNTER, NODE_COUNTER, FROM_TIMESTAMP

{langgraph_api-0.1.2 → langgraph_api-0.1.3}/langgraph_api/serde.py

@@ -104,7 +104,7 @@ _option = orjson.OPT_SERIALIZE_NUMPY | orjson.OPT_NON_STR_KEYS
 
 def json_dumpb(obj) -> bytes:
     return orjson.dumps(obj, default=default, option=_option).replace(
-        b"\u0000", b""
+        rb"\u0000", b""
     )  # null unicode char not allowed in json
 
 

{langgraph_api-0.1.2 → langgraph_api-0.1.3}/langgraph_api/server.py

@@ -5,6 +5,7 @@ import sys
 # WARNING: Keep the import above before other code runs as it
 # patches an error in the Starlette library.
 import logging
+import typing
 
 import jsonschema_rs
 import structlog
@@ -13,10 +14,13 @@ from langgraph.errors import EmptyInputError, InvalidUpdateError
 from starlette.applications import Starlette
 from starlette.middleware import Middleware
 from starlette.middleware.cors import CORSMiddleware
+
 from langgraph_api.api.openapi import set_custom_spec
+from starlette.types import Receive, Scope, Send
 
 import langgraph_api.config as config
 from langgraph_api.api import routes, meta_routes, user_router
+from starlette.routing import Mount
 from langgraph_api.errors import (
     overloaded_error_handler,
     validation_error_handler,
@@ -149,3 +153,37 @@ else:
         middleware=middleware,
         exception_handlers=exception_handlers,
     )
+
+if config.MOUNT_PREFIX:
+    prefix = config.MOUNT_PREFIX
+    if not prefix.startswith("/") or prefix.endswith("/"):
+        raise ValueError(
+            f"Invalid mount_prefix '{prefix}': Must start with '/' and must not end with '/'. "
+            f"Valid examples: '/my-api', '/v1', '/api/v1'.\nInvalid examples: 'api/', '/api/'"
+        )
+    logger.info(f"Mounting routes at prefix: {prefix}")
+    plen = len(prefix)
+    rplen = len(prefix.encode("utf-8"))
+
+    class ASGIBypassMiddleware:
+        def __init__(self, app: typing.Any, **kwargs):
+            self.app = app
+
+        async def __call__(
+            self, scope: Scope, receive: Receive, send: Send
+        ) -> typing.Any:
+            if (root_path := scope.get("root_path")) and root_path == "/noauth":
+                # The SDK initialized with None is trying to connect via
+                # ASGITransport. Ensure that it has the correct subpath prefixes
+                # so the regular router can handle it.
+                scope["path"] = f'/noauth{prefix}{scope["path"]}'
+                scope["raw_path"] = scope["path"].encode("utf-8")
+
+            return await self.app(scope, receive, send)
+
+    app = Starlette(
+        routes=[Mount(prefix, app=app)],
+        lifespan=app.router.lifespan_context,
+        middleware=[Middleware(ASGIBypassMiddleware)] + app.user_middleware,
+        exception_handlers=app.exception_handlers,
+    )

{langgraph_api-0.1.2 → langgraph_api-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.1.2"
+version = "0.1.3"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",
@@ -65,6 +65,7 @@ fastapi = "^0.115.8"
 langgraph = ">=0.3.17"
 pycryptodome = "^3.22.0"
 langgraph-runtime-inmem = { path = "../runtime_inmem", develop = true, python = ">=3.11,<4.0" }
+blockbuster = "^1.5.24"
 
 [tool.poetry_bumpversion.file."langgraph_api/__init__.py"]
 search = '__version__ = "{current_version}"'

langgraph_api-0.1.2/langgraph_api/__init__.py

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