diff --git a/dbgpt/storage/vector_store/milvus_store.py b/dbgpt/storage/vector_store/milvus_store.py index c03ede6d0..ea69b35fd 100644 --- a/dbgpt/storage/vector_store/milvus_store.py +++ b/dbgpt/storage/vector_store/milvus_store.py @@ -354,8 +354,7 @@ def _add_documents( res = self.col.insert( insert_list, partition_name=partition_name, timeout=timeout ) - # make sure data is searchable. - self.col.flush() + return res.primary_keys def load_document(self, chunks: List[Chunk]) -> List[str]: