Skip to content

Commit e31effe

Browse files
jsondaicopybara-github
authored andcommitted
fix: GenAI Client(evals) - Add datetime and bytes serialization to _json_serializer
PiperOrigin-RevId: 879315519
1 parent 3b795f6 commit e31effe

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

vertexai/_genai/_evals_visualization.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
"""Visualization utilities for GenAI Evaluation SDK."""
1616

1717
import base64
18+
import datetime
1819
import html
1920
import json
2021
import logging
@@ -44,6 +45,10 @@ def _pydantic_serializer(obj: Any) -> Any:
4445
"""Custom serializer for Pydantic models."""
4546
if hasattr(obj, "model_dump"):
4647
return obj.model_dump(mode="json")
48+
if isinstance(obj, datetime.datetime):
49+
return obj.isoformat()
50+
if isinstance(obj, bytes):
51+
return base64.b64encode(obj).decode("utf-8")
4752
raise TypeError(f"Object of type {obj.__class__.__name__} is not JSON serializable")
4853

4954

0 commit comments

Comments
 (0)