Revert metadata serde

langchain-ai · Sep 12, 2024 · fa9009a · fa9009a
1 parent 7be933f
commit fa9009a
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/libs/checkpoint-postgres/langgraph/checkpoint/postgres/base.py b/libs/checkpoint-postgres/langgraph/checkpoint/postgres/base.py
@@ -11,7 +11,6 @@
     EmptyChannelError,
     get_checkpoint_id,
 )
-from langgraph.checkpoint.serde.jsonplus import JsonPlusSerializer
 from langgraph.checkpoint.serde.types import TASKS, ChannelProtocol
 
 MetadataInput = Optional[dict[str, Any]]
@@ -131,8 +130,6 @@ class BasePostgresSaver(BaseCheckpointSaver):
     UPSERT_CHECKPOINT_WRITES_SQL = UPSERT_CHECKPOINT_WRITES_SQL
     INSERT_CHECKPOINT_WRITES_SQL = INSERT_CHECKPOINT_WRITES_SQL
 
-    jsonplus_serde = JsonPlusSerializer()
-
     def _load_checkpoint(
         self,
         checkpoint: dict[str, Any],
@@ -224,7 +221,7 @@ def _dump_writes(
         ]
 
     def _load_metadata(self, metadata: dict[str, Any]) -> dict[str, Any]:
-        return self.serde.loads(self.serde.dumps(metadata))
+        return self.jsonplus_serde.loads(self.jsonplus_serde.dumps(metadata))
 
     def _dump_metadata(self, metadata) -> str:
         serialized_metadata_type, serialized_metadata = self.jsonplus_serde.dumps_typed(