Handle class cast exceptions

Author: newtork

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

@@ -15,10 +15,10 @@
 import com.sap.ai.sdk.orchestration.model.TokenUsage;
 import com.sap.ai.sdk.orchestration.model.ToolChatMessage;
 import com.sap.ai.sdk.orchestration.model.UserChatMessage;
+import io.vavr.control.Try;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
-import java.util.Optional;
 import javax.annotation.Nonnull;
 import lombok.RequiredArgsConstructor;
 import lombok.Value;
@@ -43,20 +43,19 @@ public String getContent() throws OrchestrationOutputFilterException {
     final var choice = getChoice();
 
     if ("content_filter".equals(choice.getFinishReason())) {
-      @SuppressWarnings("unchecked")
-      final var filterDetails =
-          Optional.of(getOriginalResponse().getModuleResults().getOutputFiltering())
-              .map(outputFiltering -> (Map<String, Object>) outputFiltering.getData())
-              .map(data -> (List<Map<String, Object>>) data.get("choices"))
-              .map(choices -> choices.get(0))
-              .orElseGet(Map::of);
-
-      throw new OrchestrationOutputFilterException(
-          "Content filter filtered the output.", filterDetails);
+      final var filterDetails = Try.of(this::getOutputFilteringChoices).getOrElseGet(e -> Map.of());
+      final var message = "Content filter filtered the output.";
+      throw new OrchestrationOutputFilterException(message, filterDetails);
     }
     return choice.getMessage().getContent();
   }
 
+  @SuppressWarnings("unchecked")
+  private Map<String, Object> getOutputFilteringChoices() {
+    final var f = getOriginalResponse().getModuleResults().getOutputFiltering();
+    return ((List<Map<String, Object>>) ((Map<String, Object>) f.getData()).get("choices")).get(0);
+  }
+
   /**
    * Get the token usage.
    *

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

@@ -22,6 +22,8 @@
 import java.util.function.Supplier;
 import java.util.stream.Stream;
 import javax.annotation.Nonnull;
+
+import io.vavr.control.Try;
 import lombok.extern.slf4j.Slf4j;
 import lombok.val;
 
@@ -118,19 +120,19 @@ private static void throwOnContentFilter(@Nonnull final OrchestrationChatComplet
       throws OrchestrationOutputFilterException {
     final String finishReason = delta.getFinishReason();
     if (finishReason != null && finishReason.equals("content_filter")) {
-      @SuppressWarnings("unchecked")
-      final var filterDetails =
-          Optional.ofNullable(delta.getModuleResults().getOutputFiltering())
-              .map(outputFiltering -> (Map<String, Object>) outputFiltering.getData())
-              .map(data -> (List<Map<String, Object>>) data.get("choices"))
-              .map(choices -> choices.get(0))
-              .orElseGet(Map::of);
-
-      throw new OrchestrationOutputFilterException(
-          "Content filter filtered the output.", filterDetails);
+      final var filterDetails = Try.of(() -> getOutputFilteringChoices(delta)).getOrElseGet(e -> Map.of());
+      final var message = "Content filter filtered the output.";
+      throw new OrchestrationOutputFilterException(message, filterDetails);
     }
   }
 
+  @SuppressWarnings("unchecked")
+  private static Map<String, Object> getOutputFilteringChoices(@Nonnull final OrchestrationChatCompletionDelta delta) {
+    final var f = delta.getModuleResults().getOutputFiltering();
+    return ((List<Map<String, Object>>) ((Map<String, Object>) f.getData()).get("choices")).get(0);
+  }
+
+
   /**
    * Serializes the given request, executes it and deserializes the response.
    *

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

@@ -3,6 +3,10 @@
 import com.google.common.annotations.Beta;
 import com.sap.ai.sdk.core.common.ClientExceptionFactory;
 import com.sap.ai.sdk.orchestration.OrchestrationFilterException.OrchestrationInputFilterException;
+import com.sap.ai.sdk.orchestration.model.ErrorResponse;
+import com.sap.ai.sdk.orchestration.model.GenericModuleResult;
+import com.sap.ai.sdk.orchestration.model.ModuleResults;
+
 import java.util.Collections;
 import java.util.Map;
 import java.util.Optional;
@@ -37,9 +41,9 @@ public OrchestrationClientException buildFromClientError(
   private Map<String, Object> extractInputFilterDetails(@Nonnull final OrchestrationError error) {
 
     return Optional.of(error.getErrorResponse())
-        .flatMap(response -> Optional.ofNullable(response.getModuleResults()))
-        .flatMap(moduleResults -> Optional.ofNullable(moduleResults.getInputFiltering()))
-        .flatMap(inputFiltering -> Optional.ofNullable(inputFiltering.getData()))
+        .map(ErrorResponse::getModuleResults)
+        .map(ModuleResults::getInputFiltering)
+        .map(GenericModuleResult::getData)
         .filter(Map.class::isInstance)
         .map(map -> (Map<String, Object>) map)
         .orElseGet(Collections::emptyMap);