feat: add prediction service RPC RawPredict to aiplatform_v1beta1 feat: add tensorboard service RPCs to aiplatform_v1beta1: BatchCreateTensorboardRuns, BatchCreateTensorboardTimeSeries, WriteTensorboardExperimentData feat: add model_deployment_monitori... (#670)

* feat: add prediction service RPC RawPredict to aiplatform_v1beta1 feat: add tensorboard service RPCs to aiplatform_v1beta1: BatchCreateTensorboardRuns, BatchCreateTensorboardTimeSeries, WriteTensorboardExperimentData feat: add model_deployment_monitoring_job to Endpoint in aiplatform_v1beta1 feat: add deployment_group to DeployedIndex in aiplatform_v1beta1 feat: add ModelEvaluationExplanationSpec in aiplatform_v1beta1 Committer: @dizcology PiperOrigin-RevId: 393890669 Source-Link: googleapis/googleapis@321abab Source-Link: googleapis/googleapis-gen@629290d * 🦉 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>

Author: gcf-owl-bot[bot]

google/cloud/aiplatform_v1beta1/__init__.py

@@ -376,6 +376,7 @@
 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
@@ -398,6 +399,10 @@
 from .types.tensorboard_data import TimeSeriesDataPoint
 from .types.tensorboard_experiment import TensorboardExperiment
 from .types.tensorboard_run import TensorboardRun
+from .types.tensorboard_service import BatchCreateTensorboardRunsRequest
+from .types.tensorboard_service import BatchCreateTensorboardRunsResponse
+from .types.tensorboard_service import BatchCreateTensorboardTimeSeriesRequest
+from .types.tensorboard_service import BatchCreateTensorboardTimeSeriesResponse
 from .types.tensorboard_service import CreateTensorboardExperimentRequest
 from .types.tensorboard_service import CreateTensorboardOperationMetadata
 from .types.tensorboard_service import CreateTensorboardRequest
@@ -430,6 +435,8 @@
 from .types.tensorboard_service import UpdateTensorboardRequest
 from .types.tensorboard_service import UpdateTensorboardRunRequest
 from .types.tensorboard_service import UpdateTensorboardTimeSeriesRequest
+from .types.tensorboard_service import WriteTensorboardExperimentDataRequest
+from .types.tensorboard_service import WriteTensorboardExperimentDataResponse
 from .types.tensorboard_service import WriteTensorboardRunDataRequest
 from .types.tensorboard_service import WriteTensorboardRunDataResponse
 from .types.tensorboard_time_series import TensorboardTimeSeries
@@ -503,6 +510,10 @@
  "BatchCreateFeaturesOperationMetadata",
  "BatchCreateFeaturesRequest",
  "BatchCreateFeaturesResponse",
+ "BatchCreateTensorboardRunsRequest",
+ "BatchCreateTensorboardRunsResponse",
+ "BatchCreateTensorboardTimeSeriesRequest",
+ "BatchCreateTensorboardTimeSeriesResponse",
  "BatchDedicatedResources",
  "BatchMigrateResourcesOperationMetadata",
  "BatchMigrateResourcesRequest",
@@ -825,6 +836,7 @@
  "QueryArtifactLineageSubgraphRequest",
  "QueryContextLineageSubgraphRequest",
  "QueryExecutionInputsAndOutputsRequest",
+ "RawPredictRequest",
  "ReadFeatureValuesRequest",
  "ReadFeatureValuesResponse",
  "ReadTensorboardBlobDataRequest",
@@ -906,6 +918,8 @@
  "Value",
  "VizierServiceClient",
  "WorkerPoolSpec",
+ "WriteTensorboardExperimentDataRequest",
+ "WriteTensorboardExperimentDataResponse",
  "WriteTensorboardRunDataRequest",
  "WriteTensorboardRunDataResponse",
  "XraiAttribution",

google/cloud/aiplatform_v1beta1/gapic_metadata.json

@@ -1493,6 +1493,11 @@
  "methods": [
  "predict"
  ]
+ },
+ "RawPredict": {
+ "methods": [
+ "raw_predict"
+ ]
  }
  }
  },
@@ -1508,6 +1513,11 @@
  "methods": [
  "predict"
  ]
+ },
+ "RawPredict": {
+ "methods": [
+ "raw_predict"
+ ]
  }
  }
  }
@@ -1582,6 +1592,16 @@
  "grpc": {
  "libraryClient": "TensorboardServiceClient",
  "rpcs": {
+ "BatchCreateTensorboardRuns": {
+ "methods": [
+ "batch_create_tensorboard_runs"
+ ]
+ },
+ "BatchCreateTensorboardTimeSeries": {
+ "methods": [
+ "batch_create_tensorboard_time_series"
+ ]
+ },
  "CreateTensorboard": {
  "methods": [
  "create_tensorboard"
@@ -1697,6 +1717,11 @@
  "update_tensorboard_time_series"
  ]
  },
+ "WriteTensorboardExperimentData": {
+ "methods": [
+ "write_tensorboard_experiment_data"
+ ]
+ },
  "WriteTensorboardRunData": {
  "methods": [
  "write_tensorboard_run_data"
@@ -1707,6 +1732,16 @@
  "grpc-async": {
  "libraryClient": "TensorboardServiceAsyncClient",
  "rpcs": {
+ "BatchCreateTensorboardRuns": {
+ "methods": [
+ "batch_create_tensorboard_runs"
+ ]
+ },
+ "BatchCreateTensorboardTimeSeries": {
+ "methods": [
+ "batch_create_tensorboard_time_series"
+ ]
+ },
  "CreateTensorboard": {
  "methods": [
  "create_tensorboard"
@@ -1822,6 +1857,11 @@
  "update_tensorboard_time_series"
  ]
  },
+ "WriteTensorboardExperimentData": {
+ "methods": [
+ "write_tensorboard_experiment_data"
+ ]
+ },
  "WriteTensorboardRunData": {
  "methods": [
  "write_tensorboard_run_data"

google/cloud/aiplatform_v1beta1/services/dataset_service/async_client.py

@@ -47,7 +47,9 @@
 
 
 class DatasetServiceAsyncClient:
- """"""
+ """The service that handles the CRUD of Vertex AI Dataset and
+ its child resources.
+ """
 
  _client: DatasetServiceClient
 

google/cloud/aiplatform_v1beta1/services/dataset_service/client.py

@@ -84,7 +84,9 @@ def get_transport_class(cls, label: str = None,) -> Type[DatasetServiceTransport
 
 
 class DatasetServiceClient(metaclass=DatasetServiceClientMeta):
- """"""
+ """The service that handles the CRUD of Vertex AI Dataset and
+ its child resources.
+ """
 
  @staticmethod
  def _get_default_mtls_endpoint(api_endpoint):

google/cloud/aiplatform_v1beta1/services/dataset_service/transports/grpc.py

@@ -36,6 +36,9 @@
 class DatasetServiceGrpcTransport(DatasetServiceTransport):
  """gRPC backend transport for DatasetService.
 
+ The service that handles the CRUD of Vertex AI Dataset and
+ its child resources.
+
  This class defines the same methods as the primary client, so the
  primary client can load the underlying transport implementation
  and call it.

google/cloud/aiplatform_v1beta1/services/dataset_service/transports/grpc_asyncio.py

@@ -38,6 +38,9 @@
 class DatasetServiceGrpcAsyncIOTransport(DatasetServiceTransport):
  """gRPC AsyncIO backend transport for DatasetService.
 
+ The service that handles the CRUD of Vertex AI Dataset and
+ its child resources.
+
  This class defines the same methods as the primary client, so the
  primary client can load the underlying transport implementation
  and call it.

google/cloud/aiplatform_v1beta1/services/endpoint_service/async_client.py

@@ -43,7 +43,7 @@
 
 
 class EndpointServiceAsyncClient:
- """"""
+ """A service for managing Vertex AI's Endpoints."""
 
  _client: EndpointServiceClient
 
@@ -54,6 +54,12 @@ class EndpointServiceAsyncClient:
  parse_endpoint_path = staticmethod(EndpointServiceClient.parse_endpoint_path)
  model_path = staticmethod(EndpointServiceClient.model_path)
  parse_model_path = staticmethod(EndpointServiceClient.parse_model_path)
+ model_deployment_monitoring_job_path = staticmethod(
+ EndpointServiceClient.model_deployment_monitoring_job_path
+ )
+ parse_model_deployment_monitoring_job_path = staticmethod(
+ EndpointServiceClient.parse_model_deployment_monitoring_job_path
+ )
  network_path = staticmethod(EndpointServiceClient.network_path)
  parse_network_path = staticmethod(EndpointServiceClient.parse_network_path)
  common_billing_account_path = staticmethod(

google/cloud/aiplatform_v1beta1/services/endpoint_service/client.py

@@ -80,7 +80,7 @@ def get_transport_class(cls, label: str = None,) -> Type[EndpointServiceTranspor
 
 
 class EndpointServiceClient(metaclass=EndpointServiceClientMeta):
- """"""
+ """A service for managing Vertex AI's Endpoints."""
 
  @staticmethod
  def _get_default_mtls_endpoint(api_endpoint):
@@ -196,6 +196,26 @@ def parse_model_path(path: str) -> Dict[str, str]:
  )
  return m.groupdict() if m else {}
 
+ @staticmethod
+ def model_deployment_monitoring_job_path(
+ project: str, location: str, model_deployment_monitoring_job: str,
+ ) -> str:
+ """Returns a fully-qualified model_deployment_monitoring_job string."""
+ return "projects/{project}/locations/{location}/modelDeploymentMonitoringJobs/{model_deployment_monitoring_job}".format(
+ project=project,
+ location=location,
+ model_deployment_monitoring_job=model_deployment_monitoring_job,
+ )
+
+ @staticmethod
+ def parse_model_deployment_monitoring_job_path(path: str) -> Dict[str, str]:
+ """Parses a model_deployment_monitoring_job path into its component segments."""
+ m = re.match(
+ r"^projects/(?P<project>.+?)/locations/(?P<location>.+?)/modelDeploymentMonitoringJobs/(?P<model_deployment_monitoring_job>.+?)$",
+ path,
+ )
+ return m.groupdict() if m else {}
+
  @staticmethod
  def network_path(project: str, network: str,) -> str:
  """Returns a fully-qualified network string."""

google/cloud/aiplatform_v1beta1/services/endpoint_service/transports/grpc.py

@@ -35,6 +35,8 @@
 class EndpointServiceGrpcTransport(EndpointServiceTransport):
  """gRPC backend transport for EndpointService.
 
+ A service for managing Vertex AI's Endpoints.
+
  This class defines the same methods as the primary client, so the
  primary client can load the underlying transport implementation
  and call it.

google/cloud/aiplatform_v1beta1/services/endpoint_service/transports/grpc_asyncio.py

@@ -37,6 +37,8 @@
 class EndpointServiceGrpcAsyncIOTransport(EndpointServiceTransport):
  """gRPC AsyncIO backend transport for EndpointService.
 
+ A service for managing Vertex AI's Endpoints.
+
  This class defines the same methods as the primary client, so the
  primary client can load the underlying transport implementation
  and call it.