zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-30 20:49:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: replace unsupported default qwen rerank model (#17535)

Browse Source
This commit is contained in:
Lynn
2026-07-29 17:40:02 +08:00
committed by GitHub
parent 33bfc11fa4
commit 317363e513
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
rag/llm/rerank_model.py
View File

@@ -519,10 +519,9 @@ class QWenRerank(Base):
_FACTORY_NAME = "Tongyi-Qianwen"
def __init__(self, key, model_name="gte-rerank-v2", **kwargs):
import dashscope
self.api_key = key
self.model_name = dashscope.TextReRank.Models.gte_rerank if model_name is None else model_name
self.model_name = model_name if model_name else "gte-rerank-v2"
# Remove invalid global timeout, use official SDK per-request timeout parameter
self.request_timeout = 30.0