jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: add try catch on fenet decrypt to prevent errors (#2660)

Browse source
This commit is contained in:
Cristhian Zanforlin Lousa 2024-07-12 13:25:44 -03:00 • committed by GitHub
commit 8813143622
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/services/auth/utils.py
View file

@ -367,7 +367,11 @@ def encrypt_api_key(api_key: str, settings_service=Depends(get_settings_service)
def decrypt_api_key(encrypted_api_key: str, settings_service=Depends(get_settings_service)): def decrypt_api_key(encrypted_api_key: str, settings_service=Depends(get_settings_service)):
fernet = get_fernet(settings_service) fernet = get_fernet(settings_service)
decrypted_key = ""
# Two-way decryption # Two-way decryption
if isinstance(encrypted_api_key, str): if isinstance(encrypted_api_key, str):
decrypted_key = fernet.decrypt(encrypted_api_key).decode() try:
decrypted_key = fernet.decrypt(encrypted_api_key.encode()).decode()
except Exception:
decrypted_key = fernet.decrypt(encrypted_api_key).decode()
return decrypted_key return decrypted_key