phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 01:03:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

serialize binary data fix

Browse source
This commit is contained in:
Xi Yan 2024-12-30 13:10:37 -08:00
parent 0bb28cf20a
commit 48df492a44
2 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/apis/common/content_types.py
View file

@ -27,6 +27,9 @@ class _URLOrData(BaseModel):
return values
return {"url": values}
class Config:
json_encoders = {bytes: lambda v: str("binary data")}
@json_schema_type
class ImageContentItem(_URLOrData):

7
llama_stack/providers/utils/telemetry/trace_protocol.py
View file

@ -23,12 +23,7 @@ def serialize_value(value: Any) -> Any:
elif hasattr(value, "_name_"):
return value._name_
elif isinstance(value, BaseModel):
try:
return value.model_dump_json()
except Exception as e:
# We cannot serialize the binary bytes data,
# So we return the model dump as a string for logging
return str(value.model_dump())
return value.model_dump_json()
elif isinstance(value, (list, tuple, set)):
return [serialize_value(item) for item in value]
elif isinstance(value, dict):