Fix compilation (new Filtering classes)

Author: Jonas-Isr

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

@@ -1,7 +1,9 @@
 package com.sap.ai.sdk.orchestration;
 
-import com.sap.ai.sdk.orchestration.model.AzureContentSafety;
-import com.sap.ai.sdk.orchestration.model.AzureContentSafetyFilterConfig;
+import com.sap.ai.sdk.orchestration.model.AzureContentSafetyInput;
+import com.sap.ai.sdk.orchestration.model.AzureContentSafetyInputFilterConfig;
+import com.sap.ai.sdk.orchestration.model.AzureContentSafetyOutput;
+import com.sap.ai.sdk.orchestration.model.AzureContentSafetyOutputFilterConfig;
 import javax.annotation.Nonnull;
 import javax.annotation.Nullable;
 import lombok.NoArgsConstructor;
@@ -48,22 +50,46 @@ public class AzureContentFilter implements ContentFilter {
 
   /**
    * Converts {@code AzureContentFilter} to its serializable counterpart {@link
-   * AzureContentSafetyFilterConfig}.
+   * AzureContentSafetyInputFilterConfig}.
    *
-   * @return the corresponding {@code AzureContentSafetyFilterConfig} object.
+   * @return the corresponding {@code AzureContentSafetyInputFilterConfig} object.
    * @throws IllegalArgumentException if no policies are set.
    */
   @Override
   @Nonnull
-  public AzureContentSafetyFilterConfig createConfig() {
+  public AzureContentSafetyInputFilterConfig createInputConfig() {
     if (hate == null && selfHarm == null && sexual == null && violence == null) {
       throw new IllegalArgumentException("At least one filter category must be set");
     }
 
-    return AzureContentSafetyFilterConfig.create()
-        .type(AzureContentSafetyFilterConfig.TypeEnum.AZURE_CONTENT_SAFETY)
+    return AzureContentSafetyInputFilterConfig.create()
+        .type(AzureContentSafetyInputFilterConfig.TypeEnum.AZURE_CONTENT_SAFETY)
         .config(
-            AzureContentSafety.create()
+            AzureContentSafetyInput.create()
+                .hate(hate != null ? hate.getAzureThreshold() : null)
+                .selfHarm(selfHarm != null ? selfHarm.getAzureThreshold() : null)
+                .sexual(sexual != null ? sexual.getAzureThreshold() : null)
+                .violence(violence != null ? violence.getAzureThreshold() : null));
+  }
+
+  /**
+   * Converts {@code AzureContentFilter} to its serializable counterpart {@link
+   * AzureContentSafetyOutputFilterConfig}.
+   *
+   * @return the corresponding {@code AzureContentSafetyOutputFilterConfig} object.
+   * @throws IllegalArgumentException if no policies are set.
+   */
+  @Override
+  @Nonnull
+  public AzureContentSafetyOutputFilterConfig createOutputConfig() {
+    if (hate == null && selfHarm == null && sexual == null && violence == null) {
+      throw new IllegalArgumentException("At least one filter category must be set");
+    }
+
+    return AzureContentSafetyOutputFilterConfig.create()
+        .type(AzureContentSafetyOutputFilterConfig.TypeEnum.AZURE_CONTENT_SAFETY)
+        .config(
+            AzureContentSafetyOutput.create()
                 .hate(hate != null ? hate.getAzureThreshold() : null)
                 .selfHarm(selfHarm != null ? selfHarm.getAzureThreshold() : null)
                 .sexual(sexual != null ? sexual.getAzureThreshold() : null)

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

@@ -1,6 +1,7 @@
 package com.sap.ai.sdk.orchestration;
 
-import com.sap.ai.sdk.orchestration.model.FilterConfig;
+import com.sap.ai.sdk.orchestration.model.InputFilterConfig;
+import com.sap.ai.sdk.orchestration.model.OutputFilterConfig;
 import javax.annotation.Nonnull;
 
 /**
@@ -17,11 +18,20 @@
 public interface ContentFilter {
 
   /**
-   * A method that produces the serializable equivalent {@link FilterConfig} object from data
+   * A method that produces the serializable equivalent {@link InputFilterConfig} object from data
    * encapsulated in the {@link ContentFilter} object.
    *
-   * @return the corresponding {@code FilterConfig} object.
+   * @return the corresponding {@code InputFilterConfig} object.
    */
   @Nonnull
-  FilterConfig createConfig();
+  InputFilterConfig createInputConfig();
+
+  /**
+   * A method that produces the serializable equivalent {@link OutputFilterConfig} object from data
+   * encapsulated in the {@link ContentFilter} object.
+   *
+   * @return the corresponding {@code OutputFilterConfig} object.
+   */
+  @Nonnull
+  OutputFilterConfig createOutputConfig();
 }

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

@@ -40,7 +40,17 @@ public class LlamaGuardFilter implements ContentFilter {
 
   @Nonnull
   @Override
-  public LlamaGuard38bFilterConfig createConfig() {
+  public LlamaGuard38bFilterConfig createInputConfig() {
+    return createConfig();
+  }
+
+  @Nonnull
+  @Override
+  public LlamaGuard38bFilterConfig createOutputConfig() {
+    return createConfig();
+  }
+
+  private LlamaGuard38bFilterConfig createConfig() {
     return LlamaGuard38bFilterConfig.create().type(LLAMA_GUARD_3_8B).config(config);
   }
 }

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

@@ -160,7 +160,7 @@ public OrchestrationModuleConfig withInputFiltering(
     allFilters.addAll(Arrays.asList(contentFilters));
 
     final var filterConfigs =
-        allFilters.stream().filter(Objects::nonNull).map(ContentFilter::createConfig).toList();
+        allFilters.stream().filter(Objects::nonNull).map(ContentFilter::createInputConfig).toList();
 
     final var inputFilter = InputFilteringConfig.create().filters(filterConfigs);
 
@@ -195,7 +195,10 @@ public OrchestrationModuleConfig withOutputFiltering(
     allFilters.addAll(Arrays.asList(contentFilters));
 
     final var filterConfigs =
-        allFilters.stream().filter(Objects::nonNull).map(ContentFilter::createConfig).toList();
+        allFilters.stream()
+            .filter(Objects::nonNull)
+            .map(ContentFilter::createOutputConfig)
+            .toList();
 
     final var outputFilter = OutputFilteringConfig.create().filters(filterConfigs);