klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

entities added in proxy (#2135)

Browse Source
This commit is contained in:
spike-spiegel-21
2025-01-11 01:47:42 +05:30
committed by GitHub
parent 9c4acdcba7
commit ac8cf59473
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mem0/proxy/main.py
View File

@@ -181,8 +181,11 @@ class Completions:
def _format_query_with_memories(self, messages, relevant_memories): def _format_query_with_memories(self, messages, relevant_memories):
# Check if self.mem0_client is an instance of Memory or MemoryClient # Check if self.mem0_client is an instance of Memory or MemoryClient
entities = []
if isinstance(self.mem0_client, mem0.memory.main.Memory): if isinstance(self.mem0_client, mem0.memory.main.Memory):
memories_text = "\n".join(memory["memory"] for memory in relevant_memories["results"]) 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): elif isinstance(self.mem0_client, mem0.client.main.MemoryClient):
memories_text = "\n".join(memory["memory"] for memory in relevant_memories) 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']}"