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

fix: Add validation for MessageBase type in message input conversion (#4769)

Browse source 
Add validation for MessageBase type in message input conversion
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-11-21 17:21:33 -03:00 • committed by GitHub
commit 6fe7a58c8d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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/inputs/inputs.py
View file

@ -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)