Update gemini_sql_agent.py
Browse files- gemini_sql_agent.py +2 -1
gemini_sql_agent.py
CHANGED
@@ -46,7 +46,8 @@ def init_agent(db_type):
|
|
46 |
if db_type == "sqlite":
|
47 |
db = SQLDatabase.from_uri("sqlite:///ecommerce_system_2.db")
|
48 |
elif db_type == "mysql":
|
49 |
-
db = SQLDatabase.from_uri("mysql+pymysql://root:@localhost:3306/employee_sys")
|
|
|
50 |
# db = SQLDatabase.from_uri("mysql+pymysql://root:root@mysql_service_name:3306/employee_sys")
|
51 |
|
52 |
else:
|
|
|
46 |
if db_type == "sqlite":
|
47 |
db = SQLDatabase.from_uri("sqlite:///ecommerce_system_2.db")
|
48 |
elif db_type == "mysql":
|
49 |
+
# db = SQLDatabase.from_uri("mysql+pymysql://root:@localhost:3306/employee_sys")
|
50 |
+
db = SQDATABASE_URI = "mysql+pymysql://root:root@mysql/employee_sys"
|
51 |
# db = SQLDatabase.from_uri("mysql+pymysql://root:root@mysql_service_name:3306/employee_sys")
|
52 |
|
53 |
else:
|