1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
import logging
import os
import posthog
from r2r.telemetry.events import BaseTelemetryEvent
logger = logging.getLogger(__name__)
class PosthogClient:
"""
This is a write-only project API key, so it can only create new events. It can't
read events or any of your other data stored with PostHog, so it's safe to use in public apps.
"""
def __init__(
self, api_key: str, enabled: bool = True, debug: bool = False
):
self.enabled = enabled
self.debug = debug
if self.enabled:
logger.info(
"Initializing anonymized telemetry. To disable, set TELEMETRY_ENABLED=false in your environment."
)
posthog.project_api_key = api_key
posthog.disable_geoip = False
else:
posthog.disabled = True
if self.debug:
posthog.debug = True
logger.info(
f"Posthog telemetry {'enabled' if self.enabled else 'disabled'}, debug mode {'on' if self.debug else 'off'}"
)
def capture(self, event: BaseTelemetryEvent):
if self.enabled:
posthog.capture(event.event_id, event.event_type, event.properties)
# Initialize the telemetry client with a flag to enable or disable telemetry
telemetry_enabled = os.getenv("TELEMETRY_ENABLED", "true").lower() in (
"true",
"1",
"t",
)
debug_mode = os.getenv("DEBUG_MODE", "false").lower() in (
"true",
"1",
"t",
)
telemetry_client = PosthogClient(
api_key="phc_OPBbibOIErCGc4NDLQsOrMuYFTKDmRwXX6qxnTr6zpU",
enabled=telemetry_enabled,
)
|
