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)