ee.carlrobert.llm.client.ollama.completion.request.OllamaEmbeddingRequest 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 com.fasterxml.jackson.annotation.JsonInclude.Include;

/*
 * See ollama/api
 */
@JsonInclude(Include.NON_NULL)
public class OllamaEmbeddingRequest {

  private final String model;
  private final String prompt;
  private final OllamaParameters options;

  public OllamaEmbeddingRequest(Builder builder) {
    this.prompt = builder.prompt;
    this.model = builder.model;
    this.options = builder.options;
  }

  public String getModel() {
    return model;
  }

  public String getPrompt() {
    return prompt;
  }

  public OllamaParameters getOptions() {
    return options;
  }

  public static class Builder {

    private final String model;
    private final String prompt;
    private OllamaParameters options = null;

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

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

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