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

fix: update store_message to avoid crash (#6117)

Browse source
This commit is contained in:
Eric 2025-02-18 00:14:51 +00:00 • committed by GitHub
commit 94952c0524
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/components/helpers/store_message.py
View file

@ -61,10 +61,10 @@ class MessageStoreComponent(Component):
self.memory.session_id = message.session_id
lc_message = message.to_lc_message()
await self.memory.aadd_messages([lc_message])
stored_message = await self.memory.aget_messages()
stored_message = [Message.from_lc_message(m) for m in stored_message]
stored_messages = await self.memory.aget_messages()
stored_messages = [Message.from_lc_message(m) for m in stored_messages]
if message.sender:
stored_message = [m for m in stored_message if m.sender == message.sender]
stored_messages = [m for m in stored_messages if m.sender == message.sender]
else:
await astore_message(message, flow_id=self.graph.flow_id)
stored_messages = await aget_messages(