rag
This commit is contained in:
parent
f61bb22316
commit
7bbf3d36c0
@ -16,6 +16,12 @@
|
|||||||
"label":"文件管理",
|
"label":"文件管理",
|
||||||
"selected_row": true,
|
"selected_row": true,
|
||||||
"icon":"{{entire_url('/bricks/imgs/folder.svg')}}"
|
"icon":"{{entire_url('/bricks/imgs/folder.svg')}}"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "test",
|
||||||
|
"label": "测试知识库",
|
||||||
|
"selected_data": true,
|
||||||
|
"icon": "{{entire_url('/bricks/imgs/folder.svg')}}"
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
@ -34,6 +40,21 @@
|
|||||||
"options":{
|
"options":{
|
||||||
"url":"{{entire_url('/rag/folder')}}"
|
"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]:
|
def format_search_results(self, results: List[Dict], limit: int) -> List[Dict]:
|
||||||
"""格式化搜索结果为统一格式"""
|
"""格式化搜索结果为统一格式"""
|
||||||
formatted_results = []
|
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]:
|
for res in results[:limit]:
|
||||||
rerank_score = res.get('rerank_score', 0)
|
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)
|
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=}')
|
debug(f'{params_kw=}')
|
||||||
fiid = params_kw.ffid
|
fiids = params_kw.fiids
|
||||||
query = params_kw.query
|
query = params_kw.query
|
||||||
|
limit = params_kw.limit
|
||||||
env = DictObject(**globals())
|
env = DictObject(**globals())
|
||||||
keys = [k for k in env.keys()]
|
keys = [k for k in env.keys()]
|
||||||
debug(f'{keys=}')
|
debug(f'{keys=}')
|
||||||
x = await rfexe('fusedsearch', params_kw, **env)
|
x = await rfexe('fusedsearch', request, params_kw)
|
||||||
return x
|
return x
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user