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

🔧 chore(settings.py): add BASE_COMPONENTS_PATH constant to improve code readability and maintainability

Browse source 
🔧 chore(settings.py): add support for LANGFLOW_COMPONENT_PATH environment variable to allow custom component paths to be added
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-25 12:16:06 -03:00
commit 0095fc4753
1 changed files with 12 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/backend/langflow/settings.py
View file

@ -6,6 +6,8 @@ import yaml
from pydantic import BaseSettings, root_validator from pydantic import BaseSettings, root_validator
from langflow.utils.logger import logger from langflow.utils.logger import logger
BASE_COMPONENTS_PATH = Path(__file__).parent / "components"
class Settings(BaseSettings): class Settings(BaseSettings):
chains: dict = {} chains: dict = {}
@ -43,11 +45,18 @@ class Settings(BaseSettings):
logger.debug("No DATABASE_URL env variable, using sqlite database") logger.debug("No DATABASE_URL env variable, using sqlite database")
values["database_url"] = "sqlite:///./langflow.db" values["database_url"] = "sqlite:///./langflow.db"
values["component_path"] = [Path(__file__).parent / "components"] if not values.get("component_path"):
values["component_path"] = [BASE_COMPONENTS_PATH]
elif BASE_COMPONENTS_PATH not in values["component_path"]:
values["component_path"].append(BASE_COMPONENTS_PATH)
if os.getenv("LANGFLOW_COMPONENT_PATH"): if os.getenv("LANGFLOW_COMPONENT_PATH"):
values["component_path"].append(Path(os.getenv("LANGFLOW_COMPONENT_PATH"))) langflow_component_path = Path(os.getenv("LANGFLOW_COMPONENT_PATH"))
if (
langflow_component_path.exists()
and langflow_component_path not in values["component_path"]
):
values["component_path"].append(langflow_component_path)
return values return values
class Config: class Config: