fix: Fix typo in import statement for DocumentsToDataComponent

This commit is contained in:
Rodrigo 2024-06-15 20:15:50 -03:00
commit 2996cb726d
6 changed files with 30 additions and 38 deletions

View file

@ -1,7 +1,7 @@
from .CreateData import CreateDataComponent from .CreateData import CreateDataComponent
from .CustomComponent import Component from .CustomComponent import Component
from .ParseData import ParseDataComponent from .ParseData import ParseDataComponent
from .DocumentToData import DocumentsToDataComponent from .DocumentsToData import DocumentsToDataComponent
from .IDGenerator import UUIDGeneratorComponent from .IDGenerator import UUIDGeneratorComponent
from .UpdateData import UpdateDataComponent from .UpdateData import UpdateDataComponent

View file

@ -2,6 +2,7 @@ from langflow.base.io.chat import ChatComponent
from langflow.inputs import DropdownInput, StrInput from langflow.inputs import DropdownInput, StrInput
from langflow.schema.message import Message from langflow.schema.message import Message
from langflow.template import Output from langflow.template import Output
from langflow.field_typing import Text
class ChatInput(ChatComponent): class ChatInput(ChatComponent):
@ -40,6 +41,7 @@ class ChatInput(ChatComponent):
] ]
outputs = [ outputs = [
Output(display_name="Message", name="message", method="message_response"), Output(display_name="Message", name="message", method="message_response"),
Output(display_name="Text", name="text", method="text_response"),
] ]
def message_response(self) -> Message: def message_response(self) -> Message:
@ -53,3 +55,7 @@ class ChatInput(ChatComponent):
self.store_message(message) self.store_message(message)
self.status = message self.status = message
return message return message
def text_response(self) -> Text:
text = self.message_response().text
return text

View file

@ -1,6 +1,6 @@
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 MultilineInput, StrInput from langflow.inputs import StrInput
from langflow.template import Output from langflow.template import Output
@ -15,17 +15,10 @@ class TextInputComponent(TextComponent):
display_name="Text", display_name="Text",
info="Text to be passed as input.", info="Text to be passed as input.",
), ),
MultilineInput(
name="data_template",
display_name="Data Template",
info="Template to convert Data to Text. If left empty, it will be dynamically set to the Data's text key.",
advanced=True,
value="{text}",
),
] ]
outputs = [ outputs = [
Output(display_name="Text", name="text", method="text_response"), Output(display_name="Text", name="text", method="text_response"),
] ]
def text_response(self) -> Text: def text_response(self) -> Text:
return self.build(input_value=self.input_value, data_template=self.data_template) return self.build(input_value=self.input_value)

View file

@ -1,5 +1,5 @@
from .AmazonBedrockModel import AmazonBedrockComponent from .AmazonBedrockModel import AmazonBedrockComponent
from .AnthropicModel import AnthropicLLM from .AnthropicModel import AnthropicModelComponent
from .AzureOpenAIModel import AzureChatOpenAIComponent from .AzureOpenAIModel import AzureChatOpenAIComponent
from .BaiduQianfanChatModel import QianfanChatEndpointComponent from .BaiduQianfanChatModel import QianfanChatEndpointComponent
from .ChatLiteLLMModel import ChatLiteLLMModelComponent from .ChatLiteLLMModel import ChatLiteLLMModelComponent
@ -13,7 +13,7 @@ from .VertexAiModel import ChatVertexAIComponent
__all__ = [ __all__ = [
"ChatLiteLLMModelComponent", "ChatLiteLLMModelComponent",
"AmazonBedrockComponent", "AmazonBedrockComponent",
"AnthropicLLM", "AnthropicModelComponent",
"AzureChatOpenAIComponent", "AzureChatOpenAIComponent",
"QianfanChatEndpointComponent", "QianfanChatEndpointComponent",
"CohereComponent", "CohereComponent",

View file

@ -1,7 +1,8 @@
from langflow.base.io.chat import ChatComponent from langflow.base.io.chat import ChatComponent
from langflow.inputs import BoolInput, DropdownInput, MultilineInput, StrInput from langflow.inputs import BoolInput, DropdownInput, StrInput
from langflow.schema.message import Message from langflow.schema.message import Message
from langflow.template import Output from langflow.template import Output
from langflow.field_typing import Text
class ChatOutput(ChatComponent): class ChatOutput(ChatComponent):
@ -10,11 +11,10 @@ class ChatOutput(ChatComponent):
icon = "ChatOutput" icon = "ChatOutput"
inputs = [ inputs = [
MultilineInput( StrInput(
name="input_value", name="input_value",
display_name="Text", display_name="Text",
info="Message to be passed as output.", info="Message to be passed as output.",
input_types=["Text", "Message"],
), ),
DropdownInput( DropdownInput(
name="sender", name="sender",
@ -36,19 +36,21 @@ class ChatOutput(ChatComponent):
] ]
outputs = [ outputs = [
Output(display_name="Message", name="message", method="message_response"), Output(display_name="Message", name="message", method="message_response"),
Output(display_name="Text", name="text", method="text_response"),
] ]
def message_response(self) -> Message: def message_response(self) -> Message:
if isinstance(self.input_value, Message): message = Message(
message = self.input_value text=self.input_value,
else: sender=self.sender,
message = Message( sender_name=self.sender_name,
text=self.input_value, session_id=self.session_id,
sender=self.sender, )
sender_name=self.sender_name,
session_id=self.session_id,
)
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.status = message self.status = message
return message return message
def text_response(self) -> Text:
text = self.message_response().text
return text

View file

@ -1,6 +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.template import Input, Output from langflow.template import Output
from langflow.inputs import StrInput
class TextOutputComponent(TextComponent): class TextOutputComponent(TextComponent):
@ -9,20 +10,10 @@ class TextOutputComponent(TextComponent):
icon = "type" icon = "type"
inputs = [ inputs = [
Input( StrInput(
name="input_value", name="input_value",
type=str, display_name="Text",
display_name="Value",
info="Text or Data to be passed as output.", info="Text or Data to be passed as output.",
input_types=["Data", "Text"],
),
Input(
name="data_template",
type=str,
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.",
advanced=True,
), ),
] ]
outputs = [ outputs = [
@ -30,4 +21,4 @@ class TextOutputComponent(TextComponent):
] ]
def text_response(self) -> Text: def text_response(self) -> Text:
return self.build(input_value=self.input_value, data_template=self.data_template) return self.build(input_value=self.input_value)