test: add tests covering all methods of vector store (#3849)

2025-12-09 19:06:51 +08:00 · 2024-04-25 22:27:30 +08:00
parent 8bca908f15
commit 45dd1683fd
5 changed files with 48 additions and 4 deletions
--- a/api/core/rag/datasource/vdb/vector_base.py
+++ b/api/core/rag/datasource/vdb/vector_base.py
@@ -27,6 +27,12 @@ class BaseVector(ABC):
    def delete_by_ids(self, ids: list[str]) -> None:
        raise NotImplementedError

+    def delete_by_document_id(self, document_id: str):
+        raise NotImplementedError
+
+    def get_ids_by_metadata_field(self, key: str, value: str):
+        raise NotImplementedError
+
    @abstractmethod
    def delete_by_metadata_field(self, key: str, value: str) -> None:
        raise NotImplementedError