File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed
orchestration/src/test/java/com/sap/ai/sdk/orchestration Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -77,7 +77,11 @@ void testThrowOnEmptyFilterConfig() {
7777
7878 @ Test
7979 void testDpiMaskingConfig () {
80- var maskingConfig = DpiMasking .anonymization ().withEntities (DPIEntities .ADDRESS );
80+ var maskingConfig =
81+ DpiMasking .anonymization ()
82+ .withEntities (DPIEntities .ADDRESS )
83+ .withMaskGroundingEnabled ()
84+ .withAllowList (List .of ("Alice" ));
8185 var config =
8286 new OrchestrationModuleConfig ().withLlmConfig (GPT_4O ).withMaskingConfig (maskingConfig );
8387
@@ -87,6 +91,8 @@ void testDpiMaskingConfig() {
8791 assertThat (dpiConfig .getMethod ()).isEqualTo (DPIConfig .MethodEnum .ANONYMIZATION );
8892 assertThat (dpiConfig .getEntities ()).hasSize (1 );
8993 assertThat (dpiConfig .getEntities ().get (0 ).getType ()).isEqualTo (DPIEntities .ADDRESS );
94+ assertThat (dpiConfig .getMaskGroundingInput ().isEnabled ()).isEqualTo (true );
95+ assertThat (dpiConfig .getAllowlist ()).containsExactly ("Alice" );
9096
9197 var configModified = config .withMaskingConfig (maskingConfig );
9298 assertThat (configModified .getMaskingConfig ()).isNotNull ();
You can’t perform that action at this time.
0 commit comments