feat(ai-guard): enable tag probabilities test

manifests/java.yml

@@ -29,7 +29,10 @@ manifest:
     - weblog_declaration:
         "*": irrelevant (just one weblog is enough to test the SDK)
         "spring-boot": v1.61.0-SNAPSHOT
-  tests/ai_guard/test_ai_guard_sdk.py::Test_Tag_Probabilities: missing_feature (APPSEC-61896)
+  tests/ai_guard/test_ai_guard_sdk.py::Test_Tag_Probabilities:
+    - weblog_declaration:
+        "*": irrelevant (just one weblog is enough to test the SDK)
+        "spring-boot": v1.62.0-SNAPSHOT
   tests/apm_tracing_e2e/test_otel.py::Test_Otel_Span:
     - weblog_declaration:
         "*": v0.2.0  # real version not known

utils/build/docker/java/spring-boot/src/main/java/com/datadoghq/system_tests/springboot/ai_guard/AIGuardController.java

@@ -5,21 +5,37 @@
 import com.fasterxml.jackson.databind.JsonNode;
 import datadog.trace.api.aiguard.AIGuard;
 import datadog.trace.api.aiguard.AIGuard.Evaluation;
+import org.springframework.boot.autoconfigure.jackson.Jackson2ObjectMapperBuilderCustomizer;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestHeader;
 import org.springframework.web.bind.annotation.RestController;
 
+import java.lang.reflect.Method;
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 import java.util.stream.Collectors;
 
 
 @RestController
 public class AIGuardController {
 
+    @Configuration
+    public static class JacksonConfig {
+        @Bean
+        public Jackson2ObjectMapperBuilderCustomizer mixInCustomizer() {
+            return builder -> builder
+                    .mixIn(AIGuard.AIGuardAbortError.class, AIGuardAbortErrorMixIn.class)
+                    .mixIn(AIGuard.Evaluation.class, AIGuardEvaluationMixIn.class);
+        }
+    }
+
     @PostMapping("/ai_guard/evaluate")
     public ResponseEntity<?> evaluate(
             @RequestHeader(name = "X-AI-Guard-Block", defaultValue = "false") final boolean block,
@@ -35,7 +51,6 @@ public ResponseEntity<?> evaluate(
         }
     }
 
-
     @JsonInclude(JsonInclude.Include.NON_NULL)
     public static class Message {
         @JsonProperty("role")
@@ -194,4 +209,16 @@ public void setArguments(String arguments) {
         }
     }
 
+    public static abstract class AIGuardAbortErrorMixIn {
+
+        @JsonProperty("tag_probs")
+        abstract Object getTagProbabilities();
+    }
+
+    public static abstract class AIGuardEvaluationMixIn {
+
+        @JsonProperty("tag_probs")
+        abstract Object getTagProbabilities();
+    }
+
 }