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