Include MilvusClient

Signed-off-by: Filip Haltmayer <[email protected]>
milvus-io · Apr 27, 2023 · b25b389 · b25b389
1 parent 6d8785a
commit b25b389
Show file tree

Hide file tree

Showing 5 changed files with 1,237 additions and 0 deletions.
diff --git a/pymilvus/milvus_client/__init__.py b/pymilvus/milvus_client/__init__.py
diff --git a/pymilvus/milvus_client/defaults.py b/pymilvus/milvus_client/defaults.py
@@ -0,0 +1,12 @@
+"""Default MilvusClient args."""
+
+DEFAULT_SEARCH_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}},
+    "HNSW": {"metric_type": "L2", "params": {"ef": 10}},
+    "RHNSW_FLAT": {"metric_type": "L2", "params": {"ef": 10}},
+    "RHNSW_SQ": {"metric_type": "L2", "params": {"ef": 10}},
+    "RHNSW_PQ": {"metric_type": "L2", "params": {"ef": 10}},
+    "AUTOINDEX": {"metric_type": "L2", "params": {}},
+}