From 40a3aa31dc960b4ded67a43b70a3fbe2420e0538 Mon Sep 17 00:00:00 2001 From: Markus Klenke Date: Sat, 10 Feb 2024 22:51:47 +0000 Subject: [PATCH] Adds helper reset method to OllamaChatRequestBuilder --- .../ollama4j/core/models/chat/OllamaChatRequestBuilder.java | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/main/java/io/github/amithkoujalgi/ollama4j/core/models/chat/OllamaChatRequestBuilder.java b/src/main/java/io/github/amithkoujalgi/ollama4j/core/models/chat/OllamaChatRequestBuilder.java index 69ded1b..af28b7b 100644 --- a/src/main/java/io/github/amithkoujalgi/ollama4j/core/models/chat/OllamaChatRequestBuilder.java +++ b/src/main/java/io/github/amithkoujalgi/ollama4j/core/models/chat/OllamaChatRequestBuilder.java @@ -25,6 +25,10 @@ public class OllamaChatRequestBuilder { return request; } + public void reset(){ + request = new OllamaChatRequestModel(request.getModel(), new ArrayList<>()); + } + public OllamaChatRequestBuilder withMessage(OllamaChatMessageRole role, String content, File... images){ List messages = this.request.getMessages(); messages.add(new OllamaChatMessage(role,content,List.of(images)));