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]: