Added anonymized posthog telemetry (#22)

* added anonymized telemetry

* collect data with posthog readme

browser_use/telemetry/service.py

@@ -0,0 +1,101 @@
+import logging
+import os
+import uuid
+from pathlib import Path
+
+from dotenv import load_dotenv
+from posthog import Posthog
+
+from browser_use.telemetry.views import BaseTelemetryEvent
+from browser_use.utils import singleton
+
+load_dotenv()
+
+
+logger = logging.getLogger(__name__)
+
+POSTHOG_EVENT_SETTINGS = {'$process_person_profile': False}
+
+
+@singleton
+class ProductTelemetry:
+	"""
+	Service for capturing anonymized telemetry data.
+
+	If the environment variable `ANONYMIZED_TELEMETRY=False`, anonymized telemetry will be disabled.
+	"""
+
+	USER_ID_PATH = str(Path.home() / '.cache' / 'browser_use' / 'telemetry_user_id')
+	PROJECT_API_KEY = 'phc_F8JMNjW1i2KbGUTaW1unnDdLSPCoyc52SGRU0JecaUh'
+	HOST = 'https://eu.i.posthog.com'
+	UNKNOWN_USER_ID = 'UNKNOWN'
+
+	_curr_user_id = None
+
+	def __init__(self) -> None:
+		telemetry_disabled = os.getenv('ANONYMIZED_TELEMETRY', 'true').lower() == 'false'
+		self.debug_logging = os.getenv('BROWSER_USE_DEBUG_LOGGING', 'false').lower() == 'true'
+
+		if telemetry_disabled:
+			self._posthog_client = None
+		else:
+			logging.info(
+				'Anonymized telemetry enabled. See https://github.com/gregpr07/browser-use for more information.'
+			)
+			self._posthog_client = Posthog(
+				project_api_key=self.PROJECT_API_KEY,
+				host=self.HOST,
+			)
+
+			# Silence posthog's logging
+			if not self.debug_logging:
+				posthog_logger = logging.getLogger('posthog')
+				posthog_logger.disabled = True
+
+		if self._posthog_client is None:
+			logger.debug('Telemetry disabled')
+
+	def capture(self, event: BaseTelemetryEvent) -> None:
+		if self._posthog_client is None:
+			return
+
+		if self.debug_logging:
+			logger.debug(f'Telemetry event: {event.name} {event.properties}')
+		self._direct_capture(event)
+
+	def _direct_capture(self, event: BaseTelemetryEvent) -> None:
+		"""
+		Should not be thread blocking because posthog magically handles it
+		"""
+		if self._posthog_client is None:
+			return
+
+		try:
+			self._posthog_client.capture(
+				self.user_id,
+				event.name,
+				{**event.properties, **POSTHOG_EVENT_SETTINGS},
+			)
+		except Exception as e:
+			logger.error(f'Failed to send telemetry event {event.name}: {e}')
+
+	@property
+	def user_id(self) -> str:
+		if self._curr_user_id:
+			return self._curr_user_id
+
+		# File access may fail due to permissions or other reasons. We don't want to
+		# crash so we catch all exceptions.
+		try:
+			if not os.path.exists(self.USER_ID_PATH):
+				os.makedirs(os.path.dirname(self.USER_ID_PATH), exist_ok=True)
+				with open(self.USER_ID_PATH, 'w') as f:
+					new_user_id = str(uuid.uuid4())
+					f.write(new_user_id)
+				self._curr_user_id = new_user_id
+			else:
+				with open(self.USER_ID_PATH, 'r') as f:
+					self._curr_user_id = f.read()
+		except Exception:
+			self._curr_user_id = 'UNKNOWN_USER_ID'
+		return self._curr_user_id

browser_use/telemetry/views.py

@@ -0,0 +1,51 @@
+from abc import ABC, abstractmethod
+from dataclasses import asdict, dataclass
+from typing import Any, Dict, Optional
+
+
+@dataclass
+class BaseTelemetryEvent(ABC):
+	@property
+	@abstractmethod
+	def name(self) -> str:
+		pass
+
+	@property
+	def properties(self) -> Dict[str, Any]:
+		return {k: v for k, v in asdict(self).items() if k != 'name'}
+
+
+@dataclass
+class RegisteredFunction:
+	name: str
+	params: dict[str, Any]
+
+
+@dataclass
+class ControllerRegisteredFunctionsTelemetryEvent(BaseTelemetryEvent):
+	registered_functions: list[RegisteredFunction]
+	name: str = 'controller_registered_functions'
+
+
+@dataclass
+class AgentRunTelemetryEvent(BaseTelemetryEvent):
+	agent_id: str
+	task: str
+	name: str = 'agent_run'
+
+
+@dataclass
+class AgentStepErrorTelemetryEvent(BaseTelemetryEvent):
+	agent_id: str
+	error: str
+	name: str = 'agent_step_error'
+
+
+@dataclass
+class AgentEndTelemetryEvent(BaseTelemetryEvent):
+	agent_id: str
+	task: str
+	steps: int
+	success: bool
+	error: Optional[str] = None
+	name: str = 'agent_end'