langgraph-api 0.2.8__tar.gz → 0.2.10__tar.gz

{langgraph_api-0.2.8 → langgraph_api-0.2.10}/PKG-INFO

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

langgraph_api-0.2.10/langgraph_api/__init__.py

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

{langgraph_api-0.2.8 → langgraph_api-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-0.2.8 → langgraph_api-0.2.10}/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-0.2.8 → langgraph_api-0.2.10}/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-0.2.8 → langgraph_api-0.2.10}/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-0.2.8 → langgraph_api-0.2.10}/langgraph_api/js/tests/api.test.mts

@@ -303,7 +303,7 @@ describe("threads crud", () => {
 });
 
 describe("threads copy", () => {
-  it.concurrent("copy", async () => {
+  it.concurrent("copy", { retry: 3 }, async () => {
     const assistantId = "agent";
     const thread = await client.threads.create();
     const input = { messages: [{ type: "human", content: "foo" }] };
@@ -377,7 +377,7 @@ describe("threads copy", () => {
     }
   });
 
-  it.concurrent("copy runs", async () => {
+  it.concurrent("copy runs", { retry: 3 }, async () => {
     const assistantId = "agent";
     const thread = await client.threads.create();
 
@@ -423,7 +423,7 @@ describe("threads copy", () => {
     expect(currentOriginalThreadState).toEqual(originalThreadState);
   });
 
-  it.concurrent("get thread history", async () => {
+  it.concurrent("get thread history", { retry: 3 }, async () => {
     const assistant = await client.assistants.create({ graphId: "agent" });
     const thread = await client.threads.create();
     const input = { messages: [{ type: "human", content: "foo" }] };
@@ -467,7 +467,7 @@ describe("threads copy", () => {
     expect(filteredHistory.at(-1)?.values.messages.length).toBe(4);
   });
 
-  it.concurrent("copy update", async () => {
+  it.concurrent("copy update", { retry: 3 }, async () => {
     const assistantId = "agent";
     const thread = await client.threads.create();
     const input = {
@@ -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-0.2.8 → langgraph_api-0.2.10}/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.8 → langgraph_api-0.2.10}/langgraph_api/worker.py

@@ -275,6 +275,27 @@ async def worker(
         # let it bubble up and rollback db transaction, thus marking the run
         # as available to be picked up by another worker
 
+    # If a stateful run succeeded but no checkoint was returned, it's likely because
+    # there was a retriable exception that resumed right at the end
+    if checkpoint is None and (not temporary) and webhook and status == "success":
+        await logger.ainfo(
+            "Fetching missing checkpoint for webhook",
+            run_id=str(run_id),
+            run_attempt=attempt,
+        )
+        try:
+            state_snapshot = await Threads.State.get(
+                conn, run["kwargs"]["config"], subgraphs=True
+            )
+            checkpoint = {"values": state_snapshot.values}
+        except Exception:
+            await logger.aerror(
+                "Failed to fetch missing checkpoint for webhook. Continuing...",
+                exc_info=True,
+                run_id=str(run_id),
+                run_attempt=attempt,
+            )
+
     return WorkerResult(
         checkpoint=checkpoint,
         status=status,

{langgraph_api-0.2.8 → langgraph_api-0.2.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.2.8"
+version = "0.2.10"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",

langgraph_api-0.2.8/langgraph_api/__init__.py

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