diff --git a/src/main/java/io/github/amithkoujalgi/ollama4j/core/models/chat/OllamaChatResult.java b/src/main/java/io/github/amithkoujalgi/ollama4j/core/models/chat/OllamaChatResult.java new file mode 100644 index 0000000..6ac6578 --- /dev/null +++ b/src/main/java/io/github/amithkoujalgi/ollama4j/core/models/chat/OllamaChatResult.java @@ -0,0 +1,32 @@ +package io.github.amithkoujalgi.ollama4j.core.models.chat; + +import java.util.List; + +import io.github.amithkoujalgi.ollama4j.core.models.OllamaResult; + +/** + * Specific chat-API result that contains the chat history sent to the model and appends the answer as {@link OllamaChatResult} given by the + * {@link OllamaChatMessageRole#ASSISTANT} role. + */ +public class OllamaChatResult extends OllamaResult{ + + private List chatHistory; + + public OllamaChatResult(String response, long responseTime, int httpStatusCode, + List chatHistory) { + super(response, responseTime, httpStatusCode); + this.chatHistory = chatHistory; + appendAnswerToChatHistory(response); + } + + public List getChatHistory() { + return chatHistory; + } + + private void appendAnswerToChatHistory(String answer){ + OllamaChatMessage assistantMessage = new OllamaChatMessage(OllamaChatMessageRole.ASSISTANT, answer); + this.chatHistory.add(assistantMessage); + } + + +}