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

chore: Update ChatInput.py to include Text import and modify return type

Browse source
This commit is contained in:
Rodrigo 2024-06-11 15:27:57 -03:00
commit b7a63154c3
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/components/inputs/ChatInput.py
View file

@ -2,6 +2,8 @@ from typing import Optional
from langflow.base.io.chat import ChatComponent from langflow.base.io.chat import ChatComponent
from langflow.schema.message import Message from langflow.schema.message import Message
from langflow.field_typing import Text
from typing import Union
class ChatInput(ChatComponent): class ChatInput(ChatComponent):
@ -26,7 +28,7 @@ class ChatInput(ChatComponent):
input_value: Optional[str] = None, input_value: Optional[str] = None,
files: Optional[list[str]] = None, files: Optional[list[str]] = None,
session_id: Optional[str] = None, session_id: Optional[str] = None,
) -> Message: ) -> Union[Message, Text]:
return super().build_with_record( return super().build_with_record(
sender=sender, sender=sender,
sender_name=sender_name, sender_name=sender_name,