Skip to content

Commit a4eeeeb

Browse files
author
Malte Heuser
committed
KG-561 resolve PR comments
1 parent f04e8c1 commit a4eeeeb

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ import ai.koog.prompt.executor.clients.LLMClientException
4040
import ai.koog.prompt.executor.clients.LLMEmbeddingProvider
4141
import ai.koog.prompt.executor.clients.anthropic.AnthropicParams
4242
import ai.koog.prompt.executor.clients.anthropic.models.AnthropicThinking
43+
import ai.koog.prompt.executor.clients.google.GoogleModels
4344
import ai.koog.prompt.executor.clients.google.GoogleParams
4445
import ai.koog.prompt.executor.clients.google.models.GoogleThinkingConfig
4546
import ai.koog.prompt.executor.clients.google.models.GoogleThinkingLevel
@@ -134,7 +135,7 @@ abstract class ExecutorIntegrationTestBase {
134135
)
135136

136137
is LLMProvider.Google -> {
137-
val thinkingConfig = if (model.id.contains("gemini-3")) {
138+
val thinkingConfig = if (model.id == GoogleModels.Gemini3_Pro_Preview.id) {
138139
GoogleThinkingConfig(
139140
includeThoughts = true,
140141
thinkingLevel = GoogleThinkingLevel.LOW // with HIGH thoughts often exceed maxTokens causing test failures

0 commit comments

Comments
 (0)