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

Refactor ChatInput and ChatOutput components to prevent duplicate message storage

Browse source
This commit is contained in:
ogabrielluiz 2024-06-17 09:17:58 -03:00
commit abf1582fe6
2 changed files with 5 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/components/inputs/ChatInput.py
View file

@ -44,8 +44,6 @@ class ChatInput(ChatComponent):
Output(display_name="Text", name="text", method="text_response"), Output(display_name="Text", name="text", method="text_response"),
] ]
_message_stored = False
def message_response(self) -> Message: def message_response(self) -> Message:
message = Message( message = Message(
text=self.input_value, text=self.input_value,
@ -53,9 +51,9 @@ class ChatInput(ChatComponent):
sender_name=self.sender_name, sender_name=self.sender_name,
session_id=self.session_id, session_id=self.session_id,
) )
if self.session_id and isinstance(message, Message) and isinstance(message.text, str) and not self._message_stored: if self.session_id and isinstance(message, Message) and isinstance(message.text, str):
self.store_message(message) self.store_message(message)
self._message_stored = True self.message.value = message
self.status = message self.status = message
return message return message

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

@ -1,8 +1,8 @@
from langflow.base.io.chat import ChatComponent from langflow.base.io.chat import ChatComponent
from langflow.field_typing import Text
from langflow.inputs import BoolInput, DropdownInput, 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):
@ -39,8 +39,6 @@ class ChatOutput(ChatComponent):
Output(display_name="Text", name="text", method="text_response"), Output(display_name="Text", name="text", method="text_response"),
] ]
_message_stored = False
def message_response(self) -> Message: def message_response(self) -> Message:
message = Message( message = Message(
text=self.input_value, text=self.input_value,
@ -48,9 +46,9 @@ class ChatOutput(ChatComponent):
sender_name=self.sender_name, sender_name=self.sender_name,
session_id=self.session_id, session_id=self.session_id,
) )
if self.session_id and isinstance(message, Message) and isinstance(message.text, str) and not self._message_stored: if self.session_id and isinstance(message, Message) and isinstance(message.text, str):
self.store_message(message) self.store_message(message)
self._message_stored = True self.message.value = message
self.status = message self.status = message
return message return message