Updated so you can dump all messages for an OllamaObject, add a system message, and add messages from a list

This commit is contained in:
2026-07-05 21:19:55 +02:00
parent 9c7ba9cb4f
commit 6e31e31718
@@ -232,6 +232,36 @@ public class OllamaObject {
messages.add(message);
}
/**
* Returns the current list of messages, AND clears them.
* You will need to re-submit the System prompt if you had one
* @return an ArrayList containing all messages stored in this {@link OllamaObject}
*/
public List<OllamaMessage> dumpMessages()
{
ArrayList<OllamaMessage> messages = new ArrayList<>(this.messages);
this.messages.clear();
return messages;
}
/**
* Adds all messages from a povided list to this {@link OllamaObject}
* @param messages the message list to add
*/
public void addAllMessages(List<OllamaMessage> messages) {
this.messages.addAll(messages);
}
/**
* Adds a system message to the messages.
* <p>
* Equvalent to {@code OllamaObject#addMessage(new OllamaMessage(OllamaMessageRole.SYSTEM, <message>))}
* @param message
*/
public void addSystemMessage(String message) {
messages.add(new OllamaMessage(OllamaMessageRole.SYSTEM, message));
}
@Override
public String toString() {
JSONObject json = new JSONObject();