Skip to content

Commit b704060

Browse files
committed
Fix LLMAsJudgeNodeTest compillation
1 parent 9474b74 commit b704060

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

agents/agents-ext/src/jvmTest/kotlin/ai/koog/agents/ext/agent/LLMAsJudgeNodeTest.kt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,8 @@ class LLMAsJudgeNodeTest {
6363

6464
val initialModel = OllamaModels.Meta.LLAMA_3_2
6565

66+
val agentConfig = AIAgentConfig(prompt = prompt("id") {}, model = OpenAIModels.Chat.GPT4o, maxAgentIterations = 10)
67+
6668
val mockLLM = AIAgentLLMContext(
6769
tools = emptyList(),
6870
toolRegistry = ToolRegistry {},
@@ -71,7 +73,7 @@ class LLMAsJudgeNodeTest {
7173
responseProcessor = null,
7274
promptExecutor = mockPromptExecutor,
7375
environment = mockEnv,
74-
config = AIAgentConfig(prompt = prompt("id") {}, model = OpenAIModels.Chat.GPT4o, maxAgentIterations = 10),
76+
config = agentConfig,
7577
clock = testClock
7678
)
7779

0 commit comments

Comments
 (0)