diff --git a/requirements.txt b/requirements.txt index acac68ff3..f1ba7eb17 100644 --- a/requirements.txt +++ b/requirements.txt @@ -72,6 +72,6 @@ axolotl-contribs-mit==0.0.3 axolotl-contribs-mit==0.0.5 # telemetry -posthog>=4.2.0 +posthog==6.7.10 mistral-common==1.8.5 diff --git a/src/axolotl/telemetry/manager.py b/src/axolotl/telemetry/manager.py index 1f4088455..82d310cdc 100644 --- a/src/axolotl/telemetry/manager.py +++ b/src/axolotl/telemetry/manager.py @@ -10,10 +10,10 @@ import uuid from pathlib import Path from typing import Any +import posthog import psutil import torch import yaml -from posthog import PostHog LOG = logging.getLogger(__name__) @@ -240,7 +240,9 @@ class TelemetryManager: def _init_posthog(self): """Initialize PostHog client""" - self.posthog = PostHog(POSTHOG_WRITE_KEY, host=POSTHOG_HOST) + posthog.api_key = POSTHOG_WRITE_KEY + posthog.project_api_key = POSTHOG_WRITE_KEY + posthog.host = POSTHOG_HOST def _redact_paths(self, properties: dict[str, Any]) -> dict[str, Any]: """ @@ -389,7 +391,7 @@ class TelemetryManager: # Wrap PostHog errors in try / except to not raise errors during Axolotl usage try: # Send event via PostHog - self.posthog.capture( + posthog.capture( distinct_id=self.run_id, event=event_type, properties=properties, @@ -411,4 +413,4 @@ class TelemetryManager: def shutdown(self): """Ensure all queued events are processed before shutdown""" if self.enabled: - self.posthog.shutdown() + posthog.shutdown()