fix(test): Pass systemMessage as a parameter in test methods (#391)

# fix(test): Pass `systemMessage` as a parameter in test methods

Update SpringAi tests to pass `systemMessage` dynamically

Author: kpavlov

ai-mocks-gemini/src/jvmTest/kotlin/me/kpavlov/aimocks/gemini/springai/AbstractSpringAiTest.kt

@@ -17,10 +17,10 @@ internal abstract class AbstractSpringAiTest : AbstractMockGeminiTest() {
                     .build(),
             ).build()
 
-    protected fun prepareClientRequest(): ChatClient.ChatClientRequestSpec =
+    protected fun prepareClientRequest(systemMessage: String): ChatClient.ChatClientRequestSpec =
         chatClient
             .prompt()
-            .system("You are a helpful pirate. $seedValue")
+            .system(systemMessage)
             .user("Just say 'Hello!'")
             .options(
                 VertexAiGeminiChatOptions

ai-mocks-gemini/src/jvmTest/kotlin/me/kpavlov/aimocks/gemini/springai/ChatCompletionSpringAiTest.kt

@@ -25,7 +25,7 @@ internal class ChatCompletionSpringAiTest : AbstractSpringAiTest() {
         }
 
         val response: ChatResponse? =
-            prepareClientRequest()
+            prepareClientRequest(systemMessage)
                 .call()
                 .chatResponse()
 

ai-mocks-gemini/src/jvmTest/kotlin/me/kpavlov/aimocks/gemini/springai/StreamingChatCompletionSpringAiTest.kt

@@ -16,13 +16,14 @@ import kotlin.time.toJavaDuration
 internal class StreamingChatCompletionSpringAiTest : AbstractSpringAiTest() {
     @Test
     fun `Should respond with stream to generateContentStream`() {
+        val systemMessage = "You are a helpful pirate. $seedValue"
         gemini
             .generateContentStream {
                 temperature = temperatureValue
                 model = modelName
                 project = projectId
                 location = locationId
-                systemMessageContains("You are a helpful pirate")
+                systemMessageContains(systemMessage)
                 userMessageContains("Just say 'Hello!'")
             }.respondsStream(sse = false) {
                 responseFlow =
@@ -39,7 +40,7 @@ internal class StreamingChatCompletionSpringAiTest : AbstractSpringAiTest() {
 
         val buffer = StringBuffer()
         val chunkCount =
-            prepareClientRequest()
+            prepareClientRequest(systemMessage)
                 .stream()
                 .chatResponse()
                 .doOnNext { chunk ->

ai-mocks-ollama/src/jvmTest/kotlin/me/kpavlov/aimocks/ollama/springai/AbstractSpringAiTest.kt

@@ -20,10 +20,10 @@ internal val chatClient =
         ).build()
 
 internal abstract class AbstractSpringAiTest : AbstractMockOllamaTest() {
-    protected fun prepareClientRequest(): ChatClient.ChatClientRequestSpec =
+    protected fun prepareClientRequest(systemMessage: String): ChatClient.ChatClientRequestSpec =
         chatClient
             .prompt()
-            .system("You are a helpful pirate")
+            .system(systemMessage)
             .user("Just say 'Hello!'")
             .options(
                 OllamaOptions

ai-mocks-ollama/src/jvmTest/kotlin/me/kpavlov/aimocks/ollama/springai/ChatCompletionSpringAiTest.kt

@@ -23,7 +23,7 @@ internal class ChatCompletionSpringAiTest : AbstractSpringAiTest() {
         }
 
         val response =
-            prepareClientRequest()
+            prepareClientRequest("You are a helpful pirate")
                 .call()
                 .chatResponse()
 

ai-mocks-ollama/src/jvmTest/kotlin/me/kpavlov/aimocks/ollama/springai/StreamingChatCompletionSpringAiTest.kt

@@ -18,7 +18,7 @@ internal class StreamingChatCompletionSpringAiTest : AbstractSpringAiTest() {
             model = modelName
             topK = topKValue
             topP = topPValue
-            systemMessageContains("helpful pirate")
+            systemMessageContains("unhelpful orc")
             userMessageContains("say 'Hello!'")
             stream = true
         } respondsStream {
@@ -36,7 +36,7 @@ internal class StreamingChatCompletionSpringAiTest : AbstractSpringAiTest() {
 
         val buffer = StringBuffer()
         val chunkCount =
-            prepareClientRequest()
+            prepareClientRequest("You are a unhelpful orc")
                 .stream()
                 .chatResponse()
                 .doOnNext { chatResponse ->

ai-mocks-openai/src/jvmTest/kotlin/me/kpavlov/aimocks/openai/springai/AbstractSpringAiTest.kt

@@ -21,10 +21,10 @@ internal val chatClient =
         ).build()
 
 internal abstract class AbstractSpringAiTest : AbstractMockOpenaiTest() {
-    protected fun prepareClientRequest(): ChatClient.ChatClientRequestSpec =
+    protected fun prepareClientRequest(systemPrompt: String): ChatClient.ChatClientRequestSpec =
         chatClient
             .prompt()
-            .system("You are a helpful pirate")
+            .system(systemPrompt)
             .user("Just say 'Hello!'")
             .options(
                 OpenAiChatOptions

ai-mocks-openai/src/jvmTest/kotlin/me/kpavlov/aimocks/openai/springai/ChatCompletionSpringAiTest.kt

@@ -20,25 +20,25 @@ internal class ChatCompletionSpringAiTest : AbstractSpringAiTest() {
             seed = seedValue
             model = modelName
             maxTokens = maxCompletionTokensValue
-            systemMessageContains("helpful pirate")
+            systemMessageContains("helpful merchant")
             userMessageContains("say 'Hello!'")
             requestMatchesPredicate {
                 !it.stream
             }
         } responds {
-            assistantContent = "Ahoy there, matey! Hello!"
+            assistantContent = "Hello there! Welcome to our shop! How can I assist you today?"
             finishReason = "stop"
         }
 
         val response =
-            prepareClientRequest()
+            prepareClientRequest("You are a helpful merchant")
                 .call()
                 .chatResponse()
 
         response?.result shouldNotBe null
         response?.result?.apply {
             metadata.finishReason shouldBe "STOP"
-            output?.text shouldBe "Ahoy there, matey! Hello!"
+            output.text shouldBe "Hello there! Welcome to our shop! How can I assist you today?"
         }
     }
 
@@ -71,11 +71,11 @@ internal class ChatCompletionSpringAiTest : AbstractSpringAiTest() {
 
         val buffer = StringBuffer()
         val chunkCount =
-            prepareClientRequest()
+            prepareClientRequest("You are a helpful pirate")
                 .stream()
                 .chatResponse()
-                .doOnNext {
-                    it.result.output.text?.let {
+                .doOnNext { chunk ->
+                    chunk.result.output.text?.let {
                         buffer.append(it)
                     }
                 }.count()

ai-mocks-openai/src/jvmTest/kotlin/me/kpavlov/aimocks/openai/springai/StreamingChatCompletionSpringAiTest.kt

@@ -35,7 +35,7 @@ internal class StreamingChatCompletionSpringAiTest : AbstractSpringAiTest() {
 
         val buffer = StringBuffer()
         val chunkCount =
-            prepareClientRequest()
+            prepareClientRequest("You are a helpful pirate")
                 .stream()
                 .chatResponse()
                 .doOnNext {