conditionally define v2 serializer

Co-authored-by: Khaleel Al-Adhami <khaleel.aladhami@gmail.com>
This commit is contained in:
Masen Furer 2024-11-08 10:17:00 -08:00 committed by GitHub
parent 91437317d3
commit 6595fcf2ca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -282,17 +282,18 @@ def serialize_base_model_v1(model: BaseModelV1) -> dict:
return model.dict()
@serializer(to=dict)
def serialize_base_model_v2(model: BaseModelV2) -> dict:
"""Serialize a pydantic v2 BaseModel instance.
if BaseModelV2 is not BaseModelV1:
@serializer(to=dict)
def serialize_base_model_v2(model: BaseModelV2) -> dict:
"""Serialize a pydantic v2 BaseModel instance.
Args:
model: The BaseModel to serialize.
Args:
model: The BaseModel to serialize.
Returns:
The serialized BaseModel.
"""
return model.model_dump()
Returns:
The serialized BaseModel.
"""
return model.model_dump()
@serializer