langgraph-api 0.0.45__tar.gz → 0.0.46__tar.gz

{langgraph_api-0.0.45 → langgraph_api-0.0.46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langgraph-api
-Version: 0.0.45
+Version: 0.0.46
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -13,13 +13,13 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: blockbuster (>=1.5.24,<2.0.0)
 Requires-Dist: cloudpickle (>=3.0.0,<4.0.0)
-Requires-Dist: cryptography (>=43.0.3,<44.0.0)
+Requires-Dist: cryptography (>=42.0.0,<45.0)
 Requires-Dist: httpx (>=0.25.0)
 Requires-Dist: jsonschema-rs (>=0.20.0,<0.30)
 Requires-Dist: langchain-core (>=0.2.38,<0.4.0)
 Requires-Dist: langgraph (>=0.2.56,<0.4.0)
 Requires-Dist: langgraph-checkpoint (>=2.0.23,<3.0)
-Requires-Dist: langgraph-sdk (>=0.1.59,<0.2.0)
+Requires-Dist: langgraph-sdk (>=0.1.61,<0.2.0)
 Requires-Dist: langsmith (>=0.1.63,<0.4.0)
 Requires-Dist: orjson (>=3.9.7)
 Requires-Dist: pyjwt (>=2.9.0,<3.0.0)

{langgraph_api-0.0.45 → langgraph_api-0.0.46}/langgraph_api/api/ui.py

@@ -1,34 +1,39 @@
 import json
 import os
-from functools import lru_cache
-from pathlib import Path
 from typing import TypedDict
 
+from anyio import open_file
 from orjson import loads
 from starlette.responses import Response
 from starlette.routing import BaseRoute, Mount
 from starlette.staticfiles import StaticFiles
 
+from langgraph_api.js.ui import UI_PUBLIC_DIR, UI_SCHEMAS_FILE
 from langgraph_api.route import ApiRequest, ApiRoute
 
-# Get path to built UI assets
-UI_DIR = Path(os.path.dirname(__file__)).parent / "js" / "ui"
-SCHEMAS_FILE = Path(os.path.dirname(__file__)).parent / "js" / "client.ui.schemas.json"
-
 
 class UiSchema(TypedDict):
     name: str
     assets: list[str]
 
 
-@lru_cache(maxsize=1)
-def load_ui_schemas() -> dict[str, UiSchema]:
+_UI_SCHEMAS_CACHE: dict[str, UiSchema] | None = None
+
+
+async def load_ui_schemas() -> dict[str, UiSchema]:
     """Load and cache UI schema mappings from JSON file."""
-    if not SCHEMAS_FILE.exists():
-        return {}
+    global _UI_SCHEMAS_CACHE
+
+    if _UI_SCHEMAS_CACHE is not None:
+        return _UI_SCHEMAS_CACHE
+
+    if not UI_SCHEMAS_FILE.exists():
+        _UI_SCHEMAS_CACHE = {}
+    else:
+        async with await open_file(UI_SCHEMAS_FILE, mode="r") as f:
+            _UI_SCHEMAS_CACHE = loads(await f.read())
 
-    with open(SCHEMAS_FILE) as f:
-        return loads(f.read())
+    return _UI_SCHEMAS_CACHE
 
 
 async def handle_ui(request: ApiRequest) -> Response:
@@ -38,7 +43,7 @@ async def handle_ui(request: ApiRequest) -> Response:
     message = await request.json(schema=None)
 
     # Load UI file paths from schema
-    schemas = load_ui_schemas()
+    schemas = await load_ui_schemas()
 
     if graph_id not in schemas:
         return Response(f"UI not found for graph '{graph_id}'", status_code=404)
@@ -64,5 +69,5 @@ async def handle_ui(request: ApiRequest) -> Response:
 
 ui_routes: list[BaseRoute] = [
     ApiRoute("/ui/{graph_id}", handle_ui, methods=["POST"]),
-    Mount("/ui", StaticFiles(directory=UI_DIR, check_dir=False)),
+    Mount("/ui", StaticFiles(directory=UI_PUBLIC_DIR, check_dir=False)),
 ]

{langgraph_api-0.0.45 → langgraph_api-0.0.46}/langgraph_api/cli.py

@@ -129,6 +129,8 @@ def run_server(
     store: typing.Optional["StoreConfig"] = None,
     auth: AuthConfig | None = None,
     http: typing.Optional["HttpConfig"] = None,
+    ui: dict | None = None,
+    ui_config: dict | None = None,
     studio_url: str | None = None,
     disable_persistence: bool = False,
     allow_blocking: bool = False,
@@ -190,6 +192,9 @@ def run_server(
         LANGSMITH_LANGGRAPH_API_VARIANT="local_dev",
         LANGGRAPH_AUTH=json.dumps(auth) if auth else None,
         LANGGRAPH_HTTP=json.dumps(http) if http else None,
+        LANGGRAPH_UI=json.dumps(ui) if ui else None,
+        LANGGRAPH_UI_CONFIG=json.dumps(ui_config) if ui_config else None,
+        LANGGRAPH_UI_BUNDLER="true",
         LANGGRAPH_API_URL=local_url,
         LANGGRAPH_DISABLE_FILE_PERSISTENCE=str(disable_persistence).lower(),
         # If true, we will not raise on blocking IO calls (via blockbuster)
@@ -349,6 +354,8 @@ def main():
 
     graphs = config_data.get("graphs", {})
     auth = config_data.get("auth")
+    ui = config_data.get("ui")
+    ui_config = config_data.get("ui_config")
     run_server(
         args.host,
         args.port,
@@ -360,6 +367,8 @@ def main():
         wait_for_client=args.wait_for_client,
         env=config_data.get("env", None),
         auth=auth,
+        ui=ui,
+        ui_config=ui_config,
     )
 
 

{langgraph_api-0.0.45 → langgraph_api-0.0.46}/langgraph_api/config.py

@@ -325,3 +325,6 @@ USES_INDEXING = (
 USES_CUSTOM_APP = HTTP_CONFIG and HTTP_CONFIG.get("app")
 
 API_VARIANT = env("LANGSMITH_LANGGRAPH_API_VARIANT", cast=str, default="")
+
+# UI
+UI_USE_BUNDLER = env("LANGGRAPH_UI_BUNDLER", cast=bool, default=False)

{langgraph_api-0.0.45 → langgraph_api-0.0.46}/langgraph_api/graph.py

@@ -289,7 +289,7 @@ def _handle_exception(task: asyncio.Task) -> None:
 
 
 async def stop_remote_graphs() -> None:
-    logger.info("Cancelling remote graphs")
+    logger.info("Shutting down remote graphs")
     for task in js_bg_tasks:
         task.cancel("Stopping remote graphs.")
 

langgraph_api-0.0.46/langgraph_api/js/build.mts

@@ -0,0 +1,68 @@
+/// <reference types="./global.d.ts" />
+
+import { z } from "zod";
+import * as fs from "node:fs/promises";
+import * as path from "node:path";
+import {
+  filterValidGraphSpecs,
+  GraphSchema,
+  resolveGraph,
+  runGraphSchemaWorker,
+} from "./src/graph.mts";
+import { build } from "@langchain/langgraph-ui";
+
+const __dirname = new URL(".", import.meta.url).pathname;
+
+async function main() {
+  const specs = filterValidGraphSpecs(
+    z.record(z.string()).parse(JSON.parse(process.env.LANGSERVE_GRAPHS))
+  );
+
+  const GRAPH_SCHEMAS: Record<string, Record<string, GraphSchema> | false> = {};
+  let failed = false;
+
+  try {
+    await Promise.all(
+      specs.map(async ([graphId, rawSpec]) => {
+        console.info(`[${graphId}]: Checking for source file existence`);
+        const { resolved, ...spec } = await resolveGraph(rawSpec, {
+          onlyFilePresence: true,
+        });
+
+        try {
+          console.info(`[${graphId}]: Extracting schema`);
+          GRAPH_SCHEMAS[graphId] = await runGraphSchemaWorker(spec, {
+            timeoutMs: 120_000,
+          });
+        } catch (error) {
+          console.error(`[${graphId}]: Error extracting schema: ${error}`);
+          GRAPH_SCHEMAS[graphId] = false;
+        }
+      })
+    );
+
+    await fs.writeFile(
+      path.resolve(__dirname, "client.schemas.json"),
+      JSON.stringify(GRAPH_SCHEMAS),
+      { encoding: "utf-8" }
+    );
+  } catch (error) {
+    console.error(`Error resolving graphs: ${error}`);
+    failed = true;
+  }
+
+  // Build Gen UI assets
+  try {
+    console.info("Checking for UI assets");
+    await fs.mkdir(path.resolve(__dirname, "ui"), { recursive: true });
+
+    await build({ output: path.resolve(__dirname, "ui") });
+  } catch (error) {
+    console.error(`Error building UI: ${error}`);
+    failed = true;
+  }
+
+  if (failed) process.exit(1);
+}
+
+main();

{langgraph_api-0.0.45 → langgraph_api-0.0.46}/langgraph_api/js/package.json

@@ -24,7 +24,8 @@
     "undici": "^6.21.1",
     "uuid": "^10.0.0",
     "winston": "^3.17.0",
-    "@langchain/langgraph-api": "~0.0.19",
+    "@langchain/langgraph-api": "~0.0.20",
+    "@langchain/langgraph-ui": "~0.0.20",
     "zod": "^3.23.8"
   },
   "resolutions": {