From 949b07f25e1ad952077db19643347b5ba04e25e9 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Mon, 30 Dec 2024 11:32:20 -0800 Subject: [PATCH] serialization fix --- llama_stack/providers/utils/telemetry/trace_protocol.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/utils/telemetry/trace_protocol.py b/llama_stack/providers/utils/telemetry/trace_protocol.py index 31897c0ae..2525cd07d 100644 --- a/llama_stack/providers/utils/telemetry/trace_protocol.py +++ b/llama_stack/providers/utils/telemetry/trace_protocol.py @@ -23,7 +23,11 @@ def serialize_value(value: Any) -> Any: elif hasattr(value, "_name_"): return value._name_ elif isinstance(value, BaseModel): - return value.model_dump_json() + try: + return value.model_dump_json() + except Exception as e: + # TODO: fix this + return value.model_dump() elif isinstance(value, (list, tuple, set)): return [serialize_value(item) for item in value] elif isinstance(value, dict):