feat: Add vector database clearing

2026-05-11 09:59:50 +08:00 · 2025-09-07 16:48:14 +08:00
parent 72403d5861
commit b03f2619ca
5 changed files with 84 additions and 4 deletions
--- a/services/vector_db.py
+++ b/services/vector_db.py
@@ -75,3 +75,9 @@ class VectorDBService:
            output_fields=["path"]
        )
        return [[{'id': r['path'], 'distance': 1.0, 'entity': {'path': r['path']}} for r in results]]
+
+    def clear_all_data(self):
+        """清空所有集合的内容"""
+        collections = self.client.list_collections()
+        for collection_name in collections:
+            self.client.drop_collection(collection_name)