OptionsBuilder.java

Go to the documentation of this file.

/*
 * Ollama4j - Java library for interacting with Ollama server.
 * Copyright (c) 2025 Amith Koujalgi and contributors.
 *
 * Licensed under the MIT License (the "License");
 * you may not use this file except in compliance with the License.
 *
*/
package io.github.ollama4j.utils;
 
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;
    }
}