pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix open AI compatible rerank issue. (#3866)

Browse Source 
### What problem does this PR solve?
#3700
### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2024-12-05 10:26:21 +08:00
committed by GitHub
parent 84afb4259c
commit 78601ee1bd
2 changed files with 27 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rag/llm/rerank_model.py
View File

@ -286,7 +286,7 @@ class OpenAI_APIRerank(Base):
"Content-Type": "application/json",
"Authorization": f"Bearer {key}"
}
self.model_name = model_name
self.model_name = model_name.split("___")[0]
def similarity(self, query: str, texts: list):
# noway to config Ragflow , use fix setting