Версия Maven Central

​
Введение

Клиентская библиотека для Java предлагает простой в использовании интерфейс для взаимодействия со службами Sunra.

​
Установка

Чтобы интегрировать клиент в свой проект, добавьте зависимость с помощью Maven или Gradle: 
implementation("ai.sunra.client:sunra-client:0.1.5")

​
Функции

​
Вызов конечных точек

Sunra управляет запросами к конечным точкам через систему очередей, обеспечивая надежность и масштабируемость. Используйте метод subscribe, чтобы отправить запрос и дождаться результата. Пример: 
import ai.sunra.client.*;
import ai.sunra.client.queue.*;
import java.util.Map;
import com.google.gson.JsonObject;
import java.util.function.Consumer;

ClientConfig config = ClientConfig.builder()
    .withCredentials(CredentialsResolver.fromEnv())
    .build();

SunraClient client = SunraClient.withConfig(config);
Map<String, Object> input = Map.of(
    "prompt", "собака бежит в парке");

Consumer<QueueStatus.StatusUpdate> statusUpdateHandler = update -> {
    String status = update.getStatus().toString();
    String message = String.format("\nОбновление статуса: %s, ID запроса: %s",
        status, update.getRequestId());
    System.out.println(message);
};

SubscribeOptions<JsonObject> options = SubscribeOptions.<JsonObject>builder()
    .input(input)
    .resultType(JsonObject.class)
    .onQueueUpdate(statusUpdateHandler)
    .logs(true)
    .build();

// сохраняем запрос на подписку в переменную и выводим его
Output<JsonObject> response = client.subscribe("black-forest-labs/flux-kontext-pro/text-to-image", options);
System.out.println("Завершено!");
System.out.println(response.getData());

​
Управление очередью

Управляйте своими запросами с помощью этих методов:
Отправка запроса
Отправьте запрос и получите request_id для последующего использования. Пример: 
import ai.sunra.client.*;
import ai.sunra.client.queue.*;
import java.util.Map;
import com.google.gson.JsonObject;
import java.util.function.Consumer;

ClientConfig config = ClientConfig.builder()
    .withCredentials(CredentialsResolver.fromEnv())
    .build();

SunraClient client = SunraClient.withConfig(config);
Map<String, Object> input = Map.of(
    "prompt", "собака бежит в парке");

Consumer<QueueStatus.StatusUpdate> statusUpdateHandler = update -> {
    String status = update.getStatus().toString();
    String message = String.format("\nОбновление статуса: %s, ID запроса: %s",
        status, update.getRequestId());
    System.out.println(message);
};

SubscribeOptions<JsonObject> options = SubscribeOptions.<JsonObject>builder()
    .input(input)
    .resultType(JsonObject.class)
    .onQueueUpdate(statusUpdateHandler)
    .logs(true)
    .build();

// сохраняем запрос на подписку в переменную и выводим его
var request = client.queue().submit("black-forest-labs/flux-kontext-pro/text-to-image", options);
Проверка статуса запроса
Получите статус запроса. Пример: 
import ai.sunra.client.*;
import ai.sunra.client.queue.*;

ClientConfig config = ClientConfig.builder()
    .withCredentials(CredentialsResolver.fromEnv())
    .build();

SunraClient client = SunraClient.withConfig(config);
var job = client.queue().status(QueueStatusOptions.withRequestId("pd_eTYYuw4EqYLzRBHgnAMHA8zH"));
Получение результатов запроса
Получите результат выполненного запроса. Пример: 
import ai.sunra.client.*;
import ai.sunra.client.queue.*;

ClientConfig config = ClientConfig.builder()
    .withCredentials(CredentialsResolver.fromEnv())
    .build();

SunraClient client = SunraClient.withConfig(config);
var result = client.queue().result(QueueStatusOptions.withRequestId("pd_eTYYuw4EqYLzRBHgnAMHA8zH"));

​
Загрузка файлов

Еще не реализовано.
Максимальный размер файла: 100 МБ

​
Поддержка

Присоединяйтесь к нашему сообществу для помощи или обсуждений: Мы здесь, чтобы помочь вам!