add distanceThreshold option for nearestNeighbor operator

1 files changed, 5 insertions, 0 deletions

diff --git a/container-search/src/main/java/com/yahoo/search/query/SelectParser.java b/container-search/src/main/java/com/yahoo/search/query/SelectParser.java
index 30d741f465c..5f1f26b77e9 100644
--- a/container-search/src/main/java/com/yahoo/search/query/SelectParser.java
+++ b/container-search/src/main/java/com/yahoo/search/query/SelectParser.java
@@ -78,6 +78,7 @@ import static com.yahoo.search.yql.YqlParser.CONNECTIVITY;
 import static com.yahoo.search.yql.YqlParser.DEFAULT_TARGET_NUM_HITS;
 import static com.yahoo.search.yql.YqlParser.DESCENDING_HITS_ORDER;
 import static com.yahoo.search.yql.YqlParser.DISTANCE;
+import static com.yahoo.search.yql.YqlParser.DISTANCE_THRESHOLD;
 import static com.yahoo.search.yql.YqlParser.DOT_PRODUCT;
 import static com.yahoo.search.yql.YqlParser.EQUIV;
 import static com.yahoo.search.yql.YqlParser.FILTER;
@@ -481,6 +482,10 @@ public class SelectParser implements Parser {
                 if (TARGET_NUM_HITS.equals(annotation_name)){
                     item.setTargetNumHits((int)(annotation_value.asDouble()));
                 }
+                if (DISTANCE_THRESHOLD.equals(annotation_name)) {
+                    double distanceThreshold = annotation_value.asDouble();
+                    item.setDistanceThreshold(distanceThreshold);
+                }
                 if (HNSW_EXPLORE_ADDITIONAL_HITS.equals(annotation_name)) {
                     int hnswExploreAdditionalHits = (int)(annotation_value.asDouble());
                     item.setHnswExploreAdditionalHits(hnswExploreAdditionalHits);