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

Add checkpoint config to checkpoint metadata?

Browse Source
This commit is contained in:
Andrew Nguonly
2024-05-17 19:25:02 -07:00
parent 18addb52aa
commit 09bae3022d
3 changed files with 56 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
langgraph/pregel/__init__.py
+8 -4
View File
@@ -760,10 +760,11 @@ class Pregel(
)
)
checkpoint_config = {
**checkpoint_config,
"configurable": {
**checkpoint_config["configurable"],
"thread_ts": checkpoint["id"],
}
},
}
# increment start to 0
start += 1
@@ -885,10 +886,11 @@ class Pregel(
)
)
checkpoint_config = {
**checkpoint_config,
"configurable": {
**checkpoint_config["configurable"],
"thread_ts": checkpoint["id"],
}
},
}
# yield debug checkpoint
if stream_mode == "debug":
@@ -1031,10 +1033,11 @@ class Pregel(
)
)
checkpoint_config = {
**checkpoint_config,
"configurable": {
**checkpoint_config["configurable"],
"thread_ts": checkpoint["id"],
}
},
}
# increment start to 0
start += 1
@@ -1162,10 +1165,11 @@ class Pregel(
)
)
checkpoint_config = {
**checkpoint_config,
"configurable": {
**checkpoint_config["configurable"],
"thread_ts": checkpoint["id"],
}
},
}
# yield debug checkpoint
if stream_mode == "debug":