Skip to content

Commit 6b21e8c

Browse files
authored
Merge pull request #69 from scaleapi/dyuan/agentex/fix-datetime-in-tracing
[Agentex] Support datetime in tracing input/output.
2 parents 9303173 + ddc92bc commit 6b21e8c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/agentex/lib/utils/model_utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from collections.abc import Iterable, Mapping
2+
from datetime import datetime
23
from typing import Any, TypeVar
34

45
from pydantic import BaseModel as PydanticBaseModel
@@ -35,6 +36,9 @@ def recursive_model_dump(obj: Any) -> Any:
3536
if isinstance(obj, PydanticBaseModel):
3637
# Serialize BaseModel to dict
3738
return obj.model_dump(mode="json")
39+
elif isinstance(obj, datetime):
40+
# Serialize datetime to ISO format string
41+
return obj.isoformat()
3842
elif isinstance(obj, Mapping):
3943
# Recursively serialize dictionary values
4044
return {k: recursive_model_dump(v) for k, v in obj.items()}

0 commit comments

Comments
 (0)