fix: credentials validate compatible problem (#2170)

langgenius · Jan 24, 2024 · 784da52 · 784da52
1 parent 78524a5
commit 784da52
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/api/core/entities/provider_configuration.py b/api/core/entities/provider_configuration.py
@@ -153,8 +153,16 @@ def custom_credentials_validate(self, credentials: dict) -> Tuple[Provider, dict
 
         if provider_record:
             try:
-                original_credentials = json.loads(
-                    provider_record.encrypted_config) if provider_record.encrypted_config else {}
+                # fix origin data
+                if provider_record.encrypted_config:
+                    if not provider_record.encrypted_config.startswith("{"):
+                        original_credentials = {
+                            "openai_api_key": provider_record.encrypted_config
+                        }
+                    else:
+                        original_credentials = json.loads(provider_record.encrypted_config)
+                else:
+                    original_credentials = {}
             except JSONDecodeError:
                 original_credentials = {}