Support specifying GPU index types when configuring the Milvus vector database #3069

Closed
opened 2025-11-11 15:20:55 -06:00 by GiteaMirror · 0 comments
Owner

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:
image

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?

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: ![image](https://github.com/user-attachments/assets/8a9d0112-f158-4090-8765-20b5ad7f3d80) 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?
Sign in to join this conversation.
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: github-starred/open-webui#3069