entities added in proxy (#2135)

mem0ai · Jan 10, 2025 · ac8cf59 · ac8cf59
1 parent 9c4acdc
commit ac8cf59
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mem0/proxy/main.py b/mem0/proxy/main.py
@@ -181,8 +181,11 @@ def _fetch_relevant_memories(self, messages, user_id, agent_id, run_id, filters,
     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']}"