From 0c2d7388dbc74e1d534791df4f69f7e6a9ec1f79 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 10 Dec 2023 16:48:24 -0300 Subject: [PATCH] Fix serialization issue with range_spec in TemplateField --- src/backend/langflow/template/field/base.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/src/backend/langflow/template/field/base.py b/src/backend/langflow/template/field/base.py index 754632e98..c8e1235cc 100644 --- a/src/backend/langflow/template/field/base.py +++ b/src/backend/langflow/template/field/base.py @@ -61,7 +61,7 @@ class TemplateField(BaseModel): refresh: Optional[bool] = None """Specifies if the field should be refreshed. Defaults to False.""" - range_spec: Optional[RangeSpec] = None + range_spec: Optional[RangeSpec] = Field(None, serialization_alias="rangeSpec") """Range specification for the field. Defaults to None.""" def to_dict(self): @@ -73,8 +73,10 @@ class TemplateField(BaseModel): return value return "" - @field_serializer("range_spec") - def serialize_range_spec(self, value): - if self.field_type == "float" and value is None: - return RangeSpec().model_dump() + @field_serializer("field_type") + def serialize_field_type(self, value, _info): + if value == "float": + # check if range_spec is set + if self.range_spec is None: + self.range_spec = RangeSpec() return value