test

Author: liustve

validator/src/main/java/com/amazon/aoc/validators/TraceValidator.java

@@ -149,7 +149,13 @@ private Map<String, Object> getTrace() throws Exception {
     }
 
     if (validationConfig.getHttpPath().contains("chat")) {
-      traceFilter = String.format("id(\"%s\")", context.getTraceId());
+      // For chat endpoints, directly use the trace ID instead of searching
+      List<String> traceIdLists = Collections.singletonList(context.getTraceId());
+      List<Trace> retrievedTraceList = xrayService.listTraceByIds(traceIdLists);
+      if (retrievedTraceList == null || retrievedTraceList.isEmpty()) {
+        throw new BaseException(ExceptionCode.EMPTY_LIST);
+      }
+      return this.flattenDocument(retrievedTraceList.get(0).getSegments());
     }
     log.info("Trace Filter: {}", traceFilter);
     List<TraceSummary> retrieveTraceLists = xrayService.searchTraces(traceFilter);