From e9f5a882f5c82e00b4ffb832e8f02ee93042867c Mon Sep 17 00:00:00 2001 From: Dev Khant Date: Sat, 3 May 2025 15:17:09 +0530 Subject: [PATCH] Fix proxy for Mem0 (#2616) --- mem0/proxy/main.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/mem0/proxy/main.py b/mem0/proxy/main.py index 52776971..7668bcea 100644 --- a/mem0/proxy/main.py +++ b/mem0/proxy/main.py @@ -25,7 +25,7 @@ except ImportError: from mem0 import Memory, MemoryClient 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__) @@ -143,7 +143,10 @@ class Completions: api_key=api_key, 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 def _prepare_messages(self, messages: List[dict]) -> List[dict]: