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

Fix proxy for Mem0 (#2616)

Browse Source
This commit is contained in:
Dev Khant
2025-05-03 15:17:09 +05:30
committed by GitHub
parent 7117a94fbf
commit e9f5a882f5
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
mem0/proxy/main.py
View File

@@ -25,7 +25,7 @@ except ImportError:
from mem0 import Memory, MemoryClient from mem0 import Memory, MemoryClient
from mem0.configs.prompts import MEMORY_ANSWER_PROMPT from mem0.configs.prompts import MEMORY_ANSWER_PROMPT
from mem0.memory.telemetry import capture_client_event from mem0.memory.telemetry import capture_client_event, capture_event
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -143,7 +143,10 @@ class Completions:
api_key=api_key, api_key=api_key,
model_list=model_list, model_list=model_list,
) )
capture_client_event("mem0.chat.create", self.mem0_client) if isinstance(self.mem0_client, Memory):
capture_event("mem0.chat.create", self.mem0_client)
else:
capture_client_event("mem0.chat.create", self.mem0_client)
return response return response
def _prepare_messages(self, messages: List[dict]) -> List[dict]: def _prepare_messages(self, messages: List[dict]) -> List[dict]: