mirror of
https://github.com/open-webui/open-webui.git
synced 2026-08-30 17:25:30 -05:00
[GH-ISSUE #8017] Support specifying GPU index types when configuring the Milvus vector database #69567
Reference in New Issue
Block a user
Originally created by @qiaozhi199 on GitHub (Dec 23, 2024).
Original GitHub issue: https://github.com/open-webui/open-webui/issues/8017
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?