Procházet zdrojové kódy

Compatible model saving error (#1582)

Co-authored-by: jyong <jyong@dify.ai>
Jyong před 1 rokem
rodič
revize
cc35d0645a
1 změnil soubory, kde provedl 11 přidání a 6 odebrání
  1. 11 6
      api/controllers/console/workspace/models.py

+ 11 - 6
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 @@ 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'}