rishik1111 commited on
Commit
e856dd5
·
1 Parent(s): 4bb5758

feat: Add Qdrant Cloud API key authentication & cluster connection support

Browse files
Files changed (2) hide show
  1. config.py +2 -0
  2. 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
- if in_memory:
 
 
 
 
 
 
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: