AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-04-22 21:59:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Compatible model saving error (#1582)

Browse Source 
Co-authored-by: jyong <jyong@dify.ai>
This commit is contained in:
Jyong 2023-11-21 15:38:27 +08:00 committed by GitHub
parent c9368925a3
commit cc35d0645a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
api/controllers/console/workspace/models.py
View File

@ -1,3 +1,5 @@
import logging
from flask_login import current_user
from libs.login import login_required
from flask_restful import Resource, reqparse
@ -77,12 +79,15 @@ class DefaultModelApi(Resource):
provider_service = ProviderService()
model_settings = args['model_settings']
for model_setting in model_settings:
provider_service.update_default_model_of_model_type(
tenant_id=current_user.current_tenant_id,
model_type=model_setting['model_type'],
provider_name=model_setting['provider_name'],
model_name=model_setting['model_name']
)
try:
provider_service.update_default_model_of_model_type(
tenant_id=current_user.current_tenant_id,
model_type=model_setting['model_type'],
provider_name=model_setting['provider_name'],
model_name=model_setting['model_name']
)
except Exception:
logging.warning(f"{model_setting['model_type']} save error")
return {'result': 'success'}