dev.langchain4j.model.ollama.EmbeddingRequest Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of langchain4j-ollama Show documentation

There is a newer version: 0.36.0

package dev.langchain4j.model.ollama;

import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.databind.PropertyNamingStrategies.SnakeCaseStrategy;
import com.fasterxml.jackson.databind.annotation.JsonNaming;

import java.util.List;

import static com.fasterxml.jackson.annotation.JsonInclude.Include.NON_NULL;

@JsonIgnoreProperties(ignoreUnknown = true)
@JsonInclude(NON_NULL)
@JsonNaming(SnakeCaseStrategy.class)
class EmbeddingRequest {

    private String model;
    private List input;

    EmbeddingRequest() {
    }

    EmbeddingRequest(String model, List input) {
        this.model = model;
        this.input = input;
    }

    static Builder builder() {
        return new Builder();
    }

    public String getModel() {
        return model;
    }

    public void setModel(String model) {
        this.model = model;
    }

    public List getInput() {
        return input;
    }

    public void setInput(List input) {
        this.input = input;
    }

    static class Builder {

        private String model;
        private List input;

        Builder model(String model) {
            this.model = model;
            return this;
        }

        Builder input(List input) {
            this.input = input;
            return this;
        }

        EmbeddingRequest build() {
            return new EmbeddingRequest(model, input);
        }
    }
}