diff --git a/google/cloud/aiplatform_v1/services/endpoint_service/async_client.py b/google/cloud/aiplatform_v1/services/endpoint_service/async_client.py index 87f9a30f16..d5d02ecfa9 100644 --- a/google/cloud/aiplatform_v1/services/endpoint_service/async_client.py +++ b/google/cloud/aiplatform_v1/services/endpoint_service/async_client.py @@ -306,11 +306,19 @@ async def sample_create_endpoint(): not provided, Vertex AI will generate a value for this ID. - This value should be 1-10 characters, and valid - characters are /[0-9]/. When using HTTP/JSON, this field - is populated based on a query string argument, such as - ``?endpoint_id=12345``. This is the fallback for fields - that are not included in either the URI or the body. + If the first character is a letter, this value may be up + to 63 characters, and valid characters are + ``[a-z0-9-]``. The last character must be a letter or + number. + + If the first character is a number, this value may be up + to 9 characters, and valid characters are ``[0-9]`` with + no leading zeros. + + When using HTTP/JSON, this field is populated based on a + query string argument, such as ``?endpoint_id=12345``. + This is the fallback for fields that are not included in + either the URI or the body. This corresponds to the ``endpoint_id`` field on the ``request`` instance; if ``request`` is provided, this diff --git a/google/cloud/aiplatform_v1/services/endpoint_service/client.py b/google/cloud/aiplatform_v1/services/endpoint_service/client.py index ac7e7b97fb..3a410f8ced 100644 --- a/google/cloud/aiplatform_v1/services/endpoint_service/client.py +++ b/google/cloud/aiplatform_v1/services/endpoint_service/client.py @@ -586,11 +586,19 @@ def sample_create_endpoint(): not provided, Vertex AI will generate a value for this ID. - This value should be 1-10 characters, and valid - characters are /[0-9]/. When using HTTP/JSON, this field - is populated based on a query string argument, such as - ``?endpoint_id=12345``. This is the fallback for fields - that are not included in either the URI or the body. + If the first character is a letter, this value may be up + to 63 characters, and valid characters are + ``[a-z0-9-]``. The last character must be a letter or + number. + + If the first character is a number, this value may be up + to 9 characters, and valid characters are ``[0-9]`` with + no leading zeros. + + When using HTTP/JSON, this field is populated based on a + query string argument, such as ``?endpoint_id=12345``. + This is the fallback for fields that are not included in + either the URI or the body. This corresponds to the ``endpoint_id`` field on the ``request`` instance; if ``request`` is provided, this diff --git a/google/cloud/aiplatform_v1/services/migration_service/client.py b/google/cloud/aiplatform_v1/services/migration_service/client.py index c42427af48..b43dcee16e 100644 --- a/google/cloud/aiplatform_v1/services/migration_service/client.py +++ b/google/cloud/aiplatform_v1/services/migration_service/client.py @@ -230,40 +230,40 @@ def parse_dataset_path(path: str) -> Dict[str, str]: @staticmethod def dataset_path( project: str, + location: str, dataset: str, ) -> str: """Returns a fully-qualified dataset string.""" - return "projects/{project}/datasets/{dataset}".format( + return "projects/{project}/locations/{location}/datasets/{dataset}".format( project=project, + location=location, dataset=dataset, ) @staticmethod def parse_dataset_path(path: str) -> Dict[str, str]: """Parses a dataset path into its component segments.""" - m = re.match(r"^projects/(?P.+?)/datasets/(?P.+?)$", path) + m = re.match( + r"^projects/(?P.+?)/locations/(?P.+?)/datasets/(?P.+?)$", + path, + ) return m.groupdict() if m else {} @staticmethod def dataset_path( project: str, - location: str, dataset: str, ) -> str: """Returns a fully-qualified dataset string.""" - return "projects/{project}/locations/{location}/datasets/{dataset}".format( + return "projects/{project}/datasets/{dataset}".format( project=project, - location=location, dataset=dataset, ) @staticmethod def parse_dataset_path(path: str) -> Dict[str, str]: """Parses a dataset path into its component segments.""" - m = re.match( - r"^projects/(?P.+?)/locations/(?P.+?)/datasets/(?P.+?)$", - path, - ) + m = re.match(r"^projects/(?P.+?)/datasets/(?P.+?)$", path) return m.groupdict() if m else {} @staticmethod diff --git a/google/cloud/aiplatform_v1/types/accelerator_type.py b/google/cloud/aiplatform_v1/types/accelerator_type.py index 7d265d6234..d9d8da0546 100644 --- a/google/cloud/aiplatform_v1/types/accelerator_type.py +++ b/google/cloud/aiplatform_v1/types/accelerator_type.py @@ -30,7 +30,6 @@ class AcceleratorType(proto.Enum): r"""Represents a hardware accelerator type. - NEXT ID: 11. Values: ACCELERATOR_TYPE_UNSPECIFIED (0): diff --git a/google/cloud/aiplatform_v1/types/endpoint_service.py b/google/cloud/aiplatform_v1/types/endpoint_service.py index 0dc0a48694..3abfeff411 100644 --- a/google/cloud/aiplatform_v1/types/endpoint_service.py +++ b/google/cloud/aiplatform_v1/types/endpoint_service.py @@ -60,11 +60,18 @@ class CreateEndpointRequest(proto.Message): final component of the endpoint resource name. If not provided, Vertex AI will generate a value for this ID. - This value should be 1-10 characters, and valid characters - are /[0-9]/. When using HTTP/JSON, this field is populated - based on a query string argument, such as - ``?endpoint_id=12345``. This is the fallback for fields that - are not included in either the URI or the body. + If the first character is a letter, this value may be up to + 63 characters, and valid characters are ``[a-z0-9-]``. The + last character must be a letter or number. + + If the first character is a number, this value may be up to + 9 characters, and valid characters are ``[0-9]`` with no + leading zeros. + + When using HTTP/JSON, this field is populated based on a + query string argument, such as ``?endpoint_id=12345``. This + is the fallback for fields that are not included in either + the URI or the body. """ parent: str = proto.Field( diff --git a/google/cloud/aiplatform_v1/types/featurestore_service.py b/google/cloud/aiplatform_v1/types/featurestore_service.py index 3265420c49..0f60deadcc 100644 --- a/google/cloud/aiplatform_v1/types/featurestore_service.py +++ b/google/cloud/aiplatform_v1/types/featurestore_service.py @@ -553,11 +553,8 @@ class BatchReadFeatureValuesRequest(proto.Message): will be automatically inferred. For CSV source, the pass-through values will be passed as opaque bytes. entity_type_specs (MutableSequence[google.cloud.aiplatform_v1.types.BatchReadFeatureValuesRequest.EntityTypeSpec]): - Required. Specifies EntityType grouping Features to read - values of and settings. Each EntityType referenced in - [BatchReadFeatureValuesRequest.entity_type_specs] must have - a column specifying entity IDs in the EntityType in - [BatchReadFeatureValuesRequest.request][] . + Required. Specifies EntityType grouping + Features to read values of and settings. start_time (google.protobuf.timestamp_pb2.Timestamp): Optional. Excludes Feature values with feature generation timestamp before this diff --git a/google/cloud/aiplatform_v1/types/index_endpoint.py b/google/cloud/aiplatform_v1/types/index_endpoint.py index fcf248d06f..232a859c1f 100644 --- a/google/cloud/aiplatform_v1/types/index_endpoint.py +++ b/google/cloud/aiplatform_v1/types/index_endpoint.py @@ -111,6 +111,14 @@ class IndexEndpoint(proto.Message): and [private_service_connect_config][google.cloud.aiplatform.v1.IndexEndpoint.private_service_connect_config] are mutually exclusive. + public_endpoint_enabled (bool): + Optional. If true, the deployed index will be + accessible through public endpoint. + public_endpoint_domain_name (str): + Output only. If + [public_endpoint_enabled][google.cloud.aiplatform.v1.IndexEndpoint.public_endpoint_enabled] + is true, this field will be populated with the domain name + to use for this index endpoint. """ name: str = proto.Field( @@ -164,6 +172,14 @@ class IndexEndpoint(proto.Message): message=service_networking.PrivateServiceConnectConfig, ) ) + public_endpoint_enabled: bool = proto.Field( + proto.BOOL, + number=13, + ) + public_endpoint_domain_name: str = proto.Field( + proto.STRING, + number=14, + ) class DeployedIndex(proto.Message): @@ -264,7 +280,7 @@ class DeployedIndex(proto.Message): be deployed to any ip ranges under the provided VPC network. - The value sohuld be the name of the address + The value should be the name of the address (https://cloud.google.com/compute/docs/reference/rest/v1/addresses) Example: 'vertex-ai-ip-range'. deployment_group (str): diff --git a/google/cloud/aiplatform_v1/types/model.py b/google/cloud/aiplatform_v1/types/model.py index 48d16c4da7..d55d87f006 100644 --- a/google/cloud/aiplatform_v1/types/model.py +++ b/google/cloud/aiplatform_v1/types/model.py @@ -899,6 +899,10 @@ class ModelSourceInfo(proto.Message): Attributes: source_type (google.cloud.aiplatform_v1.types.ModelSourceInfo.ModelSourceType): Type of the model source. + copy (bool): + If this Model is copy of another Model. If true then + [source_type][google.cloud.aiplatform.v1.ModelSourceInfo.source_type] + pertains to the original. """ class ModelSourceType(proto.Enum): @@ -916,17 +920,25 @@ class ModelSourceType(proto.Enum): BQML (3): The Model is registered and sync'ed from BigQuery ML. + MODEL_GARDEN (4): + The Model is saved or tuned from Model + Garden. """ MODEL_SOURCE_TYPE_UNSPECIFIED = 0 AUTOML = 1 CUSTOM = 2 BQML = 3 + MODEL_GARDEN = 4 source_type: ModelSourceType = proto.Field( proto.ENUM, number=1, enum=ModelSourceType, ) + copy: bool = proto.Field( + proto.BOOL, + number=2, + ) __all__ = tuple(sorted(__protobuf__.manifest)) diff --git a/google/cloud/aiplatform_v1beta1/services/endpoint_service/async_client.py b/google/cloud/aiplatform_v1beta1/services/endpoint_service/async_client.py index fec395b001..0ebf7b7d92 100644 --- a/google/cloud/aiplatform_v1beta1/services/endpoint_service/async_client.py +++ b/google/cloud/aiplatform_v1beta1/services/endpoint_service/async_client.py @@ -312,11 +312,19 @@ async def sample_create_endpoint(): not provided, Vertex AI will generate a value for this ID. - This value should be 1-10 characters, and valid - characters are /[0-9]/. When using HTTP/JSON, this field - is populated based on a query string argument, such as - ``?endpoint_id=12345``. This is the fallback for fields - that are not included in either the URI or the body. + If the first character is a letter, this value may be up + to 63 characters, and valid characters are + ``[a-z0-9-]``. The last character must be a letter or + number. + + If the first character is a number, this value may be up + to 9 characters, and valid characters are ``[0-9]`` with + no leading zeros. + + When using HTTP/JSON, this field is populated based on a + query string argument, such as ``?endpoint_id=12345``. + This is the fallback for fields that are not included in + either the URI or the body. This corresponds to the ``endpoint_id`` field on the ``request`` instance; if ``request`` is provided, this diff --git a/google/cloud/aiplatform_v1beta1/services/endpoint_service/client.py b/google/cloud/aiplatform_v1beta1/services/endpoint_service/client.py index bede58c48d..46a2ebf223 100644 --- a/google/cloud/aiplatform_v1beta1/services/endpoint_service/client.py +++ b/google/cloud/aiplatform_v1beta1/services/endpoint_service/client.py @@ -608,11 +608,19 @@ def sample_create_endpoint(): not provided, Vertex AI will generate a value for this ID. - This value should be 1-10 characters, and valid - characters are /[0-9]/. When using HTTP/JSON, this field - is populated based on a query string argument, such as - ``?endpoint_id=12345``. This is the fallback for fields - that are not included in either the URI or the body. + If the first character is a letter, this value may be up + to 63 characters, and valid characters are + ``[a-z0-9-]``. The last character must be a letter or + number. + + If the first character is a number, this value may be up + to 9 characters, and valid characters are ``[0-9]`` with + no leading zeros. + + When using HTTP/JSON, this field is populated based on a + query string argument, such as ``?endpoint_id=12345``. + This is the fallback for fields that are not included in + either the URI or the body. This corresponds to the ``endpoint_id`` field on the ``request`` instance; if ``request`` is provided, this diff --git a/google/cloud/aiplatform_v1beta1/services/migration_service/client.py b/google/cloud/aiplatform_v1beta1/services/migration_service/client.py index 97bac6325c..5724af97a3 100644 --- a/google/cloud/aiplatform_v1beta1/services/migration_service/client.py +++ b/google/cloud/aiplatform_v1beta1/services/migration_service/client.py @@ -208,40 +208,40 @@ def parse_annotated_dataset_path(path: str) -> Dict[str, str]: @staticmethod def dataset_path( project: str, - location: str, dataset: str, ) -> str: """Returns a fully-qualified dataset string.""" - return "projects/{project}/locations/{location}/datasets/{dataset}".format( + return "projects/{project}/datasets/{dataset}".format( project=project, - location=location, dataset=dataset, ) @staticmethod def parse_dataset_path(path: str) -> Dict[str, str]: """Parses a dataset path into its component segments.""" - m = re.match( - r"^projects/(?P.+?)/locations/(?P.+?)/datasets/(?P.+?)$", - path, - ) + m = re.match(r"^projects/(?P.+?)/datasets/(?P.+?)$", path) return m.groupdict() if m else {} @staticmethod def dataset_path( project: str, + location: str, dataset: str, ) -> str: """Returns a fully-qualified dataset string.""" - return "projects/{project}/datasets/{dataset}".format( + return "projects/{project}/locations/{location}/datasets/{dataset}".format( project=project, + location=location, dataset=dataset, ) @staticmethod def parse_dataset_path(path: str) -> Dict[str, str]: """Parses a dataset path into its component segments.""" - m = re.match(r"^projects/(?P.+?)/datasets/(?P.+?)$", path) + m = re.match( + r"^projects/(?P.+?)/locations/(?P.+?)/datasets/(?P.+?)$", + path, + ) return m.groupdict() if m else {} @staticmethod diff --git a/google/cloud/aiplatform_v1beta1/types/accelerator_type.py b/google/cloud/aiplatform_v1beta1/types/accelerator_type.py index 7e375edad6..72b8f96d6b 100644 --- a/google/cloud/aiplatform_v1beta1/types/accelerator_type.py +++ b/google/cloud/aiplatform_v1beta1/types/accelerator_type.py @@ -30,7 +30,6 @@ class AcceleratorType(proto.Enum): r"""Represents a hardware accelerator type. - NEXT ID: 11. Values: ACCELERATOR_TYPE_UNSPECIFIED (0): diff --git a/google/cloud/aiplatform_v1beta1/types/endpoint_service.py b/google/cloud/aiplatform_v1beta1/types/endpoint_service.py index b46155dd22..bca77343c7 100644 --- a/google/cloud/aiplatform_v1beta1/types/endpoint_service.py +++ b/google/cloud/aiplatform_v1beta1/types/endpoint_service.py @@ -60,11 +60,18 @@ class CreateEndpointRequest(proto.Message): final component of the endpoint resource name. If not provided, Vertex AI will generate a value for this ID. - This value should be 1-10 characters, and valid characters - are /[0-9]/. When using HTTP/JSON, this field is populated - based on a query string argument, such as - ``?endpoint_id=12345``. This is the fallback for fields that - are not included in either the URI or the body. + If the first character is a letter, this value may be up to + 63 characters, and valid characters are ``[a-z0-9-]``. The + last character must be a letter or number. + + If the first character is a number, this value may be up to + 9 characters, and valid characters are ``[0-9]`` with no + leading zeros. + + When using HTTP/JSON, this field is populated based on a + query string argument, such as ``?endpoint_id=12345``. This + is the fallback for fields that are not included in either + the URI or the body. """ parent: str = proto.Field( diff --git a/google/cloud/aiplatform_v1beta1/types/featurestore_service.py b/google/cloud/aiplatform_v1beta1/types/featurestore_service.py index 28f48c09eb..f351a0d137 100644 --- a/google/cloud/aiplatform_v1beta1/types/featurestore_service.py +++ b/google/cloud/aiplatform_v1beta1/types/featurestore_service.py @@ -555,11 +555,8 @@ class BatchReadFeatureValuesRequest(proto.Message): will be automatically inferred. For CSV source, the pass-through values will be passed as opaque bytes. entity_type_specs (MutableSequence[google.cloud.aiplatform_v1beta1.types.BatchReadFeatureValuesRequest.EntityTypeSpec]): - Required. Specifies EntityType grouping Features to read - values of and settings. Each EntityType referenced in - [BatchReadFeatureValuesRequest.entity_type_specs] must have - a column specifying entity IDs in the EntityType in - [BatchReadFeatureValuesRequest.request][] . + Required. Specifies EntityType grouping + Features to read values of and settings. start_time (google.protobuf.timestamp_pb2.Timestamp): Optional. Excludes Feature values with feature generation timestamp before this diff --git a/google/cloud/aiplatform_v1beta1/types/index_endpoint.py b/google/cloud/aiplatform_v1beta1/types/index_endpoint.py index 086f499207..aac23e7e5d 100644 --- a/google/cloud/aiplatform_v1beta1/types/index_endpoint.py +++ b/google/cloud/aiplatform_v1beta1/types/index_endpoint.py @@ -111,6 +111,14 @@ class IndexEndpoint(proto.Message): and [private_service_connect_config][google.cloud.aiplatform.v1beta1.IndexEndpoint.private_service_connect_config] are mutually exclusive. + public_endpoint_enabled (bool): + Optional. If true, the deployed index will be + accessible through public endpoint. + public_endpoint_domain_name (str): + Output only. If + [public_endpoint_enabled][google.cloud.aiplatform.v1beta1.IndexEndpoint.public_endpoint_enabled] + is true, this field will be populated with the domain name + to use for this index endpoint. """ name: str = proto.Field( @@ -164,6 +172,14 @@ class IndexEndpoint(proto.Message): message=service_networking.PrivateServiceConnectConfig, ) ) + public_endpoint_enabled: bool = proto.Field( + proto.BOOL, + number=13, + ) + public_endpoint_domain_name: str = proto.Field( + proto.STRING, + number=14, + ) class DeployedIndex(proto.Message): @@ -264,7 +280,7 @@ class DeployedIndex(proto.Message): be deployed to any ip ranges under the provided VPC network. - The value sohuld be the name of the address + The value should be the name of the address (https://cloud.google.com/compute/docs/reference/rest/v1/addresses) Example: 'vertex-ai-ip-range'. deployment_group (str): diff --git a/google/cloud/aiplatform_v1beta1/types/model.py b/google/cloud/aiplatform_v1beta1/types/model.py index af190f6965..272280ec15 100644 --- a/google/cloud/aiplatform_v1beta1/types/model.py +++ b/google/cloud/aiplatform_v1beta1/types/model.py @@ -901,6 +901,10 @@ class ModelSourceInfo(proto.Message): Attributes: source_type (google.cloud.aiplatform_v1beta1.types.ModelSourceInfo.ModelSourceType): Type of the model source. + copy (bool): + If this Model is copy of another Model. If true then + [source_type][google.cloud.aiplatform.v1beta1.ModelSourceInfo.source_type] + pertains to the original. """ class ModelSourceType(proto.Enum): @@ -918,17 +922,25 @@ class ModelSourceType(proto.Enum): BQML (3): The Model is registered and sync'ed from BigQuery ML. + MODEL_GARDEN (4): + The Model is saved or tuned from Model + Garden. """ MODEL_SOURCE_TYPE_UNSPECIFIED = 0 AUTOML = 1 CUSTOM = 2 BQML = 3 + MODEL_GARDEN = 4 source_type: ModelSourceType = proto.Field( proto.ENUM, number=1, enum=ModelSourceType, ) + copy: bool = proto.Field( + proto.BOOL, + number=2, + ) __all__ = tuple(sorted(__protobuf__.manifest)) diff --git a/tests/unit/gapic/aiplatform_v1/test_index_endpoint_service.py b/tests/unit/gapic/aiplatform_v1/test_index_endpoint_service.py index 22b2c10298..c91226b6b3 100644 --- a/tests/unit/gapic/aiplatform_v1/test_index_endpoint_service.py +++ b/tests/unit/gapic/aiplatform_v1/test_index_endpoint_service.py @@ -1005,6 +1005,8 @@ def test_get_index_endpoint(request_type, transport: str = "grpc"): etag="etag_value", network="network_value", enable_private_service_connect=True, + public_endpoint_enabled=True, + public_endpoint_domain_name="public_endpoint_domain_name_value", ) response = client.get_index_endpoint(request) @@ -1021,6 +1023,8 @@ def test_get_index_endpoint(request_type, transport: str = "grpc"): assert response.etag == "etag_value" assert response.network == "network_value" assert response.enable_private_service_connect is True + assert response.public_endpoint_enabled is True + assert response.public_endpoint_domain_name == "public_endpoint_domain_name_value" def test_get_index_endpoint_empty_call(): @@ -1068,6 +1072,8 @@ async def test_get_index_endpoint_async( etag="etag_value", network="network_value", enable_private_service_connect=True, + public_endpoint_enabled=True, + public_endpoint_domain_name="public_endpoint_domain_name_value", ) ) response = await client.get_index_endpoint(request) @@ -1085,6 +1091,8 @@ async def test_get_index_endpoint_async( assert response.etag == "etag_value" assert response.network == "network_value" assert response.enable_private_service_connect is True + assert response.public_endpoint_enabled is True + assert response.public_endpoint_domain_name == "public_endpoint_domain_name_value" @pytest.mark.asyncio @@ -1715,6 +1723,8 @@ def test_update_index_endpoint(request_type, transport: str = "grpc"): etag="etag_value", network="network_value", enable_private_service_connect=True, + public_endpoint_enabled=True, + public_endpoint_domain_name="public_endpoint_domain_name_value", ) response = client.update_index_endpoint(request) @@ -1731,6 +1741,8 @@ def test_update_index_endpoint(request_type, transport: str = "grpc"): assert response.etag == "etag_value" assert response.network == "network_value" assert response.enable_private_service_connect is True + assert response.public_endpoint_enabled is True + assert response.public_endpoint_domain_name == "public_endpoint_domain_name_value" def test_update_index_endpoint_empty_call(): @@ -1778,6 +1790,8 @@ async def test_update_index_endpoint_async( etag="etag_value", network="network_value", enable_private_service_connect=True, + public_endpoint_enabled=True, + public_endpoint_domain_name="public_endpoint_domain_name_value", ) ) response = await client.update_index_endpoint(request) @@ -1795,6 +1809,8 @@ async def test_update_index_endpoint_async( assert response.etag == "etag_value" assert response.network == "network_value" assert response.enable_private_service_connect is True + assert response.public_endpoint_enabled is True + assert response.public_endpoint_domain_name == "public_endpoint_domain_name_value" @pytest.mark.asyncio diff --git a/tests/unit/gapic/aiplatform_v1/test_migration_service.py b/tests/unit/gapic/aiplatform_v1/test_migration_service.py index 9055645a91..8fc5db291f 100644 --- a/tests/unit/gapic/aiplatform_v1/test_migration_service.py +++ b/tests/unit/gapic/aiplatform_v1/test_migration_service.py @@ -2030,19 +2030,22 @@ def test_parse_dataset_path(): def test_dataset_path(): project = "squid" - dataset = "clam" - expected = "projects/{project}/datasets/{dataset}".format( + location = "clam" + dataset = "whelk" + expected = "projects/{project}/locations/{location}/datasets/{dataset}".format( project=project, + location=location, dataset=dataset, ) - actual = MigrationServiceClient.dataset_path(project, dataset) + actual = MigrationServiceClient.dataset_path(project, location, dataset) assert expected == actual def test_parse_dataset_path(): expected = { - "project": "whelk", - "dataset": "octopus", + "project": "octopus", + "location": "oyster", + "dataset": "nudibranch", } path = MigrationServiceClient.dataset_path(**expected) @@ -2052,22 +2055,19 @@ def test_parse_dataset_path(): def test_dataset_path(): - project = "oyster" - location = "nudibranch" - dataset = "cuttlefish" - expected = "projects/{project}/locations/{location}/datasets/{dataset}".format( + project = "cuttlefish" + dataset = "mussel" + expected = "projects/{project}/datasets/{dataset}".format( project=project, - location=location, dataset=dataset, ) - actual = MigrationServiceClient.dataset_path(project, location, dataset) + actual = MigrationServiceClient.dataset_path(project, dataset) assert expected == actual def test_parse_dataset_path(): expected = { - "project": "mussel", - "location": "winkle", + "project": "winkle", "dataset": "nautilus", } path = MigrationServiceClient.dataset_path(**expected) diff --git a/tests/unit/gapic/aiplatform_v1beta1/test_index_endpoint_service.py b/tests/unit/gapic/aiplatform_v1beta1/test_index_endpoint_service.py index afb445188a..6585f089bf 100644 --- a/tests/unit/gapic/aiplatform_v1beta1/test_index_endpoint_service.py +++ b/tests/unit/gapic/aiplatform_v1beta1/test_index_endpoint_service.py @@ -1005,6 +1005,8 @@ def test_get_index_endpoint(request_type, transport: str = "grpc"): etag="etag_value", network="network_value", enable_private_service_connect=True, + public_endpoint_enabled=True, + public_endpoint_domain_name="public_endpoint_domain_name_value", ) response = client.get_index_endpoint(request) @@ -1021,6 +1023,8 @@ def test_get_index_endpoint(request_type, transport: str = "grpc"): assert response.etag == "etag_value" assert response.network == "network_value" assert response.enable_private_service_connect is True + assert response.public_endpoint_enabled is True + assert response.public_endpoint_domain_name == "public_endpoint_domain_name_value" def test_get_index_endpoint_empty_call(): @@ -1068,6 +1072,8 @@ async def test_get_index_endpoint_async( etag="etag_value", network="network_value", enable_private_service_connect=True, + public_endpoint_enabled=True, + public_endpoint_domain_name="public_endpoint_domain_name_value", ) ) response = await client.get_index_endpoint(request) @@ -1085,6 +1091,8 @@ async def test_get_index_endpoint_async( assert response.etag == "etag_value" assert response.network == "network_value" assert response.enable_private_service_connect is True + assert response.public_endpoint_enabled is True + assert response.public_endpoint_domain_name == "public_endpoint_domain_name_value" @pytest.mark.asyncio @@ -1715,6 +1723,8 @@ def test_update_index_endpoint(request_type, transport: str = "grpc"): etag="etag_value", network="network_value", enable_private_service_connect=True, + public_endpoint_enabled=True, + public_endpoint_domain_name="public_endpoint_domain_name_value", ) response = client.update_index_endpoint(request) @@ -1731,6 +1741,8 @@ def test_update_index_endpoint(request_type, transport: str = "grpc"): assert response.etag == "etag_value" assert response.network == "network_value" assert response.enable_private_service_connect is True + assert response.public_endpoint_enabled is True + assert response.public_endpoint_domain_name == "public_endpoint_domain_name_value" def test_update_index_endpoint_empty_call(): @@ -1778,6 +1790,8 @@ async def test_update_index_endpoint_async( etag="etag_value", network="network_value", enable_private_service_connect=True, + public_endpoint_enabled=True, + public_endpoint_domain_name="public_endpoint_domain_name_value", ) ) response = await client.update_index_endpoint(request) @@ -1795,6 +1809,8 @@ async def test_update_index_endpoint_async( assert response.etag == "etag_value" assert response.network == "network_value" assert response.enable_private_service_connect is True + assert response.public_endpoint_enabled is True + assert response.public_endpoint_domain_name == "public_endpoint_domain_name_value" @pytest.mark.asyncio diff --git a/tests/unit/gapic/aiplatform_v1beta1/test_migration_service.py b/tests/unit/gapic/aiplatform_v1beta1/test_migration_service.py index 0185bdabbe..7c36d84b95 100644 --- a/tests/unit/gapic/aiplatform_v1beta1/test_migration_service.py +++ b/tests/unit/gapic/aiplatform_v1beta1/test_migration_service.py @@ -2006,22 +2006,19 @@ def test_parse_annotated_dataset_path(): def test_dataset_path(): project = "cuttlefish" - location = "mussel" - dataset = "winkle" - expected = "projects/{project}/locations/{location}/datasets/{dataset}".format( + dataset = "mussel" + expected = "projects/{project}/datasets/{dataset}".format( project=project, - location=location, dataset=dataset, ) - actual = MigrationServiceClient.dataset_path(project, location, dataset) + actual = MigrationServiceClient.dataset_path(project, dataset) assert expected == actual def test_parse_dataset_path(): expected = { - "project": "nautilus", - "location": "scallop", - "dataset": "abalone", + "project": "winkle", + "dataset": "nautilus", } path = MigrationServiceClient.dataset_path(**expected) @@ -2031,19 +2028,22 @@ def test_parse_dataset_path(): def test_dataset_path(): - project = "squid" - dataset = "clam" - expected = "projects/{project}/datasets/{dataset}".format( + project = "scallop" + location = "abalone" + dataset = "squid" + expected = "projects/{project}/locations/{location}/datasets/{dataset}".format( project=project, + location=location, dataset=dataset, ) - actual = MigrationServiceClient.dataset_path(project, dataset) + actual = MigrationServiceClient.dataset_path(project, location, dataset) assert expected == actual def test_parse_dataset_path(): expected = { - "project": "whelk", + "project": "clam", + "location": "whelk", "dataset": "octopus", } path = MigrationServiceClient.dataset_path(**expected)