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

[ Fix ]TypeError when using Chat completion (#1922)

Browse Source
This commit is contained in:
Parshva Daftari
2024-10-15 16:54:07 +05:30
committed by GitHub
parent 84eb666618
commit 721d765921
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
mem0/proxy/main.py
View File

@@ -6,6 +6,8 @@ from typing import List, Optional, Union
import httpx import httpx
import mem0
try: try:
import litellm import litellm
except ImportError: except ImportError:
@@ -178,5 +180,10 @@ class Completions:
) )
def _format_query_with_memories(self, messages, relevant_memories): def _format_query_with_memories(self, messages, relevant_memories):
memories_text = "\n".join(memory["memory"] for memory in relevant_memories) # Check if self.mem0_client is an instance of Memory or MemoryClient
if isinstance(self.mem0_client, mem0.memory.main.Memory):
memories_text = "\n".join(memory["memory"] for memory in relevant_memories['results'])
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- User Question: {messages[-1]['content']}"