pull the model for the integration tests

Author: antoniibelyshev

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

@@ -41,6 +41,7 @@ class OllamaTestFixture {
     val model = OllamaModels.Meta.LLAMA_3_2
     val visionModel = OllamaModels.Granite.GRANITE_3_2_VISION
     val moderationModel = OllamaModels.Meta.LLAMA_GUARD_3
+    val modelsWithHallucinations = listOf(OllamaModels.Meta.LLAMA_3_2, OllamaModels.Groq.LLAMA_3_GROK_TOOL_USE_8B)
 
     private lateinit var ollamaContainer: GenericContainer<*>
 
@@ -67,6 +68,7 @@ class OllamaTestFixture {
                     client.getModelOrNull(model.id, pullIfMissing = true)
                     client.getModelOrNull(visionModel.id, pullIfMissing = true)
                     client.getModelOrNull(moderationModel.id, pullIfMissing = true)
+                    modelsWithHallucinations.forEach { client.getModelOrNull(it.id, pullIfMissing = true) }
                 } catch (e: Exception) {
                     logger.error(e) { "Failed to pull models: ${e.message}" }
                     cleanContainer()

integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/agent/OllamaAgentIntegrationTest.kt

@@ -54,12 +54,11 @@ class OllamaAgentIntegrationTest : AIAgentTestBase() {
         private lateinit var fixture: OllamaTestFixture
         private val executor get() = fixture.executor
         private val model get() = fixture.model
+        private val modelsWithHallucinations get() = fixture.modelsWithHallucinations
 
         @JvmStatic
-        private fun weakOllamaModels() = Stream.of(
-            OllamaModels.Meta.LLAMA_3_2,
-            OllamaModels.Groq.LLAMA_3_GROK_TOOL_USE_8B,
-        )
+        private fun modelsWithHallucinations(): Stream<LLModel> =
+            Stream.of(*modelsWithHallucinations.toTypedArray())
     }
 
     @BeforeTest
@@ -209,8 +208,8 @@ class OllamaAgentIntegrationTest : AIAgentTestBase() {
 
     @OptIn(ResponseProcessorApi::class)
     @ParameterizedTest
-    @MethodSource("weakOllamaModels")
-    fun ollama_testLLMBasedToolCallExtractor(llmModel: LLModel) = runTest(timeout = 600.seconds) {
+    @MethodSource("modelsWithHallucinations")
+    fun ollama_testFixToolCallLLMBased(llmModel: LLModel) = runTest(timeout = 600.seconds) {
         withRetry(5) {
             val fileTools = FileOperationsTools()
             fileTools.createNewFileWithText(
@@ -222,7 +221,7 @@ class OllamaAgentIntegrationTest : AIAgentTestBase() {
                 Charlie,22,78
                 """.trimIndent()
             )
-            val toolRegistry = ToolRegistry.Companion {
+            val toolRegistry = ToolRegistry {
                 tool(fileTools.readFileContentTool)
                 tool(fileTools.createNewFileWithTextTool)
             }