Skip to content

Commit 021d59f

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
feat: Python SDK for Vertex Model Monitoring V2.
PiperOrigin-RevId: 629879834
1 parent 26657ff commit 021d59f

File tree

12 files changed

+4056
-0
lines changed

12 files changed

+4056
-0
lines changed

google/cloud/aiplatform/compat/services/__init__.py

+4
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,9 @@
6363
from google.cloud.aiplatform_v1beta1.services.model_service import (
6464
client as model_service_client_v1beta1,
6565
)
66+
from google.cloud.aiplatform_v1beta1.services.model_monitoring_service import (
67+
client as model_monitoring_service_client_v1beta1,
68+
)
6669
from google.cloud.aiplatform_v1beta1.services.persistent_resource_service import (
6770
client as persistent_resource_service_client_v1beta1,
6871
)
@@ -200,6 +203,7 @@
200203
job_service_client_v1beta1,
201204
match_service_client_v1beta1,
202205
model_garden_service_client_v1beta1,
206+
model_monitoring_service_client_v1beta1,
203207
model_service_client_v1beta1,
204208
persistent_resource_service_client_v1beta1,
205209
pipeline_service_client_v1beta1,

google/cloud/aiplatform/compat/types/__init__.py

+12
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,13 @@
7474
model_deployment_monitoring_job as model_deployment_monitoring_job_v1beta1,
7575
model_garden_service as model_garden_service_v1beta1,
7676
model_service as model_service_v1beta1,
77+
model_monitor as model_monitor_v1beta1,
7778
model_monitoring as model_monitoring_v1beta1,
79+
model_monitoring_alert as model_monitoring_alert_v1beta1,
80+
model_monitoring_job as model_monitoring_job_v1beta1,
81+
model_monitoring_service as model_monitoring_service_v1beta1,
82+
model_monitoring_spec as model_monitoring_spec_v1beta1,
83+
model_monitoring_stats as model_monitoring_stats_v1beta1,
7884
operation as operation_v1beta1,
7985
persistent_resource as persistent_resource_v1beta1,
8086
persistent_resource_service as persistent_resource_service_v1beta1,
@@ -307,7 +313,13 @@
307313
model_deployment_monitoring_job_v1beta1,
308314
model_garden_service_v1beta1,
309315
model_service_v1beta1,
316+
model_monitor_v1beta1,
310317
model_monitoring_v1beta1,
318+
model_monitoring_alert_v1beta1,
319+
model_monitoring_job_v1beta1,
320+
model_monitoring_service_v1beta1,
321+
model_monitoring_spec_v1beta1,
322+
model_monitoring_stats_v1beta1,
311323
operation_v1beta1,
312324
persistent_resource_v1beta1,
313325
persistent_resource_service_v1beta1,

google/cloud/aiplatform/utils/__init__.py

+14
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@
5151
match_service_client_v1beta1,
5252
metadata_service_client_v1beta1,
5353
model_service_client_v1beta1,
54+
model_monitoring_service_client_v1beta1,
5455
pipeline_service_client_v1beta1,
5556
prediction_service_client_v1beta1,
5657
prediction_service_async_client_v1beta1,
@@ -104,6 +105,7 @@
104105
index_service_client_v1beta1.IndexServiceClient,
105106
index_endpoint_service_client_v1beta1.IndexEndpointServiceClient,
106107
model_service_client_v1beta1.ModelServiceClient,
108+
model_monitoring_service_client_v1beta1.ModelMonitoringServiceClient,
107109
prediction_service_client_v1beta1.PredictionServiceClient,
108110
prediction_service_async_client_v1beta1.PredictionServiceAsyncClient,
109111
pipeline_service_client_v1beta1.PipelineServiceClient,
@@ -738,6 +740,17 @@ class MetadataClientWithOverride(ClientWithOverride):
738740
)
739741

740742

743+
class ModelMonitoringClientWithOverride(ClientWithOverride):
744+
_is_temporary = True
745+
_default_version = compat.V1BETA1
746+
_version_map = (
747+
(
748+
compat.V1BETA1,
749+
model_monitoring_service_client_v1beta1.ModelMonitoringServiceClient,
750+
),
751+
)
752+
753+
741754
class TensorboardClientWithOverride(ClientWithOverride):
742755
_is_temporary = False
743756
_default_version = compat.DEFAULT_VERSION
@@ -854,6 +867,7 @@ class VertexRagClientWithOverride(ClientWithOverride):
854867
PersistentResourceClientWithOverride,
855868
ReasoningEngineClientWithOverride,
856869
ReasoningEngineExecutionClientWithOverride,
870+
ModelMonitoringClientWithOverride,
857871
)
858872

859873

0 commit comments

Comments
 (0)