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.inputs.validators import CoalesceBool
from langflow.schema.data import Data from langflow.schema.data import Data
from langflow.schema.message import Message from langflow.schema.message import Message
from langflow.services.database.models.message.model import MessageBase
from langflow.template.field.base import Input from langflow.template.field.base import Input
from .input_mixin import ( from .input_mixin import (
@ -151,6 +152,8 @@ class MessageInput(StrInput, InputTraceMixin):
return v return v
if isinstance(v, str | AsyncIterator | Iterator): if isinstance(v, str | AsyncIterator | Iterator):
return Message(text=v) return Message(text=v)
if isinstance(v, MessageBase):
return Message(**v.model_dump())
msg = f"Invalid value type {type(v)}" msg = f"Invalid value type {type(v)}"
raise ValueError(msg) raise ValueError(msg)