ouyangyouzhang/QueryRewrite
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新重排序模型的API地址，

Browse Source
This commit is contained in:
ouyangyouzhang
2025-07-08 08:55:50 +08:00
parent 5a8d042360
commit 9636ed3ed2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rag2_0/tool/ModelTool.py
+2 -2
View File
@@ -163,7 +163,7 @@ class XinferenceReRankerModel:
Returns: Returns:
List[dict]: 重排序后的文档列表，每个元素包含document内容、相关性分数和原始索引 List[dict]: 重排序后的文档列表，每个元素包含document内容、相关性分数和原始索引
""" """
url = "http://172.20.0.145:9995/v1/rerank" url = "http://10.1.16.39:9995/v1/rerank"
params = {"documents": documents, "query": query, "top_n": top_k, "return_documents": True, "model": "bge-reranker-v2-m3"} params = {"documents": documents, "query": query, "top_n": top_k, "return_documents": True, "model": "bge-reranker-v2-m3"}
@@ -197,7 +197,7 @@ class XinferenceReRankerModel:
Returns: Returns:
List[dict]: 重排序后的文档列表，每个元素包含document内容、相关性分数和原始索引 List[dict]: 重排序后的文档列表，每个元素包含document内容、相关性分数和原始索引
""" """
url = "http://172.20.0.145:9995/v1/rerank" url = "http://10.1.16.39:9995/v1/rerank"
params = {"documents": documents, "query": query, "top_n": top_k, "return_documents": True, "model": "bge-reranker-v2-m3"} params = {"documents": documents, "query": query, "top_n": top_k, "return_documents": True, "model": "bge-reranker-v2-m3"}
headers = { headers = {