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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langgraph-api
-Version: 0.2.8
+Version: 0.2.9
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos

langgraph_api-0.2.9/langgraph_api/__init__.py

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

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

@@ -303,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" }] };
@@ -377,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();
 
@@ -423,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" }] };
@@ -467,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.8 → 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.8 → langgraph_api-0.2.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.2.8"
+version = "0.2.9"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",

langgraph_api-0.2.8/langgraph_api/__init__.py

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