Fix milvus truncate

Fixed the indent bug
This commit is contained in:
stevenlx96 2025-07-14 13:51:29 +08:00
parent 1e82d683cc
commit 61e5b7c03b

View File

@ -727,29 +727,29 @@ class MilvusStore(VectorStoreBase):
metadata_filter_expr = metadata_filters[0] metadata_filter_expr = metadata_filters[0]
return metadata_filter_expr return metadata_filter_expr
def truncate(self): def truncate(self):
"""检测pymilvus安装""" """检测pymilvus安装"""
try: try:
from pymilvus import Collection, utility from pymilvus import Collection, utility
except ImportError: except ImportError:
raise ValueError( raise ValueError(
"Could not import pymilvus python package. " "Could not import pymilvus python package. "
"Please install it with `pip install pymilvus`." "Please install it with `pip install pymilvus`."
) )
"""安全清空 Milvus summary collection 中所有数据,但不删除 collection 本身""" """安全清空 Milvus summary collection 中所有数据,但不删除 collection 本身"""
logger.info(f"Begin truncate Milvus collection: {self.collection_name}") logger.info(f"Begin truncate Milvus collection: {self.collection_name}")
# 先判断 collection 是否存在 # 先判断 collection 是否存在
if utility.has_collection(self.collection_name): if utility.has_collection(self.collection_name):
collection = Collection(self.collection_name) collection = Collection(self.collection_name)
# Load collection 必须调用,才能执行 delete # Load collection 必须调用,才能执行 delete
collection.load() collection.load()
# 通过pk_id删除所有数据 # 通过pk_id删除所有数据
collection.delete("pk_id >= 0") collection.delete("pk_id >= 0")
# flush 确保数据删除被提交 # flush 确保数据删除被提交
collection.flush() collection.flush()
logger.info(f"Truncate Milvus collection {self.collection_name} success") logger.info(f"Truncate Milvus collection {self.collection_name} success")
else: else:
logger.warning(f"Collection {self.collection_name} not found, skip truncate.") logger.warning(f"Collection {self.collection_name} not found, skip truncate.")
def full_text_search( def full_text_search(