1515import com .sap .ai .sdk .orchestration .model .DocumentGroundingFilter ;
1616import com .sap .ai .sdk .orchestration .model .GroundingModuleConfigConfig ;
1717import com .sap .ai .sdk .orchestration .model .GroundingModuleConfigConfigFiltersInner ;
18+ import com .sap .ai .sdk .orchestration .model .MaskingModuleConfigProviders ;
1819import com .sap .ai .sdk .orchestration .model .ResponseFormatJsonObject ;
1920import com .sap .ai .sdk .orchestration .model .ResponseFormatJsonSchema ;
2021import com .sap .ai .sdk .orchestration .model .Template ;
@@ -87,8 +88,10 @@ void testDpiMaskingConfig() {
8788 new OrchestrationModuleConfig ().withLlmConfig (GPT_4O ).withMaskingConfig (maskingConfig );
8889
8990 assertThat (config .getMaskingConfig ()).isNotNull ();
90- assertThat (config .getMaskingConfig ().getMaskingProviders ()).hasSize (1 );
91- DPIConfig dpiConfig = config .getMaskingConfig ().getMaskingProviders ().get (0 );
91+ assertThat (((MaskingModuleConfigProviders ) config .getMaskingConfig ()).getProviders ())
92+ .hasSize (1 );
93+ DPIConfig dpiConfig =
94+ ((MaskingModuleConfigProviders ) config .getMaskingConfig ()).getProviders ().get (0 );
9295 assertThat (dpiConfig .getMethod ()).isEqualTo (DPIConfig .MethodEnum .ANONYMIZATION );
9396 assertThat (dpiConfig .getEntities ()).hasSize (1 );
9497 assertThat (((DPIStandardEntity ) dpiConfig .getEntities ().get (0 )).getType ())
@@ -98,7 +101,7 @@ void testDpiMaskingConfig() {
98101
99102 var configModified = config .withMaskingConfig (maskingConfig );
100103 assertThat (configModified .getMaskingConfig ()).isNotNull ();
101- assertThat (configModified .getMaskingConfig (). getMaskingProviders ())
104+ assertThat ((( MaskingModuleConfigProviders ) configModified .getMaskingConfig ()). getProviders ())
102105 .withFailMessage ("withMaskingConfig() should overwrite the existing config and not append" )
103106 .hasSize (1 );
104107 }
0 commit comments