@@ -57,32 +57,34 @@ public class OrchestrationModuleConfig {
5757 FilteringModuleConfig filteringConfig ;
5858
5959 @ Nonnull
60- public OrchestrationModuleConfig withInputFiltering (AzureContentSafety contentFilter ) {
60+ public OrchestrationModuleConfig withInputFiltering (
61+ @ Nonnull final AzureContentSafety contentFilter ) {
6162 var azureFilter =
6263 new AzureContentSafetyFilterConfig ()
6364 .type (AzureContentSafetyFilterConfig .TypeEnum .AZURE_CONTENT_SAFETY )
6465 .config (contentFilter );
65- var inputFilters = new InputFilteringConfig ().filters (List .of (azureFilter ));
66+ var inputFilter = new InputFilteringConfig ().filters (List .of (azureFilter ));
6667
6768 var newFilteringConfig =
6869 new FilteringModuleConfig ()
69- .input (inputFilters )
70+ .input (inputFilter )
7071 .output (this .filteringConfig != null ? this .filteringConfig .getOutput () : null );
7172
7273 return this .withFilteringConfig (newFilteringConfig );
7374 }
7475
7576 @ Nonnull
76- public OrchestrationModuleConfig withOutputFiltering (AzureContentSafety safety ) {
77- var filter =
77+ public OrchestrationModuleConfig withOutputFiltering (
78+ @ Nonnull final AzureContentSafety contentFilter ) {
79+ var azureFilter =
7880 new AzureContentSafetyFilterConfig ()
7981 .type (AzureContentSafetyFilterConfig .TypeEnum .AZURE_CONTENT_SAFETY )
80- .config (safety );
81- var outputFilteringConfig = new OutputFilteringConfig ().filters (List .of (filter ));
82+ .config (contentFilter );
83+ var outputFilter = new OutputFilteringConfig ().filters (List .of (azureFilter ));
8284
8385 var newFilteringConfig =
8486 new FilteringModuleConfig ()
85- .output (outputFilteringConfig )
87+ .output (outputFilter )
8688 .input (this .filteringConfig != null ? this .filteringConfig .getInput () : null );
8789
8890 return this .withFilteringConfig (newFilteringConfig );
0 commit comments