OllamaChatResult.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.chat;
 
import static io.github.ollama4j.utils.Utils.getObjectMapper;
 
import com.fasterxml.jackson.core.JsonProcessingException;
import java.util.List;
import lombok.Getter;
 
@Getter
public class OllamaChatResult {
 
    private final List<OllamaChatMessage> chatHistory;
 
    private final OllamaChatResponseModel responseModel;
 
    public OllamaChatResult(
            OllamaChatResponseModel responseModel, List<OllamaChatMessage> chatHistory) {
        this.chatHistory = chatHistory;
        this.responseModel = responseModel;
        appendAnswerToChatHistory(responseModel);
    }
 
    private void appendAnswerToChatHistory(OllamaChatResponseModel response) {
        this.chatHistory.add(response.getMessage());
    }
 
    @Override
    public String toString() {
        try {
            return getObjectMapper().writerWithDefaultPrettyPrinter().writeValueAsString(this);
        } catch (JsonProcessingException e) {
            throw new RuntimeException(e);
        }
    }
}