github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-06-29 08:17:17 +00:00
Code Issues Projects Releases Wiki Activity

perf: Gpt3 to gpt-4o-mini

Browse Source
This commit is contained in:
feng 2024-07-19 11:50:05 +08:00 committed by feng626
parent 45291aba0c
commit 3d4fc56592
2 changed files with 8 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
apps/assets/const/protocol.py
View File

@ -295,23 +295,17 @@ class Protocol(ChoicesMixin, models.TextChoices):
'setting': { 'setting': {
'api_mode': { 'api_mode': {
'type': 'choice', 'type': 'choice',
'default': 'gpt-3.5-turbo', 'default': 'gpt-4o-mini',
'label': _('API mode'), 'label': _('API mode'),
'choices': [ 'choices': [
('gpt-3.5-turbo', 'GPT-3.5 Turbo'), ('gpt-4o-mini', 'GPT-4o-mini'),
('gpt-3.5-turbo-1106', 'GPT-3.5 Turbo 1106'), ('gpt-4o', 'GPT-4o'),
('gpt-4-turbo', 'GPT-4 Turbo'),
] ]
} }
} }
} }
} }
if settings.XPACK_LICENSE_IS_VALID:
choices = protocols[cls.chatgpt]['setting']['api_mode']['choices']
choices.extend([
('gpt-4', 'GPT-4'),
('gpt-4-turbo', 'GPT-4 Turbo'),
('gpt-4o', 'GPT-4o'),
])
return protocols return protocols
@classmethod @classmethod

23
apps/settings/serializers/feature.py
View File

@ -69,7 +69,9 @@ class VaultSettingSerializer(serializers.Serializer):
class ChatAISettingSerializer(serializers.Serializer): class ChatAISettingSerializer(serializers.Serializer):
PREFIX_TITLE = _('Chat AI') PREFIX_TITLE = _('Chat AI')
GPT_MODEL_CHOICES = [] API_MODEL = Protocol.gpt_protocols()[Protocol.chatgpt]['setting']['api_mode']
GPT_MODEL_CHOICES = API_MODEL['choices']
GPT_MODEL_DEFAULT = API_MODEL['default']
CHAT_AI_ENABLED = serializers.BooleanField( CHAT_AI_ENABLED = serializers.BooleanField(
required=False, label=_('Enable Chat AI') required=False, label=_('Enable Chat AI')
@ -84,26 +86,9 @@ class ChatAISettingSerializer(serializers.Serializer):
allow_blank=True, required=False, label=_('Proxy') allow_blank=True, required=False, label=_('Proxy')
) )
GPT_MODEL = serializers.ChoiceField( GPT_MODEL = serializers.ChoiceField(
default='', choices=GPT_MODEL_CHOICES, label=_("GPT Model"), required=False, default=GPT_MODEL_DEFAULT, choices=GPT_MODEL_CHOICES, label=_("GPT Model"), required=False,
) )
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.set_GPT_MODEL_choices()
def set_GPT_MODEL_choices(self):
field_gpt_model = self.fields.get("GPT_MODEL")
if not field_gpt_model:
return
gpt_api_model = Protocol.gpt_protocols()[Protocol.chatgpt]['setting']['api_mode']
choices = gpt_api_model['choices']
field_gpt_model.choices = choices
field_gpt_model.default = gpt_api_model['default']
cls = self.__class__
if cls.GPT_MODEL_CHOICES:
return
cls.GPT_MODEL_CHOICES.extend(choices)
class TicketSettingSerializer(serializers.Serializer): class TicketSettingSerializer(serializers.Serializer):
PREFIX_TITLE = _('Ticket') PREFIX_TITLE = _('Ticket')