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

modify: set_database_url

Browse source
This commit is contained in:
nsxshota 2023-11-27 00:06:26 +09:00
commit 90759feb2f
3 changed files with 31 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/backend/langflow/__main__.py
View file

@ -127,13 +127,13 @@ def run(
default=None, default=None,
), ),
dev: bool = typer.Option(False, help="Run in development mode (may contain bugs)"), dev: bool = typer.Option(False, help="Run in development mode (may contain bugs)"),
# This variable does not work but is set by the .env file # # This variable does not work but is set by the .env file
# and works with Pydantic # # and works with Pydantic
database_url: str = typer.Option( # database_url: str = typer.Option(
"mysql+pymysql://{}:{}@{}:3306/{}".format(os.environ["username"],os.environ["password"],os.environ["host"],os.environ["dbname"]), # "mysql+pymysql://{}:{}@{}:3306/{}".format(os.environ["username"],os.environ["password"],os.environ["host"],os.environ["dbname"]),
help="Database URL to connect to. If not provided, a local SQLite database will be used.", # help="Database URL to connect to. If not provided, a local SQLite database will be used.",
envvar="LANGFLOW_DATABASE_URL", # envvar="LANGFLOW_DATABASE_URL",
), # ),
path: str = typer.Option( path: str = typer.Option(
None, None,
help="Path to the frontend directory containing build files. This is for development purposes only.", help="Path to the frontend directory containing build files. This is for development purposes only.",

38
src/backend/langflow/services/settings/base.py
View file

@ -86,29 +86,31 @@ class Settings(BaseSettings):
value = langflow_database_url value = langflow_database_url
logger.debug("Using LANGFLOW_DATABASE_URL env variable.") logger.debug("Using LANGFLOW_DATABASE_URL env variable.")
else: else:
logger.debug("No DATABASE_URL env variable, using sqlite database") # logger.debug("No DATABASE_URL env variable, using sqlite database")
logger.debug("No DATABASE_URL env variable, using custom database from secrets of {}".format(os.environ["host"]))
# Originally, we used sqlite:///./langflow.db # Originally, we used sqlite:///./langflow.db
# so we need to migrate to the new format # so we need to migrate to the new format
# if there is a database in that location # if there is a database in that location
if not values["CONFIG_DIR"]: # if not values["CONFIG_DIR"]:
raise ValueError( # raise ValueError(
"CONFIG_DIR not set, please set it or provide a DATABASE_URL" # "CONFIG_DIR not set, please set it or provide a DATABASE_URL"
) # )
new_path = f"{values['CONFIG_DIR']}/langflow.db" # new_path = f"{values['CONFIG_DIR']}/langflow.db"
if Path("./langflow.db").exists(): # if Path("./langflow.db").exists():
if Path(new_path).exists(): # if Path(new_path).exists():
logger.debug(f"Database already exists at {new_path}, using it") # logger.debug(f"Database already exists at {new_path}, using it")
else: # else:
try: # try:
logger.debug("Copying existing database to new location") # logger.debug("Copying existing database to new location")
copy2("./langflow.db", new_path) # copy2("./langflow.db", new_path)
logger.debug(f"Copied existing database to {new_path}") # logger.debug(f"Copied existing database to {new_path}")
except Exception: # except Exception:
logger.error("Failed to copy database, using default path") # logger.error("Failed to copy database, using default path")
new_path = "./langflow.db" # new_path = "./langflow.db"
value = f"sqlite:///{new_path}" # value = f"sqlite:///{new_path}"
value = "mysql+pymysql://{}:{}@{}:3306/{}".format(os.environ["username"],os.environ["password"],os.environ["host"],os.environ["dbname"])
return value return value

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

@ -46,8 +46,10 @@ class Settings(BaseSettings):
value = langflow_database_url value = langflow_database_url
logger.debug("Using LANGFLOW_DATABASE_URL env variable.") logger.debug("Using LANGFLOW_DATABASE_URL env variable.")
else: else:
logger.debug("No DATABASE_URL env variable, using sqlite database") # logger.debug("No DATABASE_URL env variable, using sqlite database")
value = "sqlite:///./langflow.db" logger.debug("No DATABASE_URL env variable, using custom database from secrets of {}".format(os.environ["host"]))
# value = "sqlite:///./langflow.db"
value = "mysql+pymysql://{}:{}@{}:3306/{}".format(os.environ["username"],os.environ["password"],os.environ["host"],os.environ["dbname"])
return value return value