langgraph-api 0.2.7__py3-none-any.whl → 0.2.9__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.7"
+__version__ = "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/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/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/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/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.dist-info → langgraph_api-0.2.9.dist-info}/METADATA

@@ -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.7.dist-info → langgraph_api-0.2.9.dist-info}/RECORD

@@ -1,7 +1,7 @@
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api/__init__.py,sha256=XHypfHSPdgXFKmOdoewn7czU670gt8InhHhzlP5j_aA,22
+langgraph_api/__init__.py,sha256=F8OVhAhMXSkvvXYgZtbPn2SG1AQC3joK4yu-FrHt81Y,22
 langgraph_api/api/__init__.py,sha256=YVzpbn5IQotvuuLG9fhS9QMrxXfP4s4EpEMG0n4q3Nw,5625
-langgraph_api/api/assistants.py,sha256=6oYFRKlvqheJQGbWjFhQOUnnSbvsbrdMYLRJP7WtSRo,14481
+langgraph_api/api/assistants.py,sha256=mcKaVeNG8hQAV4IQDhaukS7FgVqTIVQNTyti3GfK2KI,14649
 langgraph_api/api/mcp.py,sha256=RvRYgANqRzNQzSmgjNkq4RlKTtoEJYil04ot9lsmEtE,14352
 langgraph_api/api/meta.py,sha256=sTgkhE-DaFWpERG6F7KelZfDsmJAiVc4j5dg50tDkSo,2950
 langgraph_api/api/openapi.py,sha256=WYpkjwsVV056qBkBVbGjhh-TfU9-IgZMY2au1s1PhZ4,15953
@@ -34,7 +34,7 @@ langgraph_api/js/client.http.mts,sha256=AGA-p8J85IcNh2oXZjDxHQ4PnQdJmt-LPcpZp6j0
 langgraph_api/js/client.mts,sha256=RZgkJk43CMOwnHXMXBbWIEhHh0W8WBwQ8Rm8GkJwvhM,30341
 langgraph_api/js/errors.py,sha256=Cm1TKWlUCwZReDC5AQ6SgNIVGD27Qov2xcgHyf8-GXo,361
 langgraph_api/js/global.d.ts,sha256=j4GhgtQSZ5_cHzjSPcHgMJ8tfBThxrH-pUOrrJGteOU,196
-langgraph_api/js/package.json,sha256=5gtQpj0mXh069X5fWJFWp-jt7Sb3RQUPVpe4m8Q0KHE,1289
+langgraph_api/js/package.json,sha256=J2sWtYb_gFMCb_g8661iGUR1C23lZQna3wmgl5da0Xc,1289
 langgraph_api/js/remote.py,sha256=ipAITSDyh-kNau37nfRg-PSB-6Lbtax3UJap8-lLZdw,35729
 langgraph_api/js/schema.py,sha256=7idnv7URlYUdSNMBXQcw7E4SxaPxCq_Oxwnlml8q5ik,408
 langgraph_api/js/src/graph.mts,sha256=_xKhdO2WAwsieoezFqTCcx1IDeHeN6GSMv0lyBsakKI,3485
@@ -49,7 +49,7 @@ langgraph_api/js/src/utils/importMap.mts,sha256=pX4TGOyUpuuWF82kXcxcv3-8mgusRezO
 langgraph_api/js/src/utils/pythonSchemas.mts,sha256=98IW7Z_VP7L_CHNRMb3_MsiV3BgLE2JsWQY_PQcRR3o,685
 langgraph_api/js/src/utils/serde.mts,sha256=D9o6MwTgwPezC_DEmsWS5NnLPnjPMVWIb1I1D4QPEPo,743
 langgraph_api/js/sse.py,sha256=lsfp4nyJyA1COmlKG9e2gJnTttf_HGCB5wyH8OZBER8,4105
-langgraph_api/js/tests/api.test.mts,sha256=qjHkSdgTDiBauhxRD4yNMufD5xfZh9BNZ7nD7hTgyzo,68125
+langgraph_api/js/tests/api.test.mts,sha256=LSWDHTar-AtyazqR8EbnS17MrMr3IvKWysGDZ13p-GM,69432
 langgraph_api/js/tests/auth.test.mts,sha256=mMhKe9ggJw4BgUqzSVwqYY3HLMXXEBZ23iiKK8Yq1mM,21678
 langgraph_api/js/tests/compose-postgres.auth.yml,sha256=iPfJbCeYZdV6GiRLiDn_f7qgpG4TyyGaQ4lV-ZXr6Qk,1768
 langgraph_api/js/tests/compose-postgres.yml,sha256=w4B3YRS0QEnTcZH2-MY0DYvR_c5GcER0uDa1Ga_knf8,1960
@@ -73,7 +73,7 @@ langgraph_api/js/tests/parser.test.mts,sha256=BBKUTveZnf-RI6B9XfTBLqy6tp84ddyu1t
 langgraph_api/js/tests/utils.mts,sha256=Jk1ZZmllNgSS6FJlSs9VaQxHqCEUzkqB5rRQwTSAOP4,441
 langgraph_api/js/tsconfig.json,sha256=imCYqVnqFpaBoZPx8k1nO4slHIWBFsSlmCYhO73cpBs,341
 langgraph_api/js/ui.py,sha256=XNT8iBcyT8XmbIqSQUWd-j_00HsaWB2vRTVabwFBkik,2439
-langgraph_api/js/yarn.lock,sha256=OEj5JbffHe8opUAki0eH_0XJbVmgasv9zcHhGeI0g0w,84019
+langgraph_api/js/yarn.lock,sha256=HJuG663b4R2sqQDTQOHtpfMiK615k3CiUsRIZfGDd7k,84019
 langgraph_api/logging.py,sha256=JJIzbNIgLCN6ClQ3tA-Mm5ffuBGvpRDSZsEvnIlsuu4,3693
 langgraph_api/metadata.py,sha256=ptaxwmzdx2bUBSc1KRhqgF-Xnm-Zh2gqwSiHpl8LD9c,4482
 langgraph_api/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -95,14 +95,14 @@ langgraph_api/tunneling/cloudflare.py,sha256=iKb6tj-VWPlDchHFjuQyep2Dpb-w2NGfJKt
 langgraph_api/utils.py,sha256=92mSti9GfGdMRRWyESKQW5yV-75Z9icGHnIrBYvdypU,3619
 langgraph_api/validation.py,sha256=zMuKmwUEBjBgFMwAaeLZmatwGVijKv2sOYtYg7gfRtc,4950
 langgraph_api/webhook.py,sha256=1ncwO0rIZcj-Df9sxSnFEzd1gP1bfS4okeZQS8NSRoE,1382
-langgraph_api/worker.py,sha256=e5RNsEeYDrLJx5Y13LZXgYdSR_7kP77riJqZ6Ppao_E,11281
+langgraph_api/worker.py,sha256=lLrTBfUdUzhPU9W6YQ6sCO3HOaWTfGxfLjlaVPQuOVs,12136
 langgraph_license/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_license/validation.py,sha256=ZKraAVJArAABKqrmHN-EN18ncoNUmRm500Yt1Sc7tUA,537
 langgraph_runtime/__init__.py,sha256=O4GgSmu33c-Pr8Xzxj_brcK5vkm70iNTcyxEjICFZxA,1075
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
-openapi.json,sha256=yYqTGjcV2kMhYfY8S59EWNbzDrZ4kVXchYK5T9OHK88,133203
-langgraph_api-0.2.7.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.2.7.dist-info/METADATA,sha256=c2KdHYF2bdtQI4EHeWZieTb2KzQo_3_RTdq0dZ5ckg0,4235
-langgraph_api-0.2.7.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-langgraph_api-0.2.7.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.2.7.dist-info/RECORD,,
+openapi.json,sha256=GefWJwBrbrN_jNDAEFlwNEXx1ottRtvjOmKBi965KLU,133756
+langgraph_api-0.2.9.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.2.9.dist-info/METADATA,sha256=-USWsNozeKkqmxeoMi3Aj_SkdoTp_s2p3AEwtnHq1F4,4240
+langgraph_api-0.2.9.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+langgraph_api-0.2.9.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.2.9.dist-info/RECORD,,

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",