Fix chitchat lost context (#1828)

Co-authored-by: luowei <glpat-EjySCyNjWiLqAED-YmwM> Co-authored-by: crazywoola <[email protected]> Co-authored-by: crazywoola <[email protected]>
langgenius · Dec 23, 2023 · 6b57e4e · 6b57e4e
1 parent 6da3a33
commit 6b57e4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/core/memory/read_only_conversation_token_db_buffer_shared_memory.py b/api/core/memory/read_only_conversation_token_db_buffer_shared_memory.py
@@ -27,7 +27,7 @@ def buffer(self) -> List[BaseMessage]:
         # fetch limited messages desc, and return reversed
         messages = db.session.query(Message).filter(
             Message.conversation_id == self.conversation.id,
-            Message.answer_tokens > 0
+            Message.answer != ''
         ).order_by(Message.created_at.desc()).limit(self.message_limit).all()
 
         messages = list(reversed(messages))