From 6fe7a58c8d14354547efdb2fa4e69340946b9e34 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 21 Nov 2024 17:21:33 -0300 Subject: [PATCH] fix: Add validation for MessageBase type in message input conversion (#4769) Add validation for MessageBase type in message input conversion --- src/backend/base/langflow/inputs/inputs.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/backend/base/langflow/inputs/inputs.py b/src/backend/base/langflow/inputs/inputs.py index a7346b3f2..0fc9f66f3 100644 --- a/src/backend/base/langflow/inputs/inputs.py +++ b/src/backend/base/langflow/inputs/inputs.py @@ -8,6 +8,7 @@ from pydantic import Field, field_validator from langflow.inputs.validators import CoalesceBool from langflow.schema.data import Data from langflow.schema.message import Message +from langflow.services.database.models.message.model import MessageBase from langflow.template.field.base import Input from .input_mixin import ( @@ -151,6 +152,8 @@ class MessageInput(StrInput, InputTraceMixin): return v if isinstance(v, str | AsyncIterator | Iterator): return Message(text=v) + if isinstance(v, MessageBase): + return Message(**v.model_dump()) msg = f"Invalid value type {type(v)}" raise ValueError(msg)