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

Refactor settings service and import statements

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-19 14:56:58 -03:00
commit 1d49876647
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/backend/langflow/services/settings/service.py
View file

@ -1,9 +1,10 @@
import os
import yaml
from langflow.services.base import Service from langflow.services.base import Service
from langflow.services.settings.auth import AuthSettings from langflow.services.settings.auth import AuthSettings
from langflow.services.settings.base import Settings from langflow.services.settings.base import Settings
from loguru import logger from loguru import logger
import os
import yaml
class SettingsService(Service): class SettingsService(Service):
@ -11,8 +12,8 @@ class SettingsService(Service):
def __init__(self, settings: Settings, auth_settings: AuthSettings): def __init__(self, settings: Settings, auth_settings: AuthSettings):
super().__init__() super().__init__()
self.settings = settings self.settings: Settings = settings
self.auth_settings = auth_settings self.auth_settings: AuthSettings = auth_settings
@classmethod @classmethod
def load_settings_from_yaml(cls, file_path: str) -> "SettingsService": def load_settings_from_yaml(cls, file_path: str) -> "SettingsService":
@ -30,7 +31,9 @@ class SettingsService(Service):
for key in settings_dict: for key in settings_dict:
if key not in Settings.model_fields.keys(): if key not in Settings.model_fields.keys():
raise KeyError(f"Key {key} not found in settings") raise KeyError(f"Key {key} not found in settings")
logger.debug(f"Loading {len(settings_dict[key])} {key} from {file_path}") logger.debug(
f"Loading {len(settings_dict[key])} {key} from {file_path}"
)
settings = Settings(**settings_dict) settings = Settings(**settings_dict)
if not settings.CONFIG_DIR: if not settings.CONFIG_DIR: