AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-07-10 05:21:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix retrieval resource positon missed (#9155)

Browse Source 
Co-authored-by: Bowen Liang <liangbowen@gf.com.cn>
This commit is contained in:
Jyong 2024-10-11 10:32:42 +08:00 committed by GitHub
parent a34891851b
commit cabdb4ef17
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/rag/entities/context_entities.py
View File

@ -1,3 +1,5 @@
from typing import Optional
from pydantic import BaseModel from pydantic import BaseModel
@ -7,4 +9,4 @@ class DocumentContext(BaseModel):
""" """
content: str content: str
score: float score: Optional[float] = None

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

@ -231,6 +231,9 @@ class DatasetRetrieval:
source["content"] = segment.content source["content"] = segment.content
retrieval_resource_list.append(source) retrieval_resource_list.append(source)
if hit_callback and retrieval_resource_list: if hit_callback and retrieval_resource_list:
retrieval_resource_list = sorted(retrieval_resource_list, key=lambda x: x.get("score"), reverse=True)
for position, item in enumerate(retrieval_resource_list, start=1):
item["position"] = position
hit_callback.return_retriever_resource_info(retrieval_resource_list) hit_callback.return_retriever_resource_info(retrieval_resource_list)
if document_context_list: if document_context_list:
document_context_list = sorted(document_context_list, key=lambda x: x.score, reverse=True) document_context_list = sorted(document_context_list, key=lambda x: x.score, reverse=True)