From 507fb95d25e6b10c1b6f7e6b367606b479b5a3c3 Mon Sep 17 00:00:00 2001 From: Sara Robinson Date: Fri, 26 Jun 2026 11:59:25 -0700 Subject: [PATCH] fix: Improve formatting of Enums and datetime objects in response types PiperOrigin-RevId: 938704386 --- google/genai/_common.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/google/genai/_common.py b/google/genai/_common.py index f4a6984f7..b0f720488 100644 --- a/google/genai/_common.py +++ b/google/genai/_common.py @@ -417,12 +417,16 @@ def _pretty_repr( if not items: return f'{cls_name}()' return f'{cls_name}(\n' + ',\n'.join(items) + f'\n{indent})' + elif isinstance(obj, enum.Enum): + return repr(obj.value) elif isinstance(obj, str): if '\n' in obj: escaped = obj.replace('"""', '\\"\\"\\"') # Indent the multi-line string block contents return f'"""{escaped}"""' return repr(obj) + elif isinstance(obj, datetime.datetime): + return repr(obj.isoformat()) elif isinstance(obj, bytes): if len(obj) > max_len: return f"{repr(obj[:max_len-3])[:-1]}...'"