Updated test

Author: CharlesDuboisSAP

orchestration/pom.xml

@@ -34,7 +34,7 @@
     <coverage.complexity>83%</coverage.complexity>
     <coverage.line>94%</coverage.line>
     <coverage.instruction>94%</coverage.instruction>
-    <coverage.branch>75%</coverage.branch>
+    <coverage.branch>77%</coverage.branch>
     <coverage.method>93%</coverage.method>
     <coverage.class>100%</coverage.class>
   </properties>

orchestration/src/test/java/com/sap/ai/sdk/orchestration/OrchestrationUnitTest.java

@@ -205,6 +205,7 @@ void testGrounding() throws IOException {
             "grounding_result",
             "First chunk```Second chunk```Last found chunk");
     assertThat(groundingModule.getData()).isEqualTo(groundingData);
+    assertThat(response.getOriginalResponse().getModuleResults().getInputMasking()).isNotNull();
 
     try (var requestInputStream = fileLoader.apply("groundingRequest.json")) {
       final String request = new String(requestInputStream.readAllBytes());

orchestration/src/test/resources/__files/groundingResponse.json

@@ -14,6 +14,13 @@
         "content": "Context message with embedded grounding results. First chunk```Second chunk```Last found chunk"
       }
     ],
+    "input_masking": {
+      "message": "Input to LLM and Grounding is masked successfully.",
+      "data": {
+        "masked_template": "[{\"role\": \"user\", \"content\": \"Context message with embedded grounding results. First chunk```MASKED_SENSITIVE_DATA```Last found chunk\"}]",
+        "masked_grounding_input": "[\"What does Joule do?\"]"
+      }
+    },
     "llm": {
       "id": "chatcmpl-Apz5s3CMf99jkOnxvPshH1rGLwvvU",
       "object": "chat.completion",
@@ -59,4 +66,4 @@
       "total_tokens": 380
     }
   }
-}
+}