importdatetimefromtypingimportList,Unionfromgoogle.cloudimportaiplatformdefimport_feature_values_sample(project:str,location:str,entity_type_id:str,featurestore_id:str,feature_ids:List[str],feature_time:Union[str,datetime.datetime],gcs_source_uris:Union[str,List[str]],gcs_source_type:str,):aiplatform.init(project=project,location=location)my_entity_type=aiplatform.featurestore.EntityType(entity_type_name=entity_type_id,featurestore_id=featurestore_id)my_entity_type.ingest_from_gcs(feature_ids=feature_ids,feature_time=feature_time,gcs_source_uris=gcs_source_uris,gcs_source_type=gcs_source_type,)

importcom.google.api.gax.longrunning.OperationFuture;importcom.google.cloud.aiplatform.v1.AvroSource;importcom.google.cloud.aiplatform.v1.EntityTypeName;importcom.google.cloud.aiplatform.v1.FeaturestoreServiceClient;importcom.google.cloud.aiplatform.v1.FeaturestoreServiceSettings;importcom.google.cloud.aiplatform.v1.GcsSource;importcom.google.cloud.aiplatform.v1.ImportFeatureValuesOperationMetadata;importcom.google.cloud.aiplatform.v1.ImportFeatureValuesRequest;importcom.google.cloud.aiplatform.v1.ImportFeatureValuesRequest.FeatureSpec;importcom.google.cloud.aiplatform.v1.ImportFeatureValuesResponse;importjava.io.IOException;importjava.util.ArrayList;importjava.util.List;importjava.util.concurrent.ExecutionException;importjava.util.concurrent.TimeUnit;importjava.util.concurrent.TimeoutException;publicclassImportFeatureValuesSample{publicstaticvoidmain(String[]args)throwsIOException,InterruptedException,ExecutionException,TimeoutException{// TODO(developer): Replace these variables before running the sample.Stringproject="YOUR_PROJECT_ID";StringfeaturestoreId="YOUR_FEATURESTORE_ID";StringentityTypeId="YOUR_ENTITY_TYPE_ID";StringentityIdField="YOUR_ENTITY_FIELD_ID";StringfeatureTimeField="YOUR_FEATURE_TIME_FIELD";StringgcsSourceUri="YOUR_GCS_SOURCE_URI";intworkerCount=2;Stringlocation="us-central1";Stringendpoint="us-central1-aiplatform.googleapis.com:443";inttimeout=300;importFeatureValuesSample(project,featurestoreId,entityTypeId,gcsSourceUri,entityIdField,featureTimeField,workerCount,location,endpoint,timeout);}staticvoidimportFeatureValuesSample(Stringproject,StringfeaturestoreId,StringentityTypeId,StringgcsSourceUri,StringentityIdField,StringfeatureTimeField,intworkerCount,Stringlocation,Stringendpoint,inttimeout)throwsIOException,InterruptedException,ExecutionException,TimeoutException{FeaturestoreServiceSettingsfeaturestoreServiceSettings=FeaturestoreServiceSettings.newBuilder().setEndpoint(endpoint).build();// Initialize client that will be used to send requests. This client only needs to be created// once, and can be reused for multiple requests. After completing all of your requests, call// the "close" method on the client to safely clean up any remaining background resources.try(FeaturestoreServiceClientfeaturestoreServiceClient=FeaturestoreServiceClient.create(featurestoreServiceSettings)){List<FeatureSpec>featureSpecs=newArrayList<>();featureSpecs.add(FeatureSpec.newBuilder().setId("title").build());featureSpecs.add(FeatureSpec.newBuilder().setId("genres").build());featureSpecs.add(FeatureSpec.newBuilder().setId("average_rating").build());ImportFeatureValuesRequestimportFeatureValuesRequest=ImportFeatureValuesRequest.newBuilder().setEntityType(EntityTypeName.of(project,location,featurestoreId,entityTypeId).toString()).setEntityIdField(entityIdField).setFeatureTimeField(featureTimeField).addAllFeatureSpecs(featureSpecs).setWorkerCount(workerCount).setAvroSource(AvroSource.newBuilder().setGcsSource(GcsSource.newBuilder().addUris(gcsSourceUri))).build();OperationFuture<ImportFeatureValuesResponse,ImportFeatureValuesOperationMetadata>importFeatureValuesFuture=featurestoreServiceClient.importFeatureValuesAsync(importFeatureValuesRequest);System.out.format("Operation name: %s%n",importFeatureValuesFuture.getInitialFuture().get().getName());System.out.println("Waiting for operation to finish...");ImportFeatureValuesResponseimportFeatureValuesResponse=importFeatureValuesFuture.get(timeout,TimeUnit.SECONDS);System.out.println("Import Feature Values Response");System.out.println(importFeatureValuesResponse);featurestoreServiceClient.close();}}}

/** * TODO(developer): Uncomment these variables before running the sample.\ * (Not necessary if passing values as arguments) */// const project = 'YOUR_PROJECT_ID';// const featurestoreId = 'YOUR_FEATURESTORE_ID';// const entityTypeId = 'YOUR_ENTITY_TYPE_ID';// const avroGcsUri = 'AVRO_FILE_IN_THE_GCS_URI';// const entityIdField = 'ENTITY_ID_FIELD_IN_AVRO';// const featureTimeField = 'TIMESTAMP_FIELD_IN_AVRO';// const workerCount = <NO_OF_WORKERS_FOR_INGESTION_JOB>;// const location = 'YOUR_PROJECT_LOCATION';// const apiEndpoint = 'YOUR_API_ENDPOINT';// const timeout = <TIMEOUT_IN_MILLI_SECONDS>;// Imports the Google Cloud Featurestore Service Client libraryconst{FeaturestoreServiceClient}=require('@google-cloud/aiplatform').v1;// Specifies the location of the api endpointconstclientOptions={apiEndpoint:apiEndpoint,};// Instantiates a clientconstfeaturestoreServiceClient=newFeaturestoreServiceClient(clientOptions);asyncfunctionimportFeatureValues(){// Configure the entityType resourceconstentityType=`projects/${project}/locations/${location}/featurestores/${featurestoreId}/entityTypes/${entityTypeId}`;constavroSource={gcsSource:{uris:[avroGcsUri],},};constfeatureSpecs=[{id:'age'},{id:'gender'},{id:'liked_genres'}];constrequest={entityType:entityType,avroSource:avroSource,entityIdField:entityIdField,featureSpecs:featureSpecs,featureTimeField:featureTimeField,workerCount:Number(workerCount),};// Import Feature Values Requestconst[operation]=awaitfeaturestoreServiceClient.importFeatureValues(request,{timeout:Number(timeout)});const[response]=awaitoperation.promise();console.log('Import feature values response');console.log('Raw response:');console.log(JSON.stringify(response,null,2));}importFeatureValues();