Compatible model saving error (langgenius#1582)

Co-authored-by: jyong <[email protected]>
HuberyHuV1 · Nov 21, 2023 · 6b59a15 · 6b59a15
1 parent 3a9d288
commit 6b59a15
Showing 1 changed file with 11 additions and 6 deletions.
diff --git 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'}