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 = {
|