Unit test finished

Author: CharlesDuboisSAP

orchestration/src/test/java/com/sap/ai/sdk/orchestration/spring/OrchestrationChatModelTest.java

@@ -8,6 +8,7 @@
 import static com.github.tomakehurst.wiremock.client.WireMock.stubFor;
 import static com.github.tomakehurst.wiremock.client.WireMock.urlPathEqualTo;
 import static com.github.tomakehurst.wiremock.client.WireMock.verify;
+import static com.github.tomakehurst.wiremock.stubbing.Scenario.STARTED;
 import static com.sap.ai.sdk.orchestration.OrchestrationAiModel.GPT_35_TURBO_16K;
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.assertj.core.api.Assertions.assertThatThrownBy;
@@ -145,10 +146,18 @@ void testStreamCompletion() throws IOException {
   void testToolCalls() throws IOException {
     stubFor(
         post(urlPathEqualTo("/completion"))
+            .inScenario("Tool Calls")
+            .whenScenarioStateIs(STARTED)
             .willReturn(
                 aResponse()
                     .withBodyFile("toolCallsResponse.json")
                     .withHeader("Content-Type", "application/json"))
+            .willSetStateTo("Second Call"));
+
+    stubFor(
+        post(urlPathEqualTo("/completion"))
+            .inScenario("Tool Calls")
+            .whenScenarioStateIs("Second Call")
             .willReturn(
                 aResponse()
                     .withBodyFile("toolCallsResponse2.json")

sample-code/spring-app/src/main/java/com/sap/ai/sdk/app/services/SpringAiOrchestrationService.java

@@ -34,15 +34,16 @@ public class SpringAiOrchestrationService {
    */
   @Nonnull
   public ChatResponse completion() {
-    defaultOptions.setFunctionCallbacks(
+    final OrchestrationChatOptions options = new OrchestrationChatOptions(config);
+    options.setFunctionCallbacks(
         List.of(
             FunctionCallback.builder()
                 .function(
                     "CurrentWeather", new MockWeatherService()) // (1) function name and instance
                 .description("Get the weather in location") // (2) function description
                 .inputType(MockWeatherService.Request.class) // (3) function input type
                 .build()));
-    val prompt = new Prompt("What is the weather in Potsdam and in Toulouse?", defaultOptions);
+    val prompt = new Prompt("What is the weather in Potsdam and in Toulouse?", options);
 
     return client.call(prompt);
   }