Remove stale code and events improvements (#1883)
This commit is contained in:
@@ -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()
|
||||||
|
|||||||
@@ -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()
|
||||||
|
|
||||||
|
|||||||
@@ -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 = [
|
||||||
|
|||||||
Reference in New Issue
Block a user