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

✨ (TextInput.py): Add MultilineInput class to support multiline input for data_template field

Browse source 
♻️ (TextInput.py): Remove unnecessary code for handling Message type input_value in text_response method
This commit is contained in:
ogabrielluiz 2024-06-13 00:06:51 -03:00
commit 36da9172e7
1 changed files with 3 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/backend/base/langflow/components/inputs/TextInput.py
View file

@ -1,8 +1,7 @@
from langflow.base.io.text import TextComponent from langflow.base.io.text import TextComponent
from langflow.field_typing import Text from langflow.field_typing import Text
from langflow.inputs import StrInput from langflow.inputs import MultilineInput, StrInput
from langflow.template import Output from langflow.template import Output
from langflow.schema.message import Message
class TextInput(TextComponent): class TextInput(TextComponent):
@ -13,15 +12,12 @@ class TextInput(TextComponent):
inputs = [ inputs = [
StrInput( StrInput(
name="input_value", name="input_value",
type=str,
display_name="Text", display_name="Text",
info="Text to be passed as input.", info="Text to be passed as input.",
input_types=["Text", "Message"],
), ),
StrInput( MultilineInput(
name="data_template", name="data_template",
display_name="Data Template", display_name="Data Template",
multiline=True,
info="Template to convert Data to Text. If left empty, it will be dynamically set to the Data's text key.", info="Template to convert Data to Text. If left empty, it will be dynamically set to the Data's text key.",
advanced=True, advanced=True,
value="{text}", value="{text}",
@ -32,9 +28,4 @@ class TextInput(TextComponent):
] ]
def text_response(self) -> Text: def text_response(self) -> Text:
if isinstance(self.input_value, Message): return self.build(input_value=self.input_value, data_template=self.data_template)
text = self.input_value.text
else:
text = self.input_value
return self.build(input_value=text, data_template=self.data_template)