fix(io): update tuples are ignored by pregel io (#776)

langchain-ai · Jan 15, 2025 · d650ea7 · d650ea7
1 parent 1cba1f2
commit d650ea7
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/libs/langgraph/src/pregel/io.ts b/libs/langgraph/src/pregel/io.ts
@@ -119,8 +119,15 @@ export function* mapCommand(
         "Expected cmd.update to be a dict mapping channel names to update values"
       );
     }
-    for (const [k, v] of Object.entries(cmd.update)) {
-      yield [NULL_TASK_ID, k, v];
+
+    if (Array.isArray(cmd.update)) {
+      for (const [k, v] of cmd.update) {
+        yield [NULL_TASK_ID, k, v];
+      }
+    } else {
+      for (const [k, v] of Object.entries(cmd.update)) {
+        yield [NULL_TASK_ID, k, v];
+      }
     }
   }
 }