diff --git a/mem0/proxy/main.py b/mem0/proxy/main.py index 8689ef3e..821766e1 100644 --- a/mem0/proxy/main.py +++ b/mem0/proxy/main.py @@ -181,8 +181,11 @@ class Completions: def _format_query_with_memories(self, messages, relevant_memories): # Check if self.mem0_client is an instance of Memory or MemoryClient + entities = [] if isinstance(self.mem0_client, mem0.memory.main.Memory): memories_text = "\n".join(memory["memory"] for memory in relevant_memories["results"]) + if relevant_memories.get("relations"): + entities = [entity for entity in relevant_memories["relations"]] elif isinstance(self.mem0_client, mem0.client.main.MemoryClient): memories_text = "\n".join(memory["memory"] for memory in relevant_memories) - return f"- Relevant Memories/Facts: {memories_text}\n\n- User Question: {messages[-1]['content']}" + return f"- Relevant Memories/Facts: {memories_text}\n\n- Entities: {entities}\n\n- User Question: {messages[-1]['content']}"