langgraph-api 0.0.13__py3-none-any.whl → 0.0.15__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, {
@@ -1650,3 +1650,37 @@ 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` },
+      ]);
+    }
+  );
+});

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/delay.mts

@@ -0,0 +1,25 @@
+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> => {
+  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"
   }
 }