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:
@@ -232,6 +232,36 @@ public class OllamaObject {
|
|||||||
messages.add(message);
|
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
|
@Override
|
||||||
public String toString() {
|
public String toString() {
|
||||||
JSONObject json = new JSONObject();
|
JSONObject json = new JSONObject();
|
||||||
|
|||||||
Reference in New Issue
Block a user