hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-10 03:16:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
zhuhao
2024-10-12 23:58:41 +08:00
committed by GitHub
parent ea6734f550
commit d97d3ff5fc
3 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/rag/datasource/keyword/keyword_base.py
View File

@@ -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

2
api/core/rag/datasource/vdb/elasticsearch/elasticsearch_vector.py
View File

@@ -77,7 +77,7 @@ class ElasticSearchVector(BaseVector):
raise ValueError("Elasticsearch vector database version must be greater than 8.0.0")
def get_type(self) -> str:
return "elasticsearch"
return VectorType.ELASTICSEARCH
def add_texts(self, documents: list[Document], embeddings: list[list[float]], **kwargs):
uuids = self._get_uuids(documents)