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