langgraph-api 0.2.26__py3-none-any.whl → 0.2.28__py3-none-any.whl

langgraph_api/js/tests/compose-postgres.yml

@@ -1,59 +0,0 @@
-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:${NODE_VERSION:-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", "delay": "./delay.mts:graph", "dynamic": "./dynamic.mts:graph", "command": "./command.mts:graph", "agent_simple": "./agent_simple.mts:graph"}'
-        ENV LANGGRAPH_CONFIG='{"agent": {"configurable": {"model_name": "openai"}}}'
-        ENV LANGGRAPH_HTTP='{"app": "./http.mts:app"}'
-        ENV LANGGRAPH_UI='{"agent": "./agent.ui.tsx"}'
-        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: "5"
-      LANGGRAPH_CLOUD_LICENSE_KEY: ${LANGGRAPH_CLOUD_LICENSE_KEY}
-      FF_JS_ZEROMQ_ENABLED: ${FF_JS_ZEROMQ_ENABLED}

langgraph_api/js/tests/graphs/.gitignore

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

langgraph_api/js/tests/graphs/agent.css

@@ -1 +0,0 @@
-@import "tailwindcss";

langgraph_api/js/tests/graphs/agent.mts

@@ -1,187 +0,0 @@
-import { BaseMessage, ToolMessage } from "@langchain/core/messages";
-import {
-  Annotation,
-  StateGraph,
-  START,
-  END,
-  messagesStateReducer,
-  SharedValue,
-  interrupt,
-  type LangGraphRunnableConfig,
-} from "@langchain/langgraph";
-import { FakeListChatModel } from "@langchain/core/utils/testing";
-import { ChatGenerationChunk } from "@langchain/core/outputs";
-import { v4 as uuidv4 } from "uuid";
-import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
-const GraphAnnotationOutput = Annotation.Root({
-  messages: Annotation<BaseMessage[]>({
-    reducer: messagesStateReducer,
-    default: () => [],
-  }),
-  sharedStateValue: Annotation<string | null>(),
-  interrupt: Annotation<boolean>(),
-  keyOne: Annotation<string | null>(),
-  keyTwo: Annotation<string | null>(),
-  sleep: Annotation<number | null>(),
-});
-
-const GraphAnnotationInput = Annotation.Root({
-  ...GraphAnnotationOutput.spec,
-  sharedState: SharedValue.on("user_id"),
-  sharedStateFromStoreConfig: Annotation<Record<string, any> | null>,
-});
-
-class StableFakeListChatModel extends FakeListChatModel {
-  streamMessageId: string = uuidv4();
-
-  async *_streamResponseChunks(
-    _messages: BaseMessage[],
-    options: this["ParsedCallOptions"],
-    runManager?: CallbackManagerForLLMRun,
-  ): AsyncGenerator<ChatGenerationChunk> {
-    const response = this._currentResponse();
-    this._incrementResponse();
-    this.streamMessageId = uuidv4();
-
-    if (this.emitCustomEvent) {
-      await runManager?.handleCustomEvent("some_test_event", {
-        someval: true,
-      });
-    }
-
-    for await (const text of response) {
-      await this._sleepIfRequested();
-      if (options?.thrownErrorString) {
-        throw new Error(options.thrownErrorString);
-      }
-      const chunk = this._createResponseChunk(text);
-
-      // ensure stable ID
-      chunk.message.id = this.streamMessageId;
-      chunk.message.lc_kwargs.id = this.streamMessageId;
-
-      yield chunk;
-
-      void runManager?.handleLLMNewToken(
-        text,
-        undefined,
-        undefined,
-        undefined,
-        undefined,
-        { chunk },
-      );
-    }
-  }
-}
-
-// For shared state
-const namespace = ["sharedState", "data"];
-const key = "user_id";
-
-const modelMap: Record<string, FakeListChatModel> = {};
-const getModel = (threadId: string) => {
-  modelMap[threadId] ??= new StableFakeListChatModel({
-    responses: ["begin", "end"],
-  });
-  return modelMap[threadId];
-};
-
-const agentNode = async (
-  state: typeof GraphAnnotationInput.State,
-  config: LangGraphRunnableConfig,
-) => {
-  if (state.interrupt) interrupt("i want to interrupt");
-
-  if (state.sleep != null && state.messages.at(-1)?.getType() === "human") {
-    const sleep = state.sleep;
-    await new Promise((resolve) => setTimeout(resolve, sleep * 1000));
-  }
-
-  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 {
-    interrupt: false,
-    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/agent.ui.tsx

@@ -1,10 +0,0 @@
-import "./agent.css";
-import React from "react";
-
-export function StockPrice() {
-  return <div>Stock Price</div>;
-}
-
-export default {
-  "stock-price": StockPrice,
-} as const;

langgraph_api/js/tests/graphs/agent_simple.mts

@@ -1,105 +0,0 @@
-import {
-  StateGraph,
-  END,
-  Send,
-  MessagesAnnotation,
-  Annotation,
-  START,
-  LangGraphRunnableConfig,
-} from "@langchain/langgraph";
-import { AIMessage, BaseMessage, ToolMessage } from "@langchain/core/messages";
-import { FakeListChatModel } from "@langchain/core/utils/testing";
-
-const getStableModel = (() => {
-  const cached: Record<string, FakeListChatModel> = {};
-  return (threadId: string) => {
-    cached[threadId] ??= new FakeListChatModel({
-      responses: ["begin", "end\u2028"],
-    });
-    return cached[threadId];
-  };
-})();
-
-const AgentState = Annotation.Root({
-  key_one: Annotation<string>(),
-  key_two: Annotation<string>(),
-  sleep: Annotation<number>(),
-  messages: MessagesAnnotation.spec.messages,
-  prompts: MessagesAnnotation.spec.messages,
-});
-
-async function callModel(
-  state: typeof AgentState.State,
-  config: LangGraphRunnableConfig,
-): Promise<typeof AgentState.Update> {
-  let userId: string | undefined;
-
-  if (config.configurable?.langgraph_auth_user != null) {
-    const user = config.configurable?.langgraph_auth_user as
-      | { identity: string }
-      | undefined;
-
-    userId = user?.identity;
-  }
-
-  if (config.configurable?.["x-configurable-header"] != null) {
-    return {
-      messages: [`end: ${config.configurable?.["x-configurable-header"]}`],
-    };
-  }
-
-  if (config.configurable?.["map-reduce"] != null) {
-    return { messages: ["map-reduce"] };
-  }
-
-  const model = getStableModel(config.configurable?.thread_id ?? "$");
-  const existing = await config.store?.get([userId ?? "ALL"], "key_one");
-  if (!existing) {
-    const text = state.messages.at(-1)?.content;
-    await config.store?.put([userId ?? "ALL"], "key_one", { text });
-  }
-
-  const response = await model.invoke(state.messages);
-  return { messages: [response] };
-}
-
-async function callTool(
-  message: BaseMessage,
-): Promise<typeof AgentState.Update> {
-  const response = new ToolMessage(
-    `tool_call__${message.content}`,
-    "tool_call_id",
-  );
-  return { messages: [response] };
-}
-
-function shouldContinue(
-  state: typeof AgentState.State,
-): typeof END | Send | Send[] {
-  const lastMessage = state.messages.at(-1);
-  if ((lastMessage?.content as string).startsWith("end")) return END;
-  if ((lastMessage?.content as string).includes("map-reduce")) {
-    return [
-      new Send("map-reduce", { messages: [new AIMessage("first")] }),
-      new Send("map-reduce", { messages: [new AIMessage("second")] }),
-      new Send("map-reduce", { messages: [new AIMessage("third")] }),
-    ];
-  }
-  return new Send("tool", lastMessage);
-}
-
-function callMapReduce(state: {
-  messages: BaseMessage[];
-}): typeof AgentState.Update {
-  return { prompts: state.messages.slice(-1) };
-}
-
-const workflow = new StateGraph(AgentState)
-  .addNode("agent", callModel)
-  .addNode("tool", callTool)
-  .addNode("map-reduce", callMapReduce)
-  .addEdge(START, "agent")
-  .addConditionalEdges("agent", shouldContinue)
-  .addEdge("tool", "agent");
-
-export const graph = workflow.compile();

langgraph_api/js/tests/graphs/auth.mts

@@ -1,106 +0,0 @@
-import { Auth, HTTPException } from "@langchain/langgraph-sdk/auth";
-import { JWTPayload, jwtVerify } from "jose";
-
-const SECRET_KEY = new TextEncoder().encode(
-  "09d25e094faa6ca2556c818166b7a9563b93f7099f6f0f4caa6cf63b88e8d3e7",
-);
-const ALGORITHM = "HS256";
-
-const USERS_DB: Record<
-  string,
-  {
-    username: string;
-    identity: string;
-    full_name: string;
-    email: string;
-    permissions: string[];
-    hashed_password: string;
-    disabled: boolean;
-  }
-> = {
-  johndoe: {
-    username: "johndoe",
-    identity: "johndoe",
-    full_name: "John Doe",
-    email: "johndoe@example.com",
-    permissions: ["read", "write", "assistants:write", "me"],
-    hashed_password:
-      "$2b$12$EixZaYVK1fsbw1ZfbX3OXePaWxn96p36WQoeG6Lruj3vjPGga31lW",
-    disabled: false,
-  },
-  alice: {
-    username: "alice",
-    identity: "alice",
-    full_name: "Alice Chains",
-    email: "alicechains@example.com",
-    permissions: ["read", "write", "assistants:write", "me"],
-    hashed_password:
-      "$2b$12$gSvqqUPvlXP2tfVFaWK1Be7DlH.PKZbv5H8KnzzVgXXbVxpva.pFm",
-    disabled: true,
-  },
-};
-export const auth = new Auth()
-  .authenticate(async (request) => {
-    const authorization = request.headers.get("Authorization");
-
-    const exc = new HTTPException(401, {
-      message: "Could not validate credentials",
-      headers: { "WWW-Authenticate": "Bearer" },
-    });
-
-    if (!authorization?.toLocaleLowerCase().startsWith("bearer ")) {
-      throw exc;
-    }
-
-    let payload: JWTPayload | undefined;
-    try {
-      const token = authorization.split(" ")[1];
-      const result = await jwtVerify(token, SECRET_KEY, {
-        algorithms: [ALGORITHM],
-      });
-      payload = result.payload;
-    } catch (error) {
-      throw new HTTPException(401, {
-        message: "Failed to verify JWT token",
-        cause: error,
-      });
-    }
-
-    const scopes = (payload["scopes"] ?? []) as string[];
-    const username = payload["sub"] as string | undefined;
-    const user = username ? USERS_DB[username] : null;
-    if (!user) throw exc;
-
-    let permissions = user.permissions ?? [];
-    permissions = scopes.filter((scope) => permissions.includes(scope));
-
-    return { ...user, permissions };
-  })
-  .on("*", ({ permissions }) => {
-    if (!permissions?.length) {
-      throw new HTTPException(403, { message: "Not authorized" });
-    }
-  })
-  .on("assistants:create", ({ value, user, permissions }) => {
-    if (!permissions?.includes("assistants:write")) {
-      throw new HTTPException(403, { message: "Not authorized" });
-    }
-
-    value.metadata ??= {};
-    value.metadata["owner"] = user.identity;
-  })
-  .on("assistants:search", (params) => ({ owner: params.user.identity }))
-  .on(["threads", "assistants"], ({ action, value, user }) => {
-    const filters = { owner: user.identity };
-    if (action === "create" || action === "update" || action === "create_run") {
-      value.metadata ??= {};
-      value.metadata["owner"] = user.identity;
-    }
-    return filters;
-  })
-  .on("store", ({ value, user }) => {
-    const identity = user.identity;
-    if (!identity || !value.namespace?.includes(identity)) {
-      throw new HTTPException(403, { message: "Not authorized" });
-    }
-  });

langgraph_api/js/tests/graphs/command.mts

@@ -1,48 +0,0 @@
-import {
-  Annotation,
-  Command,
-  END,
-  interrupt,
-  Send,
-  START,
-  StateGraph,
-} from "@langchain/langgraph";
-
-const StateSchema = Annotation.Root({
-  messages: Annotation<string[]>({
-    reducer: (a: string[], b: string | string[]) => [
-      ...a,
-      ...(Array.isArray(b) ? b : [b]),
-    ],
-    default: () => [],
-  }),
-});
-
-export const graph = new StateGraph(StateSchema)
-  .addNode("router", () => new Command({ goto: END }), {
-    ends: ["before_interrupt", "map", END],
-  })
-  .addNode("before_interrupt", () => ({ messages: ["before_interrupt"] }))
-  .addNode("interrupt", () => {
-    const resolved = interrupt("interrupt");
-    return { messages: [`interrupt: ${resolved}`] };
-  })
-  .addNode(
-    "map",
-    () =>
-      new Command({
-        update: { messages: ["map"] },
-        goto: [
-          new Send("task", { value: 1 }),
-          new Send("task", { value: 2 }),
-          new Send("task", { value: 3 }),
-        ],
-      }),
-    { ends: ["task"] },
-  )
-  .addNode("task", (arg: { value: number }) => ({
-    messages: [`task: ${arg.value}`],
-  }))
-  .addEdge(START, "router")
-  .addEdge("before_interrupt", "interrupt")
-  .compile();

langgraph_api/js/tests/graphs/delay.mts

@@ -1,30 +0,0 @@
-import {
-  MessagesAnnotation,
-  StateGraph,
-  END,
-  START,
-  Annotation,
-} from "@langchain/langgraph";
-
-const StateSchema = Annotation.Root({
-  ...MessagesAnnotation.spec,
-  delay: Annotation<number>(),
-});
-
-const longRunning = async (
-  state: typeof StateSchema.State,
-): Promise<typeof StateSchema.Update> => {
-  if (state.delay === -1) {
-    while (true) {
-      // hang the event loop
-    }
-  }
-  await new Promise((resolve) => setTimeout(resolve, state.delay));
-  return { messages: [`finished after ${state.delay}ms`] };
-};
-
-export const graph = new StateGraph(StateSchema)
-  .addNode("long_running", longRunning)
-  .addEdge(START, "long_running")
-  .addEdge("long_running", END)
-  .compile();

langgraph_api/js/tests/graphs/dynamic.mts

@@ -1,24 +0,0 @@
-import { StateGraph, Annotation } from "@langchain/langgraph";
-
-export const graph = async (config: {
-  configurable?: { nodeName: string };
-}) => {
-  const node = config.configurable?.nodeName ?? "default";
-
-  const state = Annotation.Root({
-    node: Annotation<string>,
-    messages: Annotation<string[]>({
-      default: () => [],
-      reducer: (a: string[], b: string[] | string) => {
-        if (Array.isArray(b)) return [...a, ...b];
-        return [...a, b];
-      },
-    }),
-  });
-
-  return new StateGraph(state)
-    .addNode(node, () => ({ node, messages: [node] }))
-    .addEdge("__start__", node)
-    .addEdge(node, "__end__")
-    .compile();
-};

langgraph_api/js/tests/graphs/error.mts

@@ -1,17 +0,0 @@
-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/http.mts

@@ -1,76 +0,0 @@
-import { Hono } from "hono";
-import { HTTPException } from "hono/http-exception";
-import { streamText } from "hono/streaming";
-import { Client } from "@langchain/langgraph-sdk";
-
-let WEBHOOK_PAYLOAD: Record<string, unknown>;
-
-export const app = new Hono<{
-  Variables: {
-    body: string | ArrayBuffer | ReadableStream | null;
-  };
-}>()
-  .use(async (c, next) => {
-    if (c.req.query("interrupt") != null) {
-      return c.json({ status: "interrupted" });
-    }
-
-    await next();
-    c.header("x-js-middleware", "true");
-  })
-  .use(async (c, next) => {
-    const runsQuery = new RegExp(
-      "^(/runs(/stream|/wait)?$|/runs/batch$|/threads/[^/]+/runs(/stream|/wait)?)$",
-    );
-
-    if (c.req.method === "POST" && c.req.path.match(runsQuery)) {
-      const value = c.req.header("x-configurable-header");
-
-      if (value != null) {
-        const body = await c.req.json();
-
-        body["config"] ??= {};
-        body["config"]["configurable"] ??= {};
-        body["config"]["configurable"]["x-configurable-header"] ??= value;
-      }
-    }
-
-    await next();
-  })
-  .get("/custom/client", async (c) => {
-    const result = await new Client().runs.wait(null, "agent_simple", {
-      input: { messages: [{ role: "human", content: "input" }] },
-    });
-
-    return c.json({ result });
-  })
-  .get("/custom/my-route", (c) =>
-    c.json(
-      { foo: "bar" },
-      {
-        headers: {
-          "x-custom-output": c.req.header("x-custom-input") as string,
-        },
-      },
-    ),
-  )
-  .get("/runs/afakeroute", (c) => c.json({ foo: "afakeroute" }))
-  .get("/custom/error", () => {
-    throw new HTTPException(400, { message: "Bad request" });
-  })
-  .get("/custom/streaming", (c) =>
-    streamText(c, async (stream) => {
-      for (let i = 0; i < 4; i++) {
-        await stream.writeln(`Count: ${i}`);
-        await new Promise((resolve) => setTimeout(resolve, 10));
-      }
-
-      await stream.close();
-    }),
-  )
-  .post("/custom/webhook", async (c) => {
-    WEBHOOK_PAYLOAD = await c.req.json();
-    return c.json({ status: "success" });
-  })
-  .get("/custom/webhook-payload", (c) => c.json(WEBHOOK_PAYLOAD))
-  .notFound((c) => c.json({ status: "not-found" }));

langgraph_api/js/tests/graphs/langgraph.json

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