From 6b59a158c4efd372f39282276531a9925bddb4d0 Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Tue, 21 Nov 2023 15:38:27 +0800 Subject: [PATCH] Compatible model saving error (#1582) Co-authored-by: jyong --- api/controllers/console/workspace/models.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/api/controllers/console/workspace/models.py b/api/controllers/console/workspace/models.py index dd10ae1ce266b4..6df99ebe1bda4f 100644 --- a/api/controllers/console/workspace/models.py +++ b/api/controllers/console/workspace/models.py @@ -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 @@ def post(self): 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'}