fix(langgraph): restore old logic

This commit is contained in:
Caspar Broekhuizen
2025-10-27 10:45:09 -07:00
parent 6669634a98
commit f02f7a5a27
+1 -8
View File
@@ -2308,15 +2308,8 @@ class Pregel(
# channel writes are saved to current checkpoint
channel_writes = [w for w in task.writes if w[0] != PUSH]
if saved and channel_writes:
# sanitize TASKS writes for storage
sanitized = [
(c, sanitize_untracked_values_in_send(v, channels))
if c == TASKS and isinstance(v, Send)
else (c, v)
for c, v in channel_writes
]
await checkpointer.aput_writes(
checkpoint_config, sanitized, task_id
checkpoint_config, channel_writes, task_id
)
# apply to checkpoint and save
apply_writes(