📝 (auth.py): refactor code to handle secret key value properly and improve readability

This commit is contained in:
ogabrielluiz 2024-06-17 21:08:34 -03:00
commit dbd9c54012

View file

@ -2,14 +2,13 @@ import secrets
from pathlib import Path
from typing import Literal
from langflow.services.settings.constants import DEFAULT_SUPERUSER, DEFAULT_SUPERUSER_PASSWORD
from langflow.services.settings.utils import read_secret_from_file, write_secret_to_file
from loguru import logger
from passlib.context import CryptContext
from pydantic import Field, SecretStr, field_validator
from pydantic_settings import BaseSettings
from langflow.services.settings.constants import DEFAULT_SUPERUSER, DEFAULT_SUPERUSER_PASSWORD
from langflow.services.settings.utils import read_secret_from_file, write_secret_to_file
class AuthSettings(BaseSettings):
# Login settings
@ -91,9 +90,9 @@ class AuthSettings(BaseSettings):
secret_key_path = Path(config_dir) / "secret_key"
if value and isinstance(value, SecretStr):
if value:
logger.debug("Secret key provided")
secret_value = value.get_secret_value()
secret_value = value.get_secret_value() if isinstance(value, SecretStr) else value
write_secret_to_file(secret_key_path, secret_value)
else:
logger.debug("No secret key provided, generating a random one")