chore: add abstract decorator and output log when query embedding fails (#9264)

2024-10-12 23:58:41 +08:00
parent ea6734f550
commit d97d3ff5fc
3 changed files with 9 additions and 2 deletions
--- a/api/core/rag/datasource/keyword/keyword_base.py
+++ b/api/core/rag/datasource/keyword/keyword_base.py
@@ -27,9 +27,11 @@ class BaseKeyword(ABC):
    def delete_by_ids(self, ids: list[str]) -> None:
        raise NotImplementedError

+    @abstractmethod
    def delete(self) -> None:
        raise NotImplementedError

+    @abstractmethod
    def search(self, query: str, **kwargs: Any) -> list[Document]:
        raise NotImplementedError