Sky Blue
commited on
Commit
·
f2ca0b5
1
Parent(s):
db99ead
fix xinference add rerank model bug (#2758)
Browse files### What problem does this PR solve?
Fix xinference add rerank model bug,
https://github.com/infiniflow/ragflow/issues/2294#issue-2510788135
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
- rag/llm/rerank_model.py +4 -2
rag/llm/rerank_model.py
CHANGED
|
@@ -15,6 +15,8 @@
|
|
| 15 |
#
|
| 16 |
import re
|
| 17 |
import threading
|
|
|
|
|
|
|
| 18 |
import requests
|
| 19 |
from huggingface_hub import snapshot_download
|
| 20 |
import os
|
|
@@ -154,8 +156,8 @@ class YoudaoRerank(DefaultRerank):
|
|
| 154 |
|
| 155 |
class XInferenceRerank(Base):
|
| 156 |
def __init__(self, key="xxxxxxx", model_name="", base_url=""):
|
| 157 |
-
if base_url.
|
| 158 |
-
base_url =
|
| 159 |
self.model_name = model_name
|
| 160 |
self.base_url = base_url
|
| 161 |
self.headers = {
|
|
|
|
| 15 |
#
|
| 16 |
import re
|
| 17 |
import threading
|
| 18 |
+
from urllib.parse import urljoin
|
| 19 |
+
|
| 20 |
import requests
|
| 21 |
from huggingface_hub import snapshot_download
|
| 22 |
import os
|
|
|
|
| 156 |
|
| 157 |
class XInferenceRerank(Base):
|
| 158 |
def __init__(self, key="xxxxxxx", model_name="", base_url=""):
|
| 159 |
+
if base_url.find("/v1") == -1:
|
| 160 |
+
base_url = urljoin(base_url, "/v1/rerank")
|
| 161 |
self.model_name = model_name
|
| 162 |
self.base_url = base_url
|
| 163 |
self.headers = {
|