From 7fbd9f0cc656e5982b0de603e1321844a103defa Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Tue, 18 Jun 2024 18:37:33 -0300 Subject: [PATCH] refactor: Add file input to ChatInput component --- src/backend/base/langflow/base/data/utils.py | 7 +------ .../base/langflow/components/inputs/ChatInput.py | 12 ++++++++++-- 2 files changed, 11 insertions(+), 8 deletions(-) diff --git a/src/backend/base/langflow/base/data/utils.py b/src/backend/base/langflow/base/data/utils.py index b1fc24cec..35ab8f98c 100644 --- a/src/backend/base/langflow/base/data/utils.py +++ b/src/backend/base/langflow/base/data/utils.py @@ -33,12 +33,7 @@ TEXT_FILE_TYPES = [ "tsx", ] -IMG_FILE_TYPES = [ - "jpg", - "jpeg", - "png", - "bmp", -] +IMG_FILE_TYPES = ["jpg", "jpeg", "png", "bmp", "image"] def normalize_text(text): diff --git a/src/backend/base/langflow/components/inputs/ChatInput.py b/src/backend/base/langflow/components/inputs/ChatInput.py index 92ee5ab45..3be3ab2ed 100644 --- a/src/backend/base/langflow/components/inputs/ChatInput.py +++ b/src/backend/base/langflow/components/inputs/ChatInput.py @@ -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.field_typing import Text -from langflow.inputs import DropdownInput, TextInput +from langflow.inputs import DropdownInput, FileInput, TextInput from langflow.schema.message import Message from langflow.template import Output @@ -37,6 +37,14 @@ class ChatInput(ChatComponent): TextInput( 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 = [ Output(display_name="Message", name="message", method="message_response"),