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