test(clients): set dummy contextLength for LLModels defined in test

ptitjes · ptitjes · commit 2fe3a1611f37 · 2025-07-18T10:58:17.000+02:00
diff --git a/agents/agents-features/agents-features-trace/src/jvmTest/kotlin/ai/koog/agents/features/tracing/writer/TraceFeatureMessageFileWriterTest.kt b/agents/agents-features/agents-features-trace/src/jvmTest/kotlin/ai/koog/agents/features/tracing/writer/TraceFeatureMessageFileWriterTest.kt
@@ -63,7 +63,8 @@ class TraceFeatureMessageFileWriterTest {
             val testModel = LLModel(
                 provider = TestLLMProvider(),
                 id = "test-llm-id",
-                capabilities = emptyList()
+                capabilities = emptyList(),
+                contextLength = 1_000,
             )
 
             var runId = ""
@@ -310,7 +311,8 @@ class TraceFeatureMessageFileWriterTest {
             val testModel = LLModel(
                 provider = TestLLMProvider(),
                 id = "test-llm-id",
-                capabilities = emptyList()
+                capabilities = emptyList(),
+                contextLength = 1_000,
             )
 
             var runId = ""
@@ -388,4 +390,4 @@ class TraceFeatureMessageFileWriterTest {
             assertContentEquals(expectedLogMessages, actualMessages)
         }
     }
-}
+}
diff --git a/agents/agents-features/agents-features-trace/src/jvmTest/kotlin/ai/koog/agents/features/tracing/writer/TraceFeatureMessageLogWriterTest.kt b/agents/agents-features/agents-features-trace/src/jvmTest/kotlin/ai/koog/agents/features/tracing/writer/TraceFeatureMessageLogWriterTest.kt
@@ -50,7 +50,8 @@ class TraceFeatureMessageLogWriterTest {
             val testModel = LLModel(
                 provider = TestLLMProvider(),
                 id = "test-llm-id",
-                capabilities = emptyList()
+                capabilities = emptyList(),
+                contextLength = 1_000,
             )
 
             var runId = ""
@@ -282,7 +283,8 @@ class TraceFeatureMessageLogWriterTest {
             val testModel = LLModel(
                 provider = TestLLMProvider(),
                 id = "test-llm-id",
-                capabilities = emptyList()
+                capabilities = emptyList(),
+                contextLength = 1_000,
             )
 
             var runId = ""
@@ -360,4 +362,4 @@ class TraceFeatureMessageLogWriterTest {
             assertContentEquals(expectedLogMessages, actualMessages)
         }
     }
-}
+}
diff --git a/agents/agents-features/agents-features-trace/src/jvmTest/kotlin/ai/koog/agents/features/tracing/writer/TraceFeatureMessageRemoteWriterTest.kt b/agents/agents-features/agents-features-trace/src/jvmTest/kotlin/ai/koog/agents/features/tracing/writer/TraceFeatureMessageRemoteWriterTest.kt
@@ -115,7 +115,8 @@ class TraceFeatureMessageRemoteWriterTest {
         val testModel = LLModel(
             provider = TestLLMProvider(),
             id = "test-llm-id",
-            capabilities = emptyList()
+            capabilities = emptyList(),
+            contextLength = 1_000,
         )
 
         val expectedLLMCallWithToolsPrompt = expectedPrompt.copy(
@@ -379,7 +380,8 @@ class TraceFeatureMessageRemoteWriterTest {
         val testModel = LLModel(
             provider = TestLLMProvider(),
             id = "test-llm-id",
-            capabilities = emptyList()
+            capabilities = emptyList(),
+            contextLength = 1_000,
         )
 
         val expectedPrompt = Prompt(
diff --git a/prompt/prompt-executor/prompt-executor-cached/src/jvmTest/kotlin/ai/koog/prompt/executor/cached/CachedPromptExecutorTest.kt b/prompt/prompt-executor/prompt-executor-cached/src/jvmTest/kotlin/ai/koog/prompt/executor/cached/CachedPromptExecutorTest.kt
@@ -25,7 +25,12 @@ class CachedPromptExecutorTest {
         private val testClock = object : Clock {
             override fun now() = testResponse.first().metaInfo.timestamp
         }
-        private val testModel = LLModel(object : LLMProvider("", "") {}, "", emptyList())
+        private val testModel = LLModel(
+            provider = object : LLMProvider("", "") {},
+            id = "",
+            capabilities = emptyList(),
+            contextLength = 1_000L,
+        )
     }
 
     // Mock implementation of PromptCache
diff --git a/prompt/prompt-executor/prompt-executor-clients/prompt-executor-bedrock-client/src/jvmTest/kotlin/ai/koog/prompt/executor/clients/bedrock/modelfamilies/ai21/BedrockAI21JambaSerializationTest.kt b/prompt/prompt-executor/prompt-executor-clients/prompt-executor-bedrock-client/src/jvmTest/kotlin/ai/koog/prompt/executor/clients/bedrock/modelfamilies/ai21/BedrockAI21JambaSerializationTest.kt
@@ -144,7 +144,8 @@ class BedrockAI21JambaSerializationTest {
         val modelWithoutTemperature = LLModel(
             provider = LLMProvider.Bedrock,
             id = "test-model",
-            capabilities = listOf(LLMCapability.Completion) // No temperature capability
+            capabilities = listOf(LLMCapability.Completion), // No temperature capability
+            contextLength = 1_000L,
         )
 
         val requestWithoutTemp = BedrockAI21JambaSerialization.createJambaRequest(
@@ -357,4 +358,4 @@ class BedrockAI21JambaSerializationTest {
         val content = BedrockAI21JambaSerialization.parseJambaStreamChunk(chunkJson)
         assertEquals("", content)
     }
-}
+}
diff --git a/prompt/prompt-executor/prompt-executor-clients/prompt-executor-bedrock-client/src/jvmTest/kotlin/ai/koog/prompt/executor/clients/bedrock/modelfamilies/amazon/BedrockAmazonNovaSerializationTest.kt b/prompt/prompt-executor/prompt-executor-clients/prompt-executor-bedrock-client/src/jvmTest/kotlin/ai/koog/prompt/executor/clients/bedrock/modelfamilies/amazon/BedrockAmazonNovaSerializationTest.kt
@@ -93,7 +93,8 @@ class BedrockAmazonNovaSerializationTest {
         val modelWithoutTemperature = LLModel(
             provider = LLMProvider.Bedrock,
             id = "test-model",
-            capabilities = listOf(LLMCapability.Completion) // No temperature capability
+            capabilities = listOf(LLMCapability.Completion), // No temperature capability
+            contextLength = 1_000L,
         )
 
         val requestWithoutTemp = BedrockAmazonNovaSerialization.createNovaRequest(
@@ -242,4 +243,4 @@ class BedrockAmazonNovaSerializationTest {
         val content = BedrockAmazonNovaSerialization.parseNovaStreamChunk(chunkJson)
         assertEquals("", content)
     }
-}
+}
diff --git a/prompt/prompt-executor/prompt-executor-clients/prompt-executor-bedrock-client/src/jvmTest/kotlin/ai/koog/prompt/executor/clients/bedrock/modelfamilies/meta/BedrockMetaLlamaSerializationTest.kt b/prompt/prompt-executor/prompt-executor-clients/prompt-executor-bedrock-client/src/jvmTest/kotlin/ai/koog/prompt/executor/clients/bedrock/modelfamilies/meta/BedrockMetaLlamaSerializationTest.kt
@@ -81,7 +81,8 @@ class BedrockMetaLlamaSerializationTest {
         val modelWithoutTemperature = LLModel(
             provider = LLMProvider.Bedrock,
             id = "test-model",
-            capabilities = listOf(LLMCapability.Completion) // No temperature capability
+            capabilities = listOf(LLMCapability.Completion), // No temperature capability
+            contextLength = 1_000L,
         )
 
         val requestWithoutTemp =
@@ -174,4 +175,4 @@ class BedrockMetaLlamaSerializationTest {
         val content = BedrockMetaLlamaSerialization.parseLlamaStreamChunk(chunkJson)
         assertEquals("", content)
     }
-}
+}

Original file line number	Diff line number	Diff line change
`@@ -63,7 +63,8 @@ class TraceFeatureMessageFileWriterTest {`
`63`	`63`	`val testModel = LLModel(`
`64`	`64`	`provider = TestLLMProvider(),`
`65`	`65`	`id = "test-llm-id",`
`66`		`- capabilities = emptyList()`
	`66`	`+ capabilities = emptyList(),`
	`67`	`+ contextLength = 1_000,`
`67`	`68`	`)`
`68`	`69`
`69`	`70`	`var runId = ""`
`@@ -310,7 +311,8 @@ class TraceFeatureMessageFileWriterTest {`
`310`	`311`	`val testModel = LLModel(`
`311`	`312`	`provider = TestLLMProvider(),`
`312`	`313`	`id = "test-llm-id",`
`313`		`- capabilities = emptyList()`
	`314`	`+ capabilities = emptyList(),`
	`315`	`+ contextLength = 1_000,`
`314`	`316`	`)`
`315`	`317`
`316`	`318`	`var runId = ""`
`@@ -388,4 +390,4 @@ class TraceFeatureMessageFileWriterTest {`
`388`	`390`	`assertContentEquals(expectedLogMessages, actualMessages)`
`389`	`391`	`}`
`390`	`392`	`}`
`391`		`-}`
	`393`	`+}`
Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,8 @@ class TraceFeatureMessageLogWriterTest {`
`50`	`50`	`val testModel = LLModel(`
`51`	`51`	`provider = TestLLMProvider(),`
`52`	`52`	`id = "test-llm-id",`
`53`		`- capabilities = emptyList()`
	`53`	`+ capabilities = emptyList(),`
	`54`	`+ contextLength = 1_000,`
`54`	`55`	`)`
`55`	`56`
`56`	`57`	`var runId = ""`
`@@ -282,7 +283,8 @@ class TraceFeatureMessageLogWriterTest {`
`282`	`283`	`val testModel = LLModel(`
`283`	`284`	`provider = TestLLMProvider(),`
`284`	`285`	`id = "test-llm-id",`
`285`		`- capabilities = emptyList()`
	`286`	`+ capabilities = emptyList(),`
	`287`	`+ contextLength = 1_000,`
`286`	`288`	`)`
`287`	`289`
`288`	`290`	`var runId = ""`
`@@ -360,4 +362,4 @@ class TraceFeatureMessageLogWriterTest {`
`360`	`362`	`assertContentEquals(expectedLogMessages, actualMessages)`
`361`	`363`	`}`
`362`	`364`	`}`
`363`		`-}`
	`365`	`+}`
Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,12 @@ class CachedPromptExecutorTest {`
`25`	`25`	`private val testClock = object : Clock {`
`26`	`26`	`override fun now() = testResponse.first().metaInfo.timestamp`
`27`	`27`	`}`
`28`		`- private val testModel = LLModel(object : LLMProvider("", "") {}, "", emptyList())`
	`28`	`+ private val testModel = LLModel(`
	`29`	`+ provider = object : LLMProvider("", "") {},`
	`30`	`+ id = "",`
	`31`	`+ capabilities = emptyList(),`
	`32`	`+ contextLength = 1_000L,`
	`33`	`+ )`
`29`	`34`	`}`
`30`	`35`
`31`	`36`	`// Mock implementation of PromptCache`
Original file line number	Diff line number	Diff line change
`@@ -144,7 +144,8 @@ class BedrockAI21JambaSerializationTest {`
`144`	`144`	`val modelWithoutTemperature = LLModel(`
`145`	`145`	`provider = LLMProvider.Bedrock,`
`146`	`146`	`id = "test-model",`
`147`		`- capabilities = listOf(LLMCapability.Completion) // No temperature capability`
	`147`	`+ capabilities = listOf(LLMCapability.Completion), // No temperature capability`
	`148`	`+ contextLength = 1_000L,`
`148`	`149`	`)`
`149`	`150`
`150`	`151`	`val requestWithoutTemp = BedrockAI21JambaSerialization.createJambaRequest(`
`@@ -357,4 +358,4 @@ class BedrockAI21JambaSerializationTest {`
`357`	`358`	`val content = BedrockAI21JambaSerialization.parseJambaStreamChunk(chunkJson)`
`358`	`359`	`assertEquals("", content)`
`359`	`360`	`}`
`360`		`-}`
	`361`	`+}`
Original file line number	Diff line number	Diff line change
`@@ -93,7 +93,8 @@ class BedrockAmazonNovaSerializationTest {`
`93`	`93`	`val modelWithoutTemperature = LLModel(`
`94`	`94`	`provider = LLMProvider.Bedrock,`
`95`	`95`	`id = "test-model",`
`96`		`- capabilities = listOf(LLMCapability.Completion) // No temperature capability`
	`96`	`+ capabilities = listOf(LLMCapability.Completion), // No temperature capability`
	`97`	`+ contextLength = 1_000L,`
`97`	`98`	`)`
`98`	`99`
`99`	`100`	`val requestWithoutTemp = BedrockAmazonNovaSerialization.createNovaRequest(`
`@@ -242,4 +243,4 @@ class BedrockAmazonNovaSerializationTest {`
`242`	`243`	`val content = BedrockAmazonNovaSerialization.parseNovaStreamChunk(chunkJson)`
`243`	`244`	`assertEquals("", content)`
`244`	`245`	`}`
`245`		`-}`
	`246`	`+}`
Original file line number	Diff line number	Diff line change
`@@ -81,7 +81,8 @@ class BedrockMetaLlamaSerializationTest {`
`81`	`81`	`val modelWithoutTemperature = LLModel(`
`82`	`82`	`provider = LLMProvider.Bedrock,`
`83`	`83`	`id = "test-model",`
`84`		`- capabilities = listOf(LLMCapability.Completion) // No temperature capability`
	`84`	`+ capabilities = listOf(LLMCapability.Completion), // No temperature capability`
	`85`	`+ contextLength = 1_000L,`
`85`	`86`	`)`
`86`	`87`
`87`	`88`	`val requestWithoutTemp =`
`@@ -174,4 +175,4 @@ class BedrockMetaLlamaSerializationTest {`
`174`	`175`	`val content = BedrockMetaLlamaSerialization.parseLlamaStreamChunk(chunkJson)`
`175`	`176`	`assertEquals("", content)`
`176`	`177`	`}`
`177`		`-}`
	`178`	`+}`