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)

Files changed (1) hide show
  1. 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.split("/")[-1] != "v1":
158
- base_url = os.path.join(base_url, "v1")
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 = {