Copybara import of the project:

-- 4eb7ad9 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add data_stats to Model feat: add data_stats to ExportDataResponse feat: add filter_split to ExportDataConfig feat: add saved_query_id to ExportDataConfig feat: add annotation_schema_uri to ExportDataConfig feat: add export_use to ExportDataConfig docs: update exported_files description in ExportDataResponse PiperOrigin-RevId: 588136649 Source-Link: googleapis/googleapis@57b57f4 Source-Link: googleapis/googleapis-gen@0d91143 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMGQ5MTE0MzdiNzY0MTA4MWRkNzU3N2UxZDYwZjAwZmRlZjhlZjE0MiJ9 -- bb69077 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 08f38ab by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add Content feat: add data_stats to ExportDataResponse feat: add display_name to DeployedIndex feat: add ExportEndpointOperationMetadata to EndpointService docs: changes name from OUTPUT_ONLY to IDENTIFIER in FeatureGroup feat: add FunctionCall and FunctionResponse feat: add REBOOTING and notebook_runtime_template to PersistentResource feat: add BatchCancelPipelineJobs to PipelineService feat: add StreamGenerateContent and GenerateContentRequest to PredictionService feat: add Tool PiperOrigin-RevId: 589253340 Source-Link: googleapis/googleapis@0ed84b3 Source-Link: googleapis/googleapis-gen@b56eb45 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYjU2ZWI0NTY5MDllOGUxMjQ5MTRlZGM2MDY0YjE0Zjg4YjFmZDI2MSJ9 -- 6bff60a by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add Content feat: add ExportUse and ExportFilterSplit to Dataset feat: add data_stats to ExportDataResponse feat: add display_name to DeployedIndex feat: add DeploymentResourcePool feat: add DeploymentResourcePoolService feat: add shared_resources to DeployedModel docs: changes name from OUTPUT_ONLY to IDENTIFIER in FeatureGroup feat: add DataStats to Model feat: add model and contents to PredictionService feat: add Tool PiperOrigin-RevId: 589254081 Source-Link: googleapis/googleapis@00931e0 Source-Link: googleapis/googleapis-gen@1393818 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMTM5MzgxODA0NTM0YzcyNDhlMjZiMGUyM2M0MThiMmUwNjdiNmJiNiJ9 -- 8273f41 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md COPYBARA_INTEGRATE_REVIEW=#3067 from googleapis:owl-bot-copy ece01ed PiperOrigin-RevId: 589301758
googleapis · Dec 9, 2023 · c9903ae · c9903ae
1 parent c9f7119
commit c9903ae
Show file tree

Hide file tree

Showing 95 changed files with 17,718 additions and 1,081 deletions.
diff --git a/docs/aiplatform_v1/deployment_resource_pool_service.rst b/docs/aiplatform_v1/deployment_resource_pool_service.rst
@@ -0,0 +1,10 @@
+DeploymentResourcePoolService
+-----------------------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1.services.deployment_resource_pool_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.aiplatform_v1.services.deployment_resource_pool_service.pagers
+    :members:
+    :inherited-members:
diff --git a/docs/aiplatform_v1/services_.rst b/docs/aiplatform_v1/services_.rst
@@ -4,6 +4,7 @@ Services for Google Cloud Aiplatform v1 API
     :maxdepth: 2
 
     dataset_service
+    deployment_resource_pool_service
     endpoint_service
     feature_online_store_admin_service
     feature_online_store_service

diff --git a/google/cloud/aiplatform_v1/__init__.py b/google/cloud/aiplatform_v1/__init__.py
@@ -20,6 +20,12 @@
 
 from .services.dataset_service import DatasetServiceClient
 from .services.dataset_service import DatasetServiceAsyncClient
+from .services.deployment_resource_pool_service import (
+    DeploymentResourcePoolServiceClient,
+)
+from .services.deployment_resource_pool_service import (
+    DeploymentResourcePoolServiceAsyncClient,
+)
 from .services.endpoint_service import EndpointServiceClient
 from .services.endpoint_service import EndpointServiceAsyncClient
 from .services.feature_online_store_admin_service import (
@@ -77,6 +83,18 @@
 from .types.artifact import Artifact
 from .types.batch_prediction_job import BatchPredictionJob
 from .types.completion_stats import CompletionStats
+from .types.content import Blob
+from .types.content import Candidate
+from .types.content import Citation
+from .types.content import CitationMetadata
+from .types.content import Content
+from .types.content import FileData
+from .types.content import GenerationConfig
+from .types.content import Part
+from .types.content import SafetyRating
+from .types.content import SafetySetting
+from .types.content import VideoMetadata
+from .types.content import HarmCategory
 from .types.context import Context
 from .types.custom_job import ContainerSpec
 from .types.custom_job import CustomJob
@@ -91,6 +109,7 @@
 from .types.data_labeling_job import TrainingConfig
 from .types.dataset import Dataset
 from .types.dataset import ExportDataConfig
+from .types.dataset import ExportFilterSplit
 from .types.dataset import ExportFractionSplit
 from .types.dataset import ImportDataConfig
 from .types.dataset_service import CreateDatasetOperationMetadata
@@ -128,6 +147,20 @@
 from .types.dataset_version import DatasetVersion
 from .types.deployed_index_ref import DeployedIndexRef
 from .types.deployed_model_ref import DeployedModelRef
+from .types.deployment_resource_pool import DeploymentResourcePool
+from .types.deployment_resource_pool_service import (
+    CreateDeploymentResourcePoolOperationMetadata,
+)
+from .types.deployment_resource_pool_service import CreateDeploymentResourcePoolRequest
+from .types.deployment_resource_pool_service import DeleteDeploymentResourcePoolRequest
+from .types.deployment_resource_pool_service import GetDeploymentResourcePoolRequest
+from .types.deployment_resource_pool_service import ListDeploymentResourcePoolsRequest
+from .types.deployment_resource_pool_service import ListDeploymentResourcePoolsResponse
+from .types.deployment_resource_pool_service import QueryDeployedModelsRequest
+from .types.deployment_resource_pool_service import QueryDeployedModelsResponse
+from .types.deployment_resource_pool_service import (
+    UpdateDeploymentResourcePoolOperationMetadata,
+)
 from .types.encryption_spec import EncryptionSpec
 from .types.endpoint import DeployedModel
 from .types.endpoint import Endpoint
@@ -508,6 +541,8 @@
 from .types.nas_job import NasJobSpec
 from .types.nas_job import NasTrial
 from .types.nas_job import NasTrialDetail
+from .types.openapi import Schema
+from .types.openapi import Type
 from .types.operation import DeleteOperationMetadata
 from .types.operation import GenericOperationMetadata
 from .types.pipeline_failure_policy import PipelineFailurePolicy
@@ -537,6 +572,8 @@
 from .types.prediction_service import DirectRawPredictResponse
 from .types.prediction_service import ExplainRequest
 from .types.prediction_service import ExplainResponse
+from .types.prediction_service import GenerateContentRequest
+from .types.prediction_service import GenerateContentResponse
 from .types.prediction_service import PredictRequest
 from .types.prediction_service import PredictResponse
 from .types.prediction_service import RawPredictRequest
@@ -627,6 +664,10 @@
 from .types.tensorboard_service import WriteTensorboardRunDataRequest
 from .types.tensorboard_service import WriteTensorboardRunDataResponse
 from .types.tensorboard_time_series import TensorboardTimeSeries
+from .types.tool import FunctionCall
+from .types.tool import FunctionDeclaration
+from .types.tool import FunctionResponse
+from .types.tool import Tool
 from .types.training_pipeline import FilterSplit
 from .types.training_pipeline import FractionSplit
 from .types.training_pipeline import InputDataConfig
@@ -667,6 +708,7 @@
 
 __all__ = (
     "DatasetServiceAsyncClient",
+    "DeploymentResourcePoolServiceAsyncClient",
     "EndpointServiceAsyncClient",
     "FeatureOnlineStoreAdminServiceAsyncClient",
     "FeatureOnlineStoreServiceAsyncClient",
@@ -727,6 +769,7 @@
     "BatchReadTensorboardTimeSeriesDataResponse",
     "BigQueryDestination",
     "BigQuerySource",
+    "Blob",
     "BlurBaselineConfig",
     "BoolArray",
     "CancelBatchPredictionJobRequest",
@@ -736,15 +779,19 @@
     "CancelNasJobRequest",
     "CancelPipelineJobRequest",
     "CancelTrainingPipelineRequest",
+    "Candidate",
     "CheckTrialEarlyStoppingStateMetatdata",
     "CheckTrialEarlyStoppingStateRequest",
     "CheckTrialEarlyStoppingStateResponse",
+    "Citation",
+    "CitationMetadata",
     "CompleteTrialRequest",
     "CompletionStats",
     "ComputeTokensRequest",
     "ComputeTokensResponse",
     "ContainerRegistryDestination",
     "ContainerSpec",
+    "Content",
     "Context",
     "CopyModelOperationMetadata",
     "CopyModelRequest",
@@ -760,6 +807,8 @@
     "CreateDatasetRequest",
     "CreateDatasetVersionOperationMetadata",
     "CreateDatasetVersionRequest",
+    "CreateDeploymentResourcePoolOperationMetadata",
+    "CreateDeploymentResourcePoolRequest",
     "CreateEndpointOperationMetadata",
     "CreateEndpointRequest",
     "CreateEntityTypeOperationMetadata",
@@ -816,6 +865,7 @@
     "DeleteDataLabelingJobRequest",
     "DeleteDatasetRequest",
     "DeleteDatasetVersionRequest",
+    "DeleteDeploymentResourcePoolRequest",
     "DeleteEndpointRequest",
     "DeleteEntityTypeRequest",
     "DeleteExecutionRequest",
@@ -859,6 +909,8 @@
     "DeployedIndexRef",
     "DeployedModel",
     "DeployedModelRef",
+    "DeploymentResourcePool",
+    "DeploymentResourcePoolServiceClient",
     "DestinationFeatureSetting",
     "DirectPredictRequest",
     "DirectPredictResponse",
@@ -895,6 +947,7 @@
     "ExportFeatureValuesOperationMetadata",
     "ExportFeatureValuesRequest",
     "ExportFeatureValuesResponse",
+    "ExportFilterSplit",
     "ExportFractionSplit",
     "ExportModelOperationMetadata",
     "ExportModelRequest",
@@ -923,12 +976,19 @@
     "FeaturestoreServiceClient",
     "FetchFeatureValuesRequest",
     "FetchFeatureValuesResponse",
+    "FileData",
     "FilterSplit",
     "FindNeighborsRequest",
     "FindNeighborsResponse",
     "FractionSplit",
+    "FunctionCall",
+    "FunctionDeclaration",
+    "FunctionResponse",
     "GcsDestination",
     "GcsSource",
+    "GenerateContentRequest",
+    "GenerateContentResponse",
+    "GenerationConfig",
     "GenericOperationMetadata",
     "GetAnnotationSpecRequest",
     "GetArtifactRequest",
@@ -938,6 +998,7 @@
     "GetDataLabelingJobRequest",
     "GetDatasetRequest",
     "GetDatasetVersionRequest",
+    "GetDeploymentResourcePoolRequest",
     "GetEndpointRequest",
     "GetEntityTypeRequest",
     "GetExecutionRequest",
@@ -969,6 +1030,7 @@
     "GetTensorboardTimeSeriesRequest",
     "GetTrainingPipelineRequest",
     "GetTrialRequest",
+    "HarmCategory",
     "HyperparameterTuningJob",
     "IdMatcher",
     "ImportDataConfig",
@@ -1011,6 +1073,8 @@
     "ListDatasetVersionsResponse",
     "ListDatasetsRequest",
     "ListDatasetsResponse",
+    "ListDeploymentResourcePoolsRequest",
+    "ListDeploymentResourcePoolsResponse",
     "ListEndpointsRequest",
     "ListEndpointsResponse",
     "ListEntityTypesRequest",
@@ -1121,6 +1185,7 @@
     "NearestNeighborSearchOperationMetadata",
     "Neighbor",
     "NfsMount",
+    "Part",
     "PauseModelDeploymentMonitoringJobRequest",
     "PauseScheduleRequest",
     "PersistentDiskSpec",
@@ -1157,6 +1222,8 @@
     "PythonPackageSpec",
     "QueryArtifactLineageSubgraphRequest",
     "QueryContextLineageSubgraphRequest",
+    "QueryDeployedModelsRequest",
+    "QueryDeployedModelsResponse",
     "QueryExecutionInputsAndOutputsRequest",
     "RawPredictRequest",
     "ReadFeatureValuesRequest",
@@ -1180,6 +1247,8 @@
     "RestoreDatasetVersionRequest",
     "ResumeModelDeploymentMonitoringJobRequest",
     "ResumeScheduleRequest",
+    "SafetyRating",
+    "SafetySetting",
     "SampleConfig",
     "SampledShapleyAttribution",
     "SamplingStrategy",
@@ -1188,6 +1257,7 @@
     "Schedule",
     "ScheduleServiceClient",
     "Scheduling",
+    "Schema",
     "SearchDataItemsRequest",
     "SearchDataItemsResponse",
     "SearchFeaturesRequest",
@@ -1230,10 +1300,12 @@
     "TimeSeriesDataPoint",
     "TimestampSplit",
     "TokensInfo",
+    "Tool",
     "TrainingConfig",
     "TrainingPipeline",
     "Trial",
     "TrialContext",
+    "Type",
     "UndeployIndexOperationMetadata",
     "UndeployIndexRequest",
     "UndeployIndexResponse",
@@ -1244,6 +1316,7 @@
     "UpdateArtifactRequest",
     "UpdateContextRequest",
     "UpdateDatasetRequest",
+    "UpdateDeploymentResourcePoolOperationMetadata",
     "UpdateEndpointRequest",
     "UpdateEntityTypeRequest",
     "UpdateExecutionRequest",
@@ -1281,6 +1354,7 @@
     "UpsertDatapointsResponse",
     "UserActionReference",
     "Value",
+    "VideoMetadata",
     "VizierServiceClient",
     "WorkerPoolSpec",
     "WriteFeatureValuesPayload",

diff --git a/google/cloud/aiplatform_v1/gapic_metadata.json b/google/cloud/aiplatform_v1/gapic_metadata.json
@@ -199,6 +199,70 @@
         }
       }
     },
+    "DeploymentResourcePoolService": {
+      "clients": {
+        "grpc": {
+          "libraryClient": "DeploymentResourcePoolServiceClient",
+          "rpcs": {
+            "CreateDeploymentResourcePool": {
+              "methods": [
+                "create_deployment_resource_pool"
+              ]
+            },
+            "DeleteDeploymentResourcePool": {
+              "methods": [
+                "delete_deployment_resource_pool"
+              ]
+            },
+            "GetDeploymentResourcePool": {
+              "methods": [
+                "get_deployment_resource_pool"
+              ]
+            },
+            "ListDeploymentResourcePools": {
+              "methods": [
+                "list_deployment_resource_pools"
+              ]
+            },
+            "QueryDeployedModels": {
+              "methods": [
+                "query_deployed_models"
+              ]
+            }
+          }
+        },
+        "grpc-async": {
+          "libraryClient": "DeploymentResourcePoolServiceAsyncClient",
+          "rpcs": {
+            "CreateDeploymentResourcePool": {
+              "methods": [
+                "create_deployment_resource_pool"
+              ]
+            },
+            "DeleteDeploymentResourcePool": {
+              "methods": [
+                "delete_deployment_resource_pool"
+              ]
+            },
+            "GetDeploymentResourcePool": {
+              "methods": [
+                "get_deployment_resource_pool"
+              ]
+            },
+            "ListDeploymentResourcePools": {
+              "methods": [
+                "list_deployment_resource_pools"
+              ]
+            },
+            "QueryDeployedModels": {
+              "methods": [
+                "query_deployed_models"
+              ]
+            }
+          }
+        }
+      }
+    },
     "EndpointService": {
       "clients": {
         "grpc": {
@@ -2188,6 +2252,11 @@
                 "server_streaming_predict"
               ]
             },
+            "StreamGenerateContent": {
+              "methods": [
+                "stream_generate_content"
+              ]
+            },
             "StreamingPredict": {
               "methods": [
                 "streaming_predict"
@@ -2233,6 +2302,11 @@
                 "server_streaming_predict"
               ]
             },
+            "StreamGenerateContent": {
+              "methods": [
+                "stream_generate_content"
+              ]
+            },
             "StreamingPredict": {
               "methods": [
                 "streaming_predict"