From 2b2d8f3a870ca3fa76a90affb4db8931ec0a3f81 Mon Sep 17 00:00:00 2001
From: nsxshota <nsxshota@amazon.co.jp>
Date: Thu, 23 Nov 2023 11:24:56 +0900
Subject: [PATCH] modify LANGFLOW_DATABASE_URL

---
 deploy/scripts/cdk/lib/construct/backend.ts |  6 +++---
 src/backend/langflow/__main__.py            | 11 ++++++-----
 2 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/deploy/scripts/cdk/lib/construct/backend.ts b/deploy/scripts/cdk/lib/construct/backend.ts
index b22ba62a8..26100d711 100644
--- a/deploy/scripts/cdk/lib/construct/backend.ts
+++ b/deploy/scripts/cdk/lib/construct/backend.ts
@@ -57,9 +57,9 @@ export class BackEndCluster extends Construct {
       }),
       environment:{
         // user:pass@endpoint:port/dbname
-        // "LANGFLOW_DATABASE_URL" : `mysql+pymysql://${ecs.Secret.fromSecretsManager(secretsDB, 'username')}:${ecs.Secret.fromSecretsManager(secretsDB, 'password')}@${ecs.Secret.fromSecretsManager(secretsDB, 'host')}:3306/${ecs.Secret.fromSecretsManager(secretsDB, 'dbname')}`,
-        "LANGFLOW_DATABASE_URL" : "sqlite:///./langflow.db",
-        "LANGFLOW_LANGCHAIN_CACHE" : "SQLiteCache",
+        // "LANGFLOW_DATABASE_URL" : `mysql+pymysql://${username}:${password}@${host}:3306/${dbname}`,
+        // "LANGFLOW_DATABASE_URL" : "sqlite:///./langflow.db",
+        // "LANGFLOW_LANGCHAIN_CACHE" : "SQLiteCache",
         "LANGFLOW_AUTO_LOGIN" : "false",
         "LANGFLOW_SUPERUSER" : "admin",
         "LANGFLOW_SUPERUSER_PASSWORD" : "1234567"
diff --git a/src/backend/langflow/__main__.py b/src/backend/langflow/__main__.py
index 1bdcf97f2..5665eb409 100644
--- a/src/backend/langflow/__main__.py
+++ b/src/backend/langflow/__main__.py
@@ -1,6 +1,7 @@
 import platform
 import socket
 import sys
+import os
 import time
 import webbrowser
 from pathlib import Path
@@ -128,11 +129,11 @@ def run(
     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
     # and works with Pydantic
-    # database_url: str = typer.Option(
-    #     None,
-    #     help="Database URL to connect to. If not provided, a local SQLite database will be used.",
-    #     envvar="LANGFLOW_DATABASE_URL",
-    # ),
+    database_url: str = typer.Option(
+        "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.",
+        envvar="LANGFLOW_DATABASE_URL",
+    ),
     path: str = typer.Option(
         None,
         help="Path to the frontend directory containing build files. This is for development purposes only.",