refactor: Add file input to ChatInput component
This commit is contained in:
parent
418b32a616
commit
7fbd9f0cc6
2 changed files with 11 additions and 8 deletions
|
|
@ -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):
|
||||||
|
|
|
||||||
|
|
@ -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"),
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue