From 60c3a253ad1eca73cab1f2162b17c98577cce337 Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Tue, 25 Mar 2025 15:01:13 +0800
Subject: [PATCH] Fix: api-key issue for xinference. (#6490)

### What problem does this PR solve?

#2792

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 rag/llm/rerank_model.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/rag/llm/rerank_model.py b/rag/llm/rerank_model.py
index 0f2078035..09631e993 100644
--- a/rag/llm/rerank_model.py
+++ b/rag/llm/rerank_model.py
@@ -208,7 +208,7 @@ class YoudaoRerank(DefaultRerank):
 
 
 class XInferenceRerank(Base):
-    def __init__(self, key="xxxxxxx", model_name="", base_url=""):
+    def __init__(self, key="x", model_name="", base_url=""):
         if base_url.find("/v1") == -1:
             base_url = urljoin(base_url, "/v1/rerank")
         if base_url.find("/rerank") == -1:
@@ -217,9 +217,10 @@ class XInferenceRerank(Base):
         self.base_url = base_url
         self.headers = {
             "Content-Type": "application/json",
-            "accept": "application/json",
-            "Authorization": f"Bearer {key}"
+            "accept": "application/json"
         }
+        if key and key != "x":
+            self.headers["Authorization"] = f"Bearer {key}"
 
     def similarity(self, query: str, texts: list):
         if len(texts) == 0: