Просмотр исходного кода

Fix a bug of SearchResutsWrapper (#758)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 1 год назад
Родитель
Сommit
0b9568ee1f
1 измененных файлов с 8 добавлено и 2 удалено
  1. 8 2
      src/main/java/io/milvus/response/SearchResultsWrapper.java

+ 8 - 2
src/main/java/io/milvus/response/SearchResultsWrapper.java

@@ -60,9 +60,15 @@ public class SearchResultsWrapper extends RowRecordWrapper {
         long topK = results.getTopK();
         List<IDScore> idScore = getIDScore(indexOfTarget);
         for (int i = 0; i < topK; ++i) {
+            IDScore score = idScore.get(i);
             QueryResultsWrapper.RowRecord record = new QueryResultsWrapper.RowRecord();
-            record.put("id", idScore.get(i).getLongID());
-            record.put("distance", idScore.get(i).getScore());
+            if (score.getStrID().isEmpty()) {
+                record.put("id", score.getLongID());
+            } else {
+                record.put("id", score.getStrID());
+            }
+
+            record.put("distance", score.getScore());
             buildRowRecord(record, i);
             records.add(record);
         }