From 05555d41d3c6e2b43abad1a9454f93581e36c1c2 Mon Sep 17 00:00:00 2001
From: CascadingRadium <rahul.rampure@couchbase.com>
Date: Fri, 5 Jan 2024 14:36:50 +0530
Subject: [PATCH] constrict K value to a max of 10000

---
 search_knn.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/search_knn.go b/search_knn.go
index cc8b25831..31a9d7f7e 100644
--- a/search_knn.go
+++ b/search_knn.go
@@ -31,6 +31,8 @@ import (
 
 type knnOperator string
 
+const MaxKValue = 10000
+
 type SearchRequest struct {
 	Query            query.Query       `json:"query"`
 	Size             int               `json:"size"`
@@ -230,6 +232,9 @@ func validateKNN(req *SearchRequest) error {
 		if q.K <= 0 || len(q.Vector) == 0 {
 			return fmt.Errorf("k must be greater than 0 and vector must be non-empty")
 		}
+		if q.K > MaxKValue {
+			return fmt.Errorf("k must be less than %d", MaxKValue)
+		}
 	}
 	switch req.KNNOperator {
 	case knnOperatorAnd, knnOperatorOr, "":