test

liustve · liustve · commit 845ef094dd52 · 2025-07-09T03:41:36.000Z
diff --git a/validator/src/main/java/com/amazon/aoc/validators/TraceValidator.java b/validator/src/main/java/com/amazon/aoc/validators/TraceValidator.java
@@ -149,10 +149,7 @@ private Map<String, Object> getTrace() throws Exception {
     }
 
     if (validationConfig.getHttpPath().contains("chat")) {
-      traceFilter = String.format("service(\"%s\")", context.getServiceName());
-      if (context.getBaggage() != null) {
-        traceFilter += String.format(" AND annotation.session_id = \"%s\"", context.getBaggage());
-      }
+      traceFilter = String.format("service(\"%s\") AND annotation.\"aws.service.type\" = \"gen_ai_agent\"", context.getServiceName());
     }
     log.info("Trace Filter: {}", traceFilter);
     List<TraceSummary> retrieveTraceLists = xrayService.searchTraces(traceFilter);

Original file line number	Diff line number	Diff line change
`@@ -149,10 +149,7 @@ private Map<String, Object> getTrace() throws Exception {`
`149`	`149`	`}`
`150`	`150`
`151`	`151`	`if (validationConfig.getHttpPath().contains("chat")) {`
`152`		`- traceFilter = String.format("service(\"%s\")", context.getServiceName());`
`153`		`- if (context.getBaggage() != null) {`
`154`		`- traceFilter += String.format(" AND annotation.session_id = \"%s\"", context.getBaggage());`
`155`		`- }`
	`152`	`+ traceFilter = String.format("service(\"%s\") AND annotation.\"aws.service.type\" = \"gen_ai_agent\"", context.getServiceName());`
`156`	`153`	`}`
`157`	`154`	`log.info("Trace Filter: {}", traceFilter);`
`158`	`155`	`List<TraceSummary> retrieveTraceLists = xrayService.searchTraces(traceFilter);`