fix: add API base URL field to AliyunBaiLianReranker and update instance creation

Author: wxg0103

apps/models_provider/impl/aliyun_bai_lian_model_provider/credential/reranker.py

@@ -5,28 +5,31 @@
 from django.utils.translation import gettext as _
 from langchain_core.documents import Document
 
+from common import forms
 from common.exception.app_exception import AppApiException
 from common.forms import BaseForm, PasswordInputField
 from models_provider.base_model_provider import BaseModelCredential, ValidCode
 from models_provider.impl.aliyun_bai_lian_model_provider.model.reranker import AliyunBaiLianReranker
 from common.utils.logger import maxkb_logger
 
+
 class AliyunBaiLianRerankerCredential(BaseForm, BaseModelCredential):
     """
     Credential class for the Aliyun BaiLian Reranker model.
     Provides validation and encryption for the model credentials.
     """
-
+    api_base = forms.TextInputField(_('API URL'), required=True,
+                                    default_value='https://dashscope.aliyuncs.com/compatible-mode/v1')
     dashscope_api_key = PasswordInputField('API Key', required=True)
 
     def is_valid(
-        self,
-        model_type: str,
-        model_name: str,
-        model_credential: Dict[str, Any],
-        model_params: Dict[str, Any],
-        provider,
-        raise_exception: bool = False
+            self,
+            model_type: str,
+            model_name: str,
+            model_credential: Dict[str, Any],
+            model_params: Dict[str, Any],
+            provider,
+            raise_exception: bool = False
     ) -> bool:
         """
         Validate the model credentials.
@@ -65,7 +68,8 @@ def is_valid(
             if raise_exception:
                 raise AppApiException(
                     ValidCode.valid_error.value,
-                    _('Verification failed, please check whether the parameters are correct: {error}').format(error=str(e))
+                    _('Verification failed, please check whether the parameters are correct: {error}').format(
+                        error=str(e))
                 )
             return False
 

apps/models_provider/impl/aliyun_bai_lian_model_provider/model/reranker.py

@@ -20,6 +20,7 @@
 class AliyunBaiLianReranker(MaxKBBaseModel, BaseDocumentCompressor):
     model: Optional[str]
     api_key: Optional[str]
+    base_url: str = ''
 
     top_n: Optional[int] = 3  # 取前 N 个最相关的结果
 
@@ -31,13 +32,14 @@ def is_cache_model():
     def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
         return AliyunBaiLianReranker(model=model_name,
                                      api_key=model_credential.get('dashscope_api_key'),
+                                     base_url=model_credential.get('api_base') or 'https://dashscope.aliyuncs.com/compatible-mode/v1',
                                      top_n=model_kwargs.get('top_n', 3))
 
     def compress_documents(self, documents: Sequence[Document], query: str, callbacks: Optional[Callbacks] = None) -> \
             Sequence[Document]:
         if not documents:
             return []
-
+        dashscope.base_http_api_url = self.base_url
         texts = [doc.page_content for doc in documents]
         resp = dashscope.TextReRank.call(
             model=self.model,