mirror of
https://github.com/open-webui/open-webui.git
synced 2026-05-07 11:28:35 -05:00
Support specifying GPU index types when configuring the Milvus vector database #3069
Reference in New Issue
Block a user
Delete Branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
Originally created by @qiaozhi199 on GitHub (Dec 23, 2024).
Feature Request
Important Notes
I want to use a GPU in the Milvus vector database to improve search efficiency, so I hope to create GPU-based indexes, such as GPU_CAGRA, GPU_IVF_FLAT, GPU_IVF_PQ, and GPU_BRUTE_FORCE. Link: https://milvus.io/docs/gpu_index.md
However, at the moment, I see that after configuring the Milvus vector database in Open-WebUI, the created index type is fixed as

HNSW. As following picture shows:I see that the index type in the Open-WebUI source code is fixed as HNSW, as shown in the source code below:
https://github.com/open-webui/open-webui/blob/main/backend/open_webui/apps/retrieval/vector/dbs/milvus.py#L99
Is it possible to specify the index type when configuring the Milvus vector database in Open-WebUI?