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