From 6595fcf2cabe45179acd8bfee1726b49ae9ae370 Mon Sep 17 00:00:00 2001 From: Masen Furer Date: Fri, 8 Nov 2024 10:17:00 -0800 Subject: [PATCH] conditionally define v2 serializer Co-authored-by: Khaleel Al-Adhami --- reflex/utils/serializers.py | 23 ++++++++++++----------- 1 file changed, 12 insertions(+), 11 deletions(-) diff --git a/reflex/utils/serializers.py b/reflex/utils/serializers.py index 480eadf35..9a0613f5a 100644 --- a/reflex/utils/serializers.py +++ b/reflex/utils/serializers.py @@ -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. - - Args: - model: The BaseModel to serialize. - - Returns: - The serialized BaseModel. - """ - return model.model_dump() +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. + + Returns: + The serialized BaseModel. + """ + return model.model_dump() @serializer