Skip to content

Commit

Permalink
feat: Python SDK for Vertex Model Monitoring V2.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 629879834
  • Loading branch information
vertex-sdk-bot authored and copybara-github committed May 2, 2024
1 parent 26657ff commit 021d59f
Show file tree
Hide file tree
Showing 12 changed files with 4,056 additions and 0 deletions.
4 changes: 4 additions & 0 deletions google/cloud/aiplatform/compat/services/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,9 @@
from google.cloud.aiplatform_v1beta1.services.model_service import (
client as model_service_client_v1beta1,
)
from google.cloud.aiplatform_v1beta1.services.model_monitoring_service import (
client as model_monitoring_service_client_v1beta1,
)
from google.cloud.aiplatform_v1beta1.services.persistent_resource_service import (
client as persistent_resource_service_client_v1beta1,
)
Expand Down Expand Up @@ -200,6 +203,7 @@
job_service_client_v1beta1,
match_service_client_v1beta1,
model_garden_service_client_v1beta1,
model_monitoring_service_client_v1beta1,
model_service_client_v1beta1,
persistent_resource_service_client_v1beta1,
pipeline_service_client_v1beta1,
Expand Down
12 changes: 12 additions & 0 deletions google/cloud/aiplatform/compat/types/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,13 @@
model_deployment_monitoring_job as model_deployment_monitoring_job_v1beta1,
model_garden_service as model_garden_service_v1beta1,
model_service as model_service_v1beta1,
model_monitor as model_monitor_v1beta1,
model_monitoring as model_monitoring_v1beta1,
model_monitoring_alert as model_monitoring_alert_v1beta1,
model_monitoring_job as model_monitoring_job_v1beta1,
model_monitoring_service as model_monitoring_service_v1beta1,
model_monitoring_spec as model_monitoring_spec_v1beta1,
model_monitoring_stats as model_monitoring_stats_v1beta1,
operation as operation_v1beta1,
persistent_resource as persistent_resource_v1beta1,
persistent_resource_service as persistent_resource_service_v1beta1,
Expand Down Expand Up @@ -307,7 +313,13 @@
model_deployment_monitoring_job_v1beta1,
model_garden_service_v1beta1,
model_service_v1beta1,
model_monitor_v1beta1,
model_monitoring_v1beta1,
model_monitoring_alert_v1beta1,
model_monitoring_job_v1beta1,
model_monitoring_service_v1beta1,
model_monitoring_spec_v1beta1,
model_monitoring_stats_v1beta1,
operation_v1beta1,
persistent_resource_v1beta1,
persistent_resource_service_v1beta1,
Expand Down
14 changes: 14 additions & 0 deletions google/cloud/aiplatform/utils/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,7 @@
match_service_client_v1beta1,
metadata_service_client_v1beta1,
model_service_client_v1beta1,
model_monitoring_service_client_v1beta1,
pipeline_service_client_v1beta1,
prediction_service_client_v1beta1,
prediction_service_async_client_v1beta1,
Expand Down Expand Up @@ -104,6 +105,7 @@
index_service_client_v1beta1.IndexServiceClient,
index_endpoint_service_client_v1beta1.IndexEndpointServiceClient,
model_service_client_v1beta1.ModelServiceClient,
model_monitoring_service_client_v1beta1.ModelMonitoringServiceClient,
prediction_service_client_v1beta1.PredictionServiceClient,
prediction_service_async_client_v1beta1.PredictionServiceAsyncClient,
pipeline_service_client_v1beta1.PipelineServiceClient,
Expand Down Expand Up @@ -738,6 +740,17 @@ class MetadataClientWithOverride(ClientWithOverride):
)


class ModelMonitoringClientWithOverride(ClientWithOverride):
_is_temporary = True
_default_version = compat.V1BETA1
_version_map = (
(
compat.V1BETA1,
model_monitoring_service_client_v1beta1.ModelMonitoringServiceClient,
),
)


class TensorboardClientWithOverride(ClientWithOverride):
_is_temporary = False
_default_version = compat.DEFAULT_VERSION
Expand Down Expand Up @@ -854,6 +867,7 @@ class VertexRagClientWithOverride(ClientWithOverride):
PersistentResourceClientWithOverride,
ReasoningEngineClientWithOverride,
ReasoningEngineExecutionClientWithOverride,
ModelMonitoringClientWithOverride,
)


Expand Down
Loading

0 comments on commit 021d59f

Please sign in to comment.