feat: add featurestore service to aiplatform v1 (#765)

* feat: add featurestore service to aiplatform v1 feat: add metadata service to aiplatform v1

Committer: @dizcology
PiperOrigin-RevId: 402573132

Source-Link: googleapis/googleapis@d706102

Source-Link: googleapis/googleapis-gen@3e9242f
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiM2U5MjQyZjY1YmUzYTA3MjVjMGNlZDU2YWUzNDFiOGIwMmMxODliNyJ9

* 🦉 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>
Co-authored-by: Yu-Han Liu <yuhanliu@google.com>

docs/aiplatform_v1/featurestore_online_serving_service.rst

@@ -0,0 +1,6 @@
+FeaturestoreOnlineServingService
+--------------------------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1.services.featurestore_online_serving_service
+    :members:
+    :inherited-members:

docs/aiplatform_v1/featurestore_service.rst

@@ -0,0 +1,10 @@
+FeaturestoreService
+-------------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1.services.featurestore_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.aiplatform_v1.services.featurestore_service.pagers
+    :members:
+    :inherited-members:

docs/aiplatform_v1/metadata_service.rst

@@ -0,0 +1,10 @@
+MetadataService
+---------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1.services.metadata_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.aiplatform_v1.services.metadata_service.pagers
+    :members:
+    :inherited-members:

docs/aiplatform_v1/services.rst

@@ -5,9 +5,12 @@ Services for Google Cloud Aiplatform v1 API
 
     dataset_service
     endpoint_service
+    featurestore_online_serving_service
+    featurestore_service
     index_endpoint_service
     index_service
     job_service
+    metadata_service
     migration_service
     model_service
     pipeline_service

google/cloud/aiplatform/v1/schema/predict/instance_v1/types/image_classification.py

@@ -24,6 +24,7 @@
 
 class ImageClassificationPredictionInstance(proto.Message):
     r"""Prediction input format for Image Classification.
+
     Attributes:
         content (str):
             The image bytes or Cloud Storage URI to make

google/cloud/aiplatform/v1/schema/predict/instance_v1/types/image_object_detection.py

@@ -24,6 +24,7 @@
 
 class ImageObjectDetectionPredictionInstance(proto.Message):
     r"""Prediction input format for Image Object Detection.
+
     Attributes:
         content (str):
             The image bytes or Cloud Storage URI to make

google/cloud/aiplatform/v1/schema/predict/instance_v1/types/image_segmentation.py

@@ -24,6 +24,7 @@
 
 class ImageSegmentationPredictionInstance(proto.Message):
     r"""Prediction input format for Image Segmentation.
+
     Attributes:
         content (str):
             The image bytes to make the predictions on.

google/cloud/aiplatform/v1/schema/predict/instance_v1/types/text_classification.py

@@ -24,6 +24,7 @@
 
 class TextClassificationPredictionInstance(proto.Message):
     r"""Prediction input format for Text Classification.
+
     Attributes:
         content (str):
             The text snippet to make the predictions on.

google/cloud/aiplatform/v1/schema/predict/instance_v1/types/text_extraction.py

@@ -24,6 +24,7 @@
 
 class TextExtractionPredictionInstance(proto.Message):
     r"""Prediction input format for Text Extraction.
+
     Attributes:
         content (str):
             The text snippet to make the predictions on.

google/cloud/aiplatform/v1/schema/predict/instance_v1/types/text_sentiment.py

@@ -24,6 +24,7 @@
 
 class TextSentimentPredictionInstance(proto.Message):
     r"""Prediction input format for Text Sentiment.
+
     Attributes:
         content (str):
             The text snippet to make the predictions on.

google/cloud/aiplatform/v1/schema/predict/instance_v1/types/video_action_recognition.py

@@ -24,6 +24,7 @@
 
 class VideoActionRecognitionPredictionInstance(proto.Message):
     r"""Prediction input format for Video Action Recognition.
+
     Attributes:
         content (str):
             The Google Cloud Storage location of the

google/cloud/aiplatform/v1/schema/predict/instance_v1/types/video_classification.py

@@ -24,6 +24,7 @@
 
 class VideoClassificationPredictionInstance(proto.Message):
     r"""Prediction input format for Video Classification.
+
     Attributes:
         content (str):
             The Google Cloud Storage location of the

google/cloud/aiplatform/v1/schema/predict/instance_v1/types/video_object_tracking.py

@@ -24,6 +24,7 @@
 
 class VideoObjectTrackingPredictionInstance(proto.Message):
     r"""Prediction input format for Video Object Tracking.
+
     Attributes:
         content (str):
             The Google Cloud Storage location of the

google/cloud/aiplatform/v1/schema/predict/params_v1/types/image_classification.py

@@ -24,6 +24,7 @@
 
 class ImageClassificationPredictionParams(proto.Message):
     r"""Prediction model parameters for Image Classification.
+
     Attributes:
         confidence_threshold (float):
             The Model only returns predictions with at

google/cloud/aiplatform/v1/schema/predict/params_v1/types/image_object_detection.py

@@ -24,6 +24,7 @@
 
 class ImageObjectDetectionPredictionParams(proto.Message):
     r"""Prediction model parameters for Image Object Detection.
+
     Attributes:
         confidence_threshold (float):
             The Model only returns predictions with at

google/cloud/aiplatform/v1/schema/predict/params_v1/types/image_segmentation.py

@@ -24,6 +24,7 @@
 
 class ImageSegmentationPredictionParams(proto.Message):
     r"""Prediction model parameters for Image Segmentation.
+
     Attributes:
         confidence_threshold (float):
             When the model predicts category of pixels of

google/cloud/aiplatform/v1/schema/predict/params_v1/types/video_action_recognition.py

@@ -24,6 +24,7 @@
 
 class VideoActionRecognitionPredictionParams(proto.Message):
     r"""Prediction model parameters for Video Action Recognition.
+
     Attributes:
         confidence_threshold (float):
             The Model only returns predictions with at

google/cloud/aiplatform/v1/schema/predict/params_v1/types/video_classification.py

@@ -24,6 +24,7 @@
 
 class VideoClassificationPredictionParams(proto.Message):
     r"""Prediction model parameters for Video Classification.
+
     Attributes:
         confidence_threshold (float):
             The Model only returns predictions with at

google/cloud/aiplatform/v1/schema/predict/params_v1/types/video_object_tracking.py

@@ -24,6 +24,7 @@
 
 class VideoObjectTrackingPredictionParams(proto.Message):
     r"""Prediction model parameters for Video Object Tracking.
+
     Attributes:
         confidence_threshold (float):
             The Model only returns predictions with at

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/classification.py

@@ -24,6 +24,7 @@
 
 class ClassificationPredictionResult(proto.Message):
     r"""Prediction output format for Image and Text Classification.
+
     Attributes:
         ids (Sequence[int]):
             The resource IDs of the AnnotationSpecs that

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/image_object_detection.py

@@ -26,6 +26,7 @@
 
 class ImageObjectDetectionPredictionResult(proto.Message):
     r"""Prediction output format for Image Object Detection.
+
     Attributes:
         ids (Sequence[int]):
             The resource IDs of the AnnotationSpecs that

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/image_segmentation.py

@@ -24,6 +24,7 @@
 
 class ImageSegmentationPredictionResult(proto.Message):
     r"""Prediction output format for Image Segmentation.
+
     Attributes:
         category_mask (str):
             A PNG image where each pixel in the mask

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/tabular_classification.py

@@ -24,6 +24,7 @@
 
 class TabularClassificationPredictionResult(proto.Message):
     r"""Prediction output format for Tabular Classification.
+
     Attributes:
         classes (Sequence[str]):
             The name of the classes being classified,

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/tabular_regression.py

@@ -24,6 +24,7 @@
 
 class TabularRegressionPredictionResult(proto.Message):
     r"""Prediction output format for Tabular Regression.
+
     Attributes:
         value (float):
             The regression value.

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/text_extraction.py

@@ -24,6 +24,7 @@
 
 class TextExtractionPredictionResult(proto.Message):
     r"""Prediction output format for Text Extraction.
+
     Attributes:
         ids (Sequence[int]):
             The resource IDs of the AnnotationSpecs that

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/text_sentiment.py

@@ -24,6 +24,7 @@
 
 class TextSentimentPredictionResult(proto.Message):
     r"""Prediction output format for Text Sentiment
+
     Attributes:
         sentiment (int):
             The integer sentiment labels between 0

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/video_action_recognition.py

@@ -27,6 +27,7 @@
 
 class VideoActionRecognitionPredictionResult(proto.Message):
     r"""Prediction output format for Video Action Recognition.
+
     Attributes:
         id (str):
             The resource ID of the AnnotationSpec that

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/video_classification.py

@@ -27,6 +27,7 @@
 
 class VideoClassificationPredictionResult(proto.Message):
     r"""Prediction output format for Video Classification.
+
     Attributes:
         id (str):
             The resource ID of the AnnotationSpec that

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/video_object_tracking.py

@@ -27,6 +27,7 @@
 
 class VideoObjectTrackingPredictionResult(proto.Message):
     r"""Prediction output format for Video Object Tracking.
+
     Attributes:
         id (str):
             The resource ID of the AnnotationSpec that

google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/automl_image_classification.py

@@ -47,6 +47,7 @@ class AutoMlImageClassification(proto.Message):
 
 class AutoMlImageClassificationInputs(proto.Message):
     r"""
+
     Attributes:
         model_type (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlImageClassificationInputs.ModelType):
 
@@ -106,6 +107,7 @@ class ModelType(proto.Enum):
 
 class AutoMlImageClassificationMetadata(proto.Message):
     r"""
+
     Attributes:
         cost_milli_node_hours (int):
             The actual training cost of creating this

google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/automl_image_object_detection.py

@@ -47,6 +47,7 @@ class AutoMlImageObjectDetection(proto.Message):
 
 class AutoMlImageObjectDetectionInputs(proto.Message):
     r"""
+
     Attributes:
         model_type (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlImageObjectDetectionInputs.ModelType):
 
@@ -93,6 +94,7 @@ class ModelType(proto.Enum):
 
 class AutoMlImageObjectDetectionMetadata(proto.Message):
     r"""
+
     Attributes:
         cost_milli_node_hours (int):
             The actual training cost of creating this

google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/automl_image_segmentation.py

@@ -47,6 +47,7 @@ class AutoMlImageSegmentation(proto.Message):
 
 class AutoMlImageSegmentationInputs(proto.Message):
     r"""
+
     Attributes:
         model_type (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlImageSegmentationInputs.ModelType):
 
@@ -87,6 +88,7 @@ class ModelType(proto.Enum):
 
 class AutoMlImageSegmentationMetadata(proto.Message):
     r"""
+
     Attributes:
         cost_milli_node_hours (int):
             The actual training cost of creating this

google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/automl_tables.py

@@ -28,6 +28,7 @@
 
 class AutoMlTables(proto.Message):
     r"""A TrainingJob that trains and uploads an AutoML Tables Model.
+
     Attributes:
         inputs (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlTablesInputs):
             The input parameters of this TrainingJob.
@@ -41,6 +42,7 @@ class AutoMlTables(proto.Message):
 
 class AutoMlTablesInputs(proto.Message):
     r"""
+
     Attributes:
         optimization_objective_recall_value (float):
             Required when optimization_objective is
@@ -146,6 +148,7 @@ class AutoMlTablesInputs(proto.Message):
 
     class Transformation(proto.Message):
         r"""
+
         Attributes:
             auto (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlTablesInputs.Transformation.AutoTransformation):
 
@@ -409,6 +412,7 @@ class TextArrayTransformation(proto.Message):
 
 class AutoMlTablesMetadata(proto.Message):
     r"""Model metadata specific to AutoML Tables.
+
     Attributes:
         train_cost_milli_node_hours (int):
             Output only. The actual training cost of the

google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/automl_text_classification.py

@@ -38,6 +38,7 @@ class AutoMlTextClassification(proto.Message):
 
 class AutoMlTextClassificationInputs(proto.Message):
     r"""
+
     Attributes:
         multi_label (bool):
 

google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/automl_text_extraction.py

@@ -35,7 +35,8 @@ class AutoMlTextExtraction(proto.Message):
 
 
 class AutoMlTextExtractionInputs(proto.Message):
-    r"""    """
+    r"""
+    """
 
 
 __all__ = tuple(sorted(__protobuf__.manifest))

google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/automl_text_sentiment.py

@@ -36,6 +36,7 @@ class AutoMlTextSentiment(proto.Message):
 
 class AutoMlTextSentimentInputs(proto.Message):
     r"""
+
     Attributes:
         sentiment_max (int):
             A sentiment is expressed as an integer

google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/automl_video_action_recognition.py

@@ -38,6 +38,7 @@ class AutoMlVideoActionRecognition(proto.Message):
 
 class AutoMlVideoActionRecognitionInputs(proto.Message):
     r"""
+
     Attributes:
         model_type (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlVideoActionRecognitionInputs.ModelType):
 

google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/automl_video_classification.py

@@ -38,6 +38,7 @@ class AutoMlVideoClassification(proto.Message):
 
 class AutoMlVideoClassificationInputs(proto.Message):
     r"""
+
     Attributes:
         model_type (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlVideoClassificationInputs.ModelType):
 

google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/automl_video_object_tracking.py

@@ -38,6 +38,7 @@ class AutoMlVideoObjectTracking(proto.Message):
 
 class AutoMlVideoObjectTrackingInputs(proto.Message):
     r"""
+
     Attributes:
         model_type (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlVideoObjectTrackingInputs.ModelType):
 

google/cloud/aiplatform/v1beta1/schema/predict/instance_v1beta1/types/image_classification.py

@@ -24,6 +24,7 @@
 
 class ImageClassificationPredictionInstance(proto.Message):
     r"""Prediction input format for Image Classification.
+
     Attributes:
         content (str):
             The image bytes or Cloud Storage URI to make

google/cloud/aiplatform/v1beta1/schema/predict/instance_v1beta1/types/image_object_detection.py

@@ -24,6 +24,7 @@
 
 class ImageObjectDetectionPredictionInstance(proto.Message):
     r"""Prediction input format for Image Object Detection.
+
     Attributes:
         content (str):
             The image bytes or Cloud Storage URI to make

google/cloud/aiplatform/v1beta1/schema/predict/instance_v1beta1/types/image_segmentation.py

@@ -24,6 +24,7 @@
 
 class ImageSegmentationPredictionInstance(proto.Message):
     r"""Prediction input format for Image Segmentation.
+
     Attributes:
         content (str):
             The image bytes to make the predictions on.