Ver código fonte

Compatible model saving error (#1582)

Co-authored-by: jyong <jyong@dify.ai>
tags/0.3.31
Jyong 1 ano atrás
pai
commit
cc35d0645a
Nenhuma conta vinculada ao e-mail do autor do commit
1 arquivos alterados com 11 adições e 6 exclusões
  1. 11
    6
      api/controllers/console/workspace/models.py

+ 11
- 6
api/controllers/console/workspace/models.py Ver arquivo

@@ -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'}


Carregando…
Cancelar
Salvar