ee.carlrobert.llm.client.ollama.completion.request.OllamaChatCompletionRequest 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.ollama.completion.request;

import com.fasterxml.jackson.annotation.JsonInclude;
import ee.carlrobert.llm.client.ollama.completion.response.OllamaResponseFormat;
import ee.carlrobert.llm.completion.CompletionRequest;
import java.util.List;

/*
 * See ollama/api
 */
@JsonInclude(JsonInclude.Include.NON_NULL)
public class OllamaChatCompletionRequest implements CompletionRequest {

  private final String model;
  private final List messages;
  private final OllamaResponseFormat format;
  private final OllamaParameters options;
  private final Boolean stream;
  private final String keepAlive;

  public OllamaChatCompletionRequest(Builder builder) {
    this.model = builder.model;
    this.messages = builder.messages;
    this.format = builder.format;
    this.options = builder.options;
    this.stream = builder.stream;
    this.keepAlive = builder.keepAlive;
  }

  public String getModel() {
    return model;
  }

  public List getMessages() {
    return messages;
  }

  public OllamaResponseFormat getFormat() {
    return format;
  }

  public OllamaParameters getOptions() {
    return options;
  }

  public Boolean getStream() {
    return stream;
  }

  public String getKeepAlive() {
    return keepAlive;
  }

  public static class Builder {

    private final String model;
    private final List messages;

    private OllamaResponseFormat format = null;
    private OllamaParameters options = null;
    private Boolean stream = null;
    private String keepAlive = null;

    public Builder(String model, List messages) {
      this.model = model;
      this.messages = messages;
    }

    public Builder setFormat(OllamaResponseFormat format) {
      this.format = format;
      return this;
    }

    public Builder setOptions(OllamaParameters options) {
      this.options = options;
      return this;
    }

    public Builder setStream(Boolean stream) {
      this.stream = stream;
      return this;
    }

    public Builder setKeepAlive(String keepAlive) {
      this.keepAlive = keepAlive;
      return this;
    }

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