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

This commit is contained in:
Parshva Daftari
2024-10-15 16:54:07 +05:30
committed by GitHub
parent 84eb666618
commit 721d765921

View File

@@ -6,6 +6,8 @@ from typing import List, Optional, Union
import httpx
import mem0
try:
import litellm
except ImportError:
@@ -178,5 +180,10 @@ class Completions:
)
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']}"