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

refactor: Update ChatOutput inputs to use specific input classes

Browse source 
Update the inputs of the ChatOutput class in ChatOutput.py to use specific input classes such as StrInput, DropdownInput, and BoolInput. This change improves the organization and separation of concerns in the codebase, making it easier to understand and maintain.
This commit is contained in:
ogabrielluiz 2024-06-12 16:14:48 -03:00
commit 8511c2ebad
1 changed files with 14 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

25
src/backend/base/langflow/components/outputs/ChatOutput.py
View file

@ -1,7 +1,8 @@
from langflow.base.io.chat import ChatComponent from langflow.base.io.chat import ChatComponent
from langflow.field_typing import Text from langflow.field_typing import Text
from langflow.inputs import BoolInput, DropdownInput, StrInput
from langflow.schema.message import Message from langflow.schema.message import Message
from langflow.template import Input, Output from langflow.template import Output
class ChatOutput(ChatComponent): class ChatOutput(ChatComponent):
@ -10,25 +11,25 @@ class ChatOutput(ChatComponent):
icon = "ChatOutput" icon = "ChatOutput"
inputs = [ inputs = [
Input( StrInput(
name="input_value", type=str, display_name="Text", multiline=True, info="Message to be passed as output." name="input_value",
display_name="Text",
multiline=True,
info="Message to be passed as output.",
input_types=["Text", "Message"],
), ),
Input( DropdownInput(
name="sender", name="sender",
type=str,
display_name="Sender Type", display_name="Sender Type",
options=["Machine", "User"], options=["Machine", "User"],
value="Machine", value="Machine",
advanced=True, advanced=True,
info="Type of sender.", info="Type of sender.",
), ),
Input(name="sender_name", type=str, display_name="Sender Name", info="Name of the sender.", value="AI"), StrInput(name="sender_name", display_name="Sender Name", info="Name of the sender.", value="AI"),
Input( StrInput(name="session_id", 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 BoolInput(
),
Input(
name="record_template", name="record_template",
type=str,
display_name="Record Template", display_name="Record Template",
value="{text}", value="{text}",
advanced=True, advanced=True,
@ -56,5 +57,7 @@ class ChatOutput(ChatComponent):
) )
if self.session_id and isinstance(message, Message) and isinstance(message.text, str): if self.session_id and isinstance(message, Message) and isinstance(message.text, str):
self.store_message(message) self.store_message(message)
self.message.value = message
self.status = message self.status = message
return message return message