Merge pull request #55 from PriorLabs/allow-passing-api-key

Allow passing API key to PostHog

Author: safaricd

src/tabpfn_common_utils/telemetry/core/service.py

@@ -35,7 +35,12 @@ class ProductTelemetry:
     # PostHog client instance
     _posthog_client: Optional[Posthog] = None
 
-    def __init__(self, max_queue_size: int = 10, flush_at: int = 10) -> None:
+    def __init__(
+        self,
+        max_queue_size: int = 10,
+        flush_at: int = 10,
+        api_key: Optional[str] = None,
+    ) -> None:
         """
         Initialize the Telemetry service.
         """
@@ -44,9 +49,12 @@ def __init__(self, max_queue_size: int = 10, flush_at: int = 10) -> None:
             self._posthog_client = None
             return
 
+        if api_key is None:
+            api_key = self.PROJECT_API_KEY
+
         # Initialize the PostHog client
         self._posthog_client = Posthog(
-            project_api_key=self.PROJECT_API_KEY,
+            project_api_key=api_key,
             host=self.HOST,
             disable_geoip=True,
             enable_exception_autocapture=False,