enrich default_search_params

Signed-off-by: ChengZi <[email protected]>
milvus-io · Dec 19, 2024 · 0f68215 · 0f68215
1 parent 7b6c076
commit 0f68215
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/milvus_haystack/document_store.py b/src/milvus_haystack/document_store.py
@@ -137,9 +137,7 @@ def __init__(
         """
         # Default search params when one is not provided.
         self.default_search_params = {
-            "GPU_IVF_FLAT": {"metric_type": "L2", "params": {"nprobe": 10}},
-            "GPU_IVF_PQ": {"metric_type": "L2", "params": {"nprobe": 10}},
-            "GPU_CAGRA": {"metric_type": "L2", "params": {"itopk_size": 128}},
+            "FLAT": {"metric_type": "L2", "params": {}},
             "IVF_FLAT": {"metric_type": "L2", "params": {"nprobe": 10}},
             "IVF_SQ8": {"metric_type": "L2", "params": {"nprobe": 10}},
             "IVF_PQ": {"metric_type": "L2", "params": {"nprobe": 10}},
@@ -149,7 +147,16 @@ def __init__(
             "RHNSW_PQ": {"metric_type": "L2", "params": {"ef": 10}},
             "IVF_HNSW": {"metric_type": "L2", "params": {"nprobe": 10, "ef": 10}},
             "ANNOY": {"metric_type": "L2", "params": {"search_k": 10}},
+            "SCANN": {"metric_type": "L2", "params": {"search_k": 10}},
             "AUTOINDEX": {"metric_type": "L2", "params": {}},
+            "GPU_CAGRA": {"metric_type": "L2", "params": {"itopk_size": 128}},
+            "GPU_IVF_FLAT": {"metric_type": "L2", "params": {"nprobe": 10}},
+            "GPU_IVF_PQ": {"metric_type": "L2", "params": {"nprobe": 10}},
+            "SPARSE_INVERTED_INDEX": {
+                "metric_type": "IP",
+                "params": {"drop_ratio_build": 0.2},
+            },
+            "SPARSE_WAND": {"metric_type": "IP", "params": {"drop_ratio_build": 0.2}},
         }
 
         self.collection_name = collection_name