mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-08-08 11:47:44 +00:00
Fix milvus truncate
Fixed the indent bug
This commit is contained in:
parent
1e82d683cc
commit
61e5b7c03b
@ -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(
|
||||||
|
Loading…
Reference in New Issue
Block a user