| import logging | |||||
| from flask_login import current_user | from flask_login import current_user | ||||
| from libs.login import login_required | from libs.login import login_required | ||||
| from flask_restful import Resource, reqparse | from flask_restful import Resource, reqparse | ||||
| provider_service = ProviderService() | provider_service = ProviderService() | ||||
| model_settings = args['model_settings'] | model_settings = args['model_settings'] | ||||
| for model_setting in 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'} | return {'result': 'success'} | ||||