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

fix score threshold limit be None (#6900)

Browse Source
This commit is contained in:
Jyong
2024-08-02 12:10:51 +08:00
committed by GitHub
parent 56af1a0adf
commit 44801df8f8
4 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/rag/retrieval/dataset_retrieval.py
View File

@@ -138,6 +138,7 @@ class DatasetRetrieval:
retrieve_config.rerank_mode,
retrieve_config.reranking_model,
retrieve_config.weights,
retrieve_config.reranking_enabled,
message_id,
)
@@ -606,7 +607,7 @@ class DatasetRetrieval:
top_k: int, score_threshold: float) -> list[Document]:
filter_documents = []
for document in all_documents:
if document.metadata['score'] >= score_threshold:
if score_threshold and document.metadata['score'] >= score_threshold:
filter_documents.append(document)
if not filter_documents:
return []