prthm11 commited on
Commit
8ef5fa9
·
verified ·
1 Parent(s): 3a58b7a

Update gemini_sql_agent.py

Browse files
Files changed (1) hide show
  1. 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: