Splitting Test and updating minor comments.

Author: n-o-u-r-h-a-n

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

@@ -34,7 +34,7 @@
 @RequiredArgsConstructor(access = AccessLevel.PRIVATE)
 public class DpiMasking implements MaskingProvider {
   @Nonnull DPIConfig.MethodEnum maskingMethod;
-  @Nonnull List<DPIEntityConfig> entitiesDTO;
+  @Nonnull List<DPIEntityConfig> entitiesConfig;
   @With boolean maskGroundingInput;
   @Nonnull List<String> allowList;
 
@@ -77,11 +77,11 @@ public static class Builder {
     @Nonnull
     public DpiMasking withEntities(
         @Nonnull final DPIEntities entity, @Nonnull final DPIEntities... entities) {
-      val entitiesDTO =
+      val entitiesConfig =
           Stream.concat(Stream.of(entity), Arrays.stream(entities))
               .map(it -> (DPIEntityConfig) DPIStandardEntity.create().type(it))
               .toList();
-      return new DpiMasking(maskingMethod, entitiesDTO, false, List.of());
+      return new DpiMasking(maskingMethod, entitiesConfig, false, List.of());
     }
 
     /**
@@ -121,7 +121,7 @@ public DpiMasking withRegex(@Nonnull final String regex, @Nonnull final String r
                 DPIMethodConstant.create()
                     .method(DPIMethodConstant.MethodEnum.CONSTANT)
                     .value(replacement));
-    val newEntities = new java.util.ArrayList<>(entitiesDTO);
+    val newEntities = new java.util.ArrayList<>(entitiesConfig);
     newEntities.add(customEntity);
     return new DpiMasking(maskingMethod, newEntities, maskGroundingInput, allowList);
   }
@@ -134,7 +134,7 @@ public DpiMasking withRegex(@Nonnull final String regex, @Nonnull final String r
    */
   @Nonnull
   public DpiMasking withAllowList(@Nonnull final List<String> allowList) {
-    return new DpiMasking(maskingMethod, entitiesDTO, maskGroundingInput, allowList);
+    return new DpiMasking(maskingMethod, entitiesConfig, maskGroundingInput, allowList);
   }
 
   @Nonnull
@@ -143,7 +143,7 @@ public DPIConfig createConfig() {
     return DPIConfig.create()
         .type(SAP_DATA_PRIVACY_INTEGRATION)
         .method(maskingMethod)
-        .entities(entitiesDTO)
+        .entities(entitiesConfig)
         .maskGroundingInput(DPIConfigMaskGroundingInput.create().enabled(maskGroundingInput))
         .allowlist(allowList);
   }

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

@@ -106,16 +106,21 @@ void testDpiMaskingConfig() {
     assertThat(((MaskingModuleConfigProviders) configModified.getMaskingConfig()).getProviders())
         .withFailMessage("withMaskingConfig() should overwrite the existing config and not append")
         .hasSize(1);
+  }
+
+  @Test
+  void testCustomRegex() {
 
     var masking =
         DpiMasking.anonymization()
             .withRegex("\\d{3}-\\d{2}-\\d{4}", "***-**-****")
             .withRegex("\\d{2}-\\d{2}-\\d{5}", "**-**-*****");
-    config = config.withMaskingConfig(masking);
+    var config = new OrchestrationModuleConfig().withLlmConfig(GPT_4O).withMaskingConfig(masking);
     assertThat(config.getMaskingConfig()).isNotNull();
     assertThat(((MaskingModuleConfigProviders) config.getMaskingConfig()).getProviders())
         .hasSize(1);
-    dpiConfig = ((MaskingModuleConfigProviders) config.getMaskingConfig()).getProviders().get(0);
+    DPIConfig dpiConfig =
+        ((MaskingModuleConfigProviders) config.getMaskingConfig()).getProviders().get(0);
     assertThat(dpiConfig.getMethod()).isEqualTo(DPIConfig.MethodEnum.ANONYMIZATION);
     assertThat(dpiConfig.getEntities()).hasSize(2);
     assertThat(((DPICustomEntity) dpiConfig.getEntities().get(0)).getRegex())