Fix backwards compat old threads

langchain-ai · Jan 29, 2024 · 0f047a7 · 0f047a7
1 parent 7b76dbc
commit 0f047a7
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/backend/app/checkpoint.py b/backend/app/checkpoint.py
@@ -71,8 +71,11 @@ def get(self, config: RunnableConfig) -> Checkpoint | None:
             value.pop("__pregel_version")
             value.pop("__pregel_ts")
             checkpoint = empty_checkpoint()
-            checkpoint["channel_values"] = value
-            for key in value:
+            if value.get("messages"):
+                checkpoint["channel_values"] = {"__root__": value["messages"][1]}
+            else:
+                checkpoint["channel_values"] = {}
+            for key in checkpoint["channel_values"]:
                 checkpoint["channel_versions"][key] = 1
             return checkpoint
         else: