Fix: url path join issue. (#8013)

### What problem does this PR solve?

Close #7980

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2025-06-03 14:18:40 +08:00
committed by GitHub
parent 37075eab98
commit 156290f8d0
4 changed files with 25 additions and 41 deletions

View File

@ -296,12 +296,11 @@ class NvidiaRerank(Base):
self.model_name = model_name
if self.model_name == "nvidia/nv-rerankqa-mistral-4b-v3":
self.base_url = os.path.join(
base_url, "nv-rerankqa-mistral-4b-v3", "reranking"
self.base_url = urljoin(base_url, "nv-rerankqa-mistral-4b-v3/reranking"
)
if self.model_name == "nvidia/rerank-qa-mistral-4b":
self.base_url = os.path.join(base_url, "reranking")
self.base_url = urljoin(base_url, "reranking")
self.model_name = "nv-rerank-qa-mistral-4b:1"
self.headers = {