refactor: remove image params
This commit is contained in:
parent
11331a60e0
commit
cc0e431605
@ -14,22 +14,7 @@ from django.utils.translation import gettext_lazy as _, gettext
|
|||||||
|
|
||||||
|
|
||||||
class AzureOpenAIImageModelParams(BaseForm):
|
class AzureOpenAIImageModelParams(BaseForm):
|
||||||
temperature = forms.SliderField(TooltipLabel(_('Temperature'),
|
pass
|
||||||
_('Higher values make the output more random, while lower values make it more focused and deterministic')),
|
|
||||||
required=True, default_value=0.7,
|
|
||||||
_min=0.1,
|
|
||||||
_max=1.0,
|
|
||||||
_step=0.01,
|
|
||||||
precision=2)
|
|
||||||
|
|
||||||
max_tokens = forms.SliderField(
|
|
||||||
TooltipLabel(_('Output the maximum Tokens'),
|
|
||||||
_('Specify the maximum number of tokens that the model can generate')),
|
|
||||||
required=True, default_value=800,
|
|
||||||
_min=1,
|
|
||||||
_max=100000,
|
|
||||||
_step=1,
|
|
||||||
precision=0)
|
|
||||||
|
|
||||||
|
|
||||||
class AzureOpenAIImageModelCredential(BaseForm, BaseModelCredential):
|
class AzureOpenAIImageModelCredential(BaseForm, BaseModelCredential):
|
||||||
|
|||||||
@ -11,25 +11,6 @@ from common.forms import BaseForm, TooltipLabel
|
|||||||
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
||||||
|
|
||||||
|
|
||||||
class GeminiImageModelParams(BaseForm):
|
|
||||||
temperature = forms.SliderField(TooltipLabel(_('Temperature'),
|
|
||||||
_('Higher values make the output more random, while lower values make it more focused and deterministic')),
|
|
||||||
required=True, default_value=0.7,
|
|
||||||
_min=0.1,
|
|
||||||
_max=1.0,
|
|
||||||
_step=0.01,
|
|
||||||
precision=2)
|
|
||||||
|
|
||||||
max_tokens = forms.SliderField(
|
|
||||||
TooltipLabel(_('Output the maximum Tokens'),
|
|
||||||
_('Specify the maximum number of tokens that the model can generate')),
|
|
||||||
required=True, default_value=800,
|
|
||||||
_min=1,
|
|
||||||
_max=100000,
|
|
||||||
_step=1,
|
|
||||||
precision=0)
|
|
||||||
|
|
||||||
|
|
||||||
class GeminiImageModelCredential(BaseForm, BaseModelCredential):
|
class GeminiImageModelCredential(BaseForm, BaseModelCredential):
|
||||||
api_key = forms.PasswordInputField('API Key', required=True)
|
api_key = forms.PasswordInputField('API Key', required=True)
|
||||||
|
|
||||||
@ -68,4 +49,4 @@ class GeminiImageModelCredential(BaseForm, BaseModelCredential):
|
|||||||
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
||||||
|
|
||||||
def get_model_params_setting_form(self, model_name):
|
def get_model_params_setting_form(self, model_name):
|
||||||
return GeminiImageModelParams()
|
pass
|
||||||
|
|||||||
@ -7,26 +7,6 @@ from common.forms import BaseForm, TooltipLabel
|
|||||||
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
||||||
from django.utils.translation import gettext_lazy as _, gettext
|
from django.utils.translation import gettext_lazy as _, gettext
|
||||||
|
|
||||||
|
|
||||||
class OllamaImageModelParams(BaseForm):
|
|
||||||
temperature = forms.SliderField(TooltipLabel(_('Temperature'),
|
|
||||||
_('Higher values make the output more random, while lower values make it more focused and deterministic')),
|
|
||||||
required=True, default_value=0.7,
|
|
||||||
_min=0.1,
|
|
||||||
_max=1.0,
|
|
||||||
_step=0.01,
|
|
||||||
precision=2)
|
|
||||||
|
|
||||||
max_tokens = forms.SliderField(
|
|
||||||
TooltipLabel(_('Output the maximum Tokens'),
|
|
||||||
_('Specify the maximum number of tokens that the model can generate')),
|
|
||||||
required=True, default_value=800,
|
|
||||||
_min=1,
|
|
||||||
_max=100000,
|
|
||||||
_step=1,
|
|
||||||
precision=0)
|
|
||||||
|
|
||||||
|
|
||||||
class OllamaImageModelCredential(BaseForm, BaseModelCredential):
|
class OllamaImageModelCredential(BaseForm, BaseModelCredential):
|
||||||
api_base = forms.TextInputField('API URL', required=True)
|
api_base = forms.TextInputField('API URL', required=True)
|
||||||
api_key = forms.PasswordInputField('API Key', required=True)
|
api_key = forms.PasswordInputField('API Key', required=True)
|
||||||
@ -53,4 +33,4 @@ class OllamaImageModelCredential(BaseForm, BaseModelCredential):
|
|||||||
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
||||||
|
|
||||||
def get_model_params_setting_form(self, model_name):
|
def get_model_params_setting_form(self, model_name):
|
||||||
return OllamaImageModelParams()
|
pass
|
||||||
|
|||||||
@ -13,25 +13,6 @@ from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
|||||||
from django.utils.translation import gettext_lazy as _, gettext
|
from django.utils.translation import gettext_lazy as _, gettext
|
||||||
|
|
||||||
|
|
||||||
class OpenAIImageModelParams(BaseForm):
|
|
||||||
temperature = forms.SliderField(TooltipLabel(_('Temperature'),
|
|
||||||
_('Higher values make the output more random, while lower values make it more focused and deterministic')),
|
|
||||||
required=True, default_value=0.7,
|
|
||||||
_min=0.1,
|
|
||||||
_max=1.0,
|
|
||||||
_step=0.01,
|
|
||||||
precision=2)
|
|
||||||
|
|
||||||
max_tokens = forms.SliderField(
|
|
||||||
TooltipLabel(_('Output the maximum Tokens'),
|
|
||||||
_('Specify the maximum number of tokens that the model can generate')),
|
|
||||||
required=True, default_value=800,
|
|
||||||
_min=1,
|
|
||||||
_max=100000,
|
|
||||||
_step=1,
|
|
||||||
precision=0)
|
|
||||||
|
|
||||||
|
|
||||||
class OpenAIImageModelCredential(BaseForm, BaseModelCredential):
|
class OpenAIImageModelCredential(BaseForm, BaseModelCredential):
|
||||||
api_base = forms.TextInputField('API URL', required=True)
|
api_base = forms.TextInputField('API URL', required=True)
|
||||||
api_key = forms.PasswordInputField('API Key', required=True)
|
api_key = forms.PasswordInputField('API Key', required=True)
|
||||||
@ -71,4 +52,4 @@ class OpenAIImageModelCredential(BaseForm, BaseModelCredential):
|
|||||||
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
||||||
|
|
||||||
def get_model_params_setting_form(self, model_name):
|
def get_model_params_setting_form(self, model_name):
|
||||||
return OpenAIImageModelParams()
|
pass
|
||||||
|
|||||||
@ -13,25 +13,6 @@ from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
|||||||
from django.utils.translation import gettext_lazy as _, gettext
|
from django.utils.translation import gettext_lazy as _, gettext
|
||||||
|
|
||||||
|
|
||||||
class SiliconCloudImageModelParams(BaseForm):
|
|
||||||
temperature = forms.SliderField(TooltipLabel(_('Temperature'),
|
|
||||||
_('Higher values make the output more random, while lower values make it more focused and deterministic')),
|
|
||||||
required=True, default_value=0.7,
|
|
||||||
_min=0.1,
|
|
||||||
_max=1.0,
|
|
||||||
_step=0.01,
|
|
||||||
precision=2)
|
|
||||||
|
|
||||||
max_tokens = forms.SliderField(
|
|
||||||
TooltipLabel(_('Output the maximum Tokens'),
|
|
||||||
_('Specify the maximum number of tokens that the model can generate')),
|
|
||||||
required=True, default_value=800,
|
|
||||||
_min=1,
|
|
||||||
_max=100000,
|
|
||||||
_step=1,
|
|
||||||
precision=0)
|
|
||||||
|
|
||||||
|
|
||||||
class SiliconCloudImageModelCredential(BaseForm, BaseModelCredential):
|
class SiliconCloudImageModelCredential(BaseForm, BaseModelCredential):
|
||||||
api_base = forms.TextInputField('API URL', required=True)
|
api_base = forms.TextInputField('API URL', required=True)
|
||||||
api_key = forms.PasswordInputField('API Key', required=True)
|
api_key = forms.PasswordInputField('API Key', required=True)
|
||||||
@ -71,4 +52,4 @@ class SiliconCloudImageModelCredential(BaseForm, BaseModelCredential):
|
|||||||
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
||||||
|
|
||||||
def get_model_params_setting_form(self, model_name):
|
def get_model_params_setting_form(self, model_name):
|
||||||
return SiliconCloudImageModelParams()
|
pass
|
||||||
|
|||||||
@ -18,25 +18,6 @@ from common.forms import BaseForm, TooltipLabel
|
|||||||
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
||||||
|
|
||||||
|
|
||||||
class QwenModelParams(BaseForm):
|
|
||||||
temperature = forms.SliderField(TooltipLabel(_('Temperature'),
|
|
||||||
_('Higher values make the output more random, while lower values make it more focused and deterministic')),
|
|
||||||
required=True, default_value=1.0,
|
|
||||||
_min=0.1,
|
|
||||||
_max=1.9,
|
|
||||||
_step=0.01,
|
|
||||||
precision=2)
|
|
||||||
|
|
||||||
max_tokens = forms.SliderField(
|
|
||||||
TooltipLabel(_('Output the maximum Tokens'),
|
|
||||||
_('Specify the maximum number of tokens that the model can generate')),
|
|
||||||
required=True, default_value=800,
|
|
||||||
_min=1,
|
|
||||||
_max=100000,
|
|
||||||
_step=1,
|
|
||||||
precision=0)
|
|
||||||
|
|
||||||
|
|
||||||
class TencentVisionModelCredential(BaseForm, BaseModelCredential):
|
class TencentVisionModelCredential(BaseForm, BaseModelCredential):
|
||||||
|
|
||||||
def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,
|
def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,
|
||||||
@ -75,4 +56,4 @@ class TencentVisionModelCredential(BaseForm, BaseModelCredential):
|
|||||||
api_key = forms.PasswordInputField('API Key', required=True)
|
api_key = forms.PasswordInputField('API Key', required=True)
|
||||||
|
|
||||||
def get_model_params_setting_form(self, model_name):
|
def get_model_params_setting_form(self, model_name):
|
||||||
return QwenModelParams()
|
pass
|
||||||
|
|||||||
@ -10,26 +10,6 @@ from common.exception.app_exception import AppApiException
|
|||||||
from common.forms import BaseForm, TooltipLabel
|
from common.forms import BaseForm, TooltipLabel
|
||||||
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
||||||
|
|
||||||
|
|
||||||
class VllmImageModelParams(BaseForm):
|
|
||||||
temperature = forms.SliderField(TooltipLabel(_('Temperature'),
|
|
||||||
_('Higher values make the output more random, while lower values make it more focused and deterministic')),
|
|
||||||
required=True, default_value=0.7,
|
|
||||||
_min=0.1,
|
|
||||||
_max=1.0,
|
|
||||||
_step=0.01,
|
|
||||||
precision=2)
|
|
||||||
|
|
||||||
max_tokens = forms.SliderField(
|
|
||||||
TooltipLabel(_('Output the maximum Tokens'),
|
|
||||||
_('Specify the maximum number of tokens that the model can generate')),
|
|
||||||
required=True, default_value=800,
|
|
||||||
_min=1,
|
|
||||||
_max=100000,
|
|
||||||
_step=1,
|
|
||||||
precision=0)
|
|
||||||
|
|
||||||
|
|
||||||
class VllmImageModelCredential(BaseForm, BaseModelCredential):
|
class VllmImageModelCredential(BaseForm, BaseModelCredential):
|
||||||
api_base = forms.TextInputField('API URL', required=True)
|
api_base = forms.TextInputField('API URL', required=True)
|
||||||
api_key = forms.PasswordInputField('API Key', required=True)
|
api_key = forms.PasswordInputField('API Key', required=True)
|
||||||
@ -69,4 +49,4 @@ class VllmImageModelCredential(BaseForm, BaseModelCredential):
|
|||||||
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
||||||
|
|
||||||
def get_model_params_setting_form(self, model_name):
|
def get_model_params_setting_form(self, model_name):
|
||||||
return VllmImageModelParams()
|
pass
|
||||||
|
|||||||
@ -10,26 +10,6 @@ from common.exception.app_exception import AppApiException
|
|||||||
from common.forms import BaseForm, TooltipLabel
|
from common.forms import BaseForm, TooltipLabel
|
||||||
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
||||||
|
|
||||||
|
|
||||||
class VolcanicEngineImageModelParams(BaseForm):
|
|
||||||
temperature = forms.SliderField(TooltipLabel(_('Temperature'),
|
|
||||||
_('Higher values make the output more random, while lower values make it more focused and deterministic')),
|
|
||||||
required=True, default_value=0.95,
|
|
||||||
_min=0.1,
|
|
||||||
_max=1.0,
|
|
||||||
_step=0.01,
|
|
||||||
precision=2)
|
|
||||||
|
|
||||||
max_tokens = forms.SliderField(
|
|
||||||
TooltipLabel(_('Output the maximum Tokens'),
|
|
||||||
_('Specify the maximum number of tokens that the model can generate')),
|
|
||||||
required=True, default_value=1024,
|
|
||||||
_min=1,
|
|
||||||
_max=100000,
|
|
||||||
_step=1,
|
|
||||||
precision=0)
|
|
||||||
|
|
||||||
|
|
||||||
class VolcanicEngineImageModelCredential(BaseForm, BaseModelCredential):
|
class VolcanicEngineImageModelCredential(BaseForm, BaseModelCredential):
|
||||||
api_key = forms.PasswordInputField('API Key', required=True)
|
api_key = forms.PasswordInputField('API Key', required=True)
|
||||||
api_base = forms.TextInputField('API URL', required=True)
|
api_base = forms.TextInputField('API URL', required=True)
|
||||||
@ -69,4 +49,4 @@ class VolcanicEngineImageModelCredential(BaseForm, BaseModelCredential):
|
|||||||
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
||||||
|
|
||||||
def get_model_params_setting_form(self, model_name):
|
def get_model_params_setting_form(self, model_name):
|
||||||
return VolcanicEngineImageModelParams()
|
pass
|
||||||
|
|||||||
@ -10,25 +10,6 @@ from common.forms import BaseForm, TooltipLabel
|
|||||||
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
||||||
|
|
||||||
|
|
||||||
class XinferenceImageModelParams(BaseForm):
|
|
||||||
temperature = forms.SliderField(TooltipLabel(_('Temperature'),
|
|
||||||
_('Higher values make the output more random, while lower values make it more focused and deterministic')),
|
|
||||||
required=True, default_value=0.7,
|
|
||||||
_min=0.1,
|
|
||||||
_max=1.0,
|
|
||||||
_step=0.01,
|
|
||||||
precision=2)
|
|
||||||
|
|
||||||
max_tokens = forms.SliderField(
|
|
||||||
TooltipLabel(_('Output the maximum Tokens'),
|
|
||||||
_('Specify the maximum number of tokens that the model can generate')),
|
|
||||||
required=True, default_value=800,
|
|
||||||
_min=1,
|
|
||||||
_max=100000,
|
|
||||||
_step=1,
|
|
||||||
precision=0)
|
|
||||||
|
|
||||||
|
|
||||||
class XinferenceImageModelCredential(BaseForm, BaseModelCredential):
|
class XinferenceImageModelCredential(BaseForm, BaseModelCredential):
|
||||||
api_base = forms.TextInputField('API URL', required=True)
|
api_base = forms.TextInputField('API URL', required=True)
|
||||||
api_key = forms.PasswordInputField('API Key', required=True)
|
api_key = forms.PasswordInputField('API Key', required=True)
|
||||||
@ -67,4 +48,4 @@ class XinferenceImageModelCredential(BaseForm, BaseModelCredential):
|
|||||||
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
||||||
|
|
||||||
def get_model_params_setting_form(self, model_name):
|
def get_model_params_setting_form(self, model_name):
|
||||||
return XinferenceImageModelParams()
|
pass
|
||||||
|
|||||||
@ -10,26 +10,6 @@ from common.exception.app_exception import AppApiException
|
|||||||
from common.forms import BaseForm, TooltipLabel
|
from common.forms import BaseForm, TooltipLabel
|
||||||
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
from models_provider.base_model_provider import BaseModelCredential, ValidCode
|
||||||
|
|
||||||
|
|
||||||
class ZhiPuImageModelParams(BaseForm):
|
|
||||||
temperature = forms.SliderField(TooltipLabel(_('Temperature'),
|
|
||||||
_('Higher values make the output more random, while lower values make it more focused and deterministic')),
|
|
||||||
required=True, default_value=0.95,
|
|
||||||
_min=0.1,
|
|
||||||
_max=1.0,
|
|
||||||
_step=0.01,
|
|
||||||
precision=2)
|
|
||||||
|
|
||||||
max_tokens = forms.SliderField(
|
|
||||||
TooltipLabel(_('Output the maximum Tokens'),
|
|
||||||
_('Specify the maximum number of tokens that the model can generate')),
|
|
||||||
required=True, default_value=1024,
|
|
||||||
_min=1,
|
|
||||||
_max=100000,
|
|
||||||
_step=1,
|
|
||||||
precision=0)
|
|
||||||
|
|
||||||
|
|
||||||
class ZhiPuImageModelCredential(BaseForm, BaseModelCredential):
|
class ZhiPuImageModelCredential(BaseForm, BaseModelCredential):
|
||||||
api_key = forms.PasswordInputField('API Key', required=True)
|
api_key = forms.PasswordInputField('API Key', required=True)
|
||||||
|
|
||||||
@ -68,4 +48,4 @@ class ZhiPuImageModelCredential(BaseForm, BaseModelCredential):
|
|||||||
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
|
||||||
|
|
||||||
def get_model_params_setting_form(self, model_name):
|
def get_model_params_setting_form(self, model_name):
|
||||||
return ZhiPuImageModelParams()
|
pass
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user