refactor: check model use model_params

Author: liuruibin

apps/setting/models_provider/__init__.py

@@ -81,7 +81,7 @@ def get_model_type_list(provider):
     return get_provider(provider).get_model_type_list()
 
 
-def is_valid_credential(provider, model_type, model_name, model_credential: Dict[str, object], raise_exception=False):
+def is_valid_credential(provider, model_type, model_name, model_credential: Dict[str, object], model_params, raise_exception=False):
     """
     校验模型认证参数
     @param provider:         供应商字符串
@@ -91,4 +91,4 @@ def is_valid_credential(provider, model_type, model_name, model_credential: Dict
     @param raise_exception:  是否抛出错误
     @return: True|False
     """
-    return get_provider(provider).is_valid_credential(model_type, model_name, model_credential, raise_exception)
+    return get_provider(provider).is_valid_credential(model_type, model_name, model_credential, model_params, raise_exception)

apps/setting/models_provider/base_model_provider.py

@@ -67,9 +67,13 @@ def get_model_credential(self, model_type, model_name):
         model_info = self.get_model_info_manage().get_model_info(model_type, model_name)
         return model_info.model_credential
 
-    def is_valid_credential(self, model_type, model_name, model_credential: Dict[str, object], raise_exception=False):
+    def get_model_params(self, model_type, model_name):
         model_info = self.get_model_info_manage().get_model_info(model_type, model_name)
-        return model_info.model_credential.is_valid(model_type, model_name, model_credential, self,
+        return model_info.model_credential
+
+    def is_valid_credential(self, model_type, model_name, model_credential: Dict[str, object], model_params: Dict[str, object], raise_exception=False):
+        model_info = self.get_model_info_manage().get_model_info(model_type, model_name)
+        return model_info.model_credential.is_valid(model_type, model_name, model_credential, model_params, self,
                                                     raise_exception=raise_exception)
 
     def get_model(self, model_type, model_name, model_credential: Dict[str, object], **model_kwargs) -> BaseModel:
@@ -105,7 +109,7 @@ def filter_optional_params(model_kwargs):
 class BaseModelCredential(ABC):
 
     @abstractmethod
-    def is_valid(self, model_type: str, model_name, model: Dict[str, object], provider, raise_exception=True):
+    def is_valid(self, model_type: str, model_name, model: Dict[str, object], model_params, provider, raise_exception=True):
         pass
 
     @abstractmethod

apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/embedding.py

@@ -17,7 +17,7 @@
 
 class AliyunBaiLianEmbeddingCredential(BaseForm, BaseModelCredential):
 
-    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], provider,
+    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,
                  raise_exception=False):
         model_type_list = provider.get_model_type_list()
         if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):

apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/image.py

@@ -37,7 +37,7 @@ class QwenModelParams(BaseForm):
 
 class QwenVLModelCredential(BaseForm, BaseModelCredential):
 
-    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], provider,
+    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,
                  raise_exception=False):
         model_type_list = provider.get_model_type_list()
         if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
@@ -49,7 +49,7 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje
                 else:
                     return False
         try:
-            model = provider.get_model(model_type, model_name, model_credential)
+            model = provider.get_model(model_type, model_name, model_credential, **model_params)
             res = model.stream([HumanMessage(content=[{"type": "text", "text": "你好"}])])
             for chunk in res:
                 print(chunk)

apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/llm.py

@@ -28,7 +28,7 @@ class BaiLianLLMModelParams(BaseForm):
 
 class BaiLianLLMModelCredential(BaseForm, BaseModelCredential):
 
-    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], provider,
+    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,
                  raise_exception=False):
         model_type_list = provider.get_model_type_list()
         if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
@@ -41,7 +41,7 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje
                 else:
                     return False
         try:
-            model = provider.get_model(model_type, model_name, model_credential)
+            model = provider.get_model(model_type, model_name, model_credential, **model_params)
             model.invoke([HumanMessage(content='你好')])
         except Exception as e:
             if isinstance(e, AppApiException):

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

@@ -19,7 +19,7 @@
 
 class AliyunBaiLianRerankerCredential(BaseForm, BaseModelCredential):
 
-    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], provider,
+    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,
                  raise_exception=False):
         if not model_type == 'RERANKER':
             raise AppApiException(ValidCode.valid_error.value, f'{model_type} 模型类型不支持')

apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/stt.py

@@ -11,7 +11,7 @@
 class AliyunBaiLianSTTModelCredential(BaseForm, BaseModelCredential):
     api_key = forms.PasswordInputField("API Key", required=True)
 
-    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], provider,
+    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,
                  raise_exception=False):
         model_type_list = provider.get_model_type_list()
         if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):

apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/tti.py

@@ -61,7 +61,7 @@ class QwenModelParams(BaseForm):
 
 class QwenTextToImageModelCredential(BaseForm, BaseModelCredential):
 
-    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], provider,
+    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,
                  raise_exception=False):
         model_type_list = provider.get_model_type_list()
         if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
@@ -73,7 +73,7 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje
                 else:
                     return False
         try:
-            model = provider.get_model(model_type, model_name, model_credential)
+            model = provider.get_model(model_type, model_name, model_credential, **model_params)
             res = model.check_auth()
             print(res)
         except Exception as e:

apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/tts.py

@@ -45,7 +45,7 @@ class AliyunBaiLianTTSModelGeneralParams(BaseForm):
 class AliyunBaiLianTTSModelCredential(BaseForm, BaseModelCredential):
     api_key = forms.PasswordInputField("API Key", required=True)
 
-    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], provider,
+    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,
                  raise_exception=False):
         model_type_list = provider.get_model_type_list()
         if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
@@ -58,7 +58,7 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje
                 else:
                     return False
         try:
-            model = provider.get_model(model_type, model_name, model_credential)
+            model = provider.get_model(model_type, model_name, model_credential, **model_params)
             model.check_auth()
         except Exception as e:
             if isinstance(e, AppApiException):

apps/setting/models_provider/impl/aws_bedrock_model_provider/credential/embedding.py

@@ -26,7 +26,7 @@ def _update_aws_credentials(profile_name, access_key_id, secret_access_key):
         with open(credentials_path, 'w') as file:
             file.write(content)
 
-    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], provider,
+    def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,
                  raise_exception=False):
         model_type_list = provider.get_model_type_list()
         if not any(mt.get('value') == model_type for mt in model_type_list):