fix: Fix checkpointer race condition (#559)

langchain-ai · Oct 8, 2024 · b9f6fd0 · b9f6fd0
1 parent cc0eab2
commit b9f6fd0
Show file tree

Hide file tree

Showing 4 changed files with 6,756 additions and 6,636 deletions.
diff --git a/libs/langgraph/src/pregel/index.ts b/libs/langgraph/src/pregel/index.ts
@@ -1156,7 +1156,7 @@ export class Pregel<
         await loop.store?.stop();
       }
       await Promise.all([
-        loop?.checkpointerPromises ?? [],
+        ...(loop?.checkpointerPromises ?? []),
         ...Array.from(managed.values()).map((mv) => mv.promises()),
       ]);
     }

diff --git a/libs/langgraph/src/tests/pregel.slow_checkpointer.int.test.ts b/libs/langgraph/src/tests/pregel.slow_checkpointer.int.test.ts
@@ -0,0 +1,4 @@
+import { runPregelTests } from "./pregel.test.js";
+import { MemorySaverAssertImmutableSlow } from "./utils.js";
+
+runPregelTests(() => new MemorySaverAssertImmutableSlow());