mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-27 20:42:52 +00:00
116 lines
5.5 KiB
Python
116 lines
5.5 KiB
Python
# coding=utf-8
|
||
|
||
from django.utils.translation import gettext_lazy as _, gettext
|
||
|
||
from common import forms
|
||
from common.exception.app_exception import AppApiException
|
||
from common.forms import BaseForm, TooltipLabel
|
||
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
||
from common.utils.logger import maxkb_logger
|
||
|
||
class TencentTTIModelParams(BaseForm):
|
||
Style = forms.SingleSelect(
|
||
TooltipLabel(_('painting style'), _('If not passed, the default value is 201 (Japanese anime style)')),
|
||
required=True,
|
||
default_value='201',
|
||
option_list=[
|
||
{'value': '000', 'label': _('Not limited to style')},
|
||
{'value': '101', 'label': _('ink painting')},
|
||
{'value': '102', 'label': _('concept art')},
|
||
{'value': '103', 'label': _('Oil painting 1')},
|
||
{'value': '118', 'label': _('Oil Painting 2 (Van Gogh)')},
|
||
{'value': '104', 'label': _('watercolor painting')},
|
||
{'value': '105', 'label': _('pixel art')},
|
||
{'value': '106', 'label': _('impasto style')},
|
||
{'value': '107', 'label': _('illustration')},
|
||
{'value': '108', 'label': _('paper cut style')},
|
||
{'value': '109', 'label': _('Impressionism 1 (Monet)')},
|
||
{'value': '119', 'label': _('Impressionism 2')},
|
||
{'value': '110', 'label': '2.5D'},
|
||
{'value': '111', 'label': _('classical portraiture')},
|
||
{'value': '112', 'label': _('black and white sketch')},
|
||
{'value': '113', 'label': _('cyberpunk')},
|
||
{'value': '114', 'label': _('science fiction style')},
|
||
{'value': '115', 'label': _('dark style')},
|
||
{'value': '116', 'label': '3D'},
|
||
{'value': '117', 'label': _('vaporwave')},
|
||
{'value': '201', 'label': _('Japanese animation')},
|
||
{'value': '202', 'label': _('monster style')},
|
||
{'value': '203', 'label': _('Beautiful ancient style')},
|
||
{'value': '204', 'label': _('retro anime')},
|
||
{'value': '301', 'label': _('Game cartoon hand drawing')},
|
||
{'value': '401', 'label': _('Universal realistic style')},
|
||
],
|
||
value_field='value',
|
||
text_field='label'
|
||
)
|
||
|
||
Resolution = forms.SingleSelect(
|
||
TooltipLabel(_('Generate image resolution'), _('If not transmitted, the default value is 768:768.')),
|
||
required=True,
|
||
default_value='768:768',
|
||
option_list=[
|
||
{'value': '768:768', 'label': '768:768（1:1）'},
|
||
{'value': '768:1024', 'label': '768:1024（3:4）'},
|
||
{'value': '1024:768', 'label': '1024:768（4:3）'},
|
||
{'value': '1024:1024', 'label': '1024:1024（1:1）'},
|
||
{'value': '720:1280', 'label': '720:1280（9:16）'},
|
||
{'value': '1280:720', 'label': '1280:720（16:9）'},
|
||
{'value': '768:1280', 'label': '768:1280（3:5）'},
|
||
{'value': '1280:768', 'label': '1280:768（5:3）'},
|
||
{'value': '1080:1920', 'label': '1080:1920（9:16）'},
|
||
{'value': '1920:1080', 'label': '1920:1080（16:9）'},
|
||
],
|
||
value_field='value',
|
||
text_field='label'
|
||
)
|
||
|
||
|
||
class TencentTTIModelCredential(BaseForm, BaseModelCredential):
|
||
REQUIRED_FIELDS = ['hunyuan_secret_id', 'hunyuan_secret_key']
|
||
|
||
@classmethod
|
||
def _validate_model_type(cls, model_type, provider, raise_exception=False):
|
||
if not any(mt['value'] == model_type for mt in provider.get_model_type_list()):
|
||
if raise_exception:
|
||
raise AppApiException(ValidCode.valid_error.value,
|
||
gettext('{model_type} Model type is not supported').format(model_type=model_type))
|
||
return False
|
||
return True
|
||
|
||
@classmethod
|
||
def _validate_credential_fields(cls, model_credential, raise_exception=False):
|
||
missing_keys = [key for key in cls.REQUIRED_FIELDS if key not in model_credential]
|
||
if missing_keys:
|
||
if raise_exception:
|
||
raise AppApiException(ValidCode.valid_error.value,
|
||
gettext('{keys} is required').format(keys=", ".join(missing_keys)))
|
||
return False
|
||
return True
|
||
|
||
def is_valid(self, model_type, model_name, model_credential, model_params, provider, raise_exception=False):
|
||
if not (self._validate_model_type(model_type, provider, raise_exception) and
|
||
self._validate_credential_fields(model_credential, raise_exception)):
|
||
return False
|
||
try:
|
||
model = provider.get_model(model_type, model_name, model_credential, **model_params)
|
||
model.check_auth()
|
||
except Exception as e:
|
||
maxkb_logger.error(f'Exception: {e}', exc_info=True)
|
||
if raise_exception:
|
||
raise AppApiException(ValidCode.valid_error.value,
|
||
gettext(
|
||
'Verification failed, please check whether the parameters are correct: {error}').format(
|
||
error=str(e)))
|
||
return False
|
||
return True
|
||
|
||
def encryption_dict(self, model):
|
||
return {**model, 'hunyuan_secret_key': super().encryption(model.get('hunyuan_secret_key', ''))}
|
||
|
||
hunyuan_secret_id = forms.PasswordInputField('SecretId', required=True)
|
||
hunyuan_secret_key = forms.PasswordInputField('SecretKey', required=True)
|
||
|
||
def get_model_params_setting_form(self, model_name):
|
||
return TencentTTIModelParams()
|