seeseemelk/ollama4j
1
0
Fork 0
Code Issues Pull Requests Actions Packages Releases 1 Activity

Adds helper reset method to OllamaChatRequestBuilder

Browse Source
This commit is contained in:
Markus Klenke
2024-02-10 22:51:47 +00:00
parent 90669b611b
commit 40a3aa31dc
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/io/github/amithkoujalgi/ollama4j/core/models/chat/OllamaChatRequestBuilder.java
View File

@@ -25,6 +25,10 @@ public class OllamaChatRequestBuilder {
return request; return request;
} }
public void reset(){
request = new OllamaChatRequestModel(request.getModel(), new ArrayList<>());
}
public OllamaChatRequestBuilder withMessage(OllamaChatMessageRole role, String content, File... images){ public OllamaChatRequestBuilder withMessage(OllamaChatMessageRole role, String content, File... images){
List<OllamaChatMessage> messages = this.request.getMessages(); List<OllamaChatMessage> messages = this.request.getMessages();
messages.add(new OllamaChatMessage(role,content,List.of(images))); messages.add(new OllamaChatMessage(role,content,List.of(images)));