Fix integration tests compilation in develop (#1283)

Author: tiginamaria

integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/executor/ExecutorIntegrationTestBase.kt

@@ -222,42 +222,6 @@ abstract class ExecutorIntegrationTestBase {
         }
     }
 
-    open fun integration_testExecuteStreamingWithTools(model: LLModel) = runTest(timeout = 300.seconds) {
-        Models.assumeAvailable(model.provider)
-        assumeTrue(model.capabilities.contains(LLMCapability.Tools), "Model $model does not support tools")
-
-        val executor = getExecutor(model)
-
-        val prompt = Prompt.build("test-streaming", LLMParams(toolChoice = ToolChoice.Required)) {
-            system("You are a helpful assistant.")
-            user("Count three times five")
-        }
-
-        withRetry(times = 3, testName = "integration_testExecuteStreamingWithTools[${model.id}]") {
-            with(StringBuilder()) {
-                val endMessages = mutableListOf<StreamFrame.End>()
-                val toolMessages = mutableListOf<StreamFrame.ToolCall>()
-
-                executor.executeStreamAndCollect(
-                    prompt = prompt,
-                    model = model,
-                    tools = listOf(SimpleCalculatorTool.descriptor),
-                    appendable = this,
-                    endMessages = endMessages,
-                    toolMessages = toolMessages
-                )
-
-                toolMessages.shouldNotBeEmpty()
-                withClue("Expected calculator tool call but got: [$toolMessages]") {
-                    toolMessages.any {
-                        it.name == SimpleCalculatorTool.name &&
-                            it.content.contains(CalculatorOperation.MULTIPLY.name, ignoreCase = true)
-                    } shouldBe true
-                }
-            }
-        }
-    }
-
     open fun integration_testToolWithRequiredParams(model: LLModel) = runTest(timeout = 300.seconds) {
         Models.assumeAvailable(model.provider)
         assumeTrue(model.capabilities.contains(LLMCapability.Tools), "Model $model does not support tools")

integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/executor/MultipleLLMPromptExecutorIntegrationTest.kt

@@ -10,7 +10,6 @@ import ai.koog.integration.tests.utils.getLLMClientForProvider
 import ai.koog.prompt.executor.llms.MultiLLMPromptExecutor
 import ai.koog.prompt.executor.model.PromptExecutor
 import ai.koog.prompt.llm.LLModel
-import org.junit.jupiter.api.Disabled
 import org.junit.jupiter.params.ParameterizedTest
 import org.junit.jupiter.params.provider.Arguments
 import org.junit.jupiter.params.provider.MethodSource
@@ -114,13 +113,6 @@ class MultipleLLMPromptExecutorIntegrationTest : ExecutorIntegrationTestBase() {
         super.integration_testExecuteStreaming(model)
     }
 
-    @Disabled("KG-616")
-    @ParameterizedTest
-    @MethodSource("allCompletionModels")
-    override fun integration_testExecuteStreamingWithTools(model: LLModel) {
-        super.integration_testExecuteStreamingWithTools(model)
-    }
-
     @ParameterizedTest
     @MethodSource("allCompletionModels")
     override fun integration_testExecuteStreamingWithTools(model: LLModel) {

integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/executor/SingleLLMPromptExecutorIntegrationTest.kt

@@ -11,7 +11,6 @@ import ai.koog.prompt.executor.llms.SingleLLMPromptExecutor
 import ai.koog.prompt.executor.model.PromptExecutor
 import ai.koog.prompt.llm.LLMProvider
 import ai.koog.prompt.llm.LLModel
-import org.junit.jupiter.api.Disabled
 import org.junit.jupiter.params.ParameterizedTest
 import org.junit.jupiter.params.provider.Arguments
 import org.junit.jupiter.params.provider.MethodSource
@@ -108,13 +107,6 @@ class SingleLLMPromptExecutorIntegrationTest : ExecutorIntegrationTestBase() {
         super.integration_testExecuteStreaming(model)
     }
 
-    @Disabled("KG-616")
-    @ParameterizedTest
-    @MethodSource("allCompletionModels")
-    override fun integration_testExecuteStreamingWithTools(model: LLModel) {
-        super.integration_testExecuteStreamingWithTools(model)
-    }
-
     @ParameterizedTest
     @MethodSource("allCompletionModels")
     override fun integration_testExecuteStreamingWithTools(model: LLModel) {