OllamaEndpointCaller.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.models.request;
 
import io.github.ollama4j.utils.Constants;
import java.net.URI;
import java.net.http.HttpRequest;
import java.time.Duration;
import lombok.Getter;
 
@Getter
public abstract class OllamaEndpointCaller {
 
    private final String host;
    private final Auth auth;
    private final long requestTimeoutSeconds;
 
    protected OllamaEndpointCaller(String host, Auth auth, long requestTimeoutSeconds) {
        this.host = host;
        this.auth = auth;
        this.requestTimeoutSeconds = requestTimeoutSeconds;
    }
 
    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;
    }
}