diff --git a/memory_module/memory_integration.py b/memory_module/memory_integration.py index 6b31986..bc4d41b 100644 --- a/memory_module/memory_integration.py +++ b/memory_module/memory_integration.py @@ -258,7 +258,7 @@ class Mem0Integration: def generate_response_with_memory(self, user_input: str, user_id: str) -> Dict[str, Any]: """Generate a response using memories and store the interaction.""" # Step 1: Search for relevant memories - memories = self.search_memories(user_input, user_id, limit=2) + memories = self.search_memories(user_input, user_id, limit=5) # Step 2: Prepare system prompt with memory injection memory_block = self.format_memories_for_prompt(memories)