OllamaEndpointCaller.java

Go to the documentation of this file.

package io.github.ollama4j.models.request;
 
import io.github.ollama4j.OllamaAPI;
import io.github.ollama4j.utils.Constants;
import lombok.Getter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
 
import java.net.URI;
import java.net.http.HttpRequest;
import java.time.Duration;

@Getter
public abstract class OllamaEndpointCaller {
 
    private static final Logger LOG = LoggerFactory.getLogger(OllamaAPI.class);
 
    private final String host;
    private final Auth auth;
    private final long requestTimeoutSeconds;
    private final boolean verbose;
 
    public OllamaEndpointCaller(String host, Auth auth, long requestTimeoutSeconds, boolean verbose) {
        this.host = host;
        this.auth = auth;
        this.requestTimeoutSeconds = requestTimeoutSeconds;
        this.verbose = verbose;
    }
 
    protected abstract String getEndpointSuffix();
 
    protected abstract boolean parseResponseAndAddToBuffer(String line, StringBuilder responseBuffer, StringBuilder thinkingBuffer);
 

    protected HttpRequest.Builder getRequestBuilderDefault(URI uri) {
        HttpRequest.Builder requestBuilder =
                HttpRequest.newBuilder(uri)
                        .header(Constants.HttpConstants.HEADER_KEY_CONTENT_TYPE, Constants.HttpConstants.APPLICATION_JSON)
                        .timeout(Duration.ofSeconds(this.requestTimeoutSeconds));
        if (isAuthCredentialsSet()) {
            requestBuilder.header("Authorization", this.auth.getAuthHeaderValue());
        }
        return requestBuilder;
    }

    protected boolean isAuthCredentialsSet() {
        return this.auth != null;
    }
 
}