bug fix: index out of bound

Signed-off-by: yuye-aws <[email protected]>
yuye-aws · Aug 6, 2024 · d48d455 · d48d455
1 parent c2d3fb4
commit d48d455
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/org/opensearch/agent/tools/RCATool.java b/src/main/java/org/opensearch/agent/tools/RCATool.java
@@ -159,7 +159,7 @@ public <T> void runOption2(
             .range(0, causes.size())
             .boxed()
             .collect(
-                Collectors.toMap(i -> causes.get(i).get("reason"), i -> responseVectors.get(i).dotProduct(expectedResponseVectors.get(i)))
+                Collectors.toMap(i -> causes.get(i).get("reason"), i -> responseVectors.get(0).dotProduct(expectedResponseVectors.get(i)))
             );
 
         Optional<Map.Entry<String, Double>> mapEntry = dotProductMap.entrySet().stream().max(Map.Entry.comparingByValue());
@@ -201,7 +201,7 @@ public <T> void run(Map<String, String> parameters, ActionListener<T> listener)
                     if (isLLMOption) {
                         runOption1(phenomenon, causes, apiToResponse, listener);
                     } else {
-                        runOption1(phenomenon, causes, apiToResponse, listener);
+                        runOption2(phenomenon, causes, apiToResponse, listener);
                     }
                 }, listener::onFailure),
                 apiList.size()