rag
This commit is contained in:
parent
f61bb22316
commit
7bbf3d36c0
@ -16,7 +16,13 @@
|
||||
"label":"文件管理",
|
||||
"selected_row": true,
|
||||
"icon":"{{entire_url('/bricks/imgs/folder.svg')}}"
|
||||
}
|
||||
},
|
||||
{
|
||||
"name": "test",
|
||||
"label": "测试知识库",
|
||||
"selected_data": true,
|
||||
"icon": "{{entire_url('/bricks/imgs/folder.svg')}}"
|
||||
}
|
||||
]
|
||||
},
|
||||
"binds":[
|
||||
@ -34,6 +40,21 @@
|
||||
"options":{
|
||||
"url":"{{entire_url('/rag/folder')}}"
|
||||
}
|
||||
},
|
||||
{
|
||||
"wid": "self",
|
||||
"event": "test",
|
||||
"actiontype": "urlwidget",
|
||||
"target": "PopupWindow",
|
||||
"popup_options": {
|
||||
"width": "80%",
|
||||
"height": "80%",
|
||||
"icon": "{{entire_url('/bricks/imgs/app.svg')}}",
|
||||
"title": "知识库测试"
|
||||
},
|
||||
"options": {
|
||||
"url": "{{entire_url('/rag/test.ui')}}"
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
@ -443,7 +443,19 @@ class RagOperations:
|
||||
def format_search_results(self, results: List[Dict], limit: int) -> List[Dict]:
|
||||
"""格式化搜索结果为统一格式"""
|
||||
formatted_results = []
|
||||
|
||||
# for res in results[:limit]:
|
||||
# score = res.get('rerank_score', res.get('distance', 0))
|
||||
#
|
||||
# content = res.get('text', '')
|
||||
# title = res.get('metadata', {}).get('filename', 'Untitled')
|
||||
# document_id = res.get('metadata', {}).get('document_id', '')
|
||||
#
|
||||
# formatted_results.append({
|
||||
# "content": content,
|
||||
# "title": title,
|
||||
# "metadata": {"document_id": document_id, "score": score},
|
||||
# })
|
||||
#得分归一化
|
||||
for res in results[:limit]:
|
||||
rerank_score = res.get('rerank_score', 0)
|
||||
score = 1 / (1 + math.exp(-rerank_score)) if rerank_score is not None else 1 - res.get('distance', 0)
|
||||
|
||||
@ -1,8 +1,9 @@
|
||||
debug(f'{params_kw=}')
|
||||
fiid = params_kw.ffid
|
||||
fiids = params_kw.fiids
|
||||
query = params_kw.query
|
||||
limit = params_kw.limit
|
||||
env = DictObject(**globals())
|
||||
keys = [k for k in env.keys()]
|
||||
debug(f'{keys=}')
|
||||
x = await rfexe('fusedsearch', params_kw, **env)
|
||||
x = await rfexe('fusedsearch', request, params_kw)
|
||||
return x
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user