|
5 | 5 | import static org.assertj.core.api.Assertions.assertThatThrownBy; |
6 | 6 |
|
7 | 7 | import com.sap.ai.sdk.orchestration.client.model.ChatMessage; |
| 8 | +import com.sap.ai.sdk.orchestration.client.model.DPIConfig; |
| 9 | +import com.sap.ai.sdk.orchestration.client.model.DPIEntities; |
8 | 10 | import com.sap.ai.sdk.orchestration.client.model.Template; |
9 | 11 | import java.util.List; |
10 | 12 | import java.util.Map; |
@@ -75,4 +77,28 @@ void testMessagesHistory() { |
75 | 77 |
|
76 | 78 | assertThat(actual.getMessagesHistory()).containsExactly(systemMessage); |
77 | 79 | } |
| 80 | + |
| 81 | + @Test |
| 82 | + void testDpiMaskingConfig() { |
| 83 | + var maskingConfig = DpiMasking.anonymization().withEntities(DPIEntities.ADDRESS); |
| 84 | + var config = |
| 85 | + new OrchestrationModuleConfig() |
| 86 | + .withLlmConfig(CUSTOM_GPT_35) |
| 87 | + .withMaskingConfig(maskingConfig); |
| 88 | + |
| 89 | + var actual = ConfigToRequestTransformer.toModuleConfigs(config); |
| 90 | + |
| 91 | + assertThat(actual.getMaskingModuleConfig()).isNotNull(); |
| 92 | + assertThat(actual.getMaskingModuleConfig().getMaskingProviders()).hasSize(1); |
| 93 | + DPIConfig dpiConfig = (DPIConfig) actual.getMaskingModuleConfig().getMaskingProviders().get(0); |
| 94 | + assertThat(dpiConfig.getMethod()).isEqualTo(DPIConfig.MethodEnum.ANONYMIZATION); |
| 95 | + assertThat(dpiConfig.getEntities()).hasSize(1); |
| 96 | + assertThat(dpiConfig.getEntities().get(0).getType()).isEqualTo(DPIEntities.ADDRESS); |
| 97 | + |
| 98 | + var configModified = config.withMaskingConfig(maskingConfig); |
| 99 | + assertThat(configModified.getMaskingConfig()).isNotNull(); |
| 100 | + assertThat(configModified.getMaskingConfig().getMaskingProviders()) |
| 101 | + .withFailMessage("withMaskingConfig() should overwrite the existing config and not append") |
| 102 | + .hasSize(1); |
| 103 | + } |
78 | 104 | } |
0 commit comments