fix: Model addition, error log printing (#2499)

This commit is contained in:
shaohuzhang1 2025-03-05 14:45:50 +08:00 committed by GitHub
parent 83d51ea866
commit ea9c5e0ee8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
65 changed files with 131 additions and 9 deletions

View File

@ -6,6 +6,7 @@
@date2024/10/16 17:01 @date2024/10/16 17:01
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -35,6 +36,7 @@ class AliyunBaiLianEmbeddingCredential(BaseForm, BaseModelCredential):
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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 18:41 @date2024/7/11 18:41
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -56,6 +57,7 @@ class QwenVLModelCredential(BaseForm, BaseModelCredential):
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
@ -48,6 +49,7 @@ class BaiLianLLMModelCredential(BaseForm, BaseModelCredential):
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=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/9/9 17:51 @date2024/9/9 17:51
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -35,6 +36,7 @@ class AliyunBaiLianRerankerCredential(BaseForm, BaseModelCredential):
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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,5 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -30,6 +30,7 @@ class AliyunBaiLianSTTModelCredential(BaseForm, BaseModelCredential):
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 18:41 @date2024/7/11 18:41
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -76,6 +77,7 @@ class QwenTextToImageModelCredential(BaseForm, BaseModelCredential):
res = model.check_auth() res = model.check_auth()
print(res) print(res)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,5 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -64,6 +64,7 @@ class AliyunBaiLianTTSModelCredential(BaseForm, BaseModelCredential):
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.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -52,6 +53,7 @@ class AnthropicImageModelCredential(BaseForm, BaseModelCredential):
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 18:32 @date2024/7/11 18:32
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
@ -55,6 +56,7 @@ class AnthropicLLMModelCredential(BaseForm, BaseModelCredential):
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.invoke([HumanMessage(content=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,3 +1,4 @@
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -35,6 +36,7 @@ class BedrockEmbeddingCredential(BaseForm, BaseModelCredential):
except AppApiException: except AppApiException:
raise raise
except Exception as e: except Exception as e:
traceback.print_exc()
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(

View File

@ -1,3 +1,4 @@
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -53,6 +54,7 @@ class BedrockLLMModelCredential(BaseForm, BaseModelCredential):
except AppApiException: except AppApiException:
raise raise
except Exception as e: except Exception as e:
traceback.print_exc()
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
gettext( gettext(

View File

@ -6,6 +6,7 @@
@date2024/7/11 17:08 @date2024/7/11 17:08
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -35,6 +36,7 @@ class AzureOpenAIEmbeddingCredential(BaseForm, BaseModelCredential):
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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,6 +1,7 @@
# coding=utf-8 # coding=utf-8
import base64 import base64
import os import os
import traceback
from typing import Dict from typing import Dict
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
@ -55,6 +56,7 @@ class AzureOpenAIImageModelCredential(BaseForm, BaseModelCredential):
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 17:08 @date2024/7/11 17:08
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
@ -55,6 +56,7 @@ class AzureLLMModelCredential(BaseForm, BaseModelCredential):
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=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -31,6 +32,7 @@ class AzureOpenAISTTModelCredential(BaseForm, BaseModelCredential):
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -67,6 +68,7 @@ class AzureOpenAITextToImageModelCredential(BaseForm, BaseModelCredential):
res = model.check_auth() res = model.check_auth()
print(res) print(res)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -49,6 +50,7 @@ class AzureOpenAITTSModelCredential(BaseForm, BaseModelCredential):
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.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 17:51 @date2024/7/11 17:51
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -55,6 +56,7 @@ class DeepSeekLLMModelCredential(BaseForm, BaseModelCredential):
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=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/12 16:45 @date2024/7/12 16:45
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -34,6 +35,7 @@ class GeminiEmbeddingCredential(BaseForm, BaseModelCredential):
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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,5 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -52,6 +52,7 @@ class GeminiImageModelCredential(BaseForm, BaseModelCredential):
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 17:57 @date2024/7/11 17:57
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -56,6 +57,7 @@ class GeminiLLMModelCredential(BaseForm, BaseModelCredential):
res = model.invoke([HumanMessage(content=gettext('Hello'))]) res = model.invoke([HumanMessage(content=gettext('Hello'))])
print(res) print(res)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -29,6 +30,7 @@ class GeminiSTTModelCredential(BaseForm, BaseModelCredential):
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 18:06 @date2024/7/11 18:06
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -55,6 +56,7 @@ class KimiLLMModelCredential(BaseForm, BaseModelCredential):
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=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 11:06 @date2024/7/11 11:06
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -34,6 +35,7 @@ class LocalEmbeddingCredential(BaseForm, BaseModelCredential):
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(gettext('Hello')) model.embed_query(gettext('Hello'))
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/9/3 14:33 @date2024/9/3 14:33
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from langchain_core.documents import Document from langchain_core.documents import Document
@ -35,6 +36,7 @@ class LocalRerankerCredential(BaseForm, BaseModelCredential):
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=gettext('Hello'))], gettext('Hello')) model.compress_documents([Document(page_content=gettext('Hello'))], gettext('Hello'))
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/12 16:45 @date2024/7/12 16:45
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -34,6 +35,7 @@ class OpenAIEmbeddingCredential(BaseForm, BaseModelCredential):
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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,6 +1,7 @@
# coding=utf-8 # coding=utf-8
import base64 import base64
import os import os
import traceback
from typing import Dict from typing import Dict
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
@ -54,6 +55,7 @@ class OpenAIImageModelCredential(BaseForm, BaseModelCredential):
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 18:32 @date2024/7/11 18:32
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -56,6 +57,7 @@ class OpenAILLMModelCredential(BaseForm, BaseModelCredential):
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=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -30,6 +31,7 @@ class OpenAISTTModelCredential(BaseForm, BaseModelCredential):
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -70,6 +71,7 @@ class OpenAITextToImageModelCredential(BaseForm, BaseModelCredential):
res = model.check_auth() res = model.check_auth()
print(res) print(res)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -48,6 +49,7 @@ class OpenAITTSModelCredential(BaseForm, BaseModelCredential):
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.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 18:41 @date2024/7/11 18:41
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -56,6 +57,7 @@ class QwenVLModelCredential(BaseForm, BaseModelCredential):
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 18:41 @date2024/7/11 18:41
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -54,6 +55,7 @@ class OpenAILLMModelCredential(BaseForm, BaseModelCredential):
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=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 18:41 @date2024/7/11 18:41
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -76,6 +77,7 @@ class QwenTextToImageModelCredential(BaseForm, BaseModelCredential):
res = model.check_auth() res = model.check_auth()
print(res) print(res)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/12 16:45 @date2024/7/12 16:45
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -34,6 +35,7 @@ class SiliconCloudEmbeddingCredential(BaseForm, BaseModelCredential):
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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,6 +1,7 @@
# coding=utf-8 # coding=utf-8
import base64 import base64
import os import os
import traceback
from typing import Dict from typing import Dict
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
@ -54,6 +55,7 @@ class SiliconCloudImageModelCredential(BaseForm, BaseModelCredential):
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 18:32 @date2024/7/11 18:32
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -56,6 +57,7 @@ class SiliconCloudLLMModelCredential(BaseForm, BaseModelCredential):
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=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/9/9 17:51 @date2024/9/9 17:51
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -35,6 +36,7 @@ class SiliconCloudRerankerCredential(BaseForm, BaseModelCredential):
model: SiliconCloudReranker = provider.get_model(model_type, model_name, model_credential) model: SiliconCloudReranker = 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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -30,6 +31,7 @@ class SiliconCloudSTTModelCredential(BaseForm, BaseModelCredential):
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -70,6 +71,7 @@ class SiliconCloudTextToImageModelCredential(BaseForm, BaseModelCredential):
res = model.check_auth() res = model.check_auth()
print(res) print(res)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -48,6 +49,7 @@ class SiliconCloudTTSModelCredential(BaseForm, BaseModelCredential):
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.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 18:32 @date2024/7/11 18:32
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -56,6 +57,7 @@ class TencentCloudLLMModelCredential(BaseForm, BaseModelCredential):
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=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,3 +1,4 @@
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -21,6 +22,7 @@ class TencentEmbeddingCredential(BaseForm, BaseModelCredential):
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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 18:41 @date2024/7/11 18:41
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -56,6 +57,7 @@ class TencentVisionModelCredential(BaseForm, BaseModelCredential):
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,6 @@
# coding=utf-8 # coding=utf-8
import traceback
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
@ -48,6 +50,7 @@ class TencentLLMModelCredential(BaseForm, BaseModelCredential):
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=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
gettext( gettext(

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -96,6 +97,7 @@ class TencentTTIModelCredential(BaseForm, BaseModelCredential):
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.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
gettext( gettext(

View File

@ -6,6 +6,7 @@
@date2024/7/12 16:45 @date2024/7/12 16:45
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -34,6 +35,7 @@ class VllmEmbeddingCredential(BaseForm, BaseModelCredential):
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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -52,6 +53,7 @@ class VllmImageModelCredential(BaseForm, BaseModelCredential):
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,5 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -49,6 +49,7 @@ class VLLMModelCredential(BaseForm, BaseModelCredential):
try: try:
res = model.invoke([HumanMessage(content=gettext('Hello'))]) res = model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,
gettext( gettext(
'Verification failed, please check whether the parameters are correct: {error}').format( 'Verification failed, please check whether the parameters are correct: {error}').format(

View File

@ -6,6 +6,7 @@
@date2024/7/12 16:45 @date2024/7/12 16:45
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -34,6 +35,7 @@ class VolcanicEmbeddingCredential(BaseForm, BaseModelCredential):
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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -52,6 +53,7 @@ class VolcanicEngineImageModelCredential(BaseForm, BaseModelCredential):
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/11 17:57 @date2024/7/11 17:57
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -56,6 +57,7 @@ class VolcanicEngineLLMModelCredential(BaseForm, BaseModelCredential):
res = model.invoke([HumanMessage(content=gettext('Hello'))]) res = model.invoke([HumanMessage(content=gettext('Hello'))])
print(res) print(res)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,5 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -34,6 +34,7 @@ class VolcanicEngineSTTModelCredential(BaseForm, BaseModelCredential):
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,5 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -50,6 +50,7 @@ class VolcanicEngineTTIModelCredential(BaseForm, BaseModelCredential):
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.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,5 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -62,6 +62,7 @@ class VolcanicEngineTTSModelCredential(BaseForm, BaseModelCredential):
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.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/10/17 15:40 @date2024/10/17 15:40
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -29,6 +30,7 @@ class QianfanEmbeddingCredential(BaseForm, BaseModelCredential):
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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/12 10:19 @date2024/7/12 10:19
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -58,6 +59,7 @@ class WenxinLLMModelCredential(BaseForm, BaseModelCredential):
model.invoke( model.invoke(
[HumanMessage(content=gettext('Hello'))]) [HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
raise e raise e
return True return True

View File

@ -6,6 +6,7 @@
@date2024/10/17 15:40 @date2024/10/17 15:40
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -29,6 +30,7 @@ class XFEmbeddingCredential(BaseForm, BaseModelCredential):
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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,6 +1,7 @@
# coding=utf-8 # coding=utf-8
import base64 import base64
import os import os
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -41,6 +42,7 @@ class XunFeiImageModelCredential(BaseForm, BaseModelCredential):
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:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/12 10:29 @date2024/7/12 10:29
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -74,6 +75,7 @@ class XunFeiLLMModelCredential(BaseForm, BaseModelCredential):
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=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,5 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -33,6 +33,7 @@ class XunFeiSTTModelCredential(BaseForm, BaseModelCredential):
model = provider.get_model(model_type, model_name, model_credential) model = provider.get_model(model_type, model_name, model_credential)
model.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,5 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -56,6 +56,7 @@ class XunFeiTTSModelCredential(BaseForm, BaseModelCredential):
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.check_auth() model.check_auth()
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -51,6 +52,7 @@ class ZhiPuImageModelCredential(BaseForm, BaseModelCredential):
for chunk in res: for chunk in res:
print(chunk) print(chunk)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -6,6 +6,7 @@
@date2024/7/12 10:46 @date2024/7/12 10:46
@desc: @desc:
""" """
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -54,6 +55,7 @@ class ZhiPuLLMModelCredential(BaseForm, BaseModelCredential):
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=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception:

View File

@ -1,4 +1,5 @@
# coding=utf-8 # coding=utf-8
import traceback
from typing import Dict from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
@ -49,6 +50,7 @@ class ZhiPuTextToImageModelCredential(BaseForm, BaseModelCredential):
res = model.check_auth() res = model.check_auth()
print(res) print(res)
except Exception as e: except Exception as e:
traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e
if raise_exception: if raise_exception: