OllamaChatResult.java

Go to the documentation of this file.

package io.github.ollama4j.models.chat;
 
import java.util.List;
 
import io.github.ollama4j.models.response.OllamaResult;
 
public class OllamaChatResult extends OllamaResult {
 
    private List<OllamaChatMessage> chatHistory;
 
    public OllamaChatResult(String response, long responseTime, int httpStatusCode, List<OllamaChatMessage> chatHistory) {
        super(response, responseTime, httpStatusCode);
        this.chatHistory = chatHistory;
        appendAnswerToChatHistory(response);
    }
 
    public List<OllamaChatMessage> getChatHistory() {
        return chatHistory;
    }
 
    private void appendAnswerToChatHistory(String answer) {
        OllamaChatMessage assistantMessage = new OllamaChatMessage(OllamaChatMessageRole.ASSISTANT, answer);
        this.chatHistory.add(assistantMessage);
    }
}