Skip to content

Commit 1fbce55

Browse files
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>
1 parent 916c1d4 commit 1fbce55

126 files changed

Lines changed: 21572 additions & 477 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.
Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
IndexEndpointService
2+
--------------------------------------
3+
4+
.. automodule:: google.cloud.aiplatform_v1.services.index_endpoint_service
5+
:members:
6+
:inherited-members:
7+
8+
.. automodule:: google.cloud.aiplatform_v1.services.index_endpoint_service.pagers
9+
:members:
10+
:inherited-members:
Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
IndexService
2+
------------------------------
3+
4+
.. automodule:: google.cloud.aiplatform_v1.services.index_service
5+
:members:
6+
:inherited-members:
7+
8+
.. automodule:: google.cloud.aiplatform_v1.services.index_service.pagers
9+
:members:
10+
:inherited-members:

docs/aiplatform_v1/services.rst

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@ Services for Google Cloud Aiplatform v1 API
55

66
dataset_service
77
endpoint_service
8+
index_endpoint_service
9+
index_service
810
job_service
911
migration_service
1012
model_service

google/cloud/aiplatform_v1/__init__.py

Lines changed: 150 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,10 @@
1818
from .services.dataset_service import DatasetServiceAsyncClient
1919
from .services.endpoint_service import EndpointServiceClient
2020
from .services.endpoint_service import EndpointServiceAsyncClient
21+
from .services.index_endpoint_service import IndexEndpointServiceClient
22+
from .services.index_endpoint_service import IndexEndpointServiceAsyncClient
23+
from .services.index_service import IndexServiceClient
24+
from .services.index_service import IndexServiceAsyncClient
2125
from .services.job_service import JobServiceClient
2226
from .services.job_service import JobServiceAsyncClient
2327
from .services.migration_service import MigrationServiceClient
@@ -70,6 +74,7 @@
7074
from .types.dataset_service import ListDatasetsRequest
7175
from .types.dataset_service import ListDatasetsResponse
7276
from .types.dataset_service import UpdateDatasetRequest
77+
from .types.deployed_index_ref import DeployedIndexRef
7378
from .types.deployed_model_ref import DeployedModelRef
7479
from .types.encryption_spec import EncryptionSpec
7580
from .types.endpoint import DeployedModel
@@ -89,7 +94,48 @@
8994
from .types.endpoint_service import UpdateEndpointRequest
9095
from .types.env_var import EnvVar
9196
from .types.execution import Execution
97+
from .types.explanation import Attribution
98+
from .types.explanation import Explanation
99+
from .types.explanation import ExplanationMetadataOverride
100+
from .types.explanation import ExplanationParameters
101+
from .types.explanation import ExplanationSpec
102+
from .types.explanation import ExplanationSpecOverride
103+
from .types.explanation import FeatureNoiseSigma
104+
from .types.explanation import IntegratedGradientsAttribution
105+
from .types.explanation import ModelExplanation
106+
from .types.explanation import SampledShapleyAttribution
107+
from .types.explanation import SmoothGradConfig
108+
from .types.explanation import XraiAttribution
109+
from .types.explanation_metadata import ExplanationMetadata
110+
from .types.feature_monitoring_stats import FeatureStatsAnomaly
92111
from .types.hyperparameter_tuning_job import HyperparameterTuningJob
112+
from .types.index import Index
113+
from .types.index_endpoint import DeployedIndex
114+
from .types.index_endpoint import DeployedIndexAuthConfig
115+
from .types.index_endpoint import IndexEndpoint
116+
from .types.index_endpoint import IndexPrivateEndpoints
117+
from .types.index_endpoint_service import CreateIndexEndpointOperationMetadata
118+
from .types.index_endpoint_service import CreateIndexEndpointRequest
119+
from .types.index_endpoint_service import DeleteIndexEndpointRequest
120+
from .types.index_endpoint_service import DeployIndexOperationMetadata
121+
from .types.index_endpoint_service import DeployIndexRequest
122+
from .types.index_endpoint_service import DeployIndexResponse
123+
from .types.index_endpoint_service import GetIndexEndpointRequest
124+
from .types.index_endpoint_service import ListIndexEndpointsRequest
125+
from .types.index_endpoint_service import ListIndexEndpointsResponse
126+
from .types.index_endpoint_service import UndeployIndexOperationMetadata
127+
from .types.index_endpoint_service import UndeployIndexRequest
128+
from .types.index_endpoint_service import UndeployIndexResponse
129+
from .types.index_endpoint_service import UpdateIndexEndpointRequest
130+
from .types.index_service import CreateIndexOperationMetadata
131+
from .types.index_service import CreateIndexRequest
132+
from .types.index_service import DeleteIndexRequest
133+
from .types.index_service import GetIndexRequest
134+
from .types.index_service import ListIndexesRequest
135+
from .types.index_service import ListIndexesResponse
136+
from .types.index_service import NearestNeighborSearchOperationMetadata
137+
from .types.index_service import UpdateIndexOperationMetadata
138+
from .types.index_service import UpdateIndexRequest
93139
from .types.io import BigQueryDestination
94140
from .types.io import BigQuerySource
95141
from .types.io import ContainerRegistryDestination
@@ -103,14 +149,17 @@
103149
from .types.job_service import CreateCustomJobRequest
104150
from .types.job_service import CreateDataLabelingJobRequest
105151
from .types.job_service import CreateHyperparameterTuningJobRequest
152+
from .types.job_service import CreateModelDeploymentMonitoringJobRequest
106153
from .types.job_service import DeleteBatchPredictionJobRequest
107154
from .types.job_service import DeleteCustomJobRequest
108155
from .types.job_service import DeleteDataLabelingJobRequest
109156
from .types.job_service import DeleteHyperparameterTuningJobRequest
157+
from .types.job_service import DeleteModelDeploymentMonitoringJobRequest
110158
from .types.job_service import GetBatchPredictionJobRequest
111159
from .types.job_service import GetCustomJobRequest
112160
from .types.job_service import GetDataLabelingJobRequest
113161
from .types.job_service import GetHyperparameterTuningJobRequest
162+
from .types.job_service import GetModelDeploymentMonitoringJobRequest
114163
from .types.job_service import ListBatchPredictionJobsRequest
115164
from .types.job_service import ListBatchPredictionJobsResponse
116165
from .types.job_service import ListCustomJobsRequest
@@ -119,6 +168,14 @@
119168
from .types.job_service import ListDataLabelingJobsResponse
120169
from .types.job_service import ListHyperparameterTuningJobsRequest
121170
from .types.job_service import ListHyperparameterTuningJobsResponse
171+
from .types.job_service import ListModelDeploymentMonitoringJobsRequest
172+
from .types.job_service import ListModelDeploymentMonitoringJobsResponse
173+
from .types.job_service import PauseModelDeploymentMonitoringJobRequest
174+
from .types.job_service import ResumeModelDeploymentMonitoringJobRequest
175+
from .types.job_service import SearchModelDeploymentMonitoringStatsAnomaliesRequest
176+
from .types.job_service import SearchModelDeploymentMonitoringStatsAnomaliesResponse
177+
from .types.job_service import UpdateModelDeploymentMonitoringJobOperationMetadata
178+
from .types.job_service import UpdateModelDeploymentMonitoringJobRequest
122179
from .types.job_state import JobState
123180
from .types.machine_resources import AutomaticResources
124181
from .types.machine_resources import AutoscalingMetricSpec
@@ -140,8 +197,26 @@
140197
from .types.model import ModelContainerSpec
141198
from .types.model import Port
142199
from .types.model import PredictSchemata
200+
from .types.model_deployment_monitoring_job import (
201+
ModelDeploymentMonitoringBigQueryTable,
202+
)
203+
from .types.model_deployment_monitoring_job import ModelDeploymentMonitoringJob
204+
from .types.model_deployment_monitoring_job import (
205+
ModelDeploymentMonitoringObjectiveConfig,
206+
)
207+
from .types.model_deployment_monitoring_job import (
208+
ModelDeploymentMonitoringScheduleConfig,
209+
)
210+
from .types.model_deployment_monitoring_job import ModelMonitoringStatsAnomalies
211+
from .types.model_deployment_monitoring_job import (
212+
ModelDeploymentMonitoringObjectiveType,
213+
)
143214
from .types.model_evaluation import ModelEvaluation
144215
from .types.model_evaluation_slice import ModelEvaluationSlice
216+
from .types.model_monitoring import ModelMonitoringAlertConfig
217+
from .types.model_monitoring import ModelMonitoringObjectiveConfig
218+
from .types.model_monitoring import SamplingStrategy
219+
from .types.model_monitoring import ThresholdConfig
145220
from .types.model_service import DeleteModelRequest
146221
from .types.model_service import ExportModelOperationMetadata
147222
from .types.model_service import ExportModelRequest
@@ -178,8 +253,11 @@
178253
from .types.pipeline_service import ListTrainingPipelinesRequest
179254
from .types.pipeline_service import ListTrainingPipelinesResponse
180255
from .types.pipeline_state import PipelineState
256+
from .types.prediction_service import ExplainRequest
257+
from .types.prediction_service import ExplainResponse
181258
from .types.prediction_service import PredictRequest
182259
from .types.prediction_service import PredictResponse
260+
from .types.prediction_service import RawPredictRequest
183261
from .types.specialist_pool import SpecialistPool
184262
from .types.specialist_pool_service import CreateSpecialistPoolOperationMetadata
185263
from .types.specialist_pool_service import CreateSpecialistPoolRequest
@@ -190,6 +268,7 @@
190268
from .types.specialist_pool_service import UpdateSpecialistPoolOperationMetadata
191269
from .types.specialist_pool_service import UpdateSpecialistPoolRequest
192270
from .types.study import Measurement
271+
from .types.study import Study
193272
from .types.study import StudySpec
194273
from .types.study import Trial
195274
from .types.training_pipeline import FilterSplit
@@ -204,6 +283,8 @@
204283
__all__ = (
205284
"DatasetServiceAsyncClient",
206285
"EndpointServiceAsyncClient",
286+
"IndexEndpointServiceAsyncClient",
287+
"IndexServiceAsyncClient",
207288
"JobServiceAsyncClient",
208289
"MigrationServiceAsyncClient",
209290
"ModelServiceAsyncClient",
@@ -215,6 +296,7 @@
215296
"Annotation",
216297
"AnnotationSpec",
217298
"Artifact",
299+
"Attribution",
218300
"AutomaticResources",
219301
"AutoscalingMetricSpec",
220302
"BatchDedicatedResources",
@@ -242,6 +324,11 @@
242324
"CreateEndpointOperationMetadata",
243325
"CreateEndpointRequest",
244326
"CreateHyperparameterTuningJobRequest",
327+
"CreateIndexEndpointOperationMetadata",
328+
"CreateIndexEndpointRequest",
329+
"CreateIndexOperationMetadata",
330+
"CreateIndexRequest",
331+
"CreateModelDeploymentMonitoringJobRequest",
245332
"CreatePipelineJobRequest",
246333
"CreateSpecialistPoolOperationMetadata",
247334
"CreateSpecialistPoolRequest",
@@ -259,14 +346,23 @@
259346
"DeleteDatasetRequest",
260347
"DeleteEndpointRequest",
261348
"DeleteHyperparameterTuningJobRequest",
349+
"DeleteIndexEndpointRequest",
350+
"DeleteIndexRequest",
351+
"DeleteModelDeploymentMonitoringJobRequest",
262352
"DeleteModelRequest",
263353
"DeleteOperationMetadata",
264354
"DeletePipelineJobRequest",
265355
"DeleteSpecialistPoolRequest",
266356
"DeleteTrainingPipelineRequest",
357+
"DeployIndexOperationMetadata",
358+
"DeployIndexRequest",
359+
"DeployIndexResponse",
267360
"DeployModelOperationMetadata",
268361
"DeployModelRequest",
269362
"DeployModelResponse",
363+
"DeployedIndex",
364+
"DeployedIndexAuthConfig",
365+
"DeployedIndexRef",
270366
"DeployedModel",
271367
"DeployedModelRef",
272368
"DiskSpec",
@@ -275,13 +371,23 @@
275371
"EndpointServiceClient",
276372
"EnvVar",
277373
"Execution",
374+
"ExplainRequest",
375+
"ExplainResponse",
376+
"Explanation",
377+
"ExplanationMetadata",
378+
"ExplanationMetadataOverride",
379+
"ExplanationParameters",
380+
"ExplanationSpec",
381+
"ExplanationSpecOverride",
278382
"ExportDataConfig",
279383
"ExportDataOperationMetadata",
280384
"ExportDataRequest",
281385
"ExportDataResponse",
282386
"ExportModelOperationMetadata",
283387
"ExportModelRequest",
284388
"ExportModelResponse",
389+
"FeatureNoiseSigma",
390+
"FeatureStatsAnomaly",
285391
"FilterSplit",
286392
"FractionSplit",
287393
"GcsDestination",
@@ -294,6 +400,9 @@
294400
"GetDatasetRequest",
295401
"GetEndpointRequest",
296402
"GetHyperparameterTuningJobRequest",
403+
"GetIndexEndpointRequest",
404+
"GetIndexRequest",
405+
"GetModelDeploymentMonitoringJobRequest",
297406
"GetModelEvaluationRequest",
298407
"GetModelEvaluationSliceRequest",
299408
"GetModelRequest",
@@ -305,7 +414,13 @@
305414
"ImportDataOperationMetadata",
306415
"ImportDataRequest",
307416
"ImportDataResponse",
417+
"Index",
418+
"IndexEndpoint",
419+
"IndexEndpointServiceClient",
420+
"IndexPrivateEndpoints",
421+
"IndexServiceClient",
308422
"InputDataConfig",
423+
"IntegratedGradientsAttribution",
309424
"JobServiceClient",
310425
"JobState",
311426
"ListAnnotationsRequest",
@@ -324,6 +439,12 @@
324439
"ListEndpointsResponse",
325440
"ListHyperparameterTuningJobsRequest",
326441
"ListHyperparameterTuningJobsResponse",
442+
"ListIndexEndpointsRequest",
443+
"ListIndexEndpointsResponse",
444+
"ListIndexesRequest",
445+
"ListIndexesResponse",
446+
"ListModelDeploymentMonitoringJobsRequest",
447+
"ListModelDeploymentMonitoringJobsResponse",
327448
"ListModelEvaluationSlicesRequest",
328449
"ListModelEvaluationSlicesResponse",
329450
"ListModelEvaluationsRequest",
@@ -345,9 +466,20 @@
345466
"MigrationServiceClient",
346467
"Model",
347468
"ModelContainerSpec",
469+
"ModelDeploymentMonitoringBigQueryTable",
470+
"ModelDeploymentMonitoringJob",
471+
"ModelDeploymentMonitoringObjectiveConfig",
472+
"ModelDeploymentMonitoringObjectiveType",
473+
"ModelDeploymentMonitoringScheduleConfig",
348474
"ModelEvaluation",
349475
"ModelEvaluationSlice",
476+
"ModelExplanation",
477+
"ModelMonitoringAlertConfig",
478+
"ModelMonitoringObjectiveConfig",
479+
"ModelMonitoringStatsAnomalies",
350480
"ModelServiceClient",
481+
"NearestNeighborSearchOperationMetadata",
482+
"PauseModelDeploymentMonitoringJobRequest",
351483
"PipelineJob",
352484
"PipelineJobDetail",
353485
"PipelineServiceClient",
@@ -361,23 +493,40 @@
361493
"PredictSchemata",
362494
"PredictionServiceClient",
363495
"PythonPackageSpec",
496+
"RawPredictRequest",
364497
"ResourcesConsumed",
498+
"ResumeModelDeploymentMonitoringJobRequest",
365499
"SampleConfig",
500+
"SampledShapleyAttribution",
501+
"SamplingStrategy",
366502
"Scheduling",
367503
"SearchMigratableResourcesRequest",
368504
"SearchMigratableResourcesResponse",
505+
"SearchModelDeploymentMonitoringStatsAnomaliesRequest",
506+
"SearchModelDeploymentMonitoringStatsAnomaliesResponse",
507+
"SmoothGradConfig",
369508
"SpecialistPool",
370509
"SpecialistPoolServiceClient",
510+
"Study",
371511
"StudySpec",
512+
"ThresholdConfig",
372513
"TimestampSplit",
373514
"TrainingConfig",
374515
"TrainingPipeline",
375516
"Trial",
517+
"UndeployIndexOperationMetadata",
518+
"UndeployIndexRequest",
519+
"UndeployIndexResponse",
376520
"UndeployModelOperationMetadata",
377521
"UndeployModelRequest",
378522
"UndeployModelResponse",
379523
"UpdateDatasetRequest",
380524
"UpdateEndpointRequest",
525+
"UpdateIndexEndpointRequest",
526+
"UpdateIndexOperationMetadata",
527+
"UpdateIndexRequest",
528+
"UpdateModelDeploymentMonitoringJobOperationMetadata",
529+
"UpdateModelDeploymentMonitoringJobRequest",
381530
"UpdateModelRequest",
382531
"UpdateSpecialistPoolOperationMetadata",
383532
"UpdateSpecialistPoolRequest",
@@ -387,4 +536,5 @@
387536
"UserActionReference",
388537
"Value",
389538
"WorkerPoolSpec",
539+
"XraiAttribution",
390540
)

0 commit comments

Comments
 (0)