Used lombok @with

Author: CharlesDuboisSAP

orchestration/src/main/java/com/sap/ai/sdk/orchestration/DpiMasking.java

@@ -17,6 +17,7 @@
 import lombok.Getter;
 import lombok.RequiredArgsConstructor;
 import lombok.Value;
+import lombok.With;
 import lombok.val;
 
 /**
@@ -32,7 +33,7 @@
 public class DpiMasking implements MaskingProvider {
   @Nonnull DPIConfig.MethodEnum maskingMethod;
   @Nonnull List<DPIEntities> entities;
-  boolean maskGroundingInput;
+  @With boolean maskGroundingInput;
   @Nonnull List<String> allowList;
 
   /**
@@ -81,16 +82,6 @@ public DpiMasking withEntities(
     }
   }
 
-  /**
-   * If grounding is used, the input text will be masked.
-   *
-   * @return A new {@link DpiMasking} instance
-   */
-  @Nonnull
-  public DpiMasking withMaskGroundingEnabled() {
-    return new DpiMasking(maskingMethod, entities, true, allowList);
-  }
-
   /**
    * Set words that should not be masked.
    *

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

@@ -80,7 +80,7 @@ void testDpiMaskingConfig() {
     var maskingConfig =
         DpiMasking.anonymization()
             .withEntities(DPIEntities.ADDRESS)
-            .withMaskGroundingEnabled()
+            .withMaskGroundingInput(true)
             .withAllowList(List.of("Alice"));
     var config =
         new OrchestrationModuleConfig().withLlmConfig(GPT_4O).withMaskingConfig(maskingConfig);

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

@@ -173,7 +173,7 @@ void testGrounding() throws IOException {
     val maskingConfig = // optional masking configuration
         DpiMasking.anonymization()
             .withEntities(DPIEntities.SENSITIVE_DATA)
-            .withMaskGroundingEnabled()
+            .withMaskGroundingInput(true)
             .withAllowList(List.of("SAP", "Joule"));
     final var configWithGrounding =
         config.withGroundingConfig(groundingConfig).withMaskingConfig(maskingConfig);
@@ -215,7 +215,7 @@ void testGrounding() throws IOException {
             Map.of(
                 "masked_template",
                 "[{\"role\": \"user\", \"content\": \"Context message with embedded grounding results. First chunk```MASKED_SENSITIVE_DATA```Last found chunk\"}]",
-                "masked_grounding_input",
+                "masked_grounding_input", // maskGroundingInput: true will make this field present
                 "[\"What does Joule do?\"]"));
 
     try (var requestInputStream = fileLoader.apply("groundingRequest.json")) {

sample-code/spring-app/src/main/java/com/sap/ai/sdk/app/controllers/OrchestrationController.java

@@ -180,11 +180,12 @@ Object maskingPseudonymization(
     return response.getContent();
   }
 
-  @GetMapping("/grounding")
+  @GetMapping("/grounding/{maskGroundingInput}")
   @Nonnull
   Object grounding(
-      @Nullable @RequestParam(value = "format", required = false) final String format) {
-    final var response = service.grounding("What does Joule do?");
+      @Nullable @RequestParam(value = "format", required = false) final String format,
+      @PathVariable("maskGroundingInput") final boolean maskGroundingInput) {
+    final var response = service.grounding("What does Joule do?", maskGroundingInput);
     if ("json".equals(format)) {
       return response;
     }

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

@@ -316,10 +316,12 @@ public OrchestrationChatResponse maskingPseudonymization(@Nonnull final DPIEntit
    * @link <a href="https://help.sap.com/docs/sap-ai-core/sap-ai-core-service-guide/grounding">SAP
    *     AI Core: Orchestration - Grounding</a>
    * @param userMessage the user message to provide grounding for
+   * @param maskGroundingInput whether to mask the request sent to the Grounding Service
    * @return the assistant response object
    */
   @Nonnull
-  public OrchestrationChatResponse grounding(@Nonnull final String userMessage) {
+  public OrchestrationChatResponse grounding(
+      @Nonnull final String userMessage, final boolean maskGroundingInput) {
     // optional filter for collections
     val documentMetadata =
         SearchDocumentKeyValueListPair.create()
@@ -338,7 +340,7 @@ public OrchestrationChatResponse grounding(@Nonnull final String userMessage) {
     val maskingConfig = // optional masking configuration
         DpiMasking.anonymization()
             .withEntities(DPIEntities.SENSITIVE_DATA)
-            .withMaskGroundingEnabled()
+            .withMaskGroundingInput(maskGroundingInput)
             .withAllowList(List.of("SAP", "Joule"));
     val configWithGrounding =
         config.withGrounding(groundingConfig).withMaskingConfig(maskingConfig);

sample-code/spring-app/src/main/resources/static/index.html

@@ -340,7 +340,7 @@ <h2>Orchestration</h2>
                             </li>
                             <li class="list-group-item">
                                 <div class="info-tooltip">
-                                    <button type="submit" formaction="/orchestration/grounding"
+                                    <button type="submit" formaction="/orchestration/grounding/false"
                                             class="link-offset-2-hover link-underline link-underline-opacity-0 link-underline-opacity-75-hover endpoint">
                                         <code>/orchestration/grounding</code>
                                     </button>
@@ -349,6 +349,17 @@ <h2>Orchestration</h2>
                                     </div>
                                 </div>
                             </li>
+                            <li class="list-group-item">
+                                <div class="info-tooltip">
+                                    <button type="submit" formaction="/orchestration/grounding/true"
+                                            class="link-offset-2-hover link-underline link-underline-opacity-0 link-underline-opacity-75-hover endpoint">
+                                        <code>/orchestration/<wbr>maskGroundingInput</code>
+                                    </button>
+                                    <div class="tooltip-content">
+                                        The request to the grounding service will be masked.
+                                    </div>
+                                </div>
+                            </li>
 
                             <li class="list-group-item">
                                 <div class="info-tooltip">

sample-code/spring-app/src/test/java/com/sap/ai/sdk/app/controllers/OrchestrationTest.java

@@ -167,7 +167,7 @@ void testMaskingPseudonymization() {
   @DisabledIfSystemProperty(named = "aicore.landscape", matches = "production")
   void testGrounding() {
     assertThat(System.getProperty("aicore.landscape")).isNotEqualTo("production");
-    var response = service.grounding("What does Joule do?");
+    var response = service.grounding("What does Joule do?", true);
     var result = response.getOriginalResponse();
     var llmChoice =
         ((LLMModuleResultSynchronous) result.getOrchestrationResult()).getChoices().get(0);