From 721d765921afdd151f26cfdb81039e6263d01c92 Mon Sep 17 00:00:00 2001
From: Parshva Daftari <89991302+parshvadaftari@users.noreply.github.com>
Date: Tue, 15 Oct 2024 16:54:07 +0530
Subject: [PATCH] [ Fix ]TypeError when using Chat completion (#1922)

---
 mem0/proxy/main.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/mem0/proxy/main.py b/mem0/proxy/main.py
index b13c681e..d1db29b2 100644
--- a/mem0/proxy/main.py
+++ b/mem0/proxy/main.py
@@ -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']}"