forked from Mirror/ollama4j
Creates OllamaChatResult extension class
This commit is contained in:
parent
d483c23c81
commit
65d852fdc9
@ -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<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);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user