WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-12 12:47:53 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix duplicate history entries

Browse Source
This commit is contained in:
Nuno Campos
2024-03-15 08:33:35 -07:00
parent da7797431a
commit ba5dbad75f
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
langgraph/pregel/__init__.py
+7 -7
View File
@@ -581,7 +581,7 @@ class Pregel(
_apply_writes_from_view(checkpoint, channels, step_output)
# with previous step's checkpoint
if do_interrupt_before := _should_interrupt(
if _should_interrupt(
checkpoint,
interrupt_before_nodes,
self.snapshot_channels_list,
@@ -610,7 +610,7 @@ class Pregel(
if (
self.checkpointer is not None
and self.checkpointer.at == CheckpointAt.END_OF_RUN
and not do_interrupt_before
and not interrupt_before_nodes
):
checkpoint = create_checkpoint(checkpoint, channels)
self.checkpointer.put(config, checkpoint)
@@ -767,7 +767,7 @@ class Pregel(
_apply_writes_from_view(checkpoint, channels, step_output)
# with previous step's checkpoint
if do_interrupt_before := _should_interrupt(
if _should_interrupt(
checkpoint,
interrupt_before_nodes,
self.snapshot_channels_list,
@@ -776,9 +776,9 @@ class Pregel(
break
# save end of step checkpoint
if (
self.checkpointer is not None
and self.checkpointer.at == CheckpointAt.END_OF_STEP
if self.checkpointer is not None and (
self.checkpointer.at == CheckpointAt.END_OF_STEP
or interrupt_before_nodes
):
checkpoint = create_checkpoint(checkpoint, channels)
await self.checkpointer.aput(config, checkpoint)
@@ -796,7 +796,7 @@ class Pregel(
if (
self.checkpointer is not None
and self.checkpointer.at == CheckpointAt.END_OF_RUN
and not do_interrupt_before
and not interrupt_before_nodes
):
checkpoint = create_checkpoint(checkpoint, channels)
await self.checkpointer.aput(config, checkpoint)