Spaces:
Paused
Paused
Commit ·
e856dd5
1
Parent(s): 4bb5758
feat: Add Qdrant Cloud API key authentication & cluster connection support
Browse files- config.py +2 -0
- vector_search/index_qdrant.py +7 -1
config.py
CHANGED
|
@@ -108,6 +108,8 @@ else:
|
|
| 108 |
# Vector Database Engine Selection ("faiss" or "qdrant")
|
| 109 |
VECTOR_STORE_BACKEND = os.getenv("VECTOR_STORE_BACKEND", "faiss").lower()
|
| 110 |
QDRANT_STORAGE_PATH = Path(os.getenv("QDRANT_STORAGE_PATH", DATA_DIR / "qdrant_db"))
|
|
|
|
|
|
|
| 111 |
|
| 112 |
HNSW_M = 32
|
| 113 |
HNSW_EF_CONSTRUCTION = 200
|
|
|
|
| 108 |
# Vector Database Engine Selection ("faiss" or "qdrant")
|
| 109 |
VECTOR_STORE_BACKEND = os.getenv("VECTOR_STORE_BACKEND", "faiss").lower()
|
| 110 |
QDRANT_STORAGE_PATH = Path(os.getenv("QDRANT_STORAGE_PATH", DATA_DIR / "qdrant_db"))
|
| 111 |
+
QDRANT_API_KEY = os.getenv("QDRANT_API_KEY", "")
|
| 112 |
+
QDRANT_URL = os.getenv("QDRANT_URL", "")
|
| 113 |
|
| 114 |
HNSW_M = 32
|
| 115 |
HNSW_EF_CONSTRUCTION = 200
|
vector_search/index_qdrant.py
CHANGED
|
@@ -180,7 +180,13 @@ class QdrantIndexManager:
|
|
| 180 |
if not HAS_QDRANT:
|
| 181 |
raise ImportError("qdrant-client package is required for Qdrant vector backend.")
|
| 182 |
|
| 183 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 184 |
self.client = QdrantClient(":memory:")
|
| 185 |
logger.info("Initialized Qdrant In-Memory Vector Storage Engine.")
|
| 186 |
else:
|
|
|
|
| 180 |
if not HAS_QDRANT:
|
| 181 |
raise ImportError("qdrant-client package is required for Qdrant vector backend.")
|
| 182 |
|
| 183 |
+
qdrant_url = getattr(config, "QDRANT_URL", "")
|
| 184 |
+
qdrant_key = getattr(config, "QDRANT_API_KEY", "")
|
| 185 |
+
|
| 186 |
+
if qdrant_url and qdrant_url.strip():
|
| 187 |
+
self.client = QdrantClient(url=qdrant_url.strip(), api_key=qdrant_key.strip() if qdrant_key else None)
|
| 188 |
+
logger.info("Initialized Qdrant Cloud Vector Engine at '%s'.", qdrant_url)
|
| 189 |
+
elif in_memory:
|
| 190 |
self.client = QdrantClient(":memory:")
|
| 191 |
logger.info("Initialized Qdrant In-Memory Vector Storage Engine.")
|
| 192 |
else:
|