feat: implement an OTel instrumentor

PiperOrigin-RevId: 703574132
googleapis · Dec 6, 2024 · f2b4719 · f2b4719
1 parent b9fecea
commit f2b4719
Show file tree

Hide file tree

Showing 3 changed files with 399 additions and 1 deletion.
diff --git a/google/cloud/aiplatform/__init__.py b/google/cloud/aiplatform/__init__.py
@@ -66,6 +66,7 @@
     TensorboardRun,
     TensorboardTimeSeries,
 )
+from google.cloud.aiplatform.aiplatform.telemetry import VertexAIInstrumentor
 from google.cloud.aiplatform.training_jobs import (
     CustomTrainingJob,
     CustomContainerTrainingJob,
@@ -183,5 +184,6 @@
     "TemporalFusionTransformerForecastingTrainingJob",
     "TimeSeriesDataset",
     "TimeSeriesDenseEncoderForecastingTrainingJob",
+    "VertexAIInstrumentor",
     "VideoDataset",
 )