|
|
|
@ -158,6 +158,7 @@ class DatasetRetrieverTool(BaseTool):
|
|
|
|
document_score_list = {}
|
|
|
|
document_score_list = {}
|
|
|
|
if dataset.indexing_technique != "economy":
|
|
|
|
if dataset.indexing_technique != "economy":
|
|
|
|
for item in documents:
|
|
|
|
for item in documents:
|
|
|
|
|
|
|
|
if 'score' in item.metadata and item.metadata['score']:
|
|
|
|
document_score_list[item.metadata['doc_id']] = item.metadata['score']
|
|
|
|
document_score_list[item.metadata['doc_id']] = item.metadata['score']
|
|
|
|
document_context_list = []
|
|
|
|
document_context_list = []
|
|
|
|
index_node_ids = [document.metadata['doc_id'] for document in documents]
|
|
|
|
index_node_ids = [document.metadata['doc_id'] for document in documents]
|
|
|
|
|