OptionsBuilder.java

Go to the documentation of this file.

package io.github.ollama4j.utils;
 
import java.io.IOException;
import java.util.HashMap;
 
public class OptionsBuilder {
 
  private final Options options;
 
  public OptionsBuilder() {
    this.options = new Options(new HashMap<>());
  }
 
  public OptionsBuilder setMirostat(int value) {
    options.getOptionsMap().put("mirostat", value);
    return this;
  }
 
  public OptionsBuilder setMirostatEta(float value) {
    options.getOptionsMap().put("mirostat_eta", value);
    return this;
  }
 
  public OptionsBuilder setMirostatTau(float value) {
    options.getOptionsMap().put("mirostat_tau", value);
    return this;
  }
 
  public OptionsBuilder setNumCtx(int value) {
    options.getOptionsMap().put("num_ctx", value);
    return this;
  }
 
  public OptionsBuilder setNumGqa(int value) {
    options.getOptionsMap().put("num_gqa", value);
    return this;
  }
 
  public OptionsBuilder setNumGpu(int value) {
    options.getOptionsMap().put("num_gpu", value);
    return this;
  }
 
  public OptionsBuilder setNumThread(int value) {
    options.getOptionsMap().put("num_thread", value);
    return this;
  }
 
  public OptionsBuilder setRepeatLastN(int value) {
    options.getOptionsMap().put("repeat_last_n", value);
    return this;
  }
 
  public OptionsBuilder setRepeatPenalty(float value) {
    options.getOptionsMap().put("repeat_penalty", value);
    return this;
  }
 
  public OptionsBuilder setTemperature(float value) {
    options.getOptionsMap().put("temperature", value);
    return this;
  }
 
  public OptionsBuilder setSeed(int value) {
    options.getOptionsMap().put("seed", value);
    return this;
  }
 
  public OptionsBuilder setStop(String value) {
    options.getOptionsMap().put("stop", value);
    return this;
  }
 
  public OptionsBuilder setTfsZ(float value) {
    options.getOptionsMap().put("tfs_z", value);
    return this;
  }
 
  public OptionsBuilder setNumPredict(int value) {
    options.getOptionsMap().put("num_predict", value);
    return this;
  }
 
  public OptionsBuilder setTopK(int value) {
    options.getOptionsMap().put("top_k", value);
    return this;
  }
 
  public OptionsBuilder setTopP(float value) {
    options.getOptionsMap().put("top_p", value);
    return this;
  }
 
  public OptionsBuilder setMinP(float value) {
    options.getOptionsMap().put("min_p", value);
    return this;
  }
 
  public OptionsBuilder setCustomOption(String name, Object value) throws IllegalArgumentException {
    if (!(value instanceof Integer || value instanceof Float || value instanceof String)) {
      throw new IllegalArgumentException("Invalid type for parameter. Allowed types are: Integer, Float, or String.");
    }
    options.getOptionsMap().put(name, value);
    return this;
  }
 
 
 
  public Options build() {
    return options;
  }
 
 
}