Serialise BaseModels

2022-07-03 16:04:27 +02:00 · 2022-07-03 16:04:27 +02:00 · 623e81cdcb
commit 623e81cdcb
parent 2f4d203446
1 changed files with 18 additions and 8 deletions
--- a/src/great_ai/great_ai/persistence/mongodb_driver.py
+++ b/src/great_ai/great_ai/persistence/mongodb_driver.py
@ -121,7 +121,7 @@ class MongodbDriver(TracingDatabaseDriver):
        return documents, len(documents)

    def update(self, id: str, new_version: Trace) -> None:
-        serialized = new_version.dict()
+        serialized = new_version.to_flat_dict()
        serialized["_id"] = new_version.trace_id
        with MongoClient(self.mongo_connection_string) as client:
            client[self.mongo_database].traces.update_one(id, new_version)