langgraph-api 0.0.1__py3-none-any.whl

langgraph_api/js/tests/compose-postgres.yml

@@ -0,0 +1,56 @@
+services:
+  langgraph-postgres:
+    image: postgres:16.3
+    ports:
+      - "5433:5432"
+    environment:
+      POSTGRES_DB: postgres
+      POSTGRES_USER: postgres
+      POSTGRES_PASSWORD: postgres
+    healthcheck:
+      test: pg_isready -U postgres
+      start_period: 10s
+      timeout: 1s
+      retries: 5
+      interval: 60s
+      start_interval: 1s
+  langgraph-redis:
+    image: redis:6
+    restart: on-failure
+    ports:
+      - "6381:6379"
+    healthcheck:
+      test: redis-cli ping
+      start_period: 10s
+      timeout: 1s
+      retries: 5
+      interval: 60s
+      start_interval: 1s
+  api:
+    build:
+      context: graphs
+      dockerfile_inline: |
+        FROM langchain/langgraphjs-api:20
+        ADD . /deps/graphs
+        WORKDIR /deps/graphs
+        RUN yarn install --frozen-lockfile
+        ENV LANGSERVE_GRAPHS='{"agent":"./agent.mts:graph", "nested": "./nested.mts:graph", "weather": "./weather.mts:graph", "error": "./error.mts:graph"}'
+        ENV LANGGRAPH_CONFIG='{"agent": {"configurable": {"model_name": "openai"}}}'
+        RUN tsx /api/langgraph_api/js/build.mts
+    depends_on:
+      langgraph-postgres:
+        condition: service_healthy
+      langgraph-redis:
+        condition: service_healthy
+    ports:
+      - "9123:8000"
+    healthcheck:
+      test: python /api/healthcheck.py
+      interval: 60s
+      start_interval: 1s
+      start_period: 10s
+    environment:
+      REDIS_URI: redis://langgraph-redis:6379
+      DATABASE_URI: postgres://postgres:postgres@langgraph-postgres:5432/postgres?sslmode=disable
+      N_JOBS_PER_WORKER: "2"
+      LANGGRAPH_CLOUD_LICENSE_KEY: ${LANGGRAPH_CLOUD_LICENSE_KEY}

langgraph_api/js/tests/graphs/.gitignore

@@ -0,0 +1 @@
+node_modules

langgraph_api/js/tests/graphs/agent.mts

@@ -0,0 +1,127 @@
+import { BaseMessage, ToolMessage } from "@langchain/core/messages";
+import {
+  Annotation,
+  StateGraph,
+  START,
+  END,
+  messagesStateReducer,
+  SharedValue,
+  LangGraphRunnableConfig,
+} from "@langchain/langgraph";
+import { FakeListChatModel } from "@langchain/core/utils/testing";
+
+const GraphAnnotationOutput = Annotation.Root({
+  messages: Annotation<BaseMessage[]>({
+    reducer: messagesStateReducer,
+    default: () => [],
+  }),
+  sharedStateValue: Annotation<string | null>(),
+});
+
+const GraphAnnotationInput = Annotation.Root({
+  ...GraphAnnotationOutput.spec,
+  sharedState: SharedValue.on("user_id"),
+  sharedStateFromStoreConfig: Annotation<Record<string, any> | null>,
+});
+
+// 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"] });
+  return modelMap[threadId];
+};
+
+const agentNode = async (
+  state: typeof GraphAnnotationInput.State,
+  config: LangGraphRunnableConfig,
+) => {
+  const model = getModel(config.configurable?.thread_id ?? "$");
+  const response = await model.invoke(state.messages);
+  const sharedStateValue = state.sharedState?.data?.user_id ?? null;
+
+  // Define in the first node
+  // Then retrieve in the second node
+  const store = config.store;
+  // Only set if it's not already set
+  if (store && !state.sharedStateFromStoreConfig) {
+    const value = { id: config?.configurable?.user_id };
+    await store.put(namespace, key, value);
+  }
+
+  return {
+    messages: [response],
+    sharedState: { data: { user_id: config?.configurable?.user_id } },
+    sharedStateValue,
+  };
+};
+
+const toolNode = async (
+  state: typeof GraphAnnotationInput.State,
+  config: LangGraphRunnableConfig,
+) => {
+  const store = config.store;
+  let sharedStateFromStoreConfig: Record<string, any> | null = null;
+  if (store) {
+    const result = await store.get(namespace, key);
+    sharedStateFromStoreConfig = result?.value ?? null;
+  }
+
+  const lastMessage = state.messages.at(-1);
+  if (!lastMessage) return { messages: [], sharedStateFromStoreConfig };
+  return {
+    messages: [
+      new ToolMessage({
+        content: `tool_call__${lastMessage.content as string}`,
+        tool_call_id: "tool_call_id",
+      }),
+    ],
+    sharedStateFromStoreConfig,
+  };
+};
+
+const checkSharedStateNode = async (
+  _: typeof GraphAnnotationInput.State,
+  config: LangGraphRunnableConfig,
+): Promise<Partial<typeof GraphAnnotationInput.State>> => {
+  const store = config.store;
+  const namespace = ["inputtedState", "data"];
+  const key = "my_key";
+  if (store) {
+    const result = await store.get(namespace, key);
+    if (!result || !result.value.isTrue) {
+      throw new Error("Value is not true");
+    }
+  }
+
+  return {};
+};
+
+const agentCondEdge = (state: typeof GraphAnnotationInput.State) => {
+  if ((state.messages[0].content as string) === "should_end") return END;
+  if ((state.messages[0].content as string) === "___check_state_value")
+    return "checkSharedState";
+
+  const lastMessage = state.messages.at(-1);
+  if (lastMessage?.content === "end") return END;
+  return "tool";
+};
+
+const workflow = new StateGraph(
+  {
+    input: GraphAnnotationInput,
+    output: GraphAnnotationOutput,
+  },
+  Annotation.Root({ model_name: Annotation<string> }),
+)
+  .addNode("agent", agentNode)
+  .addNode("tool", toolNode)
+  .addNode("checkSharedState", checkSharedStateNode)
+  .addEdge(START, "agent")
+  .addConditionalEdges("agent", agentCondEdge)
+  .addEdge("tool", "agent")
+  .addEdge("checkSharedState", END);
+
+export const graph = (async () => workflow.compile())();

langgraph_api/js/tests/graphs/error.mts

@@ -0,0 +1,17 @@
+import { StateGraph, START } from "@langchain/langgraph";
+import { MessagesAnnotation } from "@langchain/langgraph";
+
+class CustomError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = "CustomError";
+  }
+}
+
+export const graph = new StateGraph(MessagesAnnotation)
+  .addNode("error_node", async () => {
+    await new Promise((resolve) => setTimeout(resolve, 1000));
+    throw new CustomError("Boo!");
+  })
+  .addEdge(START, "error_node")
+  .compile();

langgraph_api/js/tests/graphs/langgraph.json

@@ -0,0 +1,8 @@
+{
+  "node_version": "20",
+  "graphs": {
+    "agent": "./agent.mts:graph",
+    "subgraph": "./subgraph.mts:graph",
+    "nested": "./nested.mts:graph"
+  }
+}

langgraph_api/js/tests/graphs/nested.mts

@@ -0,0 +1,44 @@
+import { Annotation, StateGraph, END, START } from "@langchain/langgraph";
+
+const child = new StateGraph(
+  Annotation.Root({
+    messages: Annotation<string[]>({
+      reducer: (a, b) => a.concat(b),
+    }),
+    child: Annotation<"child_one" | "child_two">,
+  })
+)
+  .addNode("c_one", () => ({ messages: ["Entered c_one node"] }))
+  .addNode("c_two", () => ({ messages: ["Entered c_two node"] }))
+  .addEdge(START, "c_one")
+  .addEdge("c_one", "c_two")
+  .addEdge("c_two", END);
+
+const parent = new StateGraph(
+  Annotation.Root({
+    messages: Annotation<string[]>({
+      reducer: (a, b) => a.concat(b),
+    }),
+    parent: Annotation<"parent_one" | "parent_two">,
+  })
+)
+  .addNode("p_one", () => ({ messages: ["Entered p_one node"] }))
+  .addNode("p_two", child.compile())
+  .addEdge(START, "p_one")
+  .addEdge("p_one", "p_two")
+  .addEdge("p_two", END);
+
+const grandParent = new StateGraph(
+  Annotation.Root({
+    messages: Annotation<string[]>({
+      reducer: (a, b) => a.concat(b),
+    }),
+  })
+)
+  .addNode("gp_one", () => ({ messages: ["Entered gp_one node"] }))
+  .addNode("gp_two", parent.compile())
+  .addEdge(START, "gp_one")
+  .addEdge("gp_one", "gp_two")
+  .addEdge("gp_two", END);
+
+export const graph = grandParent.compile();

langgraph_api/js/tests/graphs/package.json

@@ -0,0 +1,7 @@
+{
+  "private": true,
+  "dependencies": {
+    "@langchain/core": "^0.3.9",
+    "@langchain/langgraph": "^0.2.14"
+  }
+}

langgraph_api/js/tests/graphs/weather.mts

@@ -0,0 +1,57 @@
+import { Annotation, StateGraph, END, START } from "@langchain/langgraph";
+import { MessagesAnnotation } from "@langchain/langgraph";
+import { AIMessage } from "@langchain/core/messages";
+
+const state = MessagesAnnotation;
+
+const weatherState = Annotation.Root({
+  ...state.spec,
+  city: Annotation<string>,
+});
+
+const routerState = Annotation.Root({
+  ...state.spec,
+  route: Annotation<"weather" | "other">,
+});
+
+const weather = new StateGraph(weatherState)
+  .addNode("model_node", (state) => {
+    const llm = new AIMessage({
+      content: "",
+      tool_calls: [
+        {
+          id: "tool_call123",
+          name: "get_weather",
+          args: { city: "San Francisco" },
+        },
+      ],
+    });
+
+    return { city: llm.tool_calls![0].args.city };
+  })
+  .addNode("weather_node", async (state) => {
+    const result = `It's sunny in ${state.city}!`;
+    return { messages: [new AIMessage({ content: result })] };
+  })
+  .addEdge(START, "model_node")
+  .addEdge("model_node", "weather_node")
+  .addEdge("weather_node", END)
+  .compile({ interruptBefore: ["weather_node"] });
+
+const router = new StateGraph(routerState)
+  .addNode("router_node", async () => ({ route: "weather" }))
+  .addNode("normal_llm_node", () => ({ messages: [new AIMessage("Hello")] }))
+  .addNode("weather_graph", weather)
+  .addEdge(START, "router_node")
+  .addConditionalEdges(
+    "router_node",
+    ({ route }) => {
+      if (route === "weather") return "weather_graph";
+      return "normal_llm_node";
+    },
+    ["weather_graph", "normal_llm_node"]
+  )
+  .addEdge("weather_graph", END)
+  .addEdge("normal_llm_node", END);
+
+export const graph = router.compile();

langgraph_api/js/tests/graphs/yarn.lock

@@ -0,0 +1,159 @@
+# THIS IS AN AUTOGENERATED FILE. DO NOT EDIT THIS FILE DIRECTLY.
+# yarn lockfile v1
+
+
+"@langchain/core@^0.3.9":
+  version "0.3.9"
+  resolved "https://registry.yarnpkg.com/@langchain/core/-/core-0.3.9.tgz#1fa75be84a22d951946277bc4ad3ea656c77fafc"
+  integrity sha512-Rttr9FuFwU+CWIoEyuLqQUPYg+3pKL1YpDgo3nvoDVhinoHqwGQ7aNGzZ/Sf+qASMi76sPSLm+75pHMJwwOiWg==
+  dependencies:
+    ansi-styles "^5.0.0"
+    camelcase "6"
+    decamelize "1.2.0"
+    js-tiktoken "^1.0.12"
+    langsmith "^0.1.56"
+    mustache "^4.2.0"
+    p-queue "^6.6.2"
+    p-retry "4"
+    uuid "^10.0.0"
+    zod "^3.22.4"
+    zod-to-json-schema "^3.22.3"
+
+"@langchain/langgraph-checkpoint@~0.0.10":
+  version "0.0.10"
+  resolved "https://registry.yarnpkg.com/@langchain/langgraph-checkpoint/-/langgraph-checkpoint-0.0.10.tgz#30d896ff48a9431ece0afa7e60e420ad2234b338"
+  integrity sha512-BMfJD5Eg39pM0iJmEv50qJL5dJJI5U2oHuNXixWlQ1BKsvtbSs713+EHc21uuvcJUct1MPiv7RdfvwXycLM/aQ==
+  dependencies:
+    uuid "^10.0.0"
+
+"@langchain/langgraph@^0.2.14":
+  version "0.2.14"
+  resolved "https://registry.yarnpkg.com/@langchain/langgraph/-/langgraph-0.2.14.tgz#f698e948b923be666ffe61601da0cfbd363f8968"
+  integrity sha512-gvneCZDzYzpt+P6ye7pveiRZtlGKWFKk3XAck31yxSf5D/++lP8s6ocMY1x+UaFEfAYd5Qj6jNPI9aPp9Y75jQ==
+  dependencies:
+    "@langchain/langgraph-checkpoint" "~0.0.10"
+    double-ended-queue "^2.1.0-0"
+    uuid "^10.0.0"
+    zod "^3.23.8"
+
+"@types/retry@0.12.0":
+  version "0.12.0"
+  resolved "https://registry.yarnpkg.com/@types/retry/-/retry-0.12.0.tgz#2b35eccfcee7d38cd72ad99232fbd58bffb3c84d"
+  integrity sha512-wWKOClTTiizcZhXnPY4wikVAwmdYHp8q6DmC+EJUzAMsycb7HB32Kh9RN4+0gExjmPmZSAQjgURXIGATPegAvA==
+
+"@types/uuid@^10.0.0":
+  version "10.0.0"
+  resolved "https://registry.yarnpkg.com/@types/uuid/-/uuid-10.0.0.tgz#e9c07fe50da0f53dc24970cca94d619ff03f6f6d"
+  integrity sha512-7gqG38EyHgyP1S+7+xomFtL+ZNHcKv6DwNaCZmJmo1vgMugyF3TCnXVg4t1uk89mLNwnLtnY3TpOpCOyp1/xHQ==
+
+ansi-styles@^5.0.0:
+  version "5.2.0"
+  resolved "https://registry.yarnpkg.com/ansi-styles/-/ansi-styles-5.2.0.tgz#07449690ad45777d1924ac2abb2fc8895dba836b"
+  integrity sha512-Cxwpt2SfTzTtXcfOlzGEee8O+c+MmUgGrNiBcXnuWxuFJHe6a5Hz7qwhwe5OgaSYI0IJvkLqWX1ASG+cJOkEiA==
+
+base64-js@^1.5.1:
+  version "1.5.1"
+  resolved "https://registry.yarnpkg.com/base64-js/-/base64-js-1.5.1.tgz#1b1b440160a5bf7ad40b650f095963481903930a"
+  integrity sha512-AKpaYlHn8t4SVbOHCy+b5+KKgvR4vrsD8vbvrbiQJps7fKDTkjkDry6ji0rUJjC0kzbNePLwzxq8iypo41qeWA==
+
+camelcase@6:
+  version "6.3.0"
+  resolved "https://registry.yarnpkg.com/camelcase/-/camelcase-6.3.0.tgz#5685b95eb209ac9c0c177467778c9c84df58ba9a"
+  integrity sha512-Gmy6FhYlCY7uOElZUSbxo2UCDH8owEk996gkbrpsgGtrJLM3J7jGxl9Ic7Qwwj4ivOE5AWZWRMecDdF7hqGjFA==
+
+commander@^10.0.1:
+  version "10.0.1"
+  resolved "https://registry.yarnpkg.com/commander/-/commander-10.0.1.tgz#881ee46b4f77d1c1dccc5823433aa39b022cbe06"
+  integrity sha512-y4Mg2tXshplEbSGzx7amzPwKKOCGuoSRP/CjEdwwk0FOGlUbq6lKuoyDZTNZkmxHdJtp54hdfY/JUrdL7Xfdug==
+
+decamelize@1.2.0:
+  version "1.2.0"
+  resolved "https://registry.yarnpkg.com/decamelize/-/decamelize-1.2.0.tgz#f6534d15148269b20352e7bee26f501f9a191290"
+  integrity sha512-z2S+W9X73hAUUki+N+9Za2lBlun89zigOyGrsax+KUQ6wKW4ZoWpEYBkGhQjwAjjDCkWxhY0VKEhk8wzY7F5cA==
+
+double-ended-queue@^2.1.0-0:
+  version "2.1.0-0"
+  resolved "https://registry.yarnpkg.com/double-ended-queue/-/double-ended-queue-2.1.0-0.tgz#103d3527fd31528f40188130c841efdd78264e5c"
+  integrity sha512-+BNfZ+deCo8hMNpDqDnvT+c0XpJ5cUa6mqYq89bho2Ifze4URTqRkcwR399hWoTrTkbZ/XJYDgP6rc7pRgffEQ==
+
+eventemitter3@^4.0.4:
+  version "4.0.7"
+  resolved "https://registry.yarnpkg.com/eventemitter3/-/eventemitter3-4.0.7.tgz#2de9b68f6528d5644ef5c59526a1b4a07306169f"
+  integrity sha512-8guHBZCwKnFhYdHr2ysuRWErTwhoN2X8XELRlrRwpmfeY2jjuUN4taQMsULKUVo1K4DvZl+0pgfyoysHxvmvEw==
+
+js-tiktoken@^1.0.12:
+  version "1.0.12"
+  resolved "https://registry.yarnpkg.com/js-tiktoken/-/js-tiktoken-1.0.12.tgz#af0f5cf58e5e7318240d050c8413234019424211"
+  integrity sha512-L7wURW1fH9Qaext0VzaUDpFGVQgjkdE3Dgsy9/+yXyGEpBKnylTd0mU0bfbNkKDlXRb6TEsZkwuflu1B8uQbJQ==
+  dependencies:
+    base64-js "^1.5.1"
+
+langsmith@^0.1.56:
+  version "0.1.64"
+  resolved "https://registry.yarnpkg.com/langsmith/-/langsmith-0.1.64.tgz#d10cc8bb0e1aa38f67f385c5835eecbfd7b7d154"
+  integrity sha512-fA827iH34G6zFzZx5kzbZcevtulqq8qo1e9g/Rn+L0TXuOJSKLMEfVbDbX6WD9laus9P8ItZcp27E5/DngOFHQ==
+  dependencies:
+    "@types/uuid" "^10.0.0"
+    commander "^10.0.1"
+    p-queue "^6.6.2"
+    p-retry "4"
+    semver "^7.6.3"
+    uuid "^10.0.0"
+
+mustache@^4.2.0:
+  version "4.2.0"
+  resolved "https://registry.yarnpkg.com/mustache/-/mustache-4.2.0.tgz#e5892324d60a12ec9c2a73359edca52972bf6f64"
+  integrity sha512-71ippSywq5Yb7/tVYyGbkBggbU8H3u5Rz56fH60jGFgr8uHwxs+aSKeqmluIVzM0m0kB7xQjKS6qPfd0b2ZoqQ==
+
+p-finally@^1.0.0:
+  version "1.0.0"
+  resolved "https://registry.yarnpkg.com/p-finally/-/p-finally-1.0.0.tgz#3fbcfb15b899a44123b34b6dcc18b724336a2cae"
+  integrity sha512-LICb2p9CB7FS+0eR1oqWnHhp0FljGLZCWBE9aix0Uye9W8LTQPwMTYVGWQWIw9RdQiDg4+epXQODwIYJtSJaow==
+
+p-queue@^6.6.2:
+  version "6.6.2"
+  resolved "https://registry.yarnpkg.com/p-queue/-/p-queue-6.6.2.tgz#2068a9dcf8e67dd0ec3e7a2bcb76810faa85e426"
+  integrity sha512-RwFpb72c/BhQLEXIZ5K2e+AhgNVmIejGlTgiB9MzZ0e93GRvqZ7uSi0dvRF7/XIXDeNkra2fNHBxTyPDGySpjQ==
+  dependencies:
+    eventemitter3 "^4.0.4"
+    p-timeout "^3.2.0"
+
+p-retry@4:
+  version "4.6.2"
+  resolved "https://registry.yarnpkg.com/p-retry/-/p-retry-4.6.2.tgz#9baae7184057edd4e17231cee04264106e092a16"
+  integrity sha512-312Id396EbJdvRONlngUx0NydfrIQ5lsYu0znKVUzVvArzEIt08V1qhtyESbGVd1FGX7UKtiFp5uwKZdM8wIuQ==
+  dependencies:
+    "@types/retry" "0.12.0"
+    retry "^0.13.1"
+
+p-timeout@^3.2.0:
+  version "3.2.0"
+  resolved "https://registry.yarnpkg.com/p-timeout/-/p-timeout-3.2.0.tgz#c7e17abc971d2a7962ef83626b35d635acf23dfe"
+  integrity sha512-rhIwUycgwwKcP9yTOOFK/AKsAopjjCakVqLHePO3CC6Mir1Z99xT+R63jZxAT5lFZLa2inS5h+ZS2GvR99/FBg==
+  dependencies:
+    p-finally "^1.0.0"
+
+retry@^0.13.1:
+  version "0.13.1"
+  resolved "https://registry.yarnpkg.com/retry/-/retry-0.13.1.tgz#185b1587acf67919d63b357349e03537b2484658"
+  integrity sha512-XQBQ3I8W1Cge0Seh+6gjj03LbmRFWuoszgK9ooCpwYIrhhoO80pfq4cUkU5DkknwfOfFteRwlZ56PYOGYyFWdg==
+
+semver@^7.6.3:
+  version "7.6.3"
+  resolved "https://registry.yarnpkg.com/semver/-/semver-7.6.3.tgz#980f7b5550bc175fb4dc09403085627f9eb33143"
+  integrity sha512-oVekP1cKtI+CTDvHWYFUcMtsK/00wmAEfyqKfNdARm8u1wNVhSgaX7A8d4UuIlUI5e84iEwOhs7ZPYRmzU9U6A==
+
+uuid@^10.0.0:
+  version "10.0.0"
+  resolved "https://registry.yarnpkg.com/uuid/-/uuid-10.0.0.tgz#5a95aa454e6e002725c79055fd42aaba30ca6294"
+  integrity sha512-8XkAphELsDnEGrDxUOHB3RGvXz6TeuYSGEZBOjtTtPm2lwhGBjLgOzLHB63IUWfBpNucQjND6d3AOudO+H3RWQ==
+
+zod-to-json-schema@^3.22.3:
+  version "3.23.2"
+  resolved "https://registry.yarnpkg.com/zod-to-json-schema/-/zod-to-json-schema-3.23.2.tgz#bc7e379c8050462538383e382964c03d8fe008f9"
+  integrity sha512-uSt90Gzc/tUfyNqxnjlfBs8W6WSGpNBv0rVsNxP/BVSMHMKGdthPYff4xtCHYloJGM0CFxFsb3NbC0eqPhfImw==
+
+zod@^3.22.4, zod@^3.23.8:
+  version "3.23.8"
+  resolved "https://registry.yarnpkg.com/zod/-/zod-3.23.8.tgz#e37b957b5d52079769fb8097099b592f0ef4067d"
+  integrity sha512-XBx9AXhXktjUqnepgTiE5flcKIYWi/rme0Eaj+5Y0lftuGBq+jyRu/md4WnuxqgP1ubdpNCsYEYPxrzVHD8d6g==