Remove stale code and events improvements (#1883)

This commit is contained in:
Deshraj Yadav
2024-09-18 14:14:21 -07:00
committed by GitHub
parent fc88cae628
commit 6102aa76bb
3 changed files with 7 additions and 8 deletions

View File

@@ -6,7 +6,7 @@ from typing import Any, Dict, List, Optional, Union
import httpx 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 from mem0.memory.telemetry import capture_client_event
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -68,13 +68,14 @@ class MemoryClient:
""" """
self.api_key = api_key or os.getenv("MEM0_API_KEY") self.api_key = api_key or os.getenv("MEM0_API_KEY")
self.host = host or "https://api.mem0.ai" self.host = host or "https://api.mem0.ai"
self.user_id = get_user_id()
if not self.api_key: if not self.api_key:
raise ValueError("API Key not provided. Please provide an API Key.") raise ValueError("API Key not provided. Please provide an API Key.")
self.client = httpx.Client( self.client = httpx.Client(
base_url=self.host, 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, timeout=60,
) )
self._validate_api_key() self._validate_api_key()

View File

@@ -5,6 +5,7 @@ import sys
from posthog import Posthog from posthog import Posthog
import mem0
from mem0.memory.setup import get_user_id, setup_config from mem0.memory.setup import get_user_id, setup_config
MEM0_TELEMETRY = os.environ.get("MEM0_TELEMETRY", "True") MEM0_TELEMETRY = os.environ.get("MEM0_TELEMETRY", "True")
@@ -32,6 +33,8 @@ class AnonymousTelemetry:
if properties is None: if properties is None:
properties = {} properties = {}
properties = { properties = {
"client_source": "python",
"client_version": mem0.__version__,
"python_version": sys.version, "python_version": sys.version,
"os": sys.platform, "os": sys.platform,
"os_version": platform.version(), "os_version": platform.version(),
@@ -42,11 +45,6 @@ class AnonymousTelemetry:
} }
self.posthog.capture(distinct_id=self.user_id, event=event_name, properties=properties) 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): def close(self):
self.posthog.shutdown() self.posthog.shutdown()

View File

@@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "mem0ai" name = "mem0ai"
version = "0.1.14" version = "0.1.15"
description = "Long-term memory for AI Agents" description = "Long-term memory for AI Agents"
authors = ["Mem0 <founders@mem0.ai>"] authors = ["Mem0 <founders@mem0.ai>"]
exclude = [ exclude = [