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

update serialization method to field only

Browse source
This commit is contained in:
Xi Yan 2024-12-30 13:54:22 -08:00
parent bbbbcf747d
commit 421fa980bc
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -9,7 +9,7 @@ from typing import Annotated, List, Literal, Optional, Union
from llama_models.schema_utils import json_schema_type, register_schema from llama_models.schema_utils import json_schema_type, register_schema
from pydantic import BaseModel, Field, model_validator from pydantic import BaseModel, Field, field_serializer, model_validator
@json_schema_type @json_schema_type
@ -28,8 +28,11 @@ class _URLOrData(BaseModel):
return values return values
return {"url": values} return {"url": values}
class Config: @field_serializer("data")
json_encoders = {bytes: lambda v: base64.b64encode(v).decode("utf-8")} def serialize_data(self, data: Optional[bytes], _info):
if data is None:
return None
return base64.b64encode(data).decode("utf-8")
@json_schema_type @json_schema_type