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"),
]
_message_stored = False
def message_response(self) -> Message:
message = Message(
text=self.input_value,
@ -53,9 +51,9 @@ class ChatInput(ChatComponent):
sender_name=self.sender_name,
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._message_stored = True
self.message.value = message
self.status = 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.field_typing import Text
from langflow.inputs import BoolInput, DropdownInput, StrInput
from langflow.schema.message import Message
from langflow.template import Output
from langflow.field_typing import Text
class ChatOutput(ChatComponent):
@ -39,8 +39,6 @@ class ChatOutput(ChatComponent):
Output(display_name="Text", name="text", method="text_response"),
]
_message_stored = False
def message_response(self) -> Message:
message = Message(
text=self.input_value,
@ -48,9 +46,9 @@ class ChatOutput(ChatComponent):
sender_name=self.sender_name,
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._message_stored = True
self.message.value = message
self.status = message
return message