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

refactor: Improve value validation in TextInput

Browse source
This commit is contained in:
ogabrielluiz 2024-06-14 16:11:23 -03:00
commit 4e13c7352d
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/base/langflow/inputs/inputs.py
View file

@ -47,13 +47,14 @@ class TextInput(StrInput):
@field_validator("value") @field_validator("value")
@classmethod @classmethod
def validate_value(cls, v: Any, _info): def validate_value(cls, v: Any, _info):
value = None
if isinstance(v, str): if isinstance(v, str):
return v value = v
elif isinstance(v, Message): elif isinstance(v, Message):
return v.text value = v.text
elif isinstance(v, Data): elif isinstance(v, Data):
if v.text_key in v.data: if v.text_key in v.data:
return v.data[v.text_key] value = v.data[v.text_key]
else: else:
keys = ", ".join(v.data.keys()) keys = ", ".join(v.data.keys())
input_name = _info.data["name"] input_name = _info.data["name"]
@ -63,6 +64,9 @@ class TextInput(StrInput):
) )
else: else:
raise ValueError(f"Invalid input type {type(v)}") raise ValueError(f"Invalid input type {type(v)}")
if isinstance(value, str):
return value
raise ValueError(f"Invalid value type {type(value)}")
class MultilineInput(BaseInputMixin): class MultilineInput(BaseInputMixin):