github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-08 03:44:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix issue2321:Provide an implementation of the truncate method

Browse Source
This commit is contained in:
283569391@qq.com 2025-02-20 11:54:03 +08:00
parent 1a6ad50b3d
commit 69a20c3e7e
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
dbgpt/storage/vector_store/milvus_store.py
View File

@ -638,3 +638,15 @@ class MilvusStore(VectorStoreBase):
else: else:
metadata_filter_expr = metadata_filters[0] metadata_filter_expr = metadata_filters[0]
return metadata_filter_expr return metadata_filter_expr
def truncate(self):
"""Truncate milvus collection.
"""
logger.info(f"begin truncate milvus collection:{self.collection_name}")
from pymilvus import utility
if utility.has_collection(self.collection_name):
utility.drop_collection(self.collection_name)
logger.info(
f"truncate milvus collection {self.collection_name} success"
)