feat(db): Implement delete and drop methods for database connectors

Author: Zhe Yu

src/vectorcode/database/base.py

@@ -84,3 +84,14 @@ async def list(
         Otherwise, this method may populate only one of them to save waiting time.
         """
         pass
+
+    @abstractmethod
+    async def delete(self, collection_path: str, file_path: str | Sequence[str]):
+        pass
+
+    @abstractmethod
+    async def drop(self, collection_path: str):
+        """
+        Delete a collection from the database.
+        """
+        pass

src/vectorcode/database/chroma0.py

@@ -8,7 +8,7 @@
 import sys
 from asyncio.subprocess import Process
 from dataclasses import dataclass
-from typing import Any, Optional, cast
+from typing import Any, Optional, Sequence, cast
 from urllib.parse import urlparse
 
 import chromadb
@@ -437,3 +437,15 @@ async def list(self, collection_path, what=None) -> CollectionContent:
                 )
 
         return content
+
+    async def delete(self, collection_path: str, file_path: str | Sequence[str]):
+        collection = await self._create_or_get_collection(collection_path, False)
+        if isinstance(file_path, str):
+            file_path = [file_path]
+        await collection.delete(
+            where={"path": {"$in": [str(expand_path(i, True)) for i in file_path]}}
+        )
+
+    async def drop(self, collection_path: str):
+        async with Chroma0ClientManager().get_client(self._configs) as client:
+            await client.delete_collection(get_collection_id(collection_path))