feat: add protected_artifact_location_id to CustomJob

docs/aiplatform_v1/feature_online_store_admin_service.rst

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

docs/aiplatform_v1/feature_online_store_service.rst

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

docs/aiplatform_v1/feature_registry_service.rst

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

docs/aiplatform_v1/services_.rst

@@ -5,6 +5,9 @@ Services for Google Cloud Aiplatform v1 API
 
     dataset_service
     endpoint_service
+    feature_online_store_admin_service
+    feature_online_store_service
+    feature_registry_service
     featurestore_online_serving_service
     featurestore_service
     index_endpoint_service

google/cloud/aiplatform_v1/__init__.py

@@ -22,6 +22,16 @@
 from .services.dataset_service import DatasetServiceAsyncClient
 from .services.endpoint_service import EndpointServiceClient
 from .services.endpoint_service import EndpointServiceAsyncClient
+from .services.feature_online_store_admin_service import (
+    FeatureOnlineStoreAdminServiceClient,
+)
+from .services.feature_online_store_admin_service import (
+    FeatureOnlineStoreAdminServiceAsyncClient,
+)
+from .services.feature_online_store_service import FeatureOnlineStoreServiceClient
+from .services.feature_online_store_service import FeatureOnlineStoreServiceAsyncClient
+from .services.feature_registry_service import FeatureRegistryServiceClient
+from .services.feature_registry_service import FeatureRegistryServiceAsyncClient
 from .services.featurestore_online_serving_service import (
     FeaturestoreOnlineServingServiceClient,
 )
@@ -164,9 +174,52 @@
 from .types.explanation import XraiAttribution
 from .types.explanation_metadata import ExplanationMetadata
 from .types.feature import Feature
+from .types.feature_group import FeatureGroup
 from .types.feature_monitoring_stats import FeatureStatsAnomaly
+from .types.feature_online_store import FeatureOnlineStore
+from .types.feature_online_store_admin_service import (
+    CreateFeatureOnlineStoreOperationMetadata,
+)
+from .types.feature_online_store_admin_service import CreateFeatureOnlineStoreRequest
+from .types.feature_online_store_admin_service import CreateFeatureViewOperationMetadata
+from .types.feature_online_store_admin_service import CreateFeatureViewRequest
+from .types.feature_online_store_admin_service import DeleteFeatureOnlineStoreRequest
+from .types.feature_online_store_admin_service import DeleteFeatureViewRequest
+from .types.feature_online_store_admin_service import GetFeatureOnlineStoreRequest
+from .types.feature_online_store_admin_service import GetFeatureViewRequest
+from .types.feature_online_store_admin_service import GetFeatureViewSyncRequest
+from .types.feature_online_store_admin_service import ListFeatureOnlineStoresRequest
+from .types.feature_online_store_admin_service import ListFeatureOnlineStoresResponse
+from .types.feature_online_store_admin_service import ListFeatureViewsRequest
+from .types.feature_online_store_admin_service import ListFeatureViewsResponse
+from .types.feature_online_store_admin_service import ListFeatureViewSyncsRequest
+from .types.feature_online_store_admin_service import ListFeatureViewSyncsResponse
+from .types.feature_online_store_admin_service import SyncFeatureViewRequest
+from .types.feature_online_store_admin_service import SyncFeatureViewResponse
+from .types.feature_online_store_admin_service import (
+    UpdateFeatureOnlineStoreOperationMetadata,
+)
+from .types.feature_online_store_admin_service import UpdateFeatureOnlineStoreRequest
+from .types.feature_online_store_admin_service import UpdateFeatureViewOperationMetadata
+from .types.feature_online_store_admin_service import UpdateFeatureViewRequest
+from .types.feature_online_store_service import FeatureViewDataKey
+from .types.feature_online_store_service import FetchFeatureValuesRequest
+from .types.feature_online_store_service import FetchFeatureValuesResponse
+from .types.feature_online_store_service import FeatureViewDataFormat
+from .types.feature_registry_service import CreateFeatureGroupOperationMetadata
+from .types.feature_registry_service import CreateFeatureGroupRequest
+from .types.feature_registry_service import CreateRegistryFeatureOperationMetadata
+from .types.feature_registry_service import DeleteFeatureGroupRequest
+from .types.feature_registry_service import GetFeatureGroupRequest
+from .types.feature_registry_service import ListFeatureGroupsRequest
+from .types.feature_registry_service import ListFeatureGroupsResponse
+from .types.feature_registry_service import UpdateFeatureGroupOperationMetadata
+from .types.feature_registry_service import UpdateFeatureGroupRequest
+from .types.feature_registry_service import UpdateFeatureOperationMetadata
 from .types.feature_selector import FeatureSelector
 from .types.feature_selector import IdMatcher
+from .types.feature_view import FeatureView
+from .types.feature_view_sync import FeatureViewSync
 from .types.featurestore import Featurestore
 from .types.featurestore_monitoring import FeaturestoreMonitoringConfig
 from .types.featurestore_online_service import FeatureValue
@@ -470,6 +523,8 @@
 from .types.pipeline_service import ListTrainingPipelinesRequest
 from .types.pipeline_service import ListTrainingPipelinesResponse
 from .types.pipeline_state import PipelineState
+from .types.prediction_service import CountTokensRequest
+from .types.prediction_service import CountTokensResponse
 from .types.prediction_service import ExplainRequest
 from .types.prediction_service import ExplainResponse
 from .types.prediction_service import PredictRequest
@@ -501,6 +556,7 @@
 from .types.study import Measurement
 from .types.study import Study
 from .types.study import StudySpec
+from .types.study import StudyTimeConstraint
 from .types.study import Trial
 from .types.study import TrialContext
 from .types.tensorboard import Tensorboard
@@ -600,6 +656,9 @@
 __all__ = (
     "DatasetServiceAsyncClient",
     "EndpointServiceAsyncClient",
+    "FeatureOnlineStoreAdminServiceAsyncClient",
+    "FeatureOnlineStoreServiceAsyncClient",
+    "FeatureRegistryServiceAsyncClient",
     "FeaturestoreOnlineServingServiceAsyncClient",
     "FeaturestoreServiceAsyncClient",
     "IndexEndpointServiceAsyncClient",
@@ -675,6 +734,8 @@
     "CopyModelOperationMetadata",
     "CopyModelRequest",
     "CopyModelResponse",
+    "CountTokensRequest",
+    "CountTokensResponse",
     "CreateArtifactRequest",
     "CreateBatchPredictionJobRequest",
     "CreateContextRequest",
@@ -689,8 +750,14 @@
     "CreateEntityTypeOperationMetadata",
     "CreateEntityTypeRequest",
     "CreateExecutionRequest",
+    "CreateFeatureGroupOperationMetadata",
+    "CreateFeatureGroupRequest",
+    "CreateFeatureOnlineStoreOperationMetadata",
+    "CreateFeatureOnlineStoreRequest",
     "CreateFeatureOperationMetadata",
     "CreateFeatureRequest",
+    "CreateFeatureViewOperationMetadata",
+    "CreateFeatureViewRequest",
     "CreateFeaturestoreOperationMetadata",
     "CreateFeaturestoreRequest",
     "CreateHyperparameterTuningJobRequest",
@@ -704,6 +771,7 @@
     "CreateModelDeploymentMonitoringJobRequest",
     "CreateNasJobRequest",
     "CreatePipelineJobRequest",
+    "CreateRegistryFeatureOperationMetadata",
     "CreateScheduleRequest",
     "CreateSpecialistPoolOperationMetadata",
     "CreateSpecialistPoolRequest",
@@ -736,10 +804,13 @@
     "DeleteEndpointRequest",
     "DeleteEntityTypeRequest",
     "DeleteExecutionRequest",
+    "DeleteFeatureGroupRequest",
+    "DeleteFeatureOnlineStoreRequest",
     "DeleteFeatureRequest",
     "DeleteFeatureValuesOperationMetadata",
     "DeleteFeatureValuesRequest",
     "DeleteFeatureValuesResponse",
+    "DeleteFeatureViewRequest",
     "DeleteFeaturestoreRequest",
     "DeleteHyperparameterTuningJobRequest",
     "DeleteIndexEndpointRequest",
@@ -812,16 +883,27 @@
     "ExportTensorboardTimeSeriesDataRequest",
     "ExportTensorboardTimeSeriesDataResponse",
     "Feature",
+    "FeatureGroup",
     "FeatureNoiseSigma",
+    "FeatureOnlineStore",
+    "FeatureOnlineStoreAdminServiceClient",
+    "FeatureOnlineStoreServiceClient",
+    "FeatureRegistryServiceClient",
     "FeatureSelector",
     "FeatureStatsAnomaly",
     "FeatureValue",
     "FeatureValueDestination",
     "FeatureValueList",
+    "FeatureView",
+    "FeatureViewDataFormat",
+    "FeatureViewDataKey",
+    "FeatureViewSync",
     "Featurestore",
     "FeaturestoreMonitoringConfig",
     "FeaturestoreOnlineServingServiceClient",
     "FeaturestoreServiceClient",
+    "FetchFeatureValuesRequest",
+    "FetchFeatureValuesResponse",
     "FilterSplit",
     "FindNeighborsRequest",
     "FindNeighborsResponse",
@@ -840,7 +922,11 @@
     "GetEndpointRequest",
     "GetEntityTypeRequest",
     "GetExecutionRequest",
+    "GetFeatureGroupRequest",
+    "GetFeatureOnlineStoreRequest",
     "GetFeatureRequest",
+    "GetFeatureViewRequest",
+    "GetFeatureViewSyncRequest",
     "GetFeaturestoreRequest",
     "GetHyperparameterTuningJobRequest",
     "GetIndexEndpointRequest",
@@ -912,6 +998,14 @@
     "ListEntityTypesResponse",
     "ListExecutionsRequest",
     "ListExecutionsResponse",
+    "ListFeatureGroupsRequest",
+    "ListFeatureGroupsResponse",
+    "ListFeatureOnlineStoresRequest",
+    "ListFeatureOnlineStoresResponse",
+    "ListFeatureViewSyncsRequest",
+    "ListFeatureViewSyncsResponse",
+    "ListFeatureViewsRequest",
+    "ListFeatureViewsResponse",
     "ListFeaturesRequest",
     "ListFeaturesResponse",
     "ListFeaturestoresRequest",
@@ -1092,9 +1186,12 @@
     "StringArray",
     "Study",
     "StudySpec",
+    "StudyTimeConstraint",
     "SuggestTrialsMetadata",
     "SuggestTrialsRequest",
     "SuggestTrialsResponse",
+    "SyncFeatureViewRequest",
+    "SyncFeatureViewResponse",
     "TFRecordDestination",
     "Tensor",
     "Tensorboard",
@@ -1129,7 +1226,14 @@
     "UpdateExplanationDatasetOperationMetadata",
     "UpdateExplanationDatasetRequest",
     "UpdateExplanationDatasetResponse",
+    "UpdateFeatureGroupOperationMetadata",
+    "UpdateFeatureGroupRequest",
+    "UpdateFeatureOnlineStoreOperationMetadata",
+    "UpdateFeatureOnlineStoreRequest",
+    "UpdateFeatureOperationMetadata",
     "UpdateFeatureRequest",
+    "UpdateFeatureViewOperationMetadata",
+    "UpdateFeatureViewRequest",
     "UpdateFeaturestoreOperationMetadata",
     "UpdateFeaturestoreRequest",
     "UpdateIndexEndpointRequest",