forked from neurodock/NeuroDock
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);
|
||||
}
|
||||
|
||||
/**
|
||||
* 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();
|
||||
|
||||
Reference in New Issue
Block a user