refactor: Add file input to ChatInput component

This commit is contained in:
ogabrielluiz 2024-06-18 18:37:33 -03:00
commit 7fbd9f0cc6
2 changed files with 11 additions and 8 deletions

View file

@ -33,12 +33,7 @@ TEXT_FILE_TYPES = [
"tsx", "tsx",
] ]
IMG_FILE_TYPES = [ IMG_FILE_TYPES = ["jpg", "jpeg", "png", "bmp", "image"]
"jpg",
"jpeg",
"png",
"bmp",
]
def normalize_text(text): def normalize_text(text):

View file

@ -1,6 +1,6 @@
from langflow.base.data.utils import IMG_FILE_TYPES, TEXT_FILE_TYPES
from langflow.base.io.chat import ChatComponent from langflow.base.io.chat import ChatComponent
from langflow.field_typing import Text from langflow.inputs import DropdownInput, FileInput, TextInput
from langflow.inputs import DropdownInput, TextInput
from langflow.schema.message import Message from langflow.schema.message import Message
from langflow.template import Output from langflow.template import Output
@ -37,6 +37,14 @@ class ChatInput(ChatComponent):
TextInput( TextInput(
name="session_id", type=str, display_name="Session ID", info="Session ID for the message.", advanced=True name="session_id", type=str, display_name="Session ID", info="Session ID for the message.", advanced=True
), ),
FileInput(
name="files",
display_name="Files",
file_types=TEXT_FILE_TYPES + IMG_FILE_TYPES,
info="Files to be sent with the message.",
advanced=True,
is_list=True,
),
] ]
outputs = [ outputs = [
Output(display_name="Message", name="message", method="message_response"), Output(display_name="Message", name="message", method="message_response"),