langgraph-api 0.0.14__py3-none-any.whl → 0.0.16__py3-none-any.whl

langgraph_api/js/tests/api.test.mts

@@ -447,7 +447,7 @@ describe("runs", () => {
     await sql`DELETE FROM store`;
   });
 
-  it.skip.concurrent("list runs", async () => {
+  it.concurrent("list runs", async () => {
     const assistant = await client.assistants.create({ graphId: "agent" });
     const thread = await client.threads.create();
     await client.runs.wait(thread.thread_id, assistant.assistant_id, {
@@ -753,13 +753,13 @@ describe("runs", () => {
       messages = findLast(chunks, (i) => i.event === "values")?.data.messages;
 
       const threadAfterInterrupt = await client.threads.get(thread.thread_id);
-      expect(threadAfterInterrupt.status).toBe("idle");
+      expect(threadAfterInterrupt.status).toBe("interrupted");
 
       expect(messages.at(-1)).not.toBeNull();
-      expect(messages.at(-1)?.content).toBe("end");
+      expect(messages.at(-1)?.content).toBe("begin");
 
       const state = await client.threads.getState(thread.thread_id);
-      expect(state.next).toEqual([]);
+      expect(state.next).toEqual(["tool"]);
 
       // continue after interrupt
       chunks = await gatherIterator(
@@ -817,6 +817,7 @@ describe("runs", () => {
     });
 
     const modifiedThread = await client.threads.get(thread.thread_id);
+    expect(modifiedThread.status).toBe("interrupted");
     expect(modifiedThread.metadata?.modified).toBe(true);
 
     const stateAfterModify = await client.threads.getState<AgentState>(
@@ -836,22 +837,42 @@ describe("runs", () => {
       })
     );
 
+    const threadAfterContinue = await client.threads.get(thread.thread_id);
+    expect(threadAfterContinue.status).toBe("idle");
+
     expect(chunks.filter((i) => i.event === "error").length).toBe(0);
     messages = findLast(chunks, (i) => i.event === "values")?.data.messages;
 
-    expect(messages.length).toBe(8);
-    expect(messages[4].content).toBe(`tool_call__modified`);
+    expect(messages.length).toBe(4);
+    expect(messages[2].content).toBe(`tool_call__modified`);
     expect(messages.at(-1)?.content).toBe("end");
 
     // get the history
     const history = await client.threads.getHistory<AgentState>(
       thread.thread_id
     );
-    expect(history.length).toBe(10);
+    expect(history.length).toBe(6);
     expect(history[0].next.length).toBe(0);
-    expect(history[0].values.messages.length).toBe(8);
+    expect(history[0].values.messages.length).toBe(4);
     expect(history.at(-1)?.next).toEqual(["__start__"]);
   });
+
+  it.concurrent("interrupt before", async () => {
+    const assistant = await client.assistants.create({ graphId: "agent" });
+    let thread = await client.threads.create();
+    const input = {
+      messages: [{ type: "human", content: "foo", id: "initial-message" }],
+    };
+
+    await client.runs.wait(thread.thread_id, assistant.assistant_id, {
+      input,
+      interruptBefore: ["agent"],
+      config: globalConfig,
+    });
+
+    thread = await client.threads.get(thread.thread_id);
+    expect(thread.status).toBe("interrupted");
+  });
 });
 
 describe("shared state", () => {
@@ -1650,3 +1671,99 @@ describe("errors", () => {
     expect(runState.status).toEqual("error");
   });
 });
+
+describe("long running tasks", () => {
+  it.concurrent.for([1000, 8000, 12000])(
+    "long running task with %dms delay",
+    { timeout: 15_000 },
+    async (delay) => {
+      const assistant = await client.assistants.create({ graphId: "delay" });
+      const thread = await client.threads.create();
+
+      const run = await client.runs.create(
+        thread.thread_id,
+        assistant.assistant_id,
+        {
+          input: { messages: [], delay },
+          config: globalConfig,
+        }
+      );
+
+      await client.runs.join(thread.thread_id, run.run_id);
+
+      const runState = await client.runs.get(thread.thread_id, run.run_id);
+      expect(runState.status).toEqual("success");
+
+      const runResult = await client.threads.getState<{
+        messages: BaseMessageLike[];
+        delay: number;
+      }>(thread.thread_id);
+
+      expect(runResult.values.messages).toMatchObject([
+        { content: `finished after ${delay}ms` },
+      ]);
+    }
+  );
+});
+
+// Not implemented in JS yet
+describe.skip("command update state", () => {
+  it("updates state via commands", async () => {
+    const assistant = await client.assistants.create({ graphId: "agent" });
+    const thread = await client.threads.create();
+
+    const input = { messages: [{ role: "human", content: "foo" }] };
+
+    // dict-based updates
+    await client.runs.wait(thread.thread_id, assistant.assistant_id, {
+      input,
+      config: globalConfig,
+    });
+    let stream = await gatherIterator(
+      client.runs.stream(thread.thread_id, assistant.assistant_id, {
+        command: { update: { keyOne: "value3", keyTwo: "value4" } },
+        config: globalConfig,
+      })
+    );
+    expect(stream.filter((chunk) => chunk.event === "error")).toEqual([]);
+
+    let state = await client.threads.getState<{
+      keyOne: string;
+      keyTwo: string;
+    }>(thread.thread_id);
+
+    expect(state.values).toMatchObject({
+      keyOne: "value3",
+      keyTwo: "value4",
+    });
+
+    // list-based updates
+    await client.runs.wait(thread.thread_id, assistant.assistant_id, {
+      input,
+      config: globalConfig,
+    });
+    stream = await gatherIterator(
+      client.runs.stream(thread.thread_id, assistant.assistant_id, {
+        command: {
+          update: [
+            ["keyOne", "value1"],
+            ["keyTwo", "value2"],
+          ],
+        },
+        config: globalConfig,
+      })
+    );
+
+    expect(stream.filter((chunk) => chunk.event === "error")).toEqual([]);
+
+    state = await client.threads.getState<{
+      keyOne: string;
+      keyTwo: string;
+    }>(thread.thread_id);
+
+    expect(state.values).toMatchObject({
+      keyOne: "value1",
+      keyTwo: "value2",
+    });
+  });
+});

langgraph_api/js/tests/compose-postgres.yml

@@ -34,7 +34,7 @@ services:
         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 LANGSERVE_GRAPHS='{"agent":"./agent.mts:graph", "nested": "./nested.mts:graph", "weather": "./weather.mts:graph", "error": "./error.mts:graph", "delay": "./delay.mts:graph"}'
         ENV LANGGRAPH_CONFIG='{"agent": {"configurable": {"model_name": "openai"}}}'
         RUN tsx /api/langgraph_api/js/build.mts
     depends_on:
@@ -54,3 +54,4 @@ services:
       DATABASE_URI: postgres://postgres:postgres@langgraph-postgres:5432/postgres?sslmode=disable
       N_JOBS_PER_WORKER: "2"
       LANGGRAPH_CLOUD_LICENSE_KEY: ${LANGGRAPH_CLOUD_LICENSE_KEY}
+      FF_JS_ZEROMQ_ENABLED: ${FF_JS_ZEROMQ_ENABLED}

langgraph_api/js/tests/graphs/agent.mts

@@ -18,6 +18,8 @@ const GraphAnnotationOutput = Annotation.Root({
   }),
   sharedStateValue: Annotation<string | null>(),
   interrupt: Annotation<boolean>(),
+  keyOne: Annotation<string | null>(),
+  keyTwo: Annotation<string | null>(),
 });
 
 const GraphAnnotationInput = Annotation.Root({

langgraph_api/js/tests/graphs/delay.mts

@@ -0,0 +1,30 @@
+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/langgraph.json

@@ -3,6 +3,7 @@
   "graphs": {
     "agent": "./agent.mts:graph",
     "subgraph": "./subgraph.mts:graph",
-    "nested": "./nested.mts:graph"
+    "nested": "./nested.mts:graph",
+    "delay": "./delay.mts:graph"
   }
 }