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

fix: 处理校验图片模型的报错

Browse Source
This commit is contained in:
CaptainB 2024-11-19 16:53:32 +08:00
parent b91d9fdaf0
commit 4fd6884ad6
3 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -29,7 +29,9 @@ class OpenAIImageModelCredential(BaseForm, BaseModelCredential):
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.stream([HumanMessage(content=[{"type": "text", "text": "你好"}])]) res = model.stream([HumanMessage(content=[{"type": "text", "text": "你好"}])])
for chunk in res:
print(chunk)
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e

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

@ -50,7 +50,9 @@ class QwenVLModelCredential(BaseForm, BaseModelCredential):
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.stream([HumanMessage(content=[{"type": "text", "text": "你好"}])]) res = model.stream([HumanMessage(content=[{"type": "text", "text": "你好"}])])
for chunk in res:
print(chunk)
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e

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

@ -50,7 +50,9 @@ class TencentVisionModelCredential(BaseForm, BaseModelCredential):
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.stream([HumanMessage(content=[{"type": "text", "text": "你好"}])]) res = model.stream([HumanMessage(content=[{"type": "text", "text": "你好"}])])
for chunk in res:
print(chunk)
except Exception as e: except Exception as e:
if isinstance(e, AppApiException): if isinstance(e, AppApiException):
raise e raise e