gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge branch 'qdrant1.7' into optimize-extraction

Browse source
This commit is contained in:
yangdx 2025-09-12 16:40:53 +08:00
commit 1ee1fe895b
1 changed files with 31 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
lightrag/kg/qdrant_impl.py
View file

@ -67,9 +67,22 @@ class QdrantVectorDBStorage(BaseVectorStorage):
def create_collection_if_not_exist(
client: QdrantClient, collection_name: str, **kwargs
):
if client.collection_exists(collection_name):
return
client.create_collection(collection_name, **kwargs)
exists = False
if hasattr(client, "collection_exists"):
try:
exists = client.collection_exists(collection_name)
except Exception:
exists = False
else:
try:
client.get_collection(collection_name)
exists = True
except Exception:
exists = False
if not exists:
client.create_collection(collection_name, **kwargs)
def __post_init__(self):
# Check for QDRANT_WORKSPACE environment variable first (higher priority)
@ -464,9 +477,23 @@ class QdrantVectorDBStorage(BaseVectorStorage):
async with get_storage_lock():
try:
# Delete the collection and recreate it
if self._client.collection_exists(self.final_namespace):
exists = False
if hasattr(self._client, "collection_exists"):
try:
exists = self._client.collection_exists(self.final_namespace)
except Exception:
exists = False
else:
try:
self._client.get_collection(self.final_namespace)
exists = True
except Exception:
exists = False
if exists:
self._client.delete_collection(self.final_namespace)
# Recreate the collection
QdrantVectorDBStorage.create_collection_if_not_exist(
self._client,