diff --git a/mem0/client/main.py b/mem0/client/main.py index e94c6f83..60424f52 100644 --- a/mem0/client/main.py +++ b/mem0/client/main.py @@ -6,7 +6,7 @@ from typing import Any, Dict, List, Optional, Union import httpx -from mem0.memory.setup import setup_config +from mem0.memory.setup import get_user_id, setup_config from mem0.memory.telemetry import capture_client_event logger = logging.getLogger(__name__) @@ -68,13 +68,14 @@ class MemoryClient: """ self.api_key = api_key or os.getenv("MEM0_API_KEY") self.host = host or "https://api.mem0.ai" + self.user_id = get_user_id() if not self.api_key: raise ValueError("API Key not provided. Please provide an API Key.") self.client = httpx.Client( base_url=self.host, - headers={"Authorization": f"Token {self.api_key}"}, + headers={"Authorization": f"Token {self.api_key}", "Mem0-User-ID": self.user_id}, timeout=60, ) self._validate_api_key() diff --git a/mem0/memory/telemetry.py b/mem0/memory/telemetry.py index 6865b2fe..e4e6b863 100644 --- a/mem0/memory/telemetry.py +++ b/mem0/memory/telemetry.py @@ -5,6 +5,7 @@ import sys from posthog import Posthog +import mem0 from mem0.memory.setup import get_user_id, setup_config MEM0_TELEMETRY = os.environ.get("MEM0_TELEMETRY", "True") @@ -32,6 +33,8 @@ class AnonymousTelemetry: if properties is None: properties = {} properties = { + "client_source": "python", + "client_version": mem0.__version__, "python_version": sys.version, "os": sys.platform, "os_version": platform.version(), @@ -42,11 +45,6 @@ class AnonymousTelemetry: } self.posthog.capture(distinct_id=self.user_id, event=event_name, properties=properties) - def identify_user(self, user_id, properties=None): - if properties is None: - properties = {} - self.posthog.identify(distinct_id=user_id, properties=properties) - def close(self): self.posthog.shutdown() diff --git a/pyproject.toml b/pyproject.toml index 3f909228..ccb4fcc2 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "mem0ai" -version = "0.1.14" +version = "0.1.15" description = "Long-term memory for AI Agents" authors = ["Mem0 "] exclude = [