File tree Expand file tree Collapse file tree
integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/executor Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -40,6 +40,7 @@ import ai.koog.prompt.executor.clients.LLMClientException
4040import ai.koog.prompt.executor.clients.LLMEmbeddingProvider
4141import ai.koog.prompt.executor.clients.anthropic.AnthropicParams
4242import ai.koog.prompt.executor.clients.anthropic.models.AnthropicThinking
43+ import ai.koog.prompt.executor.clients.google.GoogleModels
4344import ai.koog.prompt.executor.clients.google.GoogleParams
4445import ai.koog.prompt.executor.clients.google.models.GoogleThinkingConfig
4546import 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
You can’t perform that action at this time.
0 commit comments