From d4f50806cd1deb12ab6a70f3abe3c3785c0d6bb2 Mon Sep 17 00:00:00 2001
From: NanoCode012 <nano@axolotl.ai>
Date: Mon, 27 Oct 2025 16:01:27 +0700
Subject: [PATCH] fix: imports

---
 requirements.txt                 |  2 +-
 src/axolotl/telemetry/manager.py | 10 ++++++----
 2 files changed, 7 insertions(+), 5 deletions(-)

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()