Fix error in docId calculation

Author: kderusso

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity/TextSimilarityRankFeaturePhaseRankCoordinatorContext.java

@@ -216,8 +216,9 @@ private float[] extractScoresFromRankedSnippets(List<RankedDocsResults.RankedDoc
 
         for (int i = 0; i < rankedDocs.size(); i++) {
             // TODO this naively assumes that we always get the requested number of snippets per ranked document
-            int docId = i / numSnippets;
-            float score = rankedDocs.get(i).relevanceScore();
+            RankedDocsResults.RankedDoc rankedDoc = rankedDocs.get(i);
+            int docId = rankedDoc.index() / numSnippets;
+            float score = rankedDoc.relevanceScore();
 
             if (hasScore[docId] == false) {
                 scores[docId] = score;
@@ -229,7 +230,7 @@ private float[] extractScoresFromRankedSnippets(List<RankedDocsResults.RankedDoc
 
         float[] result = new float[featureDocs.length];
         for (int i = 0; i < featureDocs.length; i++) {
-            result[i] = hasScore[i] ? normalizeScore(scores[i]) : 0f;
+            result[i] = hasScore[i] ? scores[i] : 0f;
         }
 
         return result;