ee.carlrobert.llm.client.google.completion.GoogleCompletionRequest Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of llm-client Show documentation

Java http client wrapped around the OkHttp3 library

There is a newer version: 0.8.28

package ee.carlrobert.llm.client.google.completion;

import ee.carlrobert.llm.completion.CompletionRequest;
import java.util.ArrayList;
import java.util.List;

public class GoogleCompletionRequest implements CompletionRequest {

  private final List contents;
  private final List safetySettings;
  private final GoogleGenerationConfig generationConfig;

  public GoogleCompletionRequest(Builder builder) {
    this.contents = builder.contents;
    this.safetySettings = builder.safetySettings;
    this.generationConfig = builder.generationConfig;
  }

  public List getContents() {
    return contents;
  }

  public List getSafetySettings() {
    return safetySettings;
  }

  public GoogleGenerationConfig getGenerationConfig() {
    return generationConfig;
  }

  /**
   * Harm Category.
   */
  public enum HarmCategory {
    HARM_CATEGORY_UNSPECIFIED,
    HARM_CATEGORY_DEROGATORY,
    HARM_CATEGORY_TOXICITY,
    HARM_CATEGORY_VIOLENCE,
    HARM_CATEGORY_SEXUAL,
    HARM_CATEGORY_MEDICAL,
    HARM_CATEGORY_DANGEROUS,
    HARM_CATEGORY_HARASSMENT,
    HARM_CATEGORY_HATE_SPEECH,
    HARM_CATEGORY_SEXUALLY_EXPLICIT,
    HARM_CATEGORY_DANGEROUS_CONTENT;
  }

  /**
   * Harm
   * Block Threshold.
   */
  public enum HarmBlockThreshold {
    HARM_BLOCK_THRESHOLD_UNSPECIFIED,
    BLOCK_LOW_AND_ABOVE,
    BLOCK_MEDIUM_AND_ABOVE,
    BLOCK_ONLY_HIGH,
    BLOCK_NONE,
  }

  public static class SafetySetting {

    private final HarmCategory category;
    private final HarmBlockThreshold threshold;

    public SafetySetting(HarmCategory category, HarmBlockThreshold threshold) {
      this.category = category;
      this.threshold = threshold;
    }

    public HarmCategory getCategory() {
      return category;
    }

    public HarmBlockThreshold getThreshold() {
      return threshold;
    }
  }

  public static class Builder {

    private List contents;
    private List safetySettings = new ArrayList<>();
    private GoogleGenerationConfig generationConfig = new GoogleGenerationConfig.Builder().build();

    public Builder(List contents) {
      this.contents = contents;
    }

    public Builder safetySettings(List safetySettings) {
      this.safetySettings = safetySettings;
      return this;
    }

    public Builder generationConfig(GoogleGenerationConfig generationConfig) {
      this.generationConfig = generationConfig;
      return this;
    }

    public GoogleCompletionRequest build() {
      return new GoogleCompletionRequest(this);
    }
  }
}