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

check rerank document is not empty (#1740)

Browse Source 
Co-authored-by: jyong <jyong@dify.ai>
This commit is contained in:
Jyong
2023-12-11 16:12:11 +08:00
committed by GitHub
parent 994fceece3
commit d5695b3170
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/model_providers/models/reranking/cohere_reranking.py
View File

@@ -24,7 +24,10 @@ class CohereReranking(BaseReranking):
super().__init__(model_provider, client, name) super().__init__(model_provider, client, name)
def rerank(self, query: str, documents: List[Document], score_threshold: Optional[float], top_k: Optional[int]) -> Optional[List[Document]]: def rerank(self, query: str, documents: List[Document], score_threshold: Optional[float], top_k: Optional[int]) -> \
Optional[List[Document]]:
if not documents:
return []
docs = [] docs = []
doc_id = [] doc_id = []
unique_documents = [] unique_documents = []

2
api/core/model_providers/models/reranking/xinference_reranking.py
View File

@@ -21,6 +21,8 @@ class XinferenceReranking(BaseReranking):
super().__init__(model_provider, client, name) super().__init__(model_provider, client, name)
def rerank(self, query: str, documents: List[Document], score_threshold: Optional[float], top_k: Optional[int]) -> Optional[List[Document]]: def rerank(self, query: str, documents: List[Document], score_threshold: Optional[float], top_k: Optional[int]) -> Optional[List[Document]]:
if not documents:
return []
docs = [] docs = []
doc_id = [] doc_id = []
unique_documents = [] unique_documents = []