Fixed exception for explain in hybrid query when partial match in sub…

…queries Signed-off-by: Martin Gaievski <gaievski@amazon.com>
opensearch-project · Jan 18, 2025 · 33d590b · 33d590b
1 parent 660f577
commit 33d590b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git a/src/main/java/org/opensearch/neuralsearch/processor/ExplanationResponseProcessor.java b/src/main/java/org/opensearch/neuralsearch/processor/ExplanationResponseProcessor.java
@@ -99,15 +99,17 @@ public SearchResponse processResponse(
                     ExplanationDetails normalizationExplanation = combinedExplainDetail.getNormalizationExplanations();
                     ExplanationDetails combinationExplanation = combinedExplainDetail.getCombinationExplanations();
                     // Create normalized explanations for each detail
-                    Explanation[] normalizedExplanation = new Explanation[queryLevelExplanation.getDetails().length];
+                    List<Explanation> normalizedExplanation = new ArrayList<>();
                     for (int i = 0; i < queryLevelExplanation.getDetails().length; i++) {
-                        normalizedExplanation[i] = Explanation.match(
-                            // normalized score
-                            normalizationExplanation.getScoreDetails().get(i).getKey(),
-                            // description of normalized score
-                            normalizationExplanation.getScoreDetails().get(i).getValue(),
-                            // shard level details
-                            queryLevelExplanation.getDetails()[i]
+                        normalizedExplanation.add(
+                            Explanation.match(
+                                // normalized score
+                                normalizationExplanation.getScoreDetails().get(i).getKey(),
+                                // description of normalized score
+                                normalizationExplanation.getScoreDetails().get(i).getValue(),
+                                // shard level details
+                                queryLevelExplanation.getDetails()[i]
+                            )
                         );
                     }
                     // Create and set final explanation combining all components

diff --git a/src/main/java/org/opensearch/neuralsearch/query/HybridQueryWeight.java b/src/main/java/org/opensearch/neuralsearch/query/HybridQueryWeight.java
@@ -160,6 +160,7 @@ public Explanation explain(LeafReaderContext context, int doc) throws IOExceptio
                 subsOnMatch.add(e);
                 max = Math.max(max, score);
             } else if (!match) {
+                subsOnMatch.add(e);
                 subsOnNoMatch.add(e);
             }
         }