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

Fix issue with file validation in Message class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 16:01:56 -03:00
commit aec4292239
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/base/langflow/schema/message.py
View file

@ -37,6 +37,8 @@ class Message(Data):
def validate_files(cls, value): def validate_files(cls, value):
if not value: if not value:
value = [] value = []
elif not isinstance(value, list):
value = [value]
return value return value
def model_post_init(self, __context: Any) -> None: def model_post_init(self, __context: Any) -> None:
@ -174,3 +176,4 @@ class Message(Data):
instance.prompt = jsonable_encoder(prompt_template.to_json()) instance.prompt = jsonable_encoder(prompt_template.to_json())
instance.messages = instance.prompt.get("kwargs", {}).get("messages", []) instance.messages = instance.prompt.get("kwargs", {}).get("messages", [])
return instance return instance
return instance