PaLM API: краткое руководство по внедрению с помощью Java

Это краткое руководство поможет вам начать использовать службу внедрения PaLM API. Вы будете использовать службу встраивания для создания вложений текста, которые можно будет использовать в различных приложениях на естественном языке.

Получить ключ API

Для начала вам потребуется получить ключ API .

Что такое вложения?

Встраивание — это метод, используемый для представления текста (например, слов, предложений или целых абзацев) в виде списка чисел с плавающей запятой в массиве. Ключевая идея заключается в том, что текст со схожим значением будет иметь схожие вложения. Вы можете использовать отношения между ними для многих важных задач.

Установите API-клиент

Эти инструкции позволят установить PaLM Java SDK в ваш локальный репозиторий Maven, чтобы вы могли добавить его в качестве зависимости к вашему проекту Gradle.

  1. Загрузите файл google-cloud-ai-generativelanguage-v1-java.tar.gz .

  2. Извлеките файлы и установите их в 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. Откройте файл конфигурации Gradle и убедитесь, что mavenLocal() указан в repositories :

    repositories {
        mavenCentral    ()
            // ...

            // Add the Maven Local repository
        mavenLocal    ()
    }

  4. Также в файле конфигурации Gradle добавьте необходимые библиотеки в блок 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")
    }

Инициализация клиента текстовой службы

В вашей программе Java инициализируйте TextServiceClient , передав свой ключ API в качестве заголовка в TransportChannelProvider , который будет использоваться 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);

Генерация вложений

В этом разделе вы увидите, как создавать встраивания для фрагмента текста с помощью метода embedText .

Чтобы использовать метод embedText , передайте имя модели, а также текст для встраивания. Вы получите список значений с плавающей запятой.

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);