Merge pull request #62 from PriorLabs/ENG-444

Detect additional runtimes + track install_id across model calls

Author: safaricd

src/tabpfn_common_utils/telemetry/core/events.py

@@ -439,6 +439,9 @@ class ModelCallEvent(BaseTelemetryEvent):
     # Task associated with the model call
     task: Literal["classification", "regression"]
 
+    # Install ID of the user
+    install_id: str = field(default_factory=_get_install_id, init=False)
+
     # Version of the PyTorch
     torch_version: str = field(default_factory=_get_torch_version, init=False)
 

src/tabpfn_common_utils/telemetry/core/runtime.py

@@ -19,6 +19,9 @@
     "aws",
     "azure",
     "databricks",
+    "modal",
+    "mlflow",
+    "kubernetes",
 ]
 
 # Static list of environment hints, purely heuristic based on env variables.
@@ -50,6 +53,15 @@
         "DATABRICKS_WORKSPACE_URL",
         "DB_IS_DRIVER",
     ],
+    "modal": ["MODAL_CLOUD_PROVIDER", "MODAL_IMAGE_ID", "MODAL_REGION"],
+    "mlflow": [
+        "MLFLOW_RUN_ID",
+        "MLFLOW_RUN_NAME",
+    ],
+    "kubernetes": [
+        "KUBERNETES_SERVICE_HOST",
+        "KUBERNETES_POD_NAME",
+    ],
     # Cloud providers
     "aws": [
         # Generic AWS environment hints