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

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

Browse source
This commit is contained in:
ogabrielluiz 2024-06-17 21:08:34 -03:00
commit dbd9c54012
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/base/langflow/services/settings/auth.py
View file

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