jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix multi retrieval with resource score issue (#1578)

Browse source 
Co-authored-by: jyong <jyong@dify.ai>
This commit is contained in:
Jyong 2023-11-21 13:47:55 +08:00 • committed by GitHub
commit 0d9ce1bab0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
api/core/tool/dataset_retriever_tool.py
View file

@ -196,10 +196,10 @@ class DatasetRetrieverTool(BaseTool):
'document_name': document.name,
'data_source_type': document.data_source_type,
'segment_id': segment.id,
'retriever_from': self.retriever_from
'retriever_from': self.retriever_from,
'score': document_score_list.get(segment.index_node_id, None)
}
if dataset.indexing_technique != "economy":
source['score'] = document_score_list.get(segment.index_node_id)
if self.retriever_from == 'dev':
source['hit_count'] = segment.hit_count
source['word_count'] = segment.word_count