zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修复混元模型失败的缺陷

Browse Source
This commit is contained in:
wxg0103 2024-08-23 15:58:41 +08:00 committed by wxg0103
parent e0c3da7c43
commit 295e794a6a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/setting/models_provider/impl/tencent_model_provider/credential/llm.py
View File

@ -7,7 +7,7 @@ from setting.models_provider.base_model_provider import BaseModelCredential, Val
class TencentLLMModelCredential(BaseForm, BaseModelCredential): class TencentLLMModelCredential(BaseForm, BaseModelCredential):
REQUIRED_FIELDS = ['hunyuan_secret_id', 'hunyuan_secret_key'] REQUIRED_FIELDS = ['hunyuan_app_id', 'hunyuan_secret_id', 'hunyuan_secret_key']
@classmethod @classmethod
def _validate_model_type(cls, model_type, provider, raise_exception=False): def _validate_model_type(cls, model_type, provider, raise_exception=False):
@ -42,6 +42,7 @@ class TencentLLMModelCredential(BaseForm, BaseModelCredential):
def encryption_dict(self, model): def encryption_dict(self, model):
return {**model, 'hunyuan_secret_key': super().encryption(model.get('hunyuan_secret_key', ''))} return {**model, 'hunyuan_secret_key': super().encryption(model.get('hunyuan_secret_key', ''))}
hunyuan_app_id = forms.TextInputField('APP ID', required=True)
hunyuan_secret_id = forms.PasswordInputField('SecretId', required=True) hunyuan_secret_id = forms.PasswordInputField('SecretId', required=True)
hunyuan_secret_key = forms.PasswordInputField('SecretKey', required=True) hunyuan_secret_key = forms.PasswordInputField('SecretKey', required=True)