User/dyadav/fix telemetry issue (#2541)

This commit is contained in:
Deshraj Yadav
2025-04-11 13:36:26 -07:00
committed by GitHub
parent d77bed2d5d
commit d33547a77a
7 changed files with 21 additions and 58 deletions

View File

@@ -5,14 +5,17 @@ from functools import wraps
from typing import Any, Dict, List, Optional, Union
import httpx
import hashlib
from mem0.memory.setup import get_user_id, setup_config
from mem0.memory.telemetry import capture_client_event
logger = logging.getLogger(__name__)
warnings.filterwarnings("default", category=DeprecationWarning)
# Setup user config
setup_config()
class APIError(Exception):
"""Exception raised for errors in the API."""
@@ -75,13 +78,11 @@ class MemoryClient:
self.host = host or "https://api.mem0.ai"
self.org_id = org_id
self.project_id = project_id
self.user_id = get_user_id()
if not self.api_key:
raise ValueError("Mem0 API Key not provided. Please provide an API Key.")
# Create MD5 hash of API key for user_id
self.user_id = hashlib.md5(self.api_key.encode()).hexdigest()
self.client = httpx.Client(
base_url=self.host,
headers={"Authorization": f"Token {self.api_key}", "Mem0-User-ID": self.user_id},