langgraph-api 0.2.7__tar.gz → 0.2.9__tar.gz

{langgraph_api-0.2.7 → langgraph_api-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langgraph-api
-Version: 0.2.7
+Version: 0.2.9
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -18,7 +18,7 @@ Requires-Dist: jsonschema-rs (>=0.20.0,<0.30)
 Requires-Dist: langchain-core (>=0.2.38) ; python_version < "4.0"
 Requires-Dist: langgraph (>=0.3.27) ; python_version < "4.0"
 Requires-Dist: langgraph-checkpoint (>=2.0.23) ; python_version < "4.0"
-Requires-Dist: langgraph-runtime-inmem (>=0.0.7)
+Requires-Dist: langgraph-runtime-inmem (>=0.0.9,<0.1)
 Requires-Dist: langgraph-sdk (>=0.1.66,<0.2.0) ; python_version < "4.0"
 Requires-Dist: langsmith (>=0.1.63)
 Requires-Dist: orjson (>=3.9.7)

langgraph_api-0.2.9/langgraph_api/__init__.py

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

{langgraph_api-0.2.7 → langgraph_api-0.2.9}/langgraph_api/api/assistants.py

@@ -138,14 +138,19 @@ async def search_assistants(
     """List assistants."""
     payload = await request.json(AssistantSearchRequest)
     async with connect() as conn:
-        assistants_iter = await Assistants.search(
+        assistants_iter, total = await Assistants.search(
             conn,
             graph_id=payload.get("graph_id"),
             metadata=payload.get("metadata"),
             limit=int(payload.get("limit") or 10),
             offset=int(payload.get("offset") or 0),
+            sort_by=payload.get("sort_by"),
+            sort_order=payload.get("sort_order"),
         )
-    return ApiResponse([assistant async for assistant in assistants_iter])
+    return ApiResponse(
+        [assistant async for assistant in assistants_iter],
+        headers={"X-Pagination-Total": str(total)},
+    )
 
 
 @retry_db

{langgraph_api-0.2.7 → langgraph_api-0.2.9}/langgraph_api/js/package.json

@@ -33,7 +33,7 @@
   },
   "devDependencies": {
     "jose": "^6.0.10",
-    "@langchain/langgraph-sdk": "^0.0.70",
+    "@langchain/langgraph-sdk": "^0.0.73",
     "@types/react": "^19.0.8",
     "@types/react-dom": "^19.0.3",
     "@types/node": "^22.2.0",

{langgraph_api-0.2.7 → langgraph_api-0.2.9}/langgraph_api/js/tests/api.test.mts

@@ -37,6 +37,8 @@ interface AgentState {
   sharedStateValue?: string | null;
 }
 
+// the way this test is set up, it instantiates the client
+// with 6 graphs.
 beforeAll(async () => {
   await sql`DELETE FROM thread`;
   await sql`DELETE FROM store`;
@@ -44,12 +46,24 @@ beforeAll(async () => {
 });
 
 describe("assistants", () => {
+  beforeEach(async () => {
+    await sql`DELETE FROM assistant WHERE metadata->>'created_by' is null OR metadata->>'created_by' != 'system'`;
+  });
+
   it("create read update delete", async () => {
     const graphId = "agent";
     const config = { configurable: { model_name: "gpt" } };
 
-    let res = await client.assistants.create({ graphId, config });
-    expect(res).toMatchObject({ graph_id: graphId, config });
+    let res = await client.assistants.create({
+      graphId,
+      config,
+      name: "assistant1",
+    });
+    expect(res).toMatchObject({
+      graph_id: graphId,
+      config,
+      name: "assistant1",
+    });
 
     const metadata = { name: "woof" };
     await client.assistants.update(res.assistant_id, { graphId, metadata });
@@ -57,6 +71,36 @@ describe("assistants", () => {
     res = await client.assistants.get(res.assistant_id);
     expect(res).toMatchObject({ graph_id: graphId, config, metadata });
 
+    const secondAssistant = await client.assistants.create({
+      graphId,
+      config,
+      name: "assistant2",
+    });
+    expect(secondAssistant).toMatchObject({
+      graph_id: graphId,
+      config,
+      name: "assistant2",
+    });
+
+    const search = await client.assistants.search();
+    const customAssistants = search.filter(
+      (a) => a.metadata?.created_by !== "system",
+    );
+    expect(customAssistants.length).toBe(2);
+    expect(customAssistants[0].name).toBe("assistant2");
+    expect(customAssistants[1].name).toBe("assistant1");
+
+    const search2 = await client.assistants.search({
+      sortBy: "name",
+      sortOrder: "desc",
+    });
+    const customAssistants2 = search2.filter(
+      (a) => a.metadata?.created_by !== "system",
+    );
+    expect(customAssistants2.length).toBe(2);
+    expect(customAssistants2[0].name).toBe("assistant2");
+    expect(customAssistants2[1].name).toBe("assistant1");
+
     await client.assistants.delete(res.assistant_id);
     await expect(() => client.assistants.get(res.assistant_id)).rejects.toThrow(
       "HTTP 404: Not Found",
@@ -134,23 +178,27 @@ describe("assistants", () => {
 
   it("list assistants", async () => {
     let search = await client.assistants.search();
-    // Greater than or equal because the vitest retries can cause multiple assistants to be created
-    expect(search.length).toBeGreaterThanOrEqual(1);
+    const numAssistants = search.length;
+
+    const customAssistants = search.filter(
+      (a) => a.metadata?.created_by !== "system",
+    );
+    expect(customAssistants.length).toBe(0);
 
     const graphid = "agent";
     const create = await client.assistants.create({ graphId: "agent" });
 
     search = await client.assistants.search();
-    expect(search.length).toBeGreaterThanOrEqual(2);
+    expect(search.length).toBe(numAssistants + 1);
 
     search = await client.assistants.search({ graphId: graphid });
-    expect(search.length).toBeGreaterThanOrEqual(2);
+    expect(search.length).toBe(2);
     expect(search.every((i) => i.graph_id === graphid)).toBe(true);
 
     search = await client.assistants.search({
       metadata: { created_by: "system" },
     });
-    expect(search.length).toBeGreaterThanOrEqual(1);
+    expect(search.length).toBe(numAssistants);
     expect(search.every((i) => i.assistant_id !== create.assistant_id)).toBe(
       true,
     );
@@ -255,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" }] };
@@ -329,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();
 
@@ -375,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" }] };
@@ -419,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 = {

{langgraph_api-0.2.7 → langgraph_api-0.2.9}/langgraph_api/js/yarn.lock

@@ -236,10 +236,10 @@
   dependencies:
     uuid "^10.0.0"
 
-"@langchain/langgraph-sdk@^0.0.70":
-  version "0.0.70"
-  resolved "https://registry.yarnpkg.com/@langchain/langgraph-sdk/-/langgraph-sdk-0.0.70.tgz#9589f984b47de5e4a669b6008cbf01427a3d41d0"
-  integrity sha512-O8I12bfeMVz5fOrXnIcK4IdRf50IqyJTO458V56wAIHLNoi4H8/JHM+2M+Y4H2PtslXIGnvomWqlBd0eY5z/Og==
+"@langchain/langgraph-sdk@^0.0.73":
+  version "0.0.73"
+  resolved "https://registry.yarnpkg.com/@langchain/langgraph-sdk/-/langgraph-sdk-0.0.73.tgz#6f5cb44fb306461182df55be4cf1d263c5ce3e16"
+  integrity sha512-V3E7Bd1KNcTpnEglZqKpnQtth62WHf+Dxq5V+0CqMxbzPHnW9I4eynCO4c1/HSY/rHgEtvgj7fMvCAb6I5R+lQ==
   dependencies:
     "@types/json-schema" "^7.0.15"
     p-queue "^6.6.2"

{langgraph_api-0.2.7 → langgraph_api-0.2.9}/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.7 → langgraph_api-0.2.9}/openapi.json

@@ -3904,6 +3904,27 @@
             "description": "The number of results to skip.",
             "default": 0,
             "minimum": 0
+          },
+          "sort_by": {
+            "type": "string",
+            "enum": [
+              "assistant_id",
+              "created_at",
+              "updated_at",
+              "name",
+              "graph_id"
+            ],
+            "title": "Sort By",
+            "description": "The field to sort by."
+          },
+          "sort_order": {
+            "type": "string",
+            "enum": [
+              "asc",
+              "desc"
+            ],
+            "title": "Sort Order",
+            "description": "The order to sort by."
           }
         },
         "type": "object",

{langgraph_api-0.2.7 → langgraph_api-0.2.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.2.7"
+version = "0.2.9"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",
@@ -50,7 +50,7 @@ pyjwt = "^2.9.0"
 cryptography = ">=42.0.0,<45.0"
 langgraph-sdk = { version = "^0.1.66", python = "<4.0" }
 cloudpickle = "^3.0.0"
-langgraph-runtime-inmem = ">=0.0.7"
+langgraph-runtime-inmem = ">=0.0.9,<0.1"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "^0.11.7"

langgraph_api-0.2.7/langgraph_api/__init__.py

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