Refactor OllamaChatRequest and OllamaGenerateRequest to remove builder classes, implement builder-like methods directly in the request classes, and enhance request handling with additional options and image support. Update integration tests to reflect these changes.

2025-11-05 02:50:51 +01:00 · 2025-10-07 18:33:22 +05:30
parent 46f2d62fed
commit 64c629775a
12 changed files with 301 additions and 415 deletions
--- a/src/test/java/io/github/ollama4j/integrationtests/OllamaIntegrationTest.java
+++ b/src/test/java/io/github/ollama4j/integrationtests/OllamaIntegrationTest.java
@@ -18,7 +18,6 @@ import io.github.ollama4j.models.chat.*;
 import io.github.ollama4j.models.embed.OllamaEmbedRequest;
 import io.github.ollama4j.models.embed.OllamaEmbedResult;
 import io.github.ollama4j.models.generate.OllamaGenerateRequest;
-import io.github.ollama4j.models.generate.OllamaGenerateRequestBuilder;
 import io.github.ollama4j.models.generate.OllamaGenerateStreamObserver;
 import io.github.ollama4j.models.response.Model;
 import io.github.ollama4j.models.response.ModelDetail;
@@ -272,7 +271,7 @@ class OllamaIntegrationTest {
        format.put("required", List.of("isNoon"));

        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(TOOLS_MODEL)
                        .withPrompt(prompt)
                        .withFormat(format)
@@ -299,7 +298,7 @@ class OllamaIntegrationTest {
        boolean raw = false;
        boolean thinking = false;
        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(GENERAL_PURPOSE_MODEL)
                        .withPrompt(
                                "What is the capital of France? And what's France's connection with"
@@ -327,7 +326,7 @@ class OllamaIntegrationTest {
        api.pullModel(GENERAL_PURPOSE_MODEL);
        boolean raw = false;
        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(GENERAL_PURPOSE_MODEL)
                        .withPrompt(
                                "What is the capital of France? And what's France's connection with"
@@ -357,8 +356,7 @@ class OllamaIntegrationTest {
    void shouldGenerateWithCustomOptions() throws OllamaException {
        api.pullModel(GENERAL_PURPOSE_MODEL);

-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(GENERAL_PURPOSE_MODEL);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(GENERAL_PURPOSE_MODEL);
        OllamaChatRequest requestModel =
                builder.withMessage(
                                OllamaChatMessageRole.SYSTEM,
@@ -390,8 +388,7 @@ class OllamaIntegrationTest {

        String expectedResponse = "Bhai";

-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(GENERAL_PURPOSE_MODEL);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(GENERAL_PURPOSE_MODEL);
        OllamaChatRequest requestModel =
                builder.withMessage(
                                OllamaChatMessageRole.SYSTEM,
@@ -429,8 +426,7 @@ class OllamaIntegrationTest {
    @Order(10)
    void shouldChatWithHistory() throws Exception {
        api.pullModel(THINKING_TOOL_MODEL);
-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(THINKING_TOOL_MODEL);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(THINKING_TOOL_MODEL);

        OllamaChatRequest requestModel =
                builder.withMessage(
@@ -481,8 +477,7 @@ class OllamaIntegrationTest {
    void shouldChatWithExplicitTool() throws OllamaException {
        String theToolModel = TOOLS_MODEL;
        api.pullModel(theToolModel);
-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(theToolModel);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(theToolModel);

        api.registerTool(EmployeeFinderToolSpec.getSpecification());

@@ -534,8 +529,7 @@ class OllamaIntegrationTest {
    void shouldChatWithExplicitToolAndUseTools() throws OllamaException {
        String theToolModel = TOOLS_MODEL;
        api.pullModel(theToolModel);
-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(theToolModel);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(theToolModel);

        api.registerTool(EmployeeFinderToolSpec.getSpecification());

@@ -579,8 +573,7 @@ class OllamaIntegrationTest {
        String theToolModel = TOOLS_MODEL;
        api.pullModel(theToolModel);

-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(theToolModel);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(theToolModel);

        api.registerTool(EmployeeFinderToolSpec.getSpecification());

@@ -633,8 +626,7 @@ class OllamaIntegrationTest {
    void shouldChatWithAnnotatedToolSingleParam() throws OllamaException {
        String theToolModel = TOOLS_MODEL;
        api.pullModel(theToolModel);
-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(theToolModel);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(theToolModel);

        api.registerAnnotatedTools();

@@ -680,8 +672,7 @@ class OllamaIntegrationTest {
    void shouldChatWithAnnotatedToolMultipleParams() throws OllamaException {
        String theToolModel = TOOLS_MODEL;
        api.pullModel(theToolModel);
-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(theToolModel);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(theToolModel);

        api.registerAnnotatedTools(new AnnotatedTool());

@@ -712,8 +703,7 @@ class OllamaIntegrationTest {
    void shouldChatWithStream() throws OllamaException {
        api.deregisterTools();
        api.pullModel(GENERAL_PURPOSE_MODEL);
-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(GENERAL_PURPOSE_MODEL);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(GENERAL_PURPOSE_MODEL);
        OllamaChatRequest requestModel =
                builder.withMessage(
                                OllamaChatMessageRole.USER,
@@ -739,8 +729,7 @@ class OllamaIntegrationTest {
    @Order(15)
    void shouldChatWithThinkingAndStream() throws OllamaException {
        api.pullModel(THINKING_TOOL_MODEL_2);
-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(THINKING_TOOL_MODEL_2);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(THINKING_TOOL_MODEL_2);
        OllamaChatRequest requestModel =
                builder.withMessage(
                                OllamaChatMessageRole.USER,
@@ -758,32 +747,6 @@ class OllamaIntegrationTest {
        assertNotNull(chatResult.getResponseModel().getMessage().getResponse());
    }

-    /**
-     * Tests chat API with an image input from a URL.
-     *
-     * <p>Scenario: Sends a user message with an image URL and verifies the assistant's response.
-     * Usage: chat, vision model, image from URL, no tools, no thinking, no streaming.
-     */
-    @Test
-    @Order(10)
-    void shouldChatWithImageFromURL() throws OllamaException, IOException, InterruptedException {
-        api.pullModel(VISION_MODEL);
-
-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(VISION_MODEL);
-        OllamaChatRequest requestModel =
-                builder.withMessage(
-                                OllamaChatMessageRole.USER,
-                                "What's in the picture?",
-                                Collections.emptyList(),
-                                "https://t3.ftcdn.net/jpg/02/96/63/80/360_F_296638053_0gUVA4WVBKceGsIr7LNqRWSnkusi07dq.jpg")
-                        .build();
-        api.registerAnnotatedTools(new OllamaIntegrationTest());
-
-        OllamaChatResult chatResult = api.chat(requestModel, null);
-        assertNotNull(chatResult);
-    }
-
    /**
     * Tests chat API with an image input from a file and multi-turn history.
     *
@@ -795,8 +758,7 @@ class OllamaIntegrationTest {
    @Order(10)
    void shouldChatWithImageFromFileAndHistory() throws OllamaException {
        api.pullModel(VISION_MODEL);
-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(VISION_MODEL);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(VISION_MODEL);
        OllamaChatRequest requestModel =
                builder.withMessage(
                                OllamaChatMessageRole.USER,
@@ -832,7 +794,7 @@ class OllamaIntegrationTest {
        api.pullModel(VISION_MODEL);
        try {
            OllamaGenerateRequest request =
-                    OllamaGenerateRequestBuilder.builder()
+                    OllamaGenerateRequest.builder()
                            .withModel(VISION_MODEL)
                            .withPrompt("What is in this image?")
                            .withRaw(false)
@@ -865,7 +827,7 @@ class OllamaIntegrationTest {
    void shouldGenerateWithImageFilesAndResponseStreamed() throws OllamaException, IOException {
        api.pullModel(VISION_MODEL);
        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(VISION_MODEL)
                        .withPrompt("What is in this image?")
                        .withRaw(false)
@@ -900,7 +862,7 @@ class OllamaIntegrationTest {
        boolean think = true;

        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(THINKING_TOOL_MODEL)
                        .withPrompt("Who are you?")
                        .withRaw(raw)
@@ -929,7 +891,7 @@ class OllamaIntegrationTest {
        api.pullModel(THINKING_TOOL_MODEL);
        boolean raw = false;
        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(THINKING_TOOL_MODEL)
                        .withPrompt("Who are you?")
                        .withRaw(raw)
@@ -967,7 +929,7 @@ class OllamaIntegrationTest {
        boolean raw = true;
        boolean thinking = false;
        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(GENERAL_PURPOSE_MODEL)
                        .withPrompt("What is 2+2?")
                        .withRaw(raw)
@@ -995,7 +957,7 @@ class OllamaIntegrationTest {
        api.pullModel(GENERAL_PURPOSE_MODEL);
        boolean raw = true;
        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(GENERAL_PURPOSE_MODEL)
                        .withPrompt("What is the largest planet in our solar system?")
                        .withRaw(raw)
@@ -1028,7 +990,7 @@ class OllamaIntegrationTest {
        // 'response' tokens
        boolean raw = true;
        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(THINKING_TOOL_MODEL)
                        .withPrompt(
                                "Count 1 to 5. Just give me the numbers and do not give any other"
@@ -1093,7 +1055,7 @@ class OllamaIntegrationTest {
        format.put("required", List.of("cities"));

        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(TOOLS_MODEL)
                        .withPrompt(prompt)
                        .withFormat(format)
@@ -1119,8 +1081,7 @@ class OllamaIntegrationTest {
    @Order(26)
    void shouldChatWithThinkingNoStream() throws OllamaException {
        api.pullModel(THINKING_TOOL_MODEL);
-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(THINKING_TOOL_MODEL);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(THINKING_TOOL_MODEL);
        OllamaChatRequest requestModel =
                builder.withMessage(
                                OllamaChatMessageRole.USER,
@@ -1149,8 +1110,7 @@ class OllamaIntegrationTest {
    void shouldChatWithCustomOptionsAndStreaming() throws OllamaException {
        api.pullModel(GENERAL_PURPOSE_MODEL);

-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(GENERAL_PURPOSE_MODEL);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(GENERAL_PURPOSE_MODEL);
        OllamaChatRequest requestModel =
                builder.withMessage(
                                OllamaChatMessageRole.USER,
@@ -1184,8 +1144,7 @@ class OllamaIntegrationTest {

        api.registerTool(EmployeeFinderToolSpec.getSpecification());

-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(THINKING_TOOL_MODEL_2);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(THINKING_TOOL_MODEL_2);
        OllamaChatRequest requestModel =
                builder.withMessage(
                                OllamaChatMessageRole.USER,
@@ -1219,8 +1178,7 @@ class OllamaIntegrationTest {
        File image1 = getImageFileFromClasspath("emoji-smile.jpeg");
        File image2 = getImageFileFromClasspath("roses.jpg");

-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(VISION_MODEL);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(VISION_MODEL);
        OllamaChatRequest requestModel =
                builder.withMessage(
                                OllamaChatMessageRole.USER,
@@ -1247,7 +1205,7 @@ class OllamaIntegrationTest {
    void shouldHandleNonExistentModel() {
        String nonExistentModel = "this-model-does-not-exist:latest";
        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(nonExistentModel)
                        .withPrompt("Hello")
                        .withRaw(false)
@@ -1274,8 +1232,7 @@ class OllamaIntegrationTest {
        api.pullModel(GENERAL_PURPOSE_MODEL);

        List<OllamaChatToolCalls> tools = Collections.emptyList();
-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(GENERAL_PURPOSE_MODEL);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(GENERAL_PURPOSE_MODEL);
        OllamaChatRequest requestModel =
                builder.withMessage(OllamaChatMessageRole.USER, "   ", tools) // whitespace only
                        .build();
@@ -1298,7 +1255,7 @@ class OllamaIntegrationTest {
    void shouldGenerateWithExtremeParameters() throws OllamaException {
        api.pullModel(GENERAL_PURPOSE_MODEL);
        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(GENERAL_PURPOSE_MODEL)
                        .withPrompt("Generate a random word")
                        .withRaw(false)
@@ -1351,8 +1308,7 @@ class OllamaIntegrationTest {
    void shouldChatWithKeepAlive() throws OllamaException {
        api.pullModel(GENERAL_PURPOSE_MODEL);

-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder().withModel(GENERAL_PURPOSE_MODEL);
+        OllamaChatRequest builder = OllamaChatRequest.builder().withModel(GENERAL_PURPOSE_MODEL);
        OllamaChatRequest requestModel =
                builder.withMessage(OllamaChatMessageRole.USER, "Hello, how are you?")
                        .withKeepAlive("5m") // Keep model loaded for 5 minutes
@@ -1376,7 +1332,7 @@ class OllamaIntegrationTest {
    void shouldGenerateWithAdvancedOptions() throws OllamaException {
        api.pullModel(GENERAL_PURPOSE_MODEL);
        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(GENERAL_PURPOSE_MODEL)
                        .withPrompt("Write a detailed explanation of machine learning")
                        .withRaw(false)
@@ -1421,8 +1377,8 @@ class OllamaIntegrationTest {
                    new Thread(
                            () -> {
                                try {
-                                    OllamaChatRequestBuilder builder =
-                                            OllamaChatRequestBuilder.builder()
+                                    OllamaChatRequest builder =
+                                            OllamaChatRequest.builder()
                                                    .withModel(GENERAL_PURPOSE_MODEL);
                                    OllamaChatRequest requestModel =
                                            builder.withMessage(
--- a/src/test/java/io/github/ollama4j/integrationtests/WithAuth.java
+++ b/src/test/java/io/github/ollama4j/integrationtests/WithAuth.java
@@ -13,7 +13,6 @@ import static org.junit.jupiter.api.Assertions.*;
 import io.github.ollama4j.Ollama;
 import io.github.ollama4j.exceptions.OllamaException;
 import io.github.ollama4j.models.generate.OllamaGenerateRequest;
-import io.github.ollama4j.models.generate.OllamaGenerateRequestBuilder;
 import io.github.ollama4j.models.generate.OllamaGenerateStreamObserver;
 import io.github.ollama4j.models.response.OllamaResult;
 import io.github.ollama4j.samples.AnnotatedTool;
@@ -205,7 +204,7 @@ public class WithAuth {
        format.put("required", List.of("isNoon"));

        OllamaGenerateRequest request =
-                OllamaGenerateRequestBuilder.builder()
+                OllamaGenerateRequest.builder()
                        .withModel(model)
                        .withPrompt(prompt)
                        .withRaw(false)
--- a/src/test/java/io/github/ollama4j/unittests/TestMockedAPIs.java
+++ b/src/test/java/io/github/ollama4j/unittests/TestMockedAPIs.java
@@ -19,7 +19,6 @@ import io.github.ollama4j.models.chat.OllamaChatMessageRole;
 import io.github.ollama4j.models.embed.OllamaEmbedRequest;
 import io.github.ollama4j.models.embed.OllamaEmbedResult;
 import io.github.ollama4j.models.generate.OllamaGenerateRequest;
-import io.github.ollama4j.models.generate.OllamaGenerateRequestBuilder;
 import io.github.ollama4j.models.generate.OllamaGenerateStreamObserver;
 import io.github.ollama4j.models.request.CustomModelRequest;
 import io.github.ollama4j.models.response.ModelDetail;
@@ -158,7 +157,7 @@ class TestMockedAPIs {
        OllamaGenerateStreamObserver observer = new OllamaGenerateStreamObserver(null, null);
        try {
            OllamaGenerateRequest request =
-                    OllamaGenerateRequestBuilder.builder()
+                    OllamaGenerateRequest.builder()
                            .withModel(model)
                            .withPrompt(prompt)
                            .withRaw(false)
@@ -180,7 +179,7 @@ class TestMockedAPIs {
        String prompt = "some prompt text";
        try {
            OllamaGenerateRequest request =
-                    OllamaGenerateRequestBuilder.builder()
+                    OllamaGenerateRequest.builder()
                            .withModel(model)
                            .withPrompt(prompt)
                            .withRaw(false)
@@ -206,7 +205,7 @@ class TestMockedAPIs {
        String prompt = "some prompt text";
        try {
            OllamaGenerateRequest request =
-                    OllamaGenerateRequestBuilder.builder()
+                    OllamaGenerateRequest.builder()
                            .withModel(model)
                            .withPrompt(prompt)
                            .withRaw(false)
--- a/src/test/java/io/github/ollama4j/unittests/TestOllamaChatRequestBuilder.java
+++ b/src/test/java/io/github/ollama4j/unittests/TestOllamaChatRequestBuilder.java
@@ -12,15 +12,14 @@ import static org.junit.jupiter.api.Assertions.*;

 import io.github.ollama4j.models.chat.OllamaChatMessageRole;
 import io.github.ollama4j.models.chat.OllamaChatRequest;
-import io.github.ollama4j.models.chat.OllamaChatRequestBuilder;
 import org.junit.jupiter.api.Test;

 class TestOllamaChatRequestBuilder {

    @Test
    void testResetClearsMessagesButKeepsModelAndThink() {
-        OllamaChatRequestBuilder builder =
-                OllamaChatRequestBuilder.builder()
+        OllamaChatRequest builder =
+                OllamaChatRequest.builder()
                        .withModel("my-model")
                        .withThinking(true)
                        .withMessage(OllamaChatMessageRole.USER, "first");
--- a/src/test/java/io/github/ollama4j/unittests/jackson/TestChatRequestSerialization.java
+++ b/src/test/java/io/github/ollama4j/unittests/jackson/TestChatRequestSerialization.java
@@ -13,7 +13,6 @@ import static org.junit.jupiter.api.Assertions.assertThrowsExactly;

 import io.github.ollama4j.models.chat.OllamaChatMessageRole;
 import io.github.ollama4j.models.chat.OllamaChatRequest;
-import io.github.ollama4j.models.chat.OllamaChatRequestBuilder;
 import io.github.ollama4j.utils.OptionsBuilder;
 import java.io.File;
 import java.util.Collections;
@@ -24,11 +23,11 @@ import org.junit.jupiter.api.Test;

 public class TestChatRequestSerialization extends AbstractSerializationTest<OllamaChatRequest> {

-    private OllamaChatRequestBuilder builder;
+    private OllamaChatRequest builder;

    @BeforeEach
    public void init() {
-        builder = OllamaChatRequestBuilder.builder().withModel("DummyModel");
+        builder = OllamaChatRequest.builder().withModel("DummyModel");
    }

    @Test
--- a/src/test/java/io/github/ollama4j/unittests/jackson/TestGenerateRequestSerialization.java
+++ b/src/test/java/io/github/ollama4j/unittests/jackson/TestGenerateRequestSerialization.java
@@ -11,7 +11,6 @@ package io.github.ollama4j.unittests.jackson;
 import static org.junit.jupiter.api.Assertions.assertEquals;

 import io.github.ollama4j.models.generate.OllamaGenerateRequest;
-import io.github.ollama4j.models.generate.OllamaGenerateRequestBuilder;
 import io.github.ollama4j.utils.OptionsBuilder;
 import org.json.JSONObject;
 import org.junit.jupiter.api.BeforeEach;
@@ -19,16 +18,17 @@ import org.junit.jupiter.api.Test;

 class TestGenerateRequestSerialization extends AbstractSerializationTest<OllamaGenerateRequest> {

-    private OllamaGenerateRequestBuilder builder;
+    private OllamaGenerateRequest builder;

    @BeforeEach
    public void init() {
-        builder = OllamaGenerateRequestBuilder.builder().withModel("Dummy Model");
+        builder = OllamaGenerateRequest.builder().withModel("Dummy Model");
    }

    @Test
    public void testRequestOnlyMandatoryFields() {
-        OllamaGenerateRequest req = builder.withPrompt("Some prompt").build();
+        OllamaGenerateRequest req =
+                builder.withPrompt("Some prompt").withModel("Dummy Model").build();

        String jsonRequest = serialize(req);
        assertEqualsAfterUnmarshalling(deserialize(jsonRequest, OllamaGenerateRequest.class), req);
@@ -38,7 +38,10 @@ class TestGenerateRequestSerialization extends AbstractSerializationTest<OllamaG
    public void testRequestWithOptions() {
        OptionsBuilder b = new OptionsBuilder();
        OllamaGenerateRequest req =
-                builder.withPrompt("Some prompt").withOptions(b.setMirostat(1).build()).build();
+                builder.withPrompt("Some prompt")
+                        .withOptions(b.setMirostat(1).build())
+                        .withModel("Dummy Model")
+                        .build();

        String jsonRequest = serialize(req);
        OllamaGenerateRequest deserializeRequest =
@@ -49,7 +52,11 @@ class TestGenerateRequestSerialization extends AbstractSerializationTest<OllamaG

    @Test
    public void testWithJsonFormat() {
-        OllamaGenerateRequest req = builder.withPrompt("Some prompt").withGetJsonResponse().build();
+        OllamaGenerateRequest req =
+                builder.withPrompt("Some prompt")
+                        .withGetJsonResponse()
+                        .withModel("Dummy Model")
+                        .build();

        String jsonRequest = serialize(req);
        System.out.printf(jsonRequest);