Add capabilities integration tests

Signed-off-by: Sergey Karpov <[email protected]>

Author: skarpovdev

integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/AIAgentIntegrationTest.kt renamed to integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/agent/AIAgentIntegrationTest.kt

@@ -1,4 +1,4 @@
-package ai.koog.integration.tests
+package ai.koog.integration.tests.agent
 
 import ai.koog.agents.core.agent.AIAgent
 import ai.koog.agents.core.agent.ToolCalls
@@ -57,6 +57,7 @@ import org.junit.jupiter.params.provider.Arguments
 import org.junit.jupiter.params.provider.MethodSource
 import java.nio.file.Path
 import java.nio.file.Paths
+import java.util.Base64
 import java.util.stream.Stream
 import kotlin.io.path.readBytes
 import kotlin.reflect.typeOf
@@ -423,7 +424,7 @@ class AIAgentIntegrationTest {
         val imageFile = testResourcesDir.resolve("test.png")
 
         val imageBytes = imageFile.readBytes()
-        val base64Image = java.util.Base64.getEncoder().encodeToString(imageBytes)
+        val base64Image = Base64.getEncoder().encodeToString(imageBytes)
 
         val promptWithImage = """
             I'm sending you an image encoded in base64 format.

integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/AIAgentMultipleLLMIntegrationTest.kt renamed to integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/agent/AIAgentMultipleLLMIntegrationTest.kt

@@ -1,4 +1,4 @@
-package ai.koog.integration.tests
+package ai.koog.integration.tests.agent
 
 import ai.koog.agents.core.agent.AIAgent
 import ai.koog.agents.core.agent.AIAgentException
@@ -22,7 +22,7 @@ import ai.koog.agents.core.tools.ToolResult
 import ai.koog.agents.features.eventHandler.feature.EventHandler
 import ai.koog.agents.features.eventHandler.feature.EventHandlerConfig
 import ai.koog.agents.features.tracing.feature.Tracing
-import ai.koog.integration.tests.ReportingLLMLLMClient.Event
+import ai.koog.integration.tests.agent.ReportingLLMLLMClient.Event
 import ai.koog.integration.tests.utils.Models
 import ai.koog.integration.tests.utils.RetryUtils.withRetry
 import ai.koog.integration.tests.utils.TestLogPrinter
@@ -60,6 +60,7 @@ import org.junit.jupiter.params.ParameterizedTest
 import org.junit.jupiter.params.provider.Arguments
 import org.junit.jupiter.params.provider.MethodSource
 import java.io.File
+import java.util.Base64
 import java.util.stream.Stream
 import kotlin.test.Ignore
 import kotlin.test.assertContains
@@ -841,7 +842,7 @@ class AIAgentMultipleLLMIntegrationTest {
         assertTrue(imageFile.exists(), "Image test file should exist")
 
         val imageBytes = imageFile.readBytes()
-        val base64Image = java.util.Base64.getEncoder().encodeToString(imageBytes)
+        val base64Image = Base64.getEncoder().encodeToString(imageBytes)
 
         withRetry {
             val agent = when (model.provider) {

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

@@ -1,4 +1,4 @@
-package ai.koog.integration.tests
+package ai.koog.integration.tests.agent
 
 import ai.koog.agents.core.agent.AIAgent
 import ai.koog.agents.core.agent.config.AIAgentConfig
@@ -13,6 +13,9 @@ import ai.koog.agents.core.dsl.extension.onAssistantMessage
 import ai.koog.agents.core.dsl.extension.onToolCall
 import ai.koog.agents.core.tools.ToolRegistry
 import ai.koog.agents.features.eventHandler.feature.EventHandler
+import ai.koog.integration.tests.InjectOllamaTestFixture
+import ai.koog.integration.tests.OllamaTestFixture
+import ai.koog.integration.tests.OllamaTestFixtureExtension
 import ai.koog.integration.tests.tools.AnswerVerificationTool
 import ai.koog.integration.tests.tools.GenericParameterTool
 import ai.koog.integration.tests.tools.GeographyQueryTool

integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/OllamaSimpleAgentIntegrationTest.kt renamed to integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/agent/OllamaSimpleAgentIntegrationTest.kt

@@ -1,10 +1,13 @@
-package ai.koog.integration.tests
+package ai.koog.integration.tests.agent
 
 import ai.koog.agents.core.agent.AIAgent
 import ai.koog.agents.core.tools.ToolRegistry
 import ai.koog.agents.ext.tool.SayToUser
 import ai.koog.agents.features.eventHandler.feature.EventHandler
 import ai.koog.agents.features.eventHandler.feature.EventHandlerConfig
+import ai.koog.integration.tests.InjectOllamaTestFixture
+import ai.koog.integration.tests.OllamaTestFixture
+import ai.koog.integration.tests.OllamaTestFixtureExtension
 import ai.koog.integration.tests.utils.annotations.Retry
 import ai.koog.integration.tests.utils.annotations.RetryExtension
 import kotlinx.coroutines.test.runTest