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

fix: Internationalization of Models (#2068)

Browse Source
This commit is contained in:
shaohuzhang1 2025-01-22 16:12:14 +08:00 committed by GitHub
parent f924827c7c
commit 13571ef615
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
81 changed files with 2242 additions and 2199 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py
View File

@ -14,7 +14,7 @@ from typing import List
from django.db.models import QuerySet from django.db.models import QuerySet
from django.http import StreamingHttpResponse from django.http import StreamingHttpResponse
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from langchain.chat_models.base import BaseChatModel from langchain.chat_models.base import BaseChatModel
from langchain.schema import BaseMessage from langchain.schema import BaseMessage
from langchain.schema.messages import HumanMessage, AIMessage from langchain.schema.messages import HumanMessage, AIMessage
@ -174,7 +174,7 @@ class BaseChatStep(IChatStep):
[AIMessageChunk(content=no_references_setting.get('value').replace('{question}', problem_text))]), False [AIMessageChunk(content=no_references_setting.get('value').replace('{question}', problem_text))]), False
if chat_model is None: if chat_model is None:
return iter([AIMessageChunk( return iter([AIMessageChunk(
__('Sorry, the AI model is not configured. Please go to the application to set up the AI model first.'))]), False _('Sorry, the AI model is not configured. Please go to the application to set up the AI model first.'))]), False
else: else:
return chat_model.stream(message_list), True return chat_model.stream(message_list), True
@ -219,7 +219,7 @@ class BaseChatStep(IChatStep):
return AIMessage(no_references_setting.get('value').replace('{question}', problem_text)), False return AIMessage(no_references_setting.get('value').replace('{question}', problem_text)), False
if chat_model is None: if chat_model is None:
return AIMessage( return AIMessage(
__('Sorry, the AI model is not configured. Please go to the application to set up the AI model first.')), False _('Sorry, the AI model is not configured. Please go to the application to set up the AI model first.')), False
else: else:
return chat_model.invoke(message_list), True return chat_model.invoke(message_list), True

4
apps/application/chat_pipeline/step/reset_problem_step/impl/base_reset_problem_step.py
View File

@ -8,7 +8,7 @@
""" """
from typing import List from typing import List
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from langchain.schema import HumanMessage from langchain.schema import HumanMessage
from application.chat_pipeline.step.reset_problem_step.i_reset_problem_step import IResetProblemStep from application.chat_pipeline.step.reset_problem_step.i_reset_problem_step import IResetProblemStep
@ -16,7 +16,7 @@ from application.models import ChatRecord
from common.util.split_model import flat_map from common.util.split_model import flat_map
from setting.models_provider.tools import get_model_instance_by_model_user_id from setting.models_provider.tools import get_model_instance_by_model_user_id
prompt = __( prompt = _(
"() contains the user's question. Answer the guessed user's question based on the context ({question}) Requirement: Output a complete question and put it in the <data></data> tag") "() contains the user's question. Answer the guessed user's question based on the context ({question}) Requirement: Output a complete question and put it in the <data></data> tag")

1165
apps/locales/en_US/LC_MESSAGES/django.po
View File

File diff suppressed because it is too large Load Diff

1283
apps/locales/zh_CN/LC_MESSAGES/django.po
View File

File diff suppressed because it is too large Load Diff

1264
apps/locales/zh_Hant/LC_MESSAGES/django.po
View File

File diff suppressed because it is too large Load Diff

10
apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/embedding.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -24,22 +24,22 @@ class AliyunBaiLianEmbeddingCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['dashscope_api_key']: for key in ['dashscope_api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model: AliyunBaiLianEmbedding = provider.get_model(model_type, model_name, model_credential) model: AliyunBaiLianEmbedding = provider.get_model(model_type, model_name, model_credential)
model.embed_query(__('Hello')) model.embed_query(_('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

9
apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/image.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -43,11 +43,11 @@ class QwenVLModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -60,7 +60,8 @@ class QwenVLModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/llm.py
View File

@ -7,7 +7,7 @@ from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
from common.forms import BaseForm, TooltipLabel from common.forms import BaseForm, TooltipLabel
from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
class BaiLianLLMModelParams(BaseForm): class BaiLianLLMModelParams(BaseForm):
@ -36,23 +36,24 @@ class BaiLianLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
model.invoke([HumanMessage(content=__('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

10
apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/reranker.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from langchain_core.documents import Document from langchain_core.documents import Document
from common import forms from common import forms
@ -24,22 +24,22 @@ class AliyunBaiLianRerankerCredential(BaseForm, BaseModelCredential):
raise_exception=False): raise_exception=False):
if not model_type == 'RERANKER': if not model_type == 'RERANKER':
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['dashscope_api_key']: for key in ['dashscope_api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model: AliyunBaiLianReranker = provider.get_model(model_type, model_name, model_credential) model: AliyunBaiLianReranker = provider.get_model(model_type, model_name, model_credential)
model.compress_documents([Document(page_content=__('Hello'))], __('Hello')) model.compress_documents([Document(page_content=_('Hello'))], _('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

8
apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/stt.py
View File

@ -2,7 +2,7 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -18,12 +18,12 @@ class AliyunBaiLianSTTModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -34,7 +34,7 @@ class AliyunBaiLianSTTModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

9
apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/tti.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -64,11 +64,11 @@ class QwenTextToImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -80,7 +80,8 @@ class QwenTextToImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

10
apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/tts.py
View File

@ -2,11 +2,12 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
from common.forms import BaseForm, TooltipLabel from common.forms import BaseForm, TooltipLabel
from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode
from django.utils.translation import gettext_lazy as _, gettext as __
class AliyunBaiLianTTSModelGeneralParams(BaseForm): class AliyunBaiLianTTSModelGeneralParams(BaseForm):
@ -51,12 +52,12 @@ class AliyunBaiLianTTSModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -67,7 +68,8 @@ class AliyunBaiLianTTSModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

4
apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/model/tti.py
View File

@ -3,7 +3,7 @@ from http import HTTPStatus
from typing import Dict from typing import Dict
from dashscope import ImageSynthesis from dashscope import ImageSynthesis
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from langchain_community.chat_models import ChatTongyi from langchain_community.chat_models import ChatTongyi
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
@ -40,7 +40,7 @@ class QwenTextToImageModel(MaxKBBaseModel, BaseTextToImage):
def check_auth(self): def check_auth(self):
chat = ChatTongyi(api_key=self.api_key, model_name='qwen-max') chat = ChatTongyi(api_key=self.api_key, model_name='qwen-max')
chat.invoke([HumanMessage([{"type": "text", "text": __('Hello')}])]) chat.invoke([HumanMessage([{"type": "text", "text": _('Hello')}])])
def generate_image(self, prompt: str, negative_prompt: str = None): def generate_image(self, prompt: str, negative_prompt: str = None):
# api_base='https://dashscope.aliyuncs.com/compatible-mode/v1', # api_base='https://dashscope.aliyuncs.com/compatible-mode/v1',

4
apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/model/tts.py
View File

@ -2,7 +2,7 @@ from typing import Dict
import dashscope import dashscope
from dashscope.audio.tts_v2 import * from dashscope.audio.tts_v2 import *
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common.util.common import _remove_empty_lines from common.util.common import _remove_empty_lines
from setting.models_provider.base_model_provider import MaxKBBaseModel from setting.models_provider.base_model_provider import MaxKBBaseModel
@ -34,7 +34,7 @@ class AliyunBaiLianTextToSpeech(MaxKBBaseModel, BaseTextToSpeech):
) )
def check_auth(self): def check_auth(self):
self.text_to_speech(__('Hello')) self.text_to_speech(_('Hello'))
def text_to_speech(self, text): def text_to_speech(self, text):
dashscope.api_key = self.api_key dashscope.api_key = self.api_key

11
apps/setting/models_provider/impl/anthropic_model_provider/credential/image.py
View File

@ -1,15 +1,13 @@
# coding=utf-8 # coding=utf-8
import base64
import os
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
from common.forms import BaseForm, TooltipLabel from common.forms import BaseForm, TooltipLabel
from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode
from django.utils.translation import gettext_lazy as _, gettext as __
class AnthropicImageModelParams(BaseForm): class AnthropicImageModelParams(BaseForm):
@ -40,12 +38,12 @@ class AnthropicImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -58,7 +56,8 @@ class AnthropicImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/anthropic_model_provider/credential/llm.py
View File

@ -14,7 +14,7 @@ from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
from common.forms import BaseForm, TooltipLabel from common.forms import BaseForm, TooltipLabel
from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
class AnthropicLLMModelParams(BaseForm): class AnthropicLLMModelParams(BaseForm):
@ -43,23 +43,24 @@ class AnthropicLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.invoke([HumanMessage(content=__('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

10
apps/setting/models_provider/impl/aws_bedrock_model_provider/credential/embedding.py
View File

@ -1,6 +1,6 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -17,27 +17,27 @@ class BedrockEmbeddingCredential(BaseForm, BaseModelCredential):
if not any(mt.get('value') == model_type for mt in model_type_list): if not any(mt.get('value') == model_type for mt in model_type_list):
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
return False return False
required_keys = ['region_name', 'access_key_id', 'secret_access_key'] required_keys = ['region_name', 'access_key_id', 'secret_access_key']
if not all(key in model_credential for key in required_keys): if not all(key in model_credential for key in required_keys):
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('The following fields are required: {keys}').format( _('The following fields are required: {keys}').format(
keys=", ".join(required_keys))) keys=", ".join(required_keys)))
return False return False
try: try:
model: BedrockEmbeddingModel = provider.get_model(model_type, model_name, model_credential) model: BedrockEmbeddingModel = provider.get_model(model_type, model_name, model_credential)
aa = model.embed_query(__('Hello')) aa = model.embed_query(_('Hello'))
print(aa) print(aa)
except AppApiException: except AppApiException:
raise raise
except Exception as e: except Exception as e:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
return False return False

9
apps/setting/models_provider/impl/aws_bedrock_model_provider/credential/llm.py
View File

@ -1,6 +1,6 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -36,14 +36,14 @@ class BedrockLLMModelCredential(BaseForm, BaseModelCredential):
if not any(mt.get('value') == model_type for mt in model_type_list): if not any(mt.get('value') == model_type for mt in model_type_list):
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
return False return False
required_keys = ['region_name', 'access_key_id', 'secret_access_key'] required_keys = ['region_name', 'access_key_id', 'secret_access_key']
if not all(key in model_credential for key in required_keys): if not all(key in model_credential for key in required_keys):
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('The following fields are required: {keys}').format( gettext('The following fields are required: {keys}').format(
keys=", ".join(required_keys))) keys=", ".join(required_keys)))
return False return False
@ -55,7 +55,8 @@ class BedrockLLMModelCredential(BaseForm, BaseModelCredential):
except Exception as e: except Exception as e:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
return False return False

10
apps/setting/models_provider/impl/azure_model_provider/credential/embedding.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -23,23 +23,23 @@ class AzureOpenAIEmbeddingCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key', 'api_version']: for key in ['api_base', 'api_key', 'api_version']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.embed_query(__('Hello')) model.embed_query(_('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct')) _('Verification failed, please check whether the parameters are correct'))
else: else:
return False return False

9
apps/setting/models_provider/impl/azure_model_provider/credential/image.py
View File

@ -9,7 +9,7 @@ from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
from common.forms import BaseForm, TooltipLabel from common.forms import BaseForm, TooltipLabel
from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
class AzureOpenAIImageModelParams(BaseForm): class AzureOpenAIImageModelParams(BaseForm):
@ -41,12 +41,12 @@ class AzureOpenAIImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key', 'api_version']: for key in ['api_base', 'api_key', 'api_version']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -59,7 +59,8 @@ class AzureOpenAIImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

8
apps/setting/models_provider/impl/azure_model_provider/credential/llm.py
View File

@ -14,7 +14,7 @@ from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
from common.forms import BaseForm, TooltipLabel from common.forms import BaseForm, TooltipLabel
from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
class AzureLLMModelParams(BaseForm): class AzureLLMModelParams(BaseForm):
@ -43,12 +43,12 @@ class AzureLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key', 'deployment_name', 'api_version']: for key in ['api_base', 'api_key', 'deployment_name', 'api_version']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -59,7 +59,7 @@ class AzureLLMModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct')) gettext('Verification failed, please check whether the parameters are correct'))
else: else:
return False return False

8
apps/setting/models_provider/impl/azure_model_provider/credential/stt.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -19,12 +19,12 @@ class AzureOpenAISTTModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key', 'api_version']: for key in ['api_base', 'api_key', 'api_version']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -35,7 +35,7 @@ class AzureOpenAISTTModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

9
apps/setting/models_provider/impl/azure_model_provider/credential/tti.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -54,12 +54,12 @@ class AzureOpenAITextToImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key', 'api_version']: for key in ['api_base', 'api_key', 'api_version']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -71,7 +71,8 @@ class AzureOpenAITextToImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

14
apps/setting/models_provider/impl/azure_model_provider/credential/tts.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -12,7 +12,8 @@ from setting.models_provider.base_model_provider import BaseModelCredential, Val
class AzureOpenAITTSModelGeneralParams(BaseForm): class AzureOpenAITTSModelGeneralParams(BaseForm):
# alloy, echo, fable, onyx, nova, shimmer # alloy, echo, fable, onyx, nova, shimmer
voice = forms.SingleSelect( voice = forms.SingleSelect(
TooltipLabel('Voice', _('Try out the different sounds (Alloy, Echo, Fable, Onyx, Nova, and Sparkle) to find one that suits your desired tone and audience. The current voiceover is optimized for English.')), TooltipLabel('Voice',
_('Try out the different sounds (Alloy, Echo, Fable, Onyx, Nova, and Sparkle) to find one that suits your desired tone and audience. The current voiceover is optimized for English.')),
required=True, default_value='alloy', required=True, default_value='alloy',
text_field='value', text_field='value',
value_field='value', value_field='value',
@ -35,12 +36,13 @@ class AzureOpenAITTSModelCredential(BaseForm, BaseModelCredential):
raise_exception=False): raise_exception=False):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, __('{model_type} Model type is not supported').format(model_type=model_type)) raise AppApiException(ValidCode.valid_error.value,
gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key', 'api_version']: for key in ['api_base', 'api_key', 'api_version']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -50,7 +52,9 @@ class AzureOpenAITTSModelCredential(BaseForm, BaseModelCredential):
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('Verification failed, please check whether the parameters are correct: {error}').format(error=str(e))) raise AppApiException(ValidCode.valid_error.value, gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e)))
else: else:
return False return False
return True return True

11
apps/setting/models_provider/impl/deepseek_model_provider/credential/llm.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -43,23 +43,24 @@ class DeepSeekLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
model.invoke([HumanMessage(content=__('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

10
apps/setting/models_provider/impl/gemini_model_provider/credential/embedding.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -22,23 +22,23 @@ class GeminiEmbeddingCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.embed_query(__('Hello')) model.embed_query(_('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/gemini_model_provider/credential/image.py
View File

@ -2,7 +2,7 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -38,17 +38,17 @@ class GeminiImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
res = model.stream([HumanMessage(content=[{"type": "text", "text": __('Hello')}])]) res = model.stream([HumanMessage(content=[{"type": "text", "text": gettext('Hello')}])])
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
@ -56,7 +56,8 @@ class GeminiImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/gemini_model_provider/credential/llm.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -43,24 +43,25 @@ class GeminiLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
res = model.invoke([HumanMessage(content=__('Hello'))]) res = model.invoke([HumanMessage(content=gettext('Hello'))])
print(res) print(res)
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

8
apps/setting/models_provider/impl/gemini_model_provider/credential/stt.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -17,12 +17,12 @@ class GeminiSTTModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -33,7 +33,7 @@ class GeminiSTTModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

6
apps/setting/models_provider/impl/gemini_model_provider/model/stt.py
View File

@ -1,6 +1,6 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from langchain_google_genai import ChatGoogleGenerativeAI from langchain_google_genai import ChatGoogleGenerativeAI
@ -40,7 +40,7 @@ class GeminiSpeechToText(MaxKBBaseModel, BaseSpeechToText):
model=self.model, model=self.model,
google_api_key=self.api_key google_api_key=self.api_key
) )
response_list = client.invoke(__('Hello')) response_list = client.invoke(_('Hello'))
# print(response_list) # print(response_list)
def speech_to_text(self, audio_file): def speech_to_text(self, audio_file):
@ -50,7 +50,7 @@ class GeminiSpeechToText(MaxKBBaseModel, BaseSpeechToText):
) )
audio_data = audio_file.read() audio_data = audio_file.read()
msg = HumanMessage(content=[ msg = HumanMessage(content=[
{'type': 'text', 'text': __('convert audio to text')}, {'type': 'text', 'text': _('convert audio to text')},
{"type": "media", 'mime_type': 'audio/mp3', "data": audio_data} {"type": "media", 'mime_type': 'audio/mp3', "data": audio_data}
]) ])
res = client.invoke([msg]) res = client.invoke([msg])

19
apps/setting/models_provider/impl/kimi_model_provider/credential/llm.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -18,7 +18,8 @@ from setting.models_provider.base_model_provider import BaseModelCredential, Val
class KimiLLMModelParams(BaseForm): class KimiLLMModelParams(BaseForm):
temperature = forms.SliderField(TooltipLabel(_('Temperature'), _('Higher values make the output more random, while lower values make it more focused and deterministic')), 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.3, required=True, default_value=0.3,
_min=0.1, _min=0.1,
_max=1.0, _max=1.0,
@ -26,7 +27,8 @@ class KimiLLMModelParams(BaseForm):
precision=2) precision=2)
max_tokens = forms.SliderField( max_tokens = forms.SliderField(
TooltipLabel(_('Output the maximum Tokens'), _('Specify the maximum number of tokens that the model can generate')), TooltipLabel(_('Output the maximum Tokens'),
_('Specify the maximum number of tokens that the model can generate')),
required=True, default_value=1024, required=True, default_value=1024,
_min=1, _min=1,
_max=100000, _max=100000,
@ -40,22 +42,25 @@ class KimiLLMModelCredential(BaseForm, BaseModelCredential):
raise_exception=False): raise_exception=False):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, __('{model_type} Model type is not supported').format(model_type=model_type)) raise AppApiException(ValidCode.valid_error.value,
gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
model.invoke([HumanMessage(content=__('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('Verification failed, please check whether the parameters are correct: {error}').format(error=str(e))) raise AppApiException(ValidCode.valid_error.value, gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e)))
else: else:
return False return False
return True return True

11
apps/setting/models_provider/impl/local_model_provider/credential/embedding.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -23,22 +23,23 @@ class LocalEmbeddingCredential(BaseForm, BaseModelCredential):
raise_exception=False): raise_exception=False):
if not model_type == 'EMBEDDING': if not model_type == 'EMBEDDING':
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['cache_folder']: for key in ['cache_folder']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model: LocalEmbedding = provider.get_model(model_type, model_name, model_credential) model: LocalEmbedding = provider.get_model(model_type, model_name, model_credential)
model.embed_query(__('Hello')) model.embed_query(gettext('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/local_model_provider/credential/reranker.py
View File

@ -15,7 +15,7 @@ from common.exception.app_exception import AppApiException
from common.forms import BaseForm from common.forms import BaseForm
from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode
from setting.models_provider.impl.local_model_provider.model.reranker import LocalBaseReranker from setting.models_provider.impl.local_model_provider.model.reranker import LocalBaseReranker
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
class LocalRerankerCredential(BaseForm, BaseModelCredential): class LocalRerankerCredential(BaseForm, BaseModelCredential):
@ -24,22 +24,23 @@ class LocalRerankerCredential(BaseForm, BaseModelCredential):
raise_exception=False): raise_exception=False):
if not model_type == 'RERANKER': if not model_type == 'RERANKER':
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['cache_dir']: for key in ['cache_dir']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model: LocalBaseReranker = provider.get_model(model_type, model_name, model_credential) model: LocalBaseReranker = provider.get_model(model_type, model_name, model_credential)
model.compress_documents([Document(page_content=__('Hello'))], __('Hello')) model.compress_documents([Document(page_content=gettext('Hello'))], gettext('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

12
apps/setting/models_provider/impl/ollama_model_provider/credential/embedding.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -23,18 +23,18 @@ class OllamaEmbeddingModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
try: try:
model_list = provider.get_base_model_list(model_credential.get('api_base')) model_list = provider.get_base_model_list(model_credential.get('api_base'))
except Exception as e: except Exception as e:
raise AppApiException(ValidCode.valid_error.value, __('API domain name is invalid')) raise AppApiException(ValidCode.valid_error.value, _('API domain name is invalid'))
exist = [model for model in (model_list.get('models') if model_list.get('models') is not None else []) if exist = [model for model in (model_list.get('models') if model_list.get('models') is not None else []) if
model.get('model') == model_name or model.get('model').replace(":latest", "") == model_name] model.get('model') == model_name or model.get('model').replace(":latest", "") == model_name]
if len(exist) == 0: if len(exist) == 0:
raise AppApiException(ValidCode.model_not_fount, raise AppApiException(ValidCode.model_not_fount,
__('The model does not exist, please download the model first')) _('The model does not exist, please download the model first'))
model: LocalEmbedding = provider.get_model(model_type, model_name, model_credential) model: LocalEmbedding = provider.get_model(model_type, model_name, model_credential)
model.embed_query(__('Hello')) model.embed_query(_('Hello'))
return True return True
def encryption_dict(self, model_info: Dict[str, object]): def encryption_dict(self, model_info: Dict[str, object]):
@ -43,7 +43,7 @@ class OllamaEmbeddingModelCredential(BaseForm, BaseModelCredential):
def build_model(self, model_info: Dict[str, object]): def build_model(self, model_info: Dict[str, object]):
for key in ['model']: for key in ['model']:
if key not in model_info: if key not in model_info:
raise AppApiException(500, __('{key} is required').format(key=key)) raise AppApiException(500, _('{key} is required').format(key=key))
return self return self
api_base = forms.TextInputField('API URL', required=True) api_base = forms.TextInputField('API URL', required=True)

8
apps/setting/models_provider/impl/ollama_model_provider/credential/image.py
View File

@ -5,7 +5,7 @@ from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
from common.forms import BaseForm, TooltipLabel from common.forms import BaseForm, TooltipLabel
from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
class OllamaImageModelParams(BaseForm): class OllamaImageModelParams(BaseForm):
@ -36,16 +36,16 @@ class OllamaImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
try: try:
model_list = provider.get_base_model_list(model_credential.get('api_base')) model_list = provider.get_base_model_list(model_credential.get('api_base'))
except Exception as e: except Exception as e:
raise AppApiException(ValidCode.valid_error.value, __('API domain name is invalid')) raise AppApiException(ValidCode.valid_error.value, gettext('API domain name is invalid'))
exist = [model for model in (model_list.get('models') if model_list.get('models') is not None else []) if exist = [model for model in (model_list.get('models') if model_list.get('models') is not None else []) if
model.get('model') == model_name or model.get('model').replace(":latest", "") == model_name] model.get('model') == model_name or model.get('model').replace(":latest", "") == model_name]
if len(exist) == 0: if len(exist) == 0:
raise AppApiException(ValidCode.model_not_fount, raise AppApiException(ValidCode.model_not_fount,
__('The model does not exist, please download the model first')) gettext('The model does not exist, please download the model first'))
return True return True

10
apps/setting/models_provider/impl/ollama_model_provider/credential/llm.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -41,16 +41,16 @@ class OllamaLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
try: try:
model_list = provider.get_base_model_list(model_credential.get('api_base')) model_list = provider.get_base_model_list(model_credential.get('api_base'))
except Exception as e: except Exception as e:
raise AppApiException(ValidCode.valid_error.value, __('API domain name is invalid')) raise AppApiException(ValidCode.valid_error.value, gettext('API domain name is invalid'))
exist = [model for model in (model_list.get('models') if model_list.get('models') is not None else []) if exist = [model for model in (model_list.get('models') if model_list.get('models') is not None else []) if
model.get('model') == model_name or model.get('model').replace(":latest", "") == model_name] model.get('model') == model_name or model.get('model').replace(":latest", "") == model_name]
if len(exist) == 0: if len(exist) == 0:
raise AppApiException(ValidCode.model_not_fount, raise AppApiException(ValidCode.model_not_fount,
__('The model does not exist, please download the model first')) gettext('The model does not exist, please download the model first'))
return True return True
def encryption_dict(self, model_info: Dict[str, object]): def encryption_dict(self, model_info: Dict[str, object]):
@ -59,7 +59,7 @@ class OllamaLLMModelCredential(BaseForm, BaseModelCredential):
def build_model(self, model_info: Dict[str, object]): def build_model(self, model_info: Dict[str, object]):
for key in ['api_key', 'model']: for key in ['api_key', 'model']:
if key not in model_info: if key not in model_info:
raise AppApiException(500, __('{key} is required').format(key=key)) raise AppApiException(500, gettext('{key} is required').format(key=key))
self.api_key = model_info.get('api_key') self.api_key = model_info.get('api_key')
return self return self

10
apps/setting/models_provider/impl/openai_model_provider/credential/embedding.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -22,23 +22,23 @@ class OpenAIEmbeddingCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.embed_query(__('Hello')) model.embed_query(_('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

9
apps/setting/models_provider/impl/openai_model_provider/credential/image.py
View File

@ -9,7 +9,7 @@ from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
from common.forms import BaseForm, TooltipLabel from common.forms import BaseForm, TooltipLabel
from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
class OpenAIImageModelParams(BaseForm): class OpenAIImageModelParams(BaseForm):
@ -40,12 +40,12 @@ class OpenAIImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -58,7 +58,8 @@ class OpenAIImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/openai_model_provider/credential/llm.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -43,24 +43,25 @@ class OpenAILLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
model.invoke([HumanMessage(content=__('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

8
apps/setting/models_provider/impl/openai_model_provider/credential/stt.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -18,12 +18,12 @@ class OpenAISTTModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -34,7 +34,7 @@ class OpenAISTTModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

9
apps/setting/models_provider/impl/openai_model_provider/credential/tti.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -57,12 +57,12 @@ class OpenAITextToImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -74,7 +74,8 @@ class OpenAITextToImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

9
apps/setting/models_provider/impl/openai_model_provider/credential/tts.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -36,12 +36,12 @@ class OpenAITTSModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -52,7 +52,8 @@ class OpenAITTSModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/qwen_model_provider/credential/image.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -43,16 +43,16 @@ class QwenVLModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
res = model.stream([HumanMessage(content=[{"type": "text", "text": __('Hello')}])]) res = model.stream([HumanMessage(content=[{"type": "text", "text": gettext('Hello')}])])
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
@ -60,7 +60,8 @@ class QwenVLModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/qwen_model_provider/credential/llm.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -43,22 +43,23 @@ class OpenAILLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
model.invoke([HumanMessage(content=__('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

9
apps/setting/models_provider/impl/qwen_model_provider/credential/tti.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -64,11 +64,11 @@ class QwenTextToImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -80,7 +80,8 @@ class QwenTextToImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

4
apps/setting/models_provider/impl/qwen_model_provider/model/tti.py
View File

@ -3,7 +3,7 @@ from http import HTTPStatus
from typing import Dict from typing import Dict
from dashscope import ImageSynthesis from dashscope import ImageSynthesis
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from langchain_community.chat_models import ChatTongyi from langchain_community.chat_models import ChatTongyi
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
@ -40,7 +40,7 @@ class QwenTextToImageModel(MaxKBBaseModel, BaseTextToImage):
def check_auth(self): def check_auth(self):
chat = ChatTongyi(api_key=self.api_key, model_name='qwen-max') chat = ChatTongyi(api_key=self.api_key, model_name='qwen-max')
chat.invoke([HumanMessage([{"type": "text", "text": __('Hello')}])]) chat.invoke([HumanMessage([{"type": "text", "text": _('Hello')}])])
def generate_image(self, prompt: str, negative_prompt: str = None): def generate_image(self, prompt: str, negative_prompt: str = None):
# api_base='https://dashscope.aliyuncs.com/compatible-mode/v1', # api_base='https://dashscope.aliyuncs.com/compatible-mode/v1',

8
apps/setting/models_provider/impl/tencent_model_provider/credential/embedding.py
View File

@ -1,6 +1,6 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -15,17 +15,17 @@ class TencentEmbeddingCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
self.valid_form(model_credential) self.valid_form(model_credential)
try: try:
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.embed_query(__('Hello')) model.embed_query(_('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/tencent_model_provider/credential/image.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -43,16 +43,16 @@ class TencentVisionModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
res = model.stream([HumanMessage(content=[{"type": "text", "text": __('Hello')}])]) res = model.stream([HumanMessage(content=[{"type": "text", "text": gettext('Hello')}])])
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
@ -60,7 +60,8 @@ class TencentVisionModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

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

@ -1,5 +1,5 @@
# coding=utf-8 # coding=utf-8
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -26,7 +26,7 @@ class TencentLLMModelCredential(BaseForm, BaseModelCredential):
if not any(mt['value'] == model_type for mt in provider.get_model_type_list()): if not any(mt['value'] == model_type for mt in provider.get_model_type_list()):
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
return False return False
return True return True
@ -36,7 +36,7 @@ class TencentLLMModelCredential(BaseForm, BaseModelCredential):
if missing_keys: if missing_keys:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{keys} is required').format(keys=", ".join(missing_keys))) gettext('{keys} is required').format(keys=", ".join(missing_keys)))
return False return False
return True return True
@ -46,11 +46,12 @@ class TencentLLMModelCredential(BaseForm, BaseModelCredential):
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
model.invoke([HumanMessage(content=__('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
return False return False
return True return True

9
apps/setting/models_provider/impl/tencent_model_provider/credential/tti.py
View File

@ -1,6 +1,6 @@
# coding=utf-8 # coding=utf-8
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -74,7 +74,7 @@ class TencentTTIModelCredential(BaseForm, BaseModelCredential):
if not any(mt['value'] == model_type for mt in provider.get_model_type_list()): if not any(mt['value'] == model_type for mt in provider.get_model_type_list()):
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
return False return False
return True return True
@ -84,7 +84,7 @@ class TencentTTIModelCredential(BaseForm, BaseModelCredential):
if missing_keys: if missing_keys:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{keys} is required').format(keys=", ".join(missing_keys))) gettext('{keys} is required').format(keys=", ".join(missing_keys)))
return False return False
return True return True
@ -98,7 +98,8 @@ class TencentTTIModelCredential(BaseForm, BaseModelCredential):
except Exception as e: except Exception as e:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
return False return False
return True return True

4
apps/setting/models_provider/impl/tencent_model_provider/model/tti.py
View File

@ -3,7 +3,7 @@
import json import json
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from tencentcloud.common import credential from tencentcloud.common import credential
from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
from tencentcloud.common.profile.client_profile import ClientProfile from tencentcloud.common.profile.client_profile import ClientProfile
@ -51,7 +51,7 @@ class TencentTextToImageModel(MaxKBBaseModel, BaseTextToImage):
hunyuan_secret_id=self.hunyuan_secret_id, hunyuan_secret_id=self.hunyuan_secret_id,
hunyuan_secret_key=self.hunyuan_secret_key, hunyuan_secret_key=self.hunyuan_secret_key,
model="hunyuan-standard") model="hunyuan-standard")
res = chat.invoke(__('Hello')) res = chat.invoke(_('Hello'))
# print(res) # print(res)
def generate_image(self, prompt: str, negative_prompt: str = None): def generate_image(self, prompt: str, negative_prompt: str = None):

10
apps/setting/models_provider/impl/vllm_model_provider/credential/embedding.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -22,23 +22,23 @@ class VllmEmbeddingCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.embed_query(__('Hello')) model.embed_query(_('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

9
apps/setting/models_provider/impl/vllm_model_provider/credential/image.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -38,12 +38,12 @@ class VllmImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -56,7 +56,8 @@ class VllmImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

12
apps/setting/models_provider/impl/vllm_model_provider/credential/llm.py
View File

@ -2,7 +2,7 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -36,18 +36,18 @@ class VLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
try: try:
model_list = provider.get_base_model_list(model_credential.get('api_base'), model_credential.get('api_key')) model_list = provider.get_base_model_list(model_credential.get('api_base'), model_credential.get('api_key'))
except Exception as e: except Exception as e:
raise AppApiException(ValidCode.valid_error.value, __('API domain name is invalid')) raise AppApiException(ValidCode.valid_error.value, gettext('API domain name is invalid'))
exist = provider.get_model_info_by_name(model_list, model_name) exist = provider.get_model_info_by_name(model_list, model_name)
if len(exist) == 0: if len(exist) == 0:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('The model does not exist, please download the model first')) gettext('The model does not exist, please download the model first'))
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
try: try:
res = model.invoke([HumanMessage(content=__('Hello'))]) res = model.invoke([HumanMessage(content=gettext('Hello'))])
print(res) print(res)
except Exception as e: except Exception as e:
print(e) print(e)
@ -59,7 +59,7 @@ class VLLMModelCredential(BaseForm, BaseModelCredential):
def build_model(self, model_info: Dict[str, object]): def build_model(self, model_info: Dict[str, object]):
for key in ['api_key', 'model']: for key in ['api_key', 'model']:
if key not in model_info: if key not in model_info:
raise AppApiException(500, __('{key} is required').format(key=key)) raise AppApiException(500, gettext('{key} is required').format(key=key))
self.api_key = model_info.get('api_key') self.api_key = model_info.get('api_key')
return self return self

10
apps/setting/models_provider/impl/volcanic_engine_model_provider/credential/embedding.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -22,23 +22,23 @@ class VolcanicEmbeddingCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.embed_query(__('Hello')) model.embed_query(_('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/volcanic_engine_model_provider/credential/image.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -38,17 +38,17 @@ class VolcanicEngineImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key', 'api_base']: for key in ['api_key', 'api_base']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
res = model.stream([HumanMessage(content=[{"type": "text", "text": __('Hello')}])]) res = model.stream([HumanMessage(content=[{"type": "text", "text": gettext('Hello')}])])
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
@ -56,7 +56,8 @@ class VolcanicEngineImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/volcanic_engine_model_provider/credential/llm.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -43,24 +43,25 @@ class VolcanicEngineLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['access_key_id', 'secret_access_key']: for key in ['access_key_id', 'secret_access_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
res = model.invoke([HumanMessage(content=__('Hello'))]) res = model.invoke([HumanMessage(content=gettext('Hello'))])
print(res) print(res)
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

8
apps/setting/models_provider/impl/volcanic_engine_model_provider/credential/stt.py
View File

@ -2,7 +2,7 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -22,12 +22,12 @@ class VolcanicEngineSTTModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['volcanic_api_url', 'volcanic_app_id', 'volcanic_token', 'volcanic_cluster']: for key in ['volcanic_api_url', 'volcanic_app_id', 'volcanic_token', 'volcanic_cluster']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -38,7 +38,7 @@ class VolcanicEngineSTTModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/volcanic_engine_model_provider/credential/tti.py
View File

@ -2,7 +2,7 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -37,12 +37,13 @@ class VolcanicEngineTTIModelCredential(BaseForm, BaseModelCredential):
raise_exception=False): raise_exception=False):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, __('{model_type} Model type is not supported').format(model_type=model_type)) raise AppApiException(ValidCode.valid_error.value,
gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['access_key', 'secret_key']: for key in ['access_key', 'secret_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -52,7 +53,9 @@ class VolcanicEngineTTIModelCredential(BaseForm, BaseModelCredential):
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('Verification failed, please check whether the parameters are correct: {error}').format(error=str(e))) raise AppApiException(ValidCode.valid_error.value, gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e)))
else: else:
return False return False
return True return True

15
apps/setting/models_provider/impl/volcanic_engine_model_provider/credential/tts.py
View File

@ -2,11 +2,12 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
from common.forms import BaseForm, TooltipLabel from common.forms import BaseForm, TooltipLabel
from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode from setting.models_provider.base_model_provider import BaseModelCredential, ValidCode
from django.utils.translation import gettext_lazy as _, gettext as __
class VolcanicEngineTTSModelGeneralParams(BaseForm): class VolcanicEngineTTSModelGeneralParams(BaseForm):
@ -38,7 +39,8 @@ class VolcanicEngineTTSModelGeneralParams(BaseForm):
class VolcanicEngineTTSModelCredential(BaseForm, BaseModelCredential): class VolcanicEngineTTSModelCredential(BaseForm, BaseModelCredential):
volcanic_api_url = forms.TextInputField('API URL', required=True, default_value='wss://openspeech.bytedance.com/api/v1/tts/ws_binary') volcanic_api_url = forms.TextInputField('API URL', required=True,
default_value='wss://openspeech.bytedance.com/api/v1/tts/ws_binary')
volcanic_app_id = forms.TextInputField('App ID', required=True) volcanic_app_id = forms.TextInputField('App ID', required=True)
volcanic_token = forms.PasswordInputField('Access Token', required=True) volcanic_token = forms.PasswordInputField('Access Token', required=True)
volcanic_cluster = forms.TextInputField('Cluster ID', required=True) volcanic_cluster = forms.TextInputField('Cluster ID', required=True)
@ -47,12 +49,13 @@ class VolcanicEngineTTSModelCredential(BaseForm, BaseModelCredential):
raise_exception=False): raise_exception=False):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, __('{model_type} Model type is not supported').format(model_type=model_type)) raise AppApiException(ValidCode.valid_error.value,
gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['volcanic_api_url', 'volcanic_app_id', 'volcanic_token', 'volcanic_cluster']: for key in ['volcanic_api_url', 'volcanic_app_id', 'volcanic_token', 'volcanic_cluster']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -62,7 +65,9 @@ class VolcanicEngineTTSModelCredential(BaseForm, BaseModelCredential):
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('Verification failed, please check whether the parameters are correct: {error}').format(error=str(e))) raise AppApiException(ValidCode.valid_error.value, gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e)))
else: else:
return False return False
return True return True

4
apps/setting/models_provider/impl/volcanic_engine_model_provider/model/tts.py
View File

@ -18,7 +18,7 @@ import uuid
from typing import Dict from typing import Dict
import websockets import websockets
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common.util.common import _remove_empty_lines from common.util.common import _remove_empty_lines
from setting.models_provider.base_model_provider import MaxKBBaseModel from setting.models_provider.base_model_provider import MaxKBBaseModel
@ -74,7 +74,7 @@ class VolcanicEngineTextToSpeech(MaxKBBaseModel, BaseTextToSpeech):
) )
def check_auth(self): def check_auth(self):
self.text_to_speech(__('Hello')) self.text_to_speech(_('Hello'))
def text_to_speech(self, text): def text_to_speech(self, text):
request_json = { request_json = {

8
apps/setting/models_provider/impl/wenxin_model_provider/credential/embedding.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -23,17 +23,17 @@ class QianfanEmbeddingCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
self.valid_form(model_credential) self.valid_form(model_credential)
try: try:
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.embed_query(__('Hello')) model.embed_query(_('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

14
apps/setting/models_provider/impl/wenxin_model_provider/credential/llm.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -42,21 +42,21 @@ class WenxinLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
model_info = [model.lower() for model in model.client.models()] model_info = [model.lower() for model in model.client.models()]
if not model_info.__contains__(model_name.lower()): if not model_info.__containsgettext(model_name.lower()):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_name} The model does not support').format(model_name=model_name)) gettext('{model_name} The model does not support').format(model_name=model_name))
for key in ['api_key', 'secret_key']: for key in ['api_key', 'secret_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model.invoke( model.invoke(
[HumanMessage(content=__('Hello'))]) [HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
raise e raise e
return True return True
@ -67,7 +67,7 @@ class WenxinLLMModelCredential(BaseForm, BaseModelCredential):
def build_model(self, model_info: Dict[str, object]): def build_model(self, model_info: Dict[str, object]):
for key in ['api_key', 'secret_key', 'model']: for key in ['api_key', 'secret_key', 'model']:
if key not in model_info: if key not in model_info:
raise AppApiException(500, __('{key} is required').format(key=key)) raise AppApiException(500, gettext('{key} is required').format(key=key))
self.api_key = model_info.get('api_key') self.api_key = model_info.get('api_key')
self.secret_key = model_info.get('secret_key') self.secret_key = model_info.get('secret_key')
return self return self

8
apps/setting/models_provider/impl/xf_model_provider/credential/embedding.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -23,17 +23,17 @@ class XFEmbeddingCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
self.valid_form(model_credential) self.valid_form(model_credential)
try: try:
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.embed_query(__('Hello')) model.embed_query(_('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

10
apps/setting/models_provider/impl/xf_model_provider/credential/image.py
View File

@ -3,7 +3,7 @@ import base64
import os import os
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -25,12 +25,12 @@ class XunFeiImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['spark_api_url', 'spark_app_id', 'spark_api_key', 'spark_api_secret']: for key in ['spark_api_url', 'spark_app_id', 'spark_api_key', 'spark_api_secret']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -38,14 +38,14 @@ class XunFeiImageModelCredential(BaseForm, BaseModelCredential):
cwd = os.path.dirname(os.path.abspath(__file__)) cwd = os.path.dirname(os.path.abspath(__file__))
with open(f'{cwd}/img_1.png', 'rb') as f: with open(f'{cwd}/img_1.png', 'rb') as f:
message_list = [ImageMessage(str(base64.b64encode(f.read()), 'utf-8')), message_list = [ImageMessage(str(base64.b64encode(f.read()), 'utf-8')),
HumanMessage(__('Please outline this picture'))] HumanMessage(_('Please outline this picture'))]
model.stream(message_list) model.stream(message_list)
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/xf_model_provider/credential/llm.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -62,23 +62,24 @@ class XunFeiLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['spark_api_url', 'spark_app_id', 'spark_api_key', 'spark_api_secret']: for key in ['spark_api_url', 'spark_app_id', 'spark_api_key', 'spark_api_secret']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
model.invoke([HumanMessage(content=__('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

8
apps/setting/models_provider/impl/xf_model_provider/credential/stt.py
View File

@ -2,7 +2,7 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -21,12 +21,12 @@ class XunFeiSTTModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['spark_api_url', 'spark_app_id', 'spark_api_key', 'spark_api_secret']: for key in ['spark_api_url', 'spark_app_id', 'spark_api_key', 'spark_api_secret']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -37,7 +37,7 @@ class XunFeiSTTModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

9
apps/setting/models_provider/impl/xf_model_provider/credential/tts.py
View File

@ -2,7 +2,7 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -44,12 +44,12 @@ class XunFeiTTSModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['spark_api_url', 'spark_app_id', 'spark_api_key', 'spark_api_secret']: for key in ['spark_api_url', 'spark_app_id', 'spark_api_key', 'spark_api_secret']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -60,7 +60,8 @@ class XunFeiTTSModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

4
apps/setting/models_provider/impl/xf_model_provider/model/tts.py
View File

@ -17,7 +17,7 @@ from typing import Dict
from urllib.parse import urlencode, urlparse from urllib.parse import urlencode, urlparse
import websockets import websockets
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common.util.common import _remove_empty_lines from common.util.common import _remove_empty_lines
from setting.models_provider.base_model_provider import MaxKBBaseModel from setting.models_provider.base_model_provider import MaxKBBaseModel
@ -98,7 +98,7 @@ class XFSparkTextToSpeech(MaxKBBaseModel, BaseTextToSpeech):
return url return url
def check_auth(self): def check_auth(self):
self.text_to_speech(__('Hello')) self.text_to_speech(_('Hello'))
def text_to_speech(self, text): def text_to_speech(self, text):

12
apps/setting/models_provider/impl/xinference_model_provider/credential/embedding.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -16,19 +16,19 @@ class XinferenceEmbeddingModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
try: try:
model_list = provider.get_base_model_list(model_credential.get('api_base'), model_credential.get('api_key'), model_list = provider.get_base_model_list(model_credential.get('api_base'), model_credential.get('api_key'),
'embedding') 'embedding')
except Exception as e: except Exception as e:
raise AppApiException(ValidCode.valid_error.value, __('API domain name is invalid')) raise AppApiException(ValidCode.valid_error.value, _('API domain name is invalid'))
exist = provider.get_model_info_by_name(model_list, model_name) exist = provider.get_model_info_by_name(model_list, model_name)
model: LocalEmbedding = provider.get_model(model_type, model_name, model_credential) model: LocalEmbedding = provider.get_model(model_type, model_name, model_credential)
if len(exist) == 0: if len(exist) == 0:
model.start_down_model_thread() model.start_down_model_thread()
raise AppApiException(ValidCode.model_not_fount, raise AppApiException(ValidCode.model_not_fount,
__('The model does not exist, please download the model first')) _('The model does not exist, please download the model first'))
model.embed_query(__('Hello')) model.embed_query(_('Hello'))
return True return True
def encryption_dict(self, model_info: Dict[str, object]): def encryption_dict(self, model_info: Dict[str, object]):
@ -37,7 +37,7 @@ class XinferenceEmbeddingModelCredential(BaseForm, BaseModelCredential):
def build_model(self, model_info: Dict[str, object]): def build_model(self, model_info: Dict[str, object]):
for key in ['model']: for key in ['model']:
if key not in model_info: if key not in model_info:
raise AppApiException(500, __('{key} is required').format(key=key)) raise AppApiException(500, _('{key} is required').format(key=key))
return self return self
api_base = forms.TextInputField('API URL', required=True) api_base = forms.TextInputField('API URL', required=True)

11
apps/setting/models_provider/impl/xinference_model_provider/credential/image.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -38,17 +38,17 @@ class XinferenceImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
res = model.stream([HumanMessage(content=[{"type": "text", "text": __('Hello')}])]) res = model.stream([HumanMessage(content=[{"type": "text", "text": gettext('Hello')}])])
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
@ -56,7 +56,8 @@ class XinferenceImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

12
apps/setting/models_provider/impl/xinference_model_provider/credential/llm.py
View File

@ -2,7 +2,7 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -36,18 +36,18 @@ class XinferenceLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
try: try:
model_list = provider.get_base_model_list(model_credential.get('api_base'), model_credential.get('api_key'), model_list = provider.get_base_model_list(model_credential.get('api_base'), model_credential.get('api_key'),
model_type) model_type)
except Exception as e: except Exception as e:
raise AppApiException(ValidCode.valid_error.value, __('API domain name is invalid')) raise AppApiException(ValidCode.valid_error.value, gettext('API domain name is invalid'))
exist = provider.get_model_info_by_name(model_list, model_name) exist = provider.get_model_info_by_name(model_list, model_name)
if len(exist) == 0: if len(exist) == 0:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('The model does not exist, please download the model first')) gettext('The model does not exist, please download the model first'))
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
model.invoke([HumanMessage(content=__('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
return True return True
def encryption_dict(self, model_info: Dict[str, object]): def encryption_dict(self, model_info: Dict[str, object]):
@ -56,7 +56,7 @@ class XinferenceLLMModelCredential(BaseForm, BaseModelCredential):
def build_model(self, model_info: Dict[str, object]): def build_model(self, model_info: Dict[str, object]):
for key in ['api_key', 'model']: for key in ['api_key', 'model']:
if key not in model_info: if key not in model_info:
raise AppApiException(500, __('{key} is required').format(key=key)) raise AppApiException(500, gettext('{key} is required').format(key=key))
self.api_key = model_info.get('api_key') self.api_key = model_info.get('api_key')
return self return self

10
apps/setting/models_provider/impl/xinference_model_provider/credential/reranker.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from langchain_core.documents import Document from langchain_core.documents import Document
from common import forms from common import forms
@ -22,22 +22,22 @@ class XInferenceRerankerModelCredential(BaseForm, BaseModelCredential):
raise_exception=True): raise_exception=True):
if not model_type == 'RERANKER': if not model_type == 'RERANKER':
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['server_url']: for key in ['server_url']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.compress_documents([Document(page_content=__('Hello'))], __('Hello')) model.compress_documents([Document(page_content=_('Hello'))], _('Hello'))
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

8
apps/setting/models_provider/impl/xinference_model_provider/credential/stt.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -18,12 +18,12 @@ class XInferenceSTTModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) _('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, _('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -34,7 +34,7 @@ class XInferenceSTTModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( _('Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

9
apps/setting/models_provider/impl/xinference_model_provider/credential/tti.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -56,12 +56,12 @@ class XinferenceTextToImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -73,7 +73,8 @@ class XinferenceTextToImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

9
apps/setting/models_provider/impl/xinference_model_provider/credential/tts.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -36,12 +36,12 @@ class XInferenceTTSModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_base', 'api_key']: for key in ['api_base', 'api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -52,7 +52,8 @@ class XInferenceTTSModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

4
apps/setting/models_provider/impl/xinference_model_provider/model/tts.py
View File

@ -6,7 +6,7 @@ from common.config.tokenizer_manage_config import TokenizerManage
from common.util.common import _remove_empty_lines from common.util.common import _remove_empty_lines
from setting.models_provider.base_model_provider import MaxKBBaseModel from setting.models_provider.base_model_provider import MaxKBBaseModel
from setting.models_provider.impl.base_tts import BaseTextToSpeech from setting.models_provider.impl.base_tts import BaseTextToSpeech
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
def custom_get_token_ids(text: str): def custom_get_token_ids(text: str):
@ -41,7 +41,7 @@ class XInferenceTextToSpeech(MaxKBBaseModel, BaseTextToSpeech):
) )
def check_auth(self): def check_auth(self):
self.text_to_speech(__('Hello')) self.text_to_speech(_('Hello'))
def text_to_speech(self, text): def text_to_speech(self, text):
client = OpenAI( client = OpenAI(

11
apps/setting/models_provider/impl/zhipu_model_provider/credential/image.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -37,17 +37,17 @@ class ZhiPuImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
res = model.stream([HumanMessage(content=[{"type": "text", "text": __('Hello')}])]) res = model.stream([HumanMessage(content=[{"type": "text", "text": gettext('Hello')}])])
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
@ -55,7 +55,8 @@ class ZhiPuImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

11
apps/setting/models_provider/impl/zhipu_model_provider/credential/llm.py
View File

@ -8,7 +8,7 @@
""" """
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from common import forms from common import forms
@ -43,22 +43,23 @@ class ZhiPuLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
model = provider.get_model(model_type, model_name, model_credential, **model_params) model = provider.get_model(model_type, model_name, model_credential, **model_params)
model.invoke([HumanMessage(content=__('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

9
apps/setting/models_provider/impl/zhipu_model_provider/credential/tti.py
View File

@ -1,7 +1,7 @@
# coding=utf-8 # coding=utf-8
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext as __ from django.utils.translation import gettext_lazy as _, gettext
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -36,12 +36,12 @@ class ZhiPuTextToImageModelCredential(BaseForm, BaseModelCredential):
model_type_list = provider.get_model_type_list() model_type_list = provider.get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))): if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('{model_type} Model type is not supported').format(model_type=model_type)) gettext('{model_type} Model type is not supported').format(model_type=model_type))
for key in ['api_key']: for key in ['api_key']:
if key not in model_credential: if key not in model_credential:
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, __('{key} is required').format(key=key)) raise AppApiException(ValidCode.valid_error.value, gettext('{key} is required').format(key=key))
else: else:
return False return False
try: try:
@ -53,7 +53,8 @@ class ZhiPuTextToImageModelCredential(BaseForm, BaseModelCredential):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
__('Verification failed, please check whether the parameters are correct: {error}').format( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format(
error=str(e))) error=str(e)))
else: else:
return False return False

4
apps/setting/models_provider/impl/zhipu_model_provider/model/tti.py
View File

@ -1,6 +1,6 @@
from typing import Dict from typing import Dict
from django.utils.translation import gettext as __ from django.utils.translation import gettext as _
from langchain_community.chat_models import ChatZhipuAI from langchain_community.chat_models import ChatZhipuAI
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from zhipuai import ZhipuAI from zhipuai import ZhipuAI
@ -46,7 +46,7 @@ class ZhiPuTextToImage(MaxKBBaseModel, BaseTextToImage):
zhipuai_api_key=self.api_key, zhipuai_api_key=self.api_key,
model_name=self.model, model_name=self.model,
) )
chat.invoke([HumanMessage([{"type": "text", "text": __('Hello')}])]) chat.invoke([HumanMessage([{"type": "text", "text": _('Hello')}])])
# self.generate_image('生成一个小猫图片') # self.generate_image('生成一个小猫图片')