From dbd9c540123169fd1caa43eed156644b28e82f71 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Mon, 17 Jun 2024 21:08:34 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=93=9D=20(auth.py):=20refactor=20code=20t?= =?UTF-8?q?o=20handle=20secret=20key=20value=20properly=20and=20improve=20?= =?UTF-8?q?readability?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/base/langflow/services/settings/auth.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/backend/base/langflow/services/settings/auth.py b/src/backend/base/langflow/services/settings/auth.py index 8e321ed19b..b88ffe8320 100644 --- a/src/backend/base/langflow/services/settings/auth.py +++ b/src/backend/base/langflow/services/settings/auth.py @@ -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")