From 0cf2229ff1f72b4dbe55b977f342c4ad83f1003a Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 29 Mar 2024 16:19:41 -0300 Subject: [PATCH] Update TextInput and ChatOutput build methods --- .../base/langflow/components/inputs/TextInput.py | 10 +++++++--- .../base/langflow/components/outputs/ChatOutput.py | 2 ++ .../base/langflow/components/outputs/TextOutput.py | 8 ++------ 3 files changed, 11 insertions(+), 9 deletions(-) diff --git a/src/backend/base/langflow/components/inputs/TextInput.py b/src/backend/base/langflow/components/inputs/TextInput.py index d804d1cf2..05e923a54 100644 --- a/src/backend/base/langflow/components/inputs/TextInput.py +++ b/src/backend/base/langflow/components/inputs/TextInput.py @@ -6,7 +6,11 @@ from langflow.field_typing import Text class TextInput(TextComponent): display_name = "Text Input" - description = "Capture and send text inputs." + description = "Capture Text or Record and send text inputs." - def build(self, input_value: Optional[str] = "") -> Text: - return super().build(input_value=input_value) + def build( + self, + input_value: Optional[str] = "", + record_template: Optional[str] = "{text}", + ) -> Text: + return super().build(input_value=input_value, record_template=record_template) diff --git a/src/backend/base/langflow/components/outputs/ChatOutput.py b/src/backend/base/langflow/components/outputs/ChatOutput.py index 24c850f80..8c1f3de77 100644 --- a/src/backend/base/langflow/components/outputs/ChatOutput.py +++ b/src/backend/base/langflow/components/outputs/ChatOutput.py @@ -17,6 +17,7 @@ class ChatOutput(ChatComponent): input_value: Optional[str] = None, session_id: Optional[str] = None, return_record: Optional[bool] = False, + record_template: Optional[str] = "{text}", ) -> Union[Text, Record]: return super().build( sender=sender, @@ -24,4 +25,5 @@ class ChatOutput(ChatComponent): input_value=input_value, session_id=session_id, return_record=return_record, + record_template=record_template, ) diff --git a/src/backend/base/langflow/components/outputs/TextOutput.py b/src/backend/base/langflow/components/outputs/TextOutput.py index 099c89b64..214d2f9ea 100644 --- a/src/backend/base/langflow/components/outputs/TextOutput.py +++ b/src/backend/base/langflow/components/outputs/TextOutput.py @@ -8,9 +8,5 @@ class TextOutput(TextComponent): display_name = "Text Output" description = "Used to send a text output." - field_config = { - "input_value": {"display_name": "Value"}, - } - - def build(self, input_value: Optional[Text] = "") -> Text: - return super().build(input_value=input_value) + def build(self, input_value: Optional[Text] = "", record_template: str = "{text}") -> Text: + return super().build(input_value=input_value, record_template=record_template)