fix: milvus collection creation issue

2024-09-26 22:59:09 +02:00 · 2024-09-26 22:59:09 +02:00 · 719f4da1dc
parent fbc9634ffc
commit 719f4da1dc
1 changed files with 4 additions and 1 deletions
--- a/backend/open_webui/apps/rag/vector/dbs/milvus.py
+++ b/backend/open_webui/apps/rag/vector/dbs/milvus.py
@ -98,7 +98,10 @@ class MilvusClient:

        index_params = self.client.prepare_index_params()
        index_params.add_index(
-            field_name="vector", index_type="HNSW", metric_type="COSINE", params={}
+            field_name="vector",
+            index_type="HNSW",
+            metric_type="COSINE",
+            params={"M": 16, "efConstruction": 100},
        )

        self.client.create_collection(