From ac8cf594737d6351ba6700d163fa258e75e928a6 Mon Sep 17 00:00:00 2001 From: spike-spiegel-21 <83648453+spike-spiegel-21@users.noreply.github.com> Date: Sat, 11 Jan 2025 01:47:42 +0530 Subject: [PATCH] entities added in proxy (#2135) --- mem0/proxy/main.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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']}"