andras/great-ai
andras/great-ai
Watch
1
0
Fork
You've already forked great-ai
 0
Code Issues Pull requests 2 Projects Releases 4 Packages 1 Wiki Activity Actions

Serialize supported Pydantic values before writing TinyDB traces

Browse source
This commit is contained in:
Andras Schmelczer 2026-10-27 19:56:13 +00:00
commit 757eb97be0
1 changed files with 39 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
great_ai/persistence/parallel_tinydb_driver.py
View file

@ -28,10 +28,10 @@ class ParallelTinyDbDriver(TracingDatabaseDriver):
path_to_db = Path(DEFAULT_TRACING_DB_FILENAME)
def save(self, trace: Trace) -> str:
return self._safe_execute(lambda db: db.insert(trace.model_dump()))
return self._safe_execute(lambda db: db.insert(trace.model_dump(mode="json")))
def save_batch(self, documents: List[Trace]) -> List[str]:
traces = [d.model_dump() for d in documents]
traces = [d.model_dump(mode="json") for d in documents]
return self._safe_execute(lambda db: db.insert_multiple(traces))
def get(self, id: str) -> Optional[Trace]:
@ -100,7 +100,7 @@ class ParallelTinyDbDriver(TracingDatabaseDriver):
def update(self, id: str, new_version: Trace) -> None:
self._safe_execute(
lambda db: db.update(
new_version.model_dump(), lambda d: d["trace_id"] == id
new_version.model_dump(mode="json"), lambda d: d["trace_id"] == id
)
)