jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

set type to str when field_type is Text

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-28 21:42:25 -03:00
commit fb2823bfd9
1 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/backend/langflow/template/field/base.py
View file

@ -68,7 +68,9 @@ class TemplateField(BaseModel):
refresh: Optional[bool] = None refresh: Optional[bool] = None
"""Specifies if the field should be refreshed. Defaults to False.""" """Specifies if the field should be refreshed. Defaults to False."""
range_spec: Optional[RangeSpec] = Field(default=None, serialization_alias="rangeSpec") range_spec: Optional[RangeSpec] = Field(
default=None, serialization_alias="rangeSpec"
)
"""Range specification for the field. Defaults to None.""" """Range specification for the field. Defaults to None."""
title_case: bool = False title_case: bool = False
@ -81,11 +83,13 @@ class TemplateField(BaseModel):
def serialize_model(self, handler): def serialize_model(self, handler):
result = handler(self) result = handler(self)
# If the field is str, we add the Text input type # If the field is str, we add the Text input type
if self.field_type == "str": if self.field_type in ["str", "Text"]:
if "input_types" not in result: if "input_types" not in result:
result["input_types"] = ["Text"] result["input_types"] = ["Text"]
else: else:
result["input_types"].append("Text") result["input_types"].append("Text")
if self.field_type == "Text":
result["type"] = "str"
return result return result
@field_serializer("file_path") @field_serializer("file_path")
@ -115,6 +119,10 @@ class TemplateField(BaseModel):
if not isinstance(value, list): if not isinstance(value, list):
raise ValueError("file_types must be a list") raise ValueError("file_types must be a list")
return [ return [
(f".{file_type}" if isinstance(file_type, str) and not file_type.startswith(".") else file_type) (
f".{file_type}"
if isinstance(file_type, str) and not file_type.startswith(".")
else file_type
)
for file_type in value for file_type in value
] ]