ee.carlrobert.llm.client.you.completion.YouCompletionRequest 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.you.completion;

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

public class YouCompletionRequest implements CompletionRequest {

  private final String prompt;
  private final List messages;
  private final UUID chatId;
  private final UUID userId;
  private final UUID queryTraceId;
  private final boolean useGPT4Model;

  private final YouCompletionMode chatMode;
  private final YouCompletionCustomModel customModel;

  public YouCompletionRequest(Builder builder) {
    this.prompt = builder.prompt;
    this.messages = builder.messages;
    this.chatId = builder.chatId;
    this.userId = builder.userId;
    this.queryTraceId = builder.queryTraceId;
    this.useGPT4Model = builder.useGPT4Model;
    this.chatMode = builder.chatMode;
    this.customModel = builder.customModel;
  }

  public String getPrompt() {
    return prompt;
  }

  public List getMessages() {
    return messages;
  }

  public UUID getChatId() {
    return chatId;
  }

  public UUID getUserId() {
    return userId;
  }

  public UUID getQueryTraceId() {
    return queryTraceId;
  }

  public boolean isUseGPT4Model() {
    return useGPT4Model;
  }

  public YouCompletionMode getChatMode() {
    return chatMode == null ? YouCompletionMode.DEFAULT : chatMode;
  }

  public YouCompletionCustomModel getCustomModel() {
    return customModel == null ? YouCompletionCustomModel.GPT_4_TURBO : customModel;
  }

  public static class Builder {

    private final String prompt;
    private List messages;
    private UUID chatId;
    private UUID userId;
    private UUID queryTraceId;
    private boolean useGPT4Model;
    private YouCompletionMode chatMode = YouCompletionMode.DEFAULT;
    private YouCompletionCustomModel customModel = YouCompletionCustomModel.GPT_4_TURBO;

    public Builder(String prompt) {
      this.prompt = prompt;
    }

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

    public Builder setChatId(UUID chatId) {
      this.chatId = chatId;
      return this;
    }

    public Builder setUserId(UUID userId) {
      this.userId = userId;
      return this;
    }

    public Builder setQueryTraceId(UUID queryTraceId) {
      this.queryTraceId = queryTraceId;
      return this;
    }

    public Builder setUseGPT4Model(boolean useGPT4Model) {
      this.useGPT4Model = useGPT4Model;
      return this;
    }

    public Builder setChatMode(YouCompletionMode chatMode) {
      this.chatMode = chatMode;
      return this;
    }

    public Builder setCustomModel(YouCompletionCustomModel customModel) {
      this.customModel = customModel;
      return this;
    }

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