OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #6954 from matthewhand/feature/qdrant-auth

Browse Source 
feat: add API key support for Qdrant Cloud authentication
This commit is contained in:
Timothy Jaeryang Baek
2024-11-16 23:42:16 -08:00
committed by GitHub
parent 034674c19c ca63ae649c
commit 24e98bb9d7
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
backend/open_webui/apps/retrieval/vector/dbs/qdrant.py
View File

@@ -5,7 +5,7 @@ from qdrant_client.http.models import PointStruct
from qdrant_client.models import models
from open_webui.apps.retrieval.vector.main import VectorItem, SearchResult, GetResult
from open_webui.config import QDRANT_URI
from open_webui.config import QDRANT_URI, QDRANT_API_KEY
NO_LIMIT = 999999999
@@ -14,7 +14,11 @@ class QdrantClient:
def __init__(self):
self.collection_prefix = "open-webui"
self.QDRANT_URI = QDRANT_URI
self.client = Qclient(url=self.QDRANT_URI) if self.QDRANT_URI else None
self.QDRANT_API_KEY = QDRANT_API_KEY
self.client = Qclient(
url=self.QDRANT_URI,
api_key=self.QDRANT_API_KEY
) if self.QDRANT_URI else None
def _result_to_get_result(self, points) -> GetResult:
ids = []