From c33d56f0783c561b1d031db109c2564accdab992 Mon Sep 17 00:00:00 2001 From: Wenjie Zhang Date: Tue, 14 Oct 2025 06:07:51 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E7=A7=BB=E9=99=A4=20collection.flush()?= =?UTF-8?q?=20Fixes=20#297?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/knowledge/implementations/milvus.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/knowledge/implementations/milvus.py b/src/knowledge/implementations/milvus.py index 36592332..41affd58 100644 --- a/src/knowledge/implementations/milvus.py +++ b/src/knowledge/implementations/milvus.py @@ -263,11 +263,10 @@ class MilvusKB(KnowledgeBase): embeddings, ] - def _insert_and_flush(): + def _insert_records(): collection.insert(entities) - collection.flush() - await asyncio.to_thread(_insert_and_flush) + await asyncio.to_thread(_insert_records) logger.info(f"Inserted {content_type} {item} into Milvus. Done.") @@ -363,7 +362,6 @@ class MilvusKB(KnowledgeBase): def _delete_from_milvus(): try: collection.delete(expr) - collection.flush() logger.info(f"Deleted chunks for file {file_id} from Milvus") except Exception as e: logger.error(f"Error deleting file {file_id} from Milvus: {e}")