fix: credentials validate compatible problem (#2170)

784da52e · takatost · GitHub · 78524a56 · 784da52e
Unverified Commit 784da52e authored Jan 24, 2024 by takatost Committed by GitHub Jan 24, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 2 deletions

provider_configuration.py api/core/entities/provider_configuration.py +10 -2

No files found.
--- a/api/core/entities/provider_configuration.py
+++ b/api/core/entities/provider_configuration.py
@@ -153,8 +153,16 @@ class ProviderConfiguration(BaseModel):

        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 = {}