langgraph-api 0.2.9__py3-none-any.whl → 0.2.10__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.9"
+__version__ = "0.2.10"

langgraph_api/js/build.mts

@@ -4,13 +4,10 @@ import "./src/preload.mjs";
 import { z } from "zod";
 import * as fs from "node:fs/promises";
 import * as path from "node:path";
-import {
-  type GraphSchema,
-  resolveGraph,
-  runGraphSchemaWorker,
-} from "./src/graph.mts";
+import { type GraphSchema, resolveGraph } from "./src/graph.mts";
 import { build } from "@langchain/langgraph-ui";
 import { checkLangGraphSemver } from "@langchain/langgraph-api/semver";
+import { getStaticGraphSchema } from "@langchain/langgraph-api/schema";
 import { filterValidExportPath } from "./src/utils/files.mts";
 
 const __dirname = new URL(".", import.meta.url).pathname;
@@ -54,7 +51,7 @@ async function main() {
 
         try {
           console.info(`[${graphId}]: Extracting schema`);
-          GRAPH_SCHEMAS[graphId] = await runGraphSchemaWorker(spec, {
+          GRAPH_SCHEMAS[graphId] = await getStaticGraphSchema(spec, {
             timeoutMs: 120_000,
           });
         } catch (error) {

langgraph_api/js/client.mts

@@ -38,7 +38,6 @@ import { BaseMessageChunk, isBaseMessage } from "@langchain/core/messages";
 import type { PyItem, PyResult } from "./src/utils/pythonSchemas.mts";
 import type { RunnableConfig } from "@langchain/core/runnables";
 import {
-  runGraphSchemaWorker,
   GraphSchema,
   resolveGraph,
   GraphSpec,
@@ -52,6 +51,10 @@ import {
   authorize,
   registerAuth,
 } from "@langchain/langgraph-api/auth";
+import {
+  getRuntimeGraphSchema,
+  getStaticGraphSchema,
+} from "@langchain/langgraph-api/schema";
 import { filterValidExportPath } from "./src/utils/files.mts";
 
 const logger = createLogger({
@@ -177,7 +180,7 @@ async function getOrExtractSchema(graphId: string) {
         // ignore
       }
 
-      GRAPH_SCHEMA[graphId] = await runGraphSchemaWorker(GRAPH_SPEC[graphId], {
+      GRAPH_SCHEMA[graphId] = await getStaticGraphSchema(GRAPH_SPEC[graphId], {
         timeoutMs,
       });
       timer.done({ message: `Extracting schema for ${graphId} finished` });
@@ -482,6 +485,15 @@ function pyItemToJs(item?: PyItem): Item | undefined {
   };
 }
 
+function isPyItem(result: unknown): result is PyItem {
+  return (
+    result != null &&
+    typeof result === "object" &&
+    "value" in result &&
+    "key" in result
+  );
+}
+
 export class RemoteStore extends BaseStore {
   async batch<Op extends Operation[]>(
     operations: Op,
@@ -492,16 +504,10 @@ export class RemoteStore extends BaseStore {
 
     return results.map((result) => {
       if (Array.isArray(result)) {
-        return result.map((item) => pyItemToJs(item));
-      } else if (
-        result &&
-        typeof result === "object" &&
-        "value" in result &&
-        "key" in result
-      ) {
-        return pyItemToJs(result);
+        return result.map((item) => (isPyItem(item) ? pyItemToJs(item) : item));
       }
-      return result;
+
+      return isPyItem(result) ? pyItemToJs(result) : result;
     }) as OperationResults<Op>;
   }
 
@@ -788,25 +794,33 @@ async function getSubgraphsRequest(
   rawPayload: z.infer<typeof GetSubgraphsPayload>,
 ) {
   const { graph_id: graphId, ...payload } = rawPayload;
-  const graphConfig = getRunnableConfig(payload.graph_config);
-  const graph = await getGraph(graphId, graphConfig, payload.graph_name);
-  const result: Array<[name: string, Record<string, any>]> = [];
 
-  const graphSchema = await getOrExtractSchema(graphId);
-  const rootGraphId = Object.keys(graphSchema).find((i) => !i.includes("|"));
+  const config = getRunnableConfig(payload.graph_config);
+  const graph = await getGraph(graphId, config, payload.graph_name);
 
-  if (!rootGraphId) throw new Error("Failed to find root graph");
+  const result: Array<[name: string, schema: Record<string, any>]> = [];
+  let graphSchemaPromise: ReturnType<typeof getOrExtractSchema> | undefined;
 
-  for await (const [name] of graph.getSubgraphsAsync(
+  for await (const [ns, subgraph] of graph.getSubgraphsAsync(
     payload.namespace ?? undefined,
     payload.recurse ?? undefined,
   )) {
-    const schema =
-      graphSchema[`${rootGraphId}|${name}`] || graphSchema[rootGraphId];
-    result.push([name, schema]);
+    const schema = await (async () => {
+      const runtimeSchema = await getRuntimeGraphSchema(subgraph);
+      if (runtimeSchema) return runtimeSchema;
+
+      graphSchemaPromise ??= getOrExtractSchema(graphId);
+      const graphSchema = await graphSchemaPromise;
+      const rootGraphId = Object.keys(graphSchema).find(
+        (i) => !i.includes("|"),
+      );
+      if (!rootGraphId) throw new Error("Failed to find root graph");
+      return graphSchema[`${rootGraphId}|${ns}`] || graphSchema[rootGraphId];
+    })();
+
+    result.push([ns, schema]);
   }
 
-  // TODO: make this a stream
   return Object.fromEntries(result);
 }
 
@@ -864,14 +878,21 @@ const GetSchemaPayload = z.object({
 
 async function getSchemaRequest(payload: z.infer<typeof GetSchemaPayload>) {
   const { graph_id: graphId } = payload;
-  // TODO: add support for schema inference with dynamic graphs (now that Zod is supported)
 
-  const schemas = await getOrExtractSchema(graphId);
-  const rootGraphId = Object.keys(schemas).find((i) => !i.includes("|"));
-  if (!rootGraphId) {
-    throw new Error("Failed to find root graph");
-  }
-  return schemas[rootGraphId];
+  const config = getRunnableConfig(payload.graph_config);
+  const graph = await getGraph(graphId, config, payload.graph_name);
+
+  const schema = await (async () => {
+    const runtimeSchema = await getRuntimeGraphSchema(graph);
+    if (runtimeSchema) return runtimeSchema;
+
+    const graphSchema = await getOrExtractSchema(graphId);
+    const rootGraphId = Object.keys(graphSchema).find((i) => !i.includes("|"));
+    if (!rootGraphId) throw new Error("Failed to find root graph");
+    return graphSchema[rootGraphId];
+  })();
+
+  return schema;
 }
 
 const GetStateHistoryPayload = z.object({

langgraph_api/js/package.json

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

langgraph_api/js/src/graph.mts

@@ -1,8 +1,7 @@
-import { Worker } from "node:worker_threads";
-import { register } from "node:module";
 import * as fs from "node:fs/promises";
-import type { CompiledGraph, Graph } from "@langchain/langgraph";
 import * as path from "node:path";
+
+import type { CompiledGraph, Graph } from "@langchain/langgraph";
 import type { JSONSchema7 } from "json-schema";
 
 export interface GraphSchema {
@@ -40,14 +39,14 @@ export async function resolveGraph(
   options?: { onlyFilePresence?: boolean },
 ) {
   const [userFile, exportSymbol] = spec.split(":", 2);
+  const sourceFile = path.resolve(process.cwd(), userFile);
 
   // validate file exists
-  await fs.stat(userFile);
+  await fs.stat(sourceFile);
   if (options?.onlyFilePresence) {
-    return { sourceFile: userFile, exportSymbol, resolved: undefined };
+    return { sourceFile, exportSymbol, resolved: undefined };
   }
 
-  const sourceFile = path.resolve(process.cwd(), userFile);
   type GraphLike = CompiledGraph<string> | Graph<string>;
 
   type GraphUnknown =
@@ -89,29 +88,3 @@ export async function resolveGraph(
 
   return { sourceFile, exportSymbol, resolved };
 }
-
-export async function runGraphSchemaWorker(
-  spec: GraphSpec,
-  options?: { timeoutMs?: number },
-) {
-  return await new Promise<Record<string, GraphSchema>>((resolve, reject) => {
-    const worker = new Worker(
-      new URL("./parser/parser.worker.mjs", import.meta.url).pathname,
-    );
-
-    // Set a timeout to reject if the worker takes too long
-    const timeoutId = setTimeout(() => {
-      worker.terminate();
-      reject(new Error("Schema extract worker timed out"));
-    }, options?.timeoutMs ?? 30_000);
-
-    worker.on("message", (result) => {
-      worker.terminate();
-      clearTimeout(timeoutId);
-      resolve(result);
-    });
-
-    worker.on("error", reject);
-    worker.postMessage(spec);
-  });
-}

langgraph_api/js/tests/api.test.mts

@@ -561,7 +561,9 @@ describe("runs", () => {
       }
 
       if (chunk.event === "values") {
-        const messageIds = chunk.data.messages.map((message) => message.id);
+        const messageIds = chunk.data.messages.map(
+          (message: any) => message.id,
+        );
         expect(messageIds.slice(0, -1)).toEqual(previousMessageIds);
         previousMessageIds = messageIds;
       }

langgraph_api/js/yarn.lock

@@ -203,15 +203,15 @@
     zod "^3.22.4"
     zod-to-json-schema "^3.22.3"
 
-"@langchain/langgraph-api@~0.0.29":
-  version "0.0.29"
-  resolved "https://registry.yarnpkg.com/@langchain/langgraph-api/-/langgraph-api-0.0.29.tgz#787e01675250720d293e3363ff579b601a9c94ab"
-  integrity sha512-WJVOWFJBEo7sCUtzchFRhFWT4+Zn3evKakmLGDJihMZ2nrNeoaordEUmbsXMLiQVOe3bP7Fpweq3mSPIwxQyyQ==
+"@langchain/langgraph-api@~0.0.32":
+  version "0.0.32"
+  resolved "https://registry.yarnpkg.com/@langchain/langgraph-api/-/langgraph-api-0.0.32.tgz#65f41373318055b330fa64e7d4ccd7c994f7dc8d"
+  integrity sha512-wYflud8iCGjwh3/GtzrTeWlwcc91kXBa49Goc2iaHcJRjMnTbl+loE3eklegDPnQGx1VKUCiB9qW/HutxuqXGA==
   dependencies:
     "@babel/code-frame" "^7.26.2"
     "@hono/node-server" "^1.12.0"
     "@hono/zod-validator" "^0.2.2"
-    "@langchain/langgraph-ui" "0.0.29"
+    "@langchain/langgraph-ui" "0.0.32"
     "@types/json-schema" "^7.0.15"
     "@typescript/vfs" "^1.6.0"
     dedent "^1.5.3"
@@ -256,10 +256,10 @@
     p-retry "4"
     uuid "^9.0.0"
 
-"@langchain/langgraph-ui@0.0.29", "@langchain/langgraph-ui@~0.0.29":
-  version "0.0.29"
-  resolved "https://registry.yarnpkg.com/@langchain/langgraph-ui/-/langgraph-ui-0.0.29.tgz#98659cc56f0e8e074a9a0b2e33e967dc5f30e79f"
-  integrity sha512-FADBt6B1/egKSpZFmTa6vxonBK9lF6EmeKoBEWh/XTkO6lD5zsejWoRgT+LR2x22CHJvfvCQtXxgP5y8AtxsQw==
+"@langchain/langgraph-ui@0.0.32", "@langchain/langgraph-ui@~0.0.32":
+  version "0.0.32"
+  resolved "https://registry.yarnpkg.com/@langchain/langgraph-ui/-/langgraph-ui-0.0.32.tgz#7e0cab750ba1e65f47d7b1d2adf955a36e50664e"
+  integrity sha512-65QZpouvw4P+hCpklYebNiwR88M84h7awjNB6FaKWHvCZa6NcSzlruFa/PnvU1o/kRK5NNpXNPyDUQLhT3tmTA==
   dependencies:
     "@commander-js/extra-typings" "^13.0.0"
     commander "^13.0.0"

{langgraph_api-0.2.9.dist-info → langgraph_api-0.2.10.dist-info}/METADATA

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

{langgraph_api-0.2.9.dist-info → langgraph_api-0.2.10.dist-info}/RECORD

@@ -1,5 +1,5 @@
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api/__init__.py,sha256=F8OVhAhMXSkvvXYgZtbPn2SG1AQC3joK4yu-FrHt81Y,22
+langgraph_api/__init__.py,sha256=waXgc7p-jgGCsUjdVfO_KjlVZblnCvrzf4A0dsBj_lg,23
 langgraph_api/api/__init__.py,sha256=YVzpbn5IQotvuuLG9fhS9QMrxXfP4s4EpEMG0n4q3Nw,5625
 langgraph_api/api/assistants.py,sha256=mcKaVeNG8hQAV4IQDhaukS7FgVqTIVQNTyti3GfK2KI,14649
 langgraph_api/api/mcp.py,sha256=RvRYgANqRzNQzSmgjNkq4RlKTtoEJYil04ot9lsmEtE,14352
@@ -29,27 +29,23 @@ langgraph_api/js/.gitignore,sha256=l5yI6G_V6F1600I1IjiUKn87f4uYIrBAYU1MOyBBhg4,5
 langgraph_api/js/.prettierrc,sha256=0es3ovvyNIqIw81rPQsdt1zCQcOdBqyR_DMbFE4Ifms,19
 langgraph_api/js/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/js/base.py,sha256=gjY6K8avI03OrI-Hy6a311fQ_EG5r_x8hUYlc7uqxdE,534
-langgraph_api/js/build.mts,sha256=ceHRr_Io_9otkDOSgHuKbGnOUu4AX2jPj8APTTa9UjM,2741
+langgraph_api/js/build.mts,sha256=05bkP9mpxf1JVuJ3at0y56p13QBkHgf7Q6eG6pOZsIg,2784
 langgraph_api/js/client.http.mts,sha256=AGA-p8J85IcNh2oXZjDxHQ4PnQdJmt-LPcpZp6j0Cws,4687
-langgraph_api/js/client.mts,sha256=RZgkJk43CMOwnHXMXBbWIEhHh0W8WBwQ8Rm8GkJwvhM,30341
+langgraph_api/js/client.mts,sha256=N9CTH7mbXGSD-gpv-XyruYsHI-rgrObL8cQoAp5s3_U,30986
 langgraph_api/js/errors.py,sha256=Cm1TKWlUCwZReDC5AQ6SgNIVGD27Qov2xcgHyf8-GXo,361
 langgraph_api/js/global.d.ts,sha256=j4GhgtQSZ5_cHzjSPcHgMJ8tfBThxrH-pUOrrJGteOU,196
-langgraph_api/js/package.json,sha256=J2sWtYb_gFMCb_g8661iGUR1C23lZQna3wmgl5da0Xc,1289
+langgraph_api/js/package.json,sha256=craKzuSIEnt-WQpKZDaACwdm75lPqpRQde6l9LfS-eI,1289
 langgraph_api/js/remote.py,sha256=ipAITSDyh-kNau37nfRg-PSB-6Lbtax3UJap8-lLZdw,35729
 langgraph_api/js/schema.py,sha256=7idnv7URlYUdSNMBXQcw7E4SxaPxCq_Oxwnlml8q5ik,408
-langgraph_api/js/src/graph.mts,sha256=_xKhdO2WAwsieoezFqTCcx1IDeHeN6GSMv0lyBsakKI,3485
+langgraph_api/js/src/graph.mts,sha256=9zTQNdtanI_CFnOwNRoamoCVHHQHGbNlbm91aRxDeOc,2675
 langgraph_api/js/src/load.hooks.mjs,sha256=xNVHq75W0Lk6MUKl1pQYrx-wtQ8_neiUyI6SO-k0ecM,2235
-langgraph_api/js/src/parser/parser.mts,sha256=iW5G-YIVIuwuFsfn_GS3_CZsjpa00SxypICfPf2SN9Q,14156
-langgraph_api/js/src/parser/parser.worker.mjs,sha256=Byo65U2aI3FtZaUwRRq65xBpwornoR6ldak333jxH4c,387
-langgraph_api/js/src/parser/schema/types.mts,sha256=Vx34tLAncRnITwAVe1wABFgM86TpByMLQ5y5pnit4tI,62693
-langgraph_api/js/src/parser/schema/types.template.mts,sha256=Dbjj_8d-OubqH4QY_OaxSu8ocZ4dVjI94oncL20fqtk,2235
 langgraph_api/js/src/preload.mjs,sha256=ORV7xwMuZcXWL6jQxNAcCYp8GZVYIvVJbUhmle8jbno,759
 langgraph_api/js/src/utils/files.mts,sha256=MXC-3gy0pkS82AjPBoUN83jY_qg37WSAPHOA7DwfB4M,141
 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=D9o6MwTgwPezC_DEmsWS5NnLPnjPMVWIb1I1D4QPEPo,743
 langgraph_api/js/sse.py,sha256=lsfp4nyJyA1COmlKG9e2gJnTttf_HGCB5wyH8OZBER8,4105
-langgraph_api/js/tests/api.test.mts,sha256=LSWDHTar-AtyazqR8EbnS17MrMr3IvKWysGDZ13p-GM,69432
+langgraph_api/js/tests/api.test.mts,sha256=UT5UotPULGYUMQ5ZQ5yLBhRP-0cfPmQ3-4-a5GQ9p58,69458
 langgraph_api/js/tests/auth.test.mts,sha256=mMhKe9ggJw4BgUqzSVwqYY3HLMXXEBZ23iiKK8Yq1mM,21678
 langgraph_api/js/tests/compose-postgres.auth.yml,sha256=iPfJbCeYZdV6GiRLiDn_f7qgpG4TyyGaQ4lV-ZXr6Qk,1768
 langgraph_api/js/tests/compose-postgres.yml,sha256=w4B3YRS0QEnTcZH2-MY0DYvR_c5GcER0uDa1Ga_knf8,1960
@@ -69,11 +65,10 @@ langgraph_api/js/tests/graphs/nested.mts,sha256=e06kFzcX2zEJlb7e5B6kMnuZn6sfvsO7
 langgraph_api/js/tests/graphs/package.json,sha256=8kgqWdZJCwekCqjsSrhbLrAPZ2vEy1DmcC8EQnwJMDU,262
 langgraph_api/js/tests/graphs/weather.mts,sha256=Gef9dCXxoVgNa4Ba0AcoptodYV85Ed2CGcleUB9BRMk,1656
 langgraph_api/js/tests/graphs/yarn.lock,sha256=HDLJKx47Y-csPzA5eYUMVHWE8fMKrZgrc4SEkQAYYCE,11201
-langgraph_api/js/tests/parser.test.mts,sha256=BBKUTveZnf-RI6B9XfTBLqy6tp84ddyu1tN3z041IAs,27900
 langgraph_api/js/tests/utils.mts,sha256=Jk1ZZmllNgSS6FJlSs9VaQxHqCEUzkqB5rRQwTSAOP4,441
 langgraph_api/js/tsconfig.json,sha256=imCYqVnqFpaBoZPx8k1nO4slHIWBFsSlmCYhO73cpBs,341
 langgraph_api/js/ui.py,sha256=XNT8iBcyT8XmbIqSQUWd-j_00HsaWB2vRTVabwFBkik,2439
-langgraph_api/js/yarn.lock,sha256=HJuG663b4R2sqQDTQOHtpfMiK615k3CiUsRIZfGDd7k,84019
+langgraph_api/js/yarn.lock,sha256=qtnJQjwX8OUQVzsh51N1imtjYGg5RCI9xkg5n5VZMKI,84019
 langgraph_api/logging.py,sha256=JJIzbNIgLCN6ClQ3tA-Mm5ffuBGvpRDSZsEvnIlsuu4,3693
 langgraph_api/metadata.py,sha256=ptaxwmzdx2bUBSc1KRhqgF-Xnm-Zh2gqwSiHpl8LD9c,4482
 langgraph_api/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -101,8 +96,8 @@ langgraph_license/validation.py,sha256=ZKraAVJArAABKqrmHN-EN18ncoNUmRm500Yt1Sc7t
 langgraph_runtime/__init__.py,sha256=O4GgSmu33c-Pr8Xzxj_brcK5vkm70iNTcyxEjICFZxA,1075
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=GefWJwBrbrN_jNDAEFlwNEXx1ottRtvjOmKBi965KLU,133756
-langgraph_api-0.2.9.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.2.9.dist-info/METADATA,sha256=-USWsNozeKkqmxeoMi3Aj_SkdoTp_s2p3AEwtnHq1F4,4240
-langgraph_api-0.2.9.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-langgraph_api-0.2.9.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.2.9.dist-info/RECORD,,
+langgraph_api-0.2.10.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.2.10.dist-info/METADATA,sha256=UCAG8IhwMEOsdWD6CZgSX3h4aaz7Semsq8dTf-cA5lQ,4241
+langgraph_api-0.2.10.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+langgraph_api-0.2.10.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.2.10.dist-info/RECORD,,