def _Run(self, args, version):
validation.ValidateDisplayName(args.display_name)
index_endpoint_ref = args.CONCEPTS.index_endpoint.Parse()
region = index_endpoint_ref.AsDict()['locationsId']
with endpoint_util.AiplatformEndpointOverrides(version, region=region):
index_endpoint_client = client.IndexEndpointsClient(
version=version)
try:
if version == constants.GA_VERSION:
result = index_endpoint_client.Patch(
index_endpoint_ref, args)
else:
result = index_endpoint_client.PatchBeta(
index_endpoint_ref, args)
except errors.NoFieldsSpecifiedError:
available_update_args = [
'display_name', 'description', 'update_labels',
'clear_labels', 'remove_labels'
]
if not any(
args.IsSpecified(arg)
for arg in available_update_args):
raise
log.status.Print('No update to perform.')
return None
else:
log.UpdatedResource(result.name,
kind='Vertex AI index endpoint')
return result
def _Run(self, args, version):
validation.ValidateDisplayName(args.display_name)
index_endpoint_ref = args.CONCEPTS.index_endpoint.Parse()
project_id = index_endpoint_ref.AsDict()['projectsId']
region = index_endpoint_ref.AsDict()['locationsId']
with endpoint_util.AiplatformEndpointOverrides(version, region=region):
index_endpoint_client = client.IndexEndpointsClient(
version=version)
if version == constants.GA_VERSION:
operation = index_endpoint_client.DeployIndex(
index_endpoint_ref, args)
else:
operation = index_endpoint_client.DeployIndexBeta(
index_endpoint_ref, args)
op_ref = index_endpoints_util.ParseIndexEndpointOperation(
operation.name)
# TODO(b/208506223): Support `--async` flag.
index_endpoint_id = op_ref.AsDict()['indexEndpointsId']
log.status.Print(
constants.OPERATION_CREATION_DISPLAY_MESSAGE.format(
name=operation.name,
verb='deploy index',
id=op_ref.Name(),
sub_commands='--index-endpoint={} [--project={}]'.format(
index_endpoint_id, project_id)))
return operation
def _Run(self, args, version):
index_endpoint_ref = args.CONCEPTS.index_endpoint.Parse()
region = index_endpoint_ref.AsDict()['locationsId']
with endpoint_util.AiplatformEndpointOverrides(version, region=region):
operation = client.IndexEndpointsClient().UndeployIndexBeta(
index_endpoint_ref, args)
return operations_util.WaitForOpMaybe(
operations_client=operations.OperationsClient(),
op=operation,
op_ref=index_endpoints_util.ParseIndexEndpointOperation(
operation.name))
def _Run(self, args, version):
index_endpoint_ref = args.CONCEPTS.index_endpoint.Parse()
region = index_endpoint_ref.AsDict()['locationsId']
index_endpoint_id = index_endpoint_ref.AsDict()['indexEndpointsId']
with endpoint_util.AiplatformEndpointOverrides(version, region=region):
console_io.PromptContinue(
'This will delete index endpoint [{}]...'.format(index_endpoint_id),
cancel_on_no=True)
operation = client.IndexEndpointsClient(
version=version).Delete(index_endpoint_ref)
return operations_util.WaitForOpMaybe(
operations_client=operations.OperationsClient(version=version),
op=operation,
op_ref=index_endpoints_util.ParseIndexEndpointOperation(
operation.name))
def _Run(self, args, version):
validation.ValidateDisplayName(args.display_name)
region_ref = args.CONCEPTS.region.Parse()
region = region_ref.AsDict()['locationsId']
with endpoint_util.AiplatformEndpointOverrides(version, region=region):
operation = client.IndexEndpointsClient().CreateBeta(
region_ref, args)
response_msg = operations_util.WaitForOpMaybe(
operations_client=operations.OperationsClient(),
op=operation,
op_ref=index_endpoints_util.ParseIndexEndpointOperation(
operation.name))
if response_msg is not None:
response = encoding.MessageToPyValue(response_msg)
if 'name' in response:
log.status.Print(
('Created AI Platform index endpoint: {}.').format(
response['name']))
return response_msg
def _Run(self, args, version):
validation.ValidateDisplayName(args.display_name)
index_endpoint_ref = args.CONCEPTS.index_endpoint.Parse()
region = index_endpoint_ref.AsDict()['locationsId']
with endpoint_util.AiplatformEndpointOverrides(version, region=region):
operation = client.IndexEndpointsClient().DeployIndexBeta(
index_endpoint_ref, args)
response_msg = operations_util.WaitForOpMaybe(
operations_client=operations.OperationsClient(),
op=operation,
op_ref=index_endpoints_util.ParseIndexEndpointOperation(
operation.name))
if response_msg is not None:
response = encoding.MessageToPyValue(response_msg)
if 'deployedIndex' in response and 'id' in response[
'deployedIndex']:
log.status.Print(('Id of the deployed index: {}.').format(
response['deployedIndex']['id']))
return response_msg
def _Run(self, args, version):
region_ref = args.CONCEPTS.region.Parse()
region = region_ref.AsDict()['locationsId']
with endpoint_util.AiplatformEndpointOverrides(version, region=region):
return client.IndexEndpointsClient(version=version).List(
region_ref=region_ref)