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

Format code with ruff

Browse source
This commit is contained in:
Yasiru Rangana 2025-11-07 12:25:07 +11:00
parent 71a185112b
commit 598de90d93
1 changed files with 10 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
lightrag/kg/qdrant_impl.py
View file

@ -71,23 +71,23 @@ def workspace_filter_condition(workspace: str) -> models.FieldCondition:
class QdrantVectorDBStorage(BaseVectorStorage): class QdrantVectorDBStorage(BaseVectorStorage):
""" """
Qdrant vector database storage implementation. Qdrant vector database storage implementation.
This class provides a storage backend for vector embeddings using Qdrant. This class provides a storage backend for vector embeddings using Qdrant.
It supports multi-tenant isolation through workspace-based filtering and It supports multi-tenant isolation through workspace-based filtering and
optional collection suffixes for different embedding dimensions or other purposes. optional collection suffixes for different embedding dimensions or other purposes.
Configuration: Configuration:
- Standard parameters: namespace, workspace, embedding_func, etc. - Standard parameters: namespace, workspace, embedding_func, etc.
- Qdrant-specific parameters in vector_db_storage_cls_kwargs: - Qdrant-specific parameters in vector_db_storage_cls_kwargs:
- cosine_better_than_threshold: Required similarity threshold - cosine_better_than_threshold: Required similarity threshold
- collection_suffix: Optional suffix for collection names - collection_suffix: Optional suffix for collection names
Note on collection_suffix: Note on collection_suffix:
If specified, this suffix will be appended to all collection names. If specified, this suffix will be appended to all collection names.
This allows creating separate sets of collections for different purposes. This allows creating separate sets of collections for different purposes.
To access this data later, you must use the same suffix in all LightRAG To access this data later, you must use the same suffix in all LightRAG
instances that need to access this data. instances that need to access this data.
Examples of collection_suffix usage: Examples of collection_suffix usage:
- Embedding dimensions: "768d", "1536d", "3072d" - Embedding dimensions: "768d", "1536d", "3072d"
- Environments: "dev", "staging", "prod" - Environments: "dev", "staging", "prod"
@ -96,7 +96,7 @@ class QdrantVectorDBStorage(BaseVectorStorage):
- Models: "ada002", "e5large", "bge" - Models: "ada002", "e5large", "bge"
- Special purposes: "filtered", "augmented", "synthetic" - Special purposes: "filtered", "augmented", "synthetic"
""" """
def __init__( def __init__(
self, namespace, global_config, embedding_func, workspace=None, meta_fields=None self, namespace, global_config, embedding_func, workspace=None, meta_fields=None
): ):
@ -319,11 +319,13 @@ class QdrantVectorDBStorage(BaseVectorStorage):
kwargs = self.global_config.get("vector_db_storage_cls_kwargs", {}) kwargs = self.global_config.get("vector_db_storage_cls_kwargs", {})
collection_suffix = kwargs.get("collection_suffix", "") collection_suffix = kwargs.get("collection_suffix", "")
cosine_threshold = kwargs.get("cosine_better_than_threshold") cosine_threshold = kwargs.get("cosine_better_than_threshold")
# Get legacy namespace for data migration from old version # Get legacy namespace for data migration from old version
if effective_workspace: if effective_workspace:
if collection_suffix: if collection_suffix:
self.legacy_namespace = f"{effective_workspace}_{self.namespace}_{collection_suffix}" self.legacy_namespace = (
f"{effective_workspace}_{self.namespace}_{collection_suffix}"
)
else: else:
self.legacy_namespace = f"{effective_workspace}_{self.namespace}" self.legacy_namespace = f"{effective_workspace}_{self.namespace}"
else: else:
@ -345,7 +347,7 @@ class QdrantVectorDBStorage(BaseVectorStorage):
) )
else: else:
self.final_namespace = f"lightrag_vdb_{self.namespace}" self.final_namespace = f"lightrag_vdb_{self.namespace}"
logger.debug( logger.debug(
f"Using shared collection '{self.final_namespace}' with workspace '{self.effective_workspace}' for payload-based partitioning" f"Using shared collection '{self.final_namespace}' with workspace '{self.effective_workspace}' for payload-based partitioning"
) )