From aec42922398964790de51c80d338242cf5472c4d Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 19 Jun 2024 16:01:56 -0300 Subject: [PATCH] Fix issue with file validation in Message class --- src/backend/base/langflow/schema/message.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/backend/base/langflow/schema/message.py b/src/backend/base/langflow/schema/message.py index 9f96e10b6..4a436a3a3 100644 --- a/src/backend/base/langflow/schema/message.py +++ b/src/backend/base/langflow/schema/message.py @@ -37,6 +37,8 @@ class Message(Data): def validate_files(cls, value): if not value: value = [] + elif not isinstance(value, list): + value = [value] return value def model_post_init(self, __context: Any) -> None: @@ -174,3 +176,4 @@ class Message(Data): instance.prompt = jsonable_encoder(prompt_template.to_json()) instance.messages = instance.prompt.get("kwargs", {}).get("messages", []) return instance + return instance