feat: add api_base field to VolcanicEngineTTIModelCredential and update related classes

Author: wxg0103

apps/models_provider/impl/volcanic_engine_model_provider/credential/tti.py

@@ -9,6 +9,7 @@
 from models_provider.base_model_provider import BaseModelCredential, ValidCode
 from common.utils.logger import maxkb_logger
 
+
 class VolcanicEngineTTIModelGeneralParams(BaseForm):
     size = forms.SingleSelect(
         TooltipLabel(_('Image size'),
@@ -32,6 +33,8 @@ class VolcanicEngineTTIModelGeneralParams(BaseForm):
 
 
 class VolcanicEngineTTIModelCredential(BaseForm, BaseModelCredential):
+    volcanic_api_url = forms.TextInputField('API URL', required=True,
+                                            default_value='https://ark.cn-beijing.volces.com/api/v3')
     api_key = forms.PasswordInputField('Api key', required=True)
 
     def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,

apps/models_provider/impl/volcanic_engine_model_provider/model/tti.py

@@ -16,12 +16,14 @@
 
 class VolcanicEngineTextToImage(MaxKBBaseModel, BaseTextToImage):
     api_key: str
+    api_base: str
     model_version: str
     params: dict
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         self.api_key = kwargs.get('api_key')
+        self.api_base = kwargs.get('api_base')
         self.model_version = kwargs.get('model_version')
         self.params = kwargs.get('params')
 
@@ -38,6 +40,7 @@ def new_instance(model_type, model_name, model_credential: Dict[str, object], **
         return VolcanicEngineTextToImage(
             model_version=model_name,
             api_key=model_credential.get('api_key'),
+            api_base=model_credential.get('api_base') or 'https://ark-api.volcengine.com',
             **optional_params
         )
 
@@ -47,7 +50,7 @@ def check_auth(self):
     def generate_image(self, prompt: str, negative_prompt: str = None):
         client = Ark(
             # 此为默认路径，您可根据业务所在地域进行配置
-            base_url="https://ark.cn-beijing.volces.com/api/v3",
+            base_url=self.api_base,
             # 从环境变量中获取您的 API Key。此为默认方式，您可根据需要进行修改
             api_key=self.api_key,
         )