API PaLM: Guide de démarrage rapide sur les représentations vectorielles continues avec Java

Ce guide de démarrage rapide vous aide à utiliser le service de représentations vectorielles continues de l'API PaLM. Vous utiliserez le service de représentations vectorielles continues pour générer des représentations vectorielles continues de texte utilisables dans diverses applications de langage naturel.

Obtenir une clé API

Pour commencer, vous devez obtenir une clé API.

Que sont les représentations vectorielles continues ?

La technique des représentations vectorielles continues permet de représenter du texte (comme des mots, des phrases ou des paragraphes entiers) sous forme de liste de nombres à virgule flottante dans un tableau. L'idée clé est que les textes ayant des significations similaires auront des représentations vectorielles continues similaires. Vous pouvez utiliser cette relation pour de nombreuses tâches importantes.

Installer le client API

Ces instructions installeront le SDK Java PaLM dans votre dépôt Maven local afin que vous puissiez l'ajouter en tant que dépendance à votre projet Gradle.

  1. Téléchargez le fichier google-cloud-ai-generativelanguage-v1-java.tar.gz.
  2. Extrayez les fichiers et installez-les dans mavenLocal:

    # Extract the files
    tar -xzvf google-cloud-ai-generativelanguage-v1-java.tar.gz
    cd google-cloud-ai-generativelanguage-v1-java
    
    # Install to mavenLocal
    ./gradlew publishToMavenLocal
    
  3. Ouvrez le fichier de configuration Gradle et assurez-vous que mavenLocal() figure sous repositories:

    repositories {
        mavenCentral()
        // ...
    
        // Add the Maven Local repository
        mavenLocal()
    }
    
  4. Toujours dans votre fichier de configuration Gradle, ajoutez les bibliothèques nécessaires au bloc dependencies:

    dependencies {
        // ...
    
        // Add these dependencies to use Generative AI
        implementation("com.google.cloud:gapic-google-cloud-ai-generativelanguage-v1-java:0.0.0-SNAPSHOT")
        implementation("io.grpc:grpc-okhttp:1.53.0")
    }
    

Initialiser le client de service de texte

Dans votre programme Java, initialisez un TextServiceClient en transmettant votre clé API en tant qu'en-tête à TransportChannelProvider, qui sera utilisée par TextServiceSettings:

HashMap<String, String> headers = new HashMap<>();
headers.put("x-goog-api-key", "API_KEY");

TransportChannelProvider provider = InstantiatingGrpcChannelProvider.newBuilder()
    .setHeaderProvider(FixedHeaderProvider.create(headers))
    .build();

TextServiceSettings settings = TextServiceSettings.newBuilder()
    .setTransportChannelProvider(provider)
    .setCredentialsProvider(FixedCredentialsProvider.create(null))
    .build();

TextServiceClient client = TextServiceClient.create(settings);

Générer des représentations vectorielles continues

Dans cette section, vous allez découvrir comment générer des représentations vectorielles continues pour un texte à l'aide de la méthode embedText.

Pour utiliser la méthode embedText, transmettez le nom du modèle ainsi que du texte à intégrer. Vous obtenez une liste de valeurs à virgule flottante.

EmbedTextRequest request = EmbedTextRequest.newBuilder()
    .setModel("models/embedding-gecko-001") // Required, which model to use to generate the result
    .setText("What do squirrels eat?") // Required
    .build();

EmbedTextResponse response = client.embedText(request);

List<Float> embedding = response.getEmbedding().getValueList();

System.out.println(embedding);