Fix AIAgentMemoryTest

Author: Ololoshechkin

agents/agents-features/agents-features-memory/src/jvmTest/kotlin/ai/koog/agents/memory/AIAgentMemoryTest.kt

@@ -93,7 +93,7 @@ class AIAgentMemoryTest {
         every { response.content } returns "Test fact"
 
         coEvery {
-            promptExecutor.execute(any(), any(), any())
+            promptExecutor.execute(any(), any())
         } returns listOf(response)
 
         coEvery {
@@ -174,7 +174,7 @@ class AIAgentMemoryTest {
         every { response.content } returns "OK"
 
         coEvery {
-            promptExecutor.execute(any(), any(), any())
+            promptExecutor.execute(any(), any())
         } returns listOf(response)
 
         val llm = AIAgentLLMContext(
@@ -285,7 +285,7 @@ class AIAgentMemoryTest {
         val response = mockk<Message.Response>()
         every { response.content } returns "Test fact"
         coEvery {
-            promptExecutor.execute(any(), any(), any())
+            promptExecutor.execute(any(), any())
         } returns listOf(response)
 
         // Mock memory feature to capture saved facts
@@ -366,7 +366,7 @@ class AIAgentMemoryTest {
 
         val capturedModels = mutableListOf<LLModel>()
         coEvery {
-            promptExecutor.execute(any(), capture(capturedModels), any())
+            promptExecutor.execute(any(), capture(capturedModels))
         } returns listOf(response)
 
         coEvery {
@@ -426,7 +426,7 @@ class AIAgentMemoryTest {
         }
 
         coVerify(exactly = 1) {
-            promptExecutor.execute(any(), customModel, any())
+            promptExecutor.execute(any(), customModel)
         }
     }
 
@@ -454,7 +454,7 @@ class AIAgentMemoryTest {
 
         val capturedModels = mutableListOf<LLModel>()
         coEvery {
-            promptExecutor.execute(any(), capture(capturedModels), any())
+            promptExecutor.execute(any(), capture(capturedModels))
         } returns listOf(response)
 
         coEvery {
@@ -541,7 +541,7 @@ class AIAgentMemoryTest {
         every { response.content } returns "OK"
 
         coEvery {
-            promptExecutor.execute(any(), any(), any())
+            promptExecutor.execute(any(), any())
         } returns listOf(response)
 
         val llm = AIAgentLLMContext(