checkpoint-postgres: use correct metadata serialization (#1255)

This commit is contained in:
Vadym Barda
2024-08-07 12:13:44 -04:00
committed by GitHub
parent fb8390e138
commit ca4b65e1a6
3 changed files with 21 additions and 6 deletions
@@ -111,7 +111,7 @@ class PostgresSaver(BasePostgresSaver):
**self._load_checkpoint(value["checkpoint"]),
"channel_values": self._load_blobs(value["channel_values"]),
},
value["metadata"],
self._load_metadata(value["metadata"]),
{
"configurable": {
"thread_id": value["thread_id"],
@@ -155,7 +155,7 @@ class PostgresSaver(BasePostgresSaver):
**self._load_checkpoint(value["checkpoint"]),
"channel_values": self._load_blobs(value["channel_values"]),
},
value["metadata"],
self._load_metadata(value["metadata"]),
{
"configurable": {
"thread_id": thread_id,
@@ -210,7 +210,7 @@ class PostgresSaver(BasePostgresSaver):
checkpoint["id"],
checkpoint_id,
Jsonb(self._dump_checkpoint(copy)),
Jsonb(metadata),
self._dump_metadata(metadata),
),
)
return next_config