enhance: [2.5] support hints param (#2442)

issue: milvus-io/milvus#37360 pr: #2408 Signed-off-by: chasingegg <[email protected]>
milvus-io · Dec 13, 2024 · be184b5 · be184b5
1 parent 6fcbb49
commit be184b5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/pymilvus/client/constants.py b/pymilvus/client/constants.py
@@ -17,6 +17,7 @@
 ITERATOR_FIELD = "iterator"
 ITERATOR_SESSION_TS_FIELD = "iterator_session_ts"
 PAGE_RETAIN_ORDER_FIELD = "page_retain_order"
+HINTS = "hints"
 
 RANKER_TYPE_RRF = "rrf"
 RANKER_TYPE_WEIGHTED = "weighted"

diff --git a/pymilvus/client/prepare.py b/pymilvus/client/prepare.py
@@ -19,6 +19,7 @@
     DYNAMIC_FIELD_NAME,
     GROUP_BY_FIELD,
     GROUP_SIZE,
+    HINTS,
     ITERATOR_FIELD,
     PAGE_RETAIN_ORDER_FIELD,
     RANK_GROUP_SCORER,
@@ -975,6 +976,9 @@ def search_requests_with_expr(
         if anns_field:
             search_params["anns_field"] = anns_field
 
+        if param.get(HINTS) is not None:
+            search_params[HINTS] = param[HINTS]
+
         req_params = [
             common_types.KeyValuePair(key=str(key), value=utils.dumps(value))
             for key, value in search_params.items()