andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Serialize BaseModel

Browse source
This commit is contained in:
Andras Schmelczer 2022-07-03 16:17:27 +02:00
parent 554deb9c60
commit 497bc85795
No known key found for this signature in database
GPG key ID: 0EA1BC97D0AB076E
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/great_ai/great_ai/views/trace.py
View file

@ -2,7 +2,7 @@ from pprint import pformat
from typing import Any, Dict, Generic, List, Optional, TypeVar from typing import Any, Dict, Generic, List, Optional, TypeVar
from uuid import uuid4 from uuid import uuid4
from pydantic import Extra, validator from pydantic import BaseModel, Extra, validator
from ..helper import HashableBaseModel from ..helper import HashableBaseModel
from .model import Model from .model import Model
@ -77,7 +77,10 @@ class Trace(Generic[T], HashableBaseModel):
"original_execution_time_ms": self.original_execution_time_ms, "original_execution_time_ms": self.original_execution_time_ms,
} }
), ),
**self.logged_values, **{
k: v.dict() if isinstance(v, BaseModel) else v
for k, v in self.logged_values.items()
},
"models_flat": self.models_flat, "models_flat": self.models_flat,
"exception_flat": self.exception_flat, "exception_flat": self.exception_flat,
"output_flat": self.output_flat, "output_flat": self.output_flat,