test(ollama): request options customization

Also expose baseUrl to allow custom clients.

Author: ptitjes

integration-tests/src/jvmMain/kotlin/ai/koog/integration/tests/OllamaTestFixture.kt

@@ -18,6 +18,7 @@ class OllamaTestFixture {
 
     private lateinit var ollamaContainer: GenericContainer<*>
 
+    lateinit var baseUrl: String
     lateinit var client: OllamaClient
     lateinit var executor: SingleLLMPromptExecutor
     val model = OllamaModels.Meta.LLAMA_3_2
@@ -31,7 +32,7 @@ class OllamaTestFixture {
 
         val host = ollamaContainer.host
         val port = ollamaContainer.getMappedPort(PORT)
-        val baseUrl = "http://$host:$port"
+        baseUrl = "http://$host:$port"
         waitForOllamaServer(baseUrl)
 
         client = OllamaClient(baseUrl)

integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/OllamaClientIntegrationTest.kt

@@ -4,14 +4,17 @@ import ai.koog.agents.core.tools.ToolDescriptor
 import ai.koog.agents.core.tools.ToolParameterDescriptor
 import ai.koog.agents.core.tools.ToolParameterType
 import ai.koog.prompt.dsl.Prompt
+import ai.koog.prompt.executor.llms.SingleLLMPromptExecutor
 import ai.koog.prompt.executor.model.PromptExecutorExt.execute
+import ai.koog.prompt.executor.ollama.client.OllamaClient
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.flow
 import kotlinx.coroutines.test.runTest
 import kotlinx.serialization.Serializable
 import org.junit.jupiter.api.Disabled
 import org.junit.jupiter.api.extension.ExtendWith
 import kotlin.test.Test
+import kotlin.test.assertEquals
 import kotlin.test.assertTrue
 import kotlin.time.Duration.Companion.seconds
 
@@ -20,6 +23,7 @@ class OllamaClientIntegrationTest {
     companion object {
         @field:InjectOllamaTestFixture
         private lateinit var fixture: OllamaTestFixture
+        private val baseUrl get() = fixture.baseUrl
         private val executor get() = fixture.executor
         private val model get() = fixture.model
     }
@@ -37,6 +41,30 @@ class OllamaClientIntegrationTest {
         assertTrue(response.content.contains("Paris"), "Response should contain 'Paris'")
     }
 
+    @Test
+    fun `ollama_test execute simple prompt with options`() = runTest(timeout = 600.seconds) {
+        // Create a custom client
+        val client = OllamaClient(
+            baseUrl = baseUrl,
+            requestBuilderAction = { _, _ ->
+                seed = 0
+                numCtx = 100
+                numPredict = 10
+            },
+        )
+        val executor = SingleLLMPromptExecutor(client)
+
+        val prompt = Prompt.build("test-with-options") {
+            system("You are a helpful assistant.")
+            user("What is the capital of France?")
+        }
+
+        val response = executor.execute(prompt = prompt, model = model)
+
+        assertTrue(response.content.isNotEmpty(), "Response should not be empty")
+        assertEquals("The capital of France is Paris.", response.content)
+    }
+
     @Test
     fun `ollama_test execute tools with required parameters`() = runTest(timeout = 600.seconds) {
         val searchTool = ToolDescriptor(
@@ -607,4 +635,4 @@ class OllamaClientIntegrationTest {
             println()
         }
     }
-}
+}