From 8b378f01c8aa449977cd55b9857e630f52c4cba3 Mon Sep 17 00:00:00 2001
From: jzonthemtn <jeff.zemerick@mtnfog.com>
Date: Wed, 11 Dec 2024 08:27:41 -0500
Subject: [PATCH] Allowing id field.

Signed-off-by: jzonthemtn <jeff.zemerick@mtnfog.com>
---
 .../eval/runners/OpenSearchQuerySetRunner.java         | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/opensearch-search-quality-evaluation-plugin/src/main/java/org/opensearch/eval/runners/OpenSearchQuerySetRunner.java b/opensearch-search-quality-evaluation-plugin/src/main/java/org/opensearch/eval/runners/OpenSearchQuerySetRunner.java
index f792943..cd58b39 100644
--- a/opensearch-search-quality-evaluation-plugin/src/main/java/org/opensearch/eval/runners/OpenSearchQuerySetRunner.java
+++ b/opensearch-search-quality-evaluation-plugin/src/main/java/org/opensearch/eval/runners/OpenSearchQuerySetRunner.java
@@ -104,8 +104,14 @@ public void onResponse(final SearchResponse searchResponse) {
 
                             for (final SearchHit hit : searchResponse.getHits().getHits()) {
 
-                                final Map<String, Object> sourceAsMap = hit.getSourceAsMap();
-                                final String documentId = sourceAsMap.get(idField).toString();
+                                final String documentId;
+
+                                if("_id".equals(idField)) {
+                                    documentId = hit.getId();
+                                } else {
+                                    final Map<String, Object> sourceAsMap = hit.getSourceAsMap();
+                                    documentId = sourceAsMap.get(idField).toString();
+                                }
 
                                 orderedDocumentIds.add(documentId);