mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2025-09-02 02:30:07 +00:00
71 lines
1.7 KiB
Python
71 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
import importlib.metadata
|
|
import platform
|
|
from typing import Any, Dict
|
|
|
|
import typer
|
|
from posthog import Posthog
|
|
|
|
from skyvern.forge.sdk.settings_manager import SettingsManager
|
|
|
|
posthog = Posthog(
|
|
"phc_bVT2ugnZhMHRWqMvSRHPdeTjaPxQqT3QSsI3r5FlQR5",
|
|
host="https://app.posthog.com",
|
|
disable_geoip=False,
|
|
)
|
|
|
|
DISTINCT_ID = "oss"
|
|
|
|
|
|
def get_oss_version() -> str:
|
|
try:
|
|
return importlib.metadata.version("skyvern")
|
|
except Exception:
|
|
return "unknown"
|
|
|
|
|
|
def analytics_metadata() -> Dict[str, Any]:
|
|
return {
|
|
"os": platform.system().lower(),
|
|
"oss_version": get_oss_version(),
|
|
"machine": platform.machine(),
|
|
"platform": platform.platform(),
|
|
"python_version": platform.python_version(),
|
|
"environment": SettingsManager.get_settings().ENV,
|
|
}
|
|
|
|
|
|
def capture(
|
|
event: str,
|
|
data: dict[str, Any] | None = None,
|
|
) -> None:
|
|
# If telemetry is disabled, don't send any data
|
|
if not SettingsManager.get_settings().SKYVERN_TELEMETRY:
|
|
return
|
|
|
|
distinct_id = SettingsManager.get_settings().ANALYTICS_ID
|
|
|
|
payload: dict[str, Any] = data or {}
|
|
try:
|
|
posthog.capture(distinct_id=distinct_id, event=event, properties=payload)
|
|
except Exception as e:
|
|
payload.update(
|
|
{
|
|
"capture_error": str(e),
|
|
}
|
|
)
|
|
posthog.capture(
|
|
distinct_id=distinct_id,
|
|
event="failure",
|
|
properties=payload,
|
|
)
|
|
|
|
|
|
# This is the main function that will be called by the typer CLI. This is separate from capture because typer
|
|
# doesn't support dict type input arguments.
|
|
def capture_simple(event: str) -> None:
|
|
capture(event)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
typer.run(capture_simple)
|