feat: add XAI, model monitoring, and index services to aiplatform v1 (#…

…668) * feat: add XAI, model monitoring, and index services to aiplatform v1 Committer: @dizcology PiperOrigin-RevId: 393860361 Source-Link: googleapis/googleapis@8031cc2 Source-Link: googleapis/googleapis-gen@6dc8099 * 🦉 Updates from OwlBot See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
googleapis · Aug 30, 2021 · 1fbce55 · 1fbce55
1 parent 916c1d4
commit 1fbce55
Show file tree

Hide file tree

Showing 126 changed files with 21,572 additions and 477 deletions.
diff --git a/docs/aiplatform_v1/index_endpoint_service.rst b/docs/aiplatform_v1/index_endpoint_service.rst
@@ -0,0 +1,10 @@
+IndexEndpointService
+--------------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1.services.index_endpoint_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.aiplatform_v1.services.index_endpoint_service.pagers
+    :members:
+    :inherited-members:
diff --git a/docs/aiplatform_v1/index_service.rst b/docs/aiplatform_v1/index_service.rst
@@ -0,0 +1,10 @@
+IndexService
+------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1.services.index_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.aiplatform_v1.services.index_service.pagers
+    :members:
+    :inherited-members:
diff --git a/docs/aiplatform_v1/services.rst b/docs/aiplatform_v1/services.rst
@@ -5,6 +5,8 @@ Services for Google Cloud Aiplatform v1 API
 
     dataset_service
     endpoint_service
+    index_endpoint_service
+    index_service
     job_service
     migration_service
     model_service

diff --git a/google/cloud/aiplatform_v1/__init__.py b/google/cloud/aiplatform_v1/__init__.py
@@ -18,6 +18,10 @@
 from .services.dataset_service import DatasetServiceAsyncClient
 from .services.endpoint_service import EndpointServiceClient
 from .services.endpoint_service import EndpointServiceAsyncClient
+from .services.index_endpoint_service import IndexEndpointServiceClient
+from .services.index_endpoint_service import IndexEndpointServiceAsyncClient
+from .services.index_service import IndexServiceClient
+from .services.index_service import IndexServiceAsyncClient
 from .services.job_service import JobServiceClient
 from .services.job_service import JobServiceAsyncClient
 from .services.migration_service import MigrationServiceClient
@@ -70,6 +74,7 @@
 from .types.dataset_service import ListDatasetsRequest
 from .types.dataset_service import ListDatasetsResponse
 from .types.dataset_service import UpdateDatasetRequest
+from .types.deployed_index_ref import DeployedIndexRef
 from .types.deployed_model_ref import DeployedModelRef
 from .types.encryption_spec import EncryptionSpec
 from .types.endpoint import DeployedModel
@@ -89,7 +94,48 @@
 from .types.endpoint_service import UpdateEndpointRequest
 from .types.env_var import EnvVar
 from .types.execution import Execution
+from .types.explanation import Attribution
+from .types.explanation import Explanation
+from .types.explanation import ExplanationMetadataOverride
+from .types.explanation import ExplanationParameters
+from .types.explanation import ExplanationSpec
+from .types.explanation import ExplanationSpecOverride
+from .types.explanation import FeatureNoiseSigma
+from .types.explanation import IntegratedGradientsAttribution
+from .types.explanation import ModelExplanation
+from .types.explanation import SampledShapleyAttribution
+from .types.explanation import SmoothGradConfig
+from .types.explanation import XraiAttribution
+from .types.explanation_metadata import ExplanationMetadata
+from .types.feature_monitoring_stats import FeatureStatsAnomaly
 from .types.hyperparameter_tuning_job import HyperparameterTuningJob
+from .types.index import Index
+from .types.index_endpoint import DeployedIndex
+from .types.index_endpoint import DeployedIndexAuthConfig
+from .types.index_endpoint import IndexEndpoint
+from .types.index_endpoint import IndexPrivateEndpoints
+from .types.index_endpoint_service import CreateIndexEndpointOperationMetadata
+from .types.index_endpoint_service import CreateIndexEndpointRequest
+from .types.index_endpoint_service import DeleteIndexEndpointRequest
+from .types.index_endpoint_service import DeployIndexOperationMetadata
+from .types.index_endpoint_service import DeployIndexRequest
+from .types.index_endpoint_service import DeployIndexResponse
+from .types.index_endpoint_service import GetIndexEndpointRequest
+from .types.index_endpoint_service import ListIndexEndpointsRequest
+from .types.index_endpoint_service import ListIndexEndpointsResponse
+from .types.index_endpoint_service import UndeployIndexOperationMetadata
+from .types.index_endpoint_service import UndeployIndexRequest
+from .types.index_endpoint_service import UndeployIndexResponse
+from .types.index_endpoint_service import UpdateIndexEndpointRequest
+from .types.index_service import CreateIndexOperationMetadata
+from .types.index_service import CreateIndexRequest
+from .types.index_service import DeleteIndexRequest
+from .types.index_service import GetIndexRequest
+from .types.index_service import ListIndexesRequest
+from .types.index_service import ListIndexesResponse
+from .types.index_service import NearestNeighborSearchOperationMetadata
+from .types.index_service import UpdateIndexOperationMetadata
+from .types.index_service import UpdateIndexRequest
 from .types.io import BigQueryDestination
 from .types.io import BigQuerySource
 from .types.io import ContainerRegistryDestination
@@ -103,14 +149,17 @@
 from .types.job_service import CreateCustomJobRequest
 from .types.job_service import CreateDataLabelingJobRequest
 from .types.job_service import CreateHyperparameterTuningJobRequest
+from .types.job_service import CreateModelDeploymentMonitoringJobRequest
 from .types.job_service import DeleteBatchPredictionJobRequest
 from .types.job_service import DeleteCustomJobRequest
 from .types.job_service import DeleteDataLabelingJobRequest
 from .types.job_service import DeleteHyperparameterTuningJobRequest
+from .types.job_service import DeleteModelDeploymentMonitoringJobRequest
 from .types.job_service import GetBatchPredictionJobRequest
 from .types.job_service import GetCustomJobRequest
 from .types.job_service import GetDataLabelingJobRequest
 from .types.job_service import GetHyperparameterTuningJobRequest
+from .types.job_service import GetModelDeploymentMonitoringJobRequest
 from .types.job_service import ListBatchPredictionJobsRequest
 from .types.job_service import ListBatchPredictionJobsResponse
 from .types.job_service import ListCustomJobsRequest
@@ -119,6 +168,14 @@
 from .types.job_service import ListDataLabelingJobsResponse
 from .types.job_service import ListHyperparameterTuningJobsRequest
 from .types.job_service import ListHyperparameterTuningJobsResponse
+from .types.job_service import ListModelDeploymentMonitoringJobsRequest
+from .types.job_service import ListModelDeploymentMonitoringJobsResponse
+from .types.job_service import PauseModelDeploymentMonitoringJobRequest
+from .types.job_service import ResumeModelDeploymentMonitoringJobRequest
+from .types.job_service import SearchModelDeploymentMonitoringStatsAnomaliesRequest
+from .types.job_service import SearchModelDeploymentMonitoringStatsAnomaliesResponse
+from .types.job_service import UpdateModelDeploymentMonitoringJobOperationMetadata
+from .types.job_service import UpdateModelDeploymentMonitoringJobRequest
 from .types.job_state import JobState
 from .types.machine_resources import AutomaticResources
 from .types.machine_resources import AutoscalingMetricSpec
@@ -140,8 +197,26 @@
 from .types.model import ModelContainerSpec
 from .types.model import Port
 from .types.model import PredictSchemata
+from .types.model_deployment_monitoring_job import (
+    ModelDeploymentMonitoringBigQueryTable,
+)
+from .types.model_deployment_monitoring_job import ModelDeploymentMonitoringJob
+from .types.model_deployment_monitoring_job import (
+    ModelDeploymentMonitoringObjectiveConfig,
+)
+from .types.model_deployment_monitoring_job import (
+    ModelDeploymentMonitoringScheduleConfig,
+)
+from .types.model_deployment_monitoring_job import ModelMonitoringStatsAnomalies
+from .types.model_deployment_monitoring_job import (
+    ModelDeploymentMonitoringObjectiveType,
+)
 from .types.model_evaluation import ModelEvaluation
 from .types.model_evaluation_slice import ModelEvaluationSlice
+from .types.model_monitoring import ModelMonitoringAlertConfig
+from .types.model_monitoring import ModelMonitoringObjectiveConfig
+from .types.model_monitoring import SamplingStrategy
+from .types.model_monitoring import ThresholdConfig
 from .types.model_service import DeleteModelRequest
 from .types.model_service import ExportModelOperationMetadata
 from .types.model_service import ExportModelRequest
@@ -178,8 +253,11 @@
 from .types.pipeline_service import ListTrainingPipelinesRequest
 from .types.pipeline_service import ListTrainingPipelinesResponse
 from .types.pipeline_state import PipelineState
+from .types.prediction_service import ExplainRequest
+from .types.prediction_service import ExplainResponse
 from .types.prediction_service import PredictRequest
 from .types.prediction_service import PredictResponse
+from .types.prediction_service import RawPredictRequest
 from .types.specialist_pool import SpecialistPool
 from .types.specialist_pool_service import CreateSpecialistPoolOperationMetadata
 from .types.specialist_pool_service import CreateSpecialistPoolRequest
@@ -190,6 +268,7 @@
 from .types.specialist_pool_service import UpdateSpecialistPoolOperationMetadata
 from .types.specialist_pool_service import UpdateSpecialistPoolRequest
 from .types.study import Measurement
+from .types.study import Study
 from .types.study import StudySpec
 from .types.study import Trial
 from .types.training_pipeline import FilterSplit
@@ -204,6 +283,8 @@
 __all__ = (
     "DatasetServiceAsyncClient",
     "EndpointServiceAsyncClient",
+    "IndexEndpointServiceAsyncClient",
+    "IndexServiceAsyncClient",
     "JobServiceAsyncClient",
     "MigrationServiceAsyncClient",
     "ModelServiceAsyncClient",
@@ -215,6 +296,7 @@
     "Annotation",
     "AnnotationSpec",
     "Artifact",
+    "Attribution",
     "AutomaticResources",
     "AutoscalingMetricSpec",
     "BatchDedicatedResources",
@@ -242,6 +324,11 @@
     "CreateEndpointOperationMetadata",
     "CreateEndpointRequest",
     "CreateHyperparameterTuningJobRequest",
+    "CreateIndexEndpointOperationMetadata",
+    "CreateIndexEndpointRequest",
+    "CreateIndexOperationMetadata",
+    "CreateIndexRequest",
+    "CreateModelDeploymentMonitoringJobRequest",
     "CreatePipelineJobRequest",
     "CreateSpecialistPoolOperationMetadata",
     "CreateSpecialistPoolRequest",
@@ -259,14 +346,23 @@
     "DeleteDatasetRequest",
     "DeleteEndpointRequest",
     "DeleteHyperparameterTuningJobRequest",
+    "DeleteIndexEndpointRequest",
+    "DeleteIndexRequest",
+    "DeleteModelDeploymentMonitoringJobRequest",
     "DeleteModelRequest",
     "DeleteOperationMetadata",
     "DeletePipelineJobRequest",
     "DeleteSpecialistPoolRequest",
     "DeleteTrainingPipelineRequest",
+    "DeployIndexOperationMetadata",
+    "DeployIndexRequest",
+    "DeployIndexResponse",
     "DeployModelOperationMetadata",
     "DeployModelRequest",
     "DeployModelResponse",
+    "DeployedIndex",
+    "DeployedIndexAuthConfig",
+    "DeployedIndexRef",
     "DeployedModel",
     "DeployedModelRef",
     "DiskSpec",
@@ -275,13 +371,23 @@
     "EndpointServiceClient",
     "EnvVar",
     "Execution",
+    "ExplainRequest",
+    "ExplainResponse",
+    "Explanation",
+    "ExplanationMetadata",
+    "ExplanationMetadataOverride",
+    "ExplanationParameters",
+    "ExplanationSpec",
+    "ExplanationSpecOverride",
     "ExportDataConfig",
     "ExportDataOperationMetadata",
     "ExportDataRequest",
     "ExportDataResponse",
     "ExportModelOperationMetadata",
     "ExportModelRequest",
     "ExportModelResponse",
+    "FeatureNoiseSigma",
+    "FeatureStatsAnomaly",
     "FilterSplit",
     "FractionSplit",
     "GcsDestination",
@@ -294,6 +400,9 @@
     "GetDatasetRequest",
     "GetEndpointRequest",
     "GetHyperparameterTuningJobRequest",
+    "GetIndexEndpointRequest",
+    "GetIndexRequest",
+    "GetModelDeploymentMonitoringJobRequest",
     "GetModelEvaluationRequest",
     "GetModelEvaluationSliceRequest",
     "GetModelRequest",
@@ -305,7 +414,13 @@
     "ImportDataOperationMetadata",
     "ImportDataRequest",
     "ImportDataResponse",
+    "Index",
+    "IndexEndpoint",
+    "IndexEndpointServiceClient",
+    "IndexPrivateEndpoints",
+    "IndexServiceClient",
     "InputDataConfig",
+    "IntegratedGradientsAttribution",
     "JobServiceClient",
     "JobState",
     "ListAnnotationsRequest",
@@ -324,6 +439,12 @@
     "ListEndpointsResponse",
     "ListHyperparameterTuningJobsRequest",
     "ListHyperparameterTuningJobsResponse",
+    "ListIndexEndpointsRequest",
+    "ListIndexEndpointsResponse",
+    "ListIndexesRequest",
+    "ListIndexesResponse",
+    "ListModelDeploymentMonitoringJobsRequest",
+    "ListModelDeploymentMonitoringJobsResponse",
     "ListModelEvaluationSlicesRequest",
     "ListModelEvaluationSlicesResponse",
     "ListModelEvaluationsRequest",
@@ -345,9 +466,20 @@
     "MigrationServiceClient",
     "Model",
     "ModelContainerSpec",
+    "ModelDeploymentMonitoringBigQueryTable",
+    "ModelDeploymentMonitoringJob",
+    "ModelDeploymentMonitoringObjectiveConfig",
+    "ModelDeploymentMonitoringObjectiveType",
+    "ModelDeploymentMonitoringScheduleConfig",
     "ModelEvaluation",
     "ModelEvaluationSlice",
+    "ModelExplanation",
+    "ModelMonitoringAlertConfig",
+    "ModelMonitoringObjectiveConfig",
+    "ModelMonitoringStatsAnomalies",
     "ModelServiceClient",
+    "NearestNeighborSearchOperationMetadata",
+    "PauseModelDeploymentMonitoringJobRequest",
     "PipelineJob",
     "PipelineJobDetail",
     "PipelineServiceClient",
@@ -361,23 +493,40 @@
     "PredictSchemata",
     "PredictionServiceClient",
     "PythonPackageSpec",
+    "RawPredictRequest",
     "ResourcesConsumed",
+    "ResumeModelDeploymentMonitoringJobRequest",
     "SampleConfig",
+    "SampledShapleyAttribution",
+    "SamplingStrategy",
     "Scheduling",
     "SearchMigratableResourcesRequest",
     "SearchMigratableResourcesResponse",
+    "SearchModelDeploymentMonitoringStatsAnomaliesRequest",
+    "SearchModelDeploymentMonitoringStatsAnomaliesResponse",
+    "SmoothGradConfig",
     "SpecialistPool",
     "SpecialistPoolServiceClient",
+    "Study",
     "StudySpec",
+    "ThresholdConfig",
     "TimestampSplit",
     "TrainingConfig",
     "TrainingPipeline",
     "Trial",
+    "UndeployIndexOperationMetadata",
+    "UndeployIndexRequest",
+    "UndeployIndexResponse",
     "UndeployModelOperationMetadata",
     "UndeployModelRequest",
     "UndeployModelResponse",
     "UpdateDatasetRequest",
     "UpdateEndpointRequest",
+    "UpdateIndexEndpointRequest",
+    "UpdateIndexOperationMetadata",
+    "UpdateIndexRequest",
+    "UpdateModelDeploymentMonitoringJobOperationMetadata",
+    "UpdateModelDeploymentMonitoringJobRequest",
     "UpdateModelRequest",
     "UpdateSpecialistPoolOperationMetadata",
     "UpdateSpecialistPoolRequest",
@@ -387,4 +536,5 @@
     "UserActionReference",
     "Value",
     "WorkerPoolSpec",
+    "XraiAttribution",
 )