Skip to content

Commit

Permalink
Remove stale code and events improvements (#1883)
Browse files Browse the repository at this point in the history
  • Loading branch information
deshraj committed Sep 18, 2024
1 parent fc88cae commit 6102aa7
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 8 deletions.
5 changes: 3 additions & 2 deletions mem0/client/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@

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__)
Expand Down Expand Up @@ -68,13 +68,14 @@ def __init__(self, api_key: Optional[str] = None, host: Optional[str] = None):
"""
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()
Expand Down
8 changes: 3 additions & 5 deletions mem0/memory/telemetry.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@

from posthog import Posthog

import mem0
from mem0.memory.setup import get_user_id, setup_config

MEM0_TELEMETRY = os.environ.get("MEM0_TELEMETRY", "True")
Expand Down Expand Up @@ -32,6 +33,8 @@ def capture_event(self, event_name, properties=None):
if properties is None:
properties = {}
properties = {
"client_source": "python",
"client_version": mem0.__version__,
"python_version": sys.version,
"os": sys.platform,
"os_version": platform.version(),
Expand All @@ -42,11 +45,6 @@ def capture_event(self, event_name, properties=None):
}
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()

Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -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 <[email protected]>"]
exclude = [
Expand Down

0 comments on commit 6102aa7

Please sign in to comment.