Kshitijk20 commited on
Commit
c60ebf3
·
1 Parent(s): 4db8795

refactor: update import statement for DatabaseTools

Browse files
Files changed (1) hide show
  1. app/agents/sql_agent.py +1 -1
app/agents/sql_agent.py CHANGED
@@ -353,7 +353,7 @@ from IPython.display import display, Image
353
  from langchain_google_genai import ChatGoogleGenerativeAI
354
  # from app.tools.database_tools import DatabaseTools
355
  from app.utils.database_connection import DatabaseConnection
356
- from app.tools.database_tools_v2 import DatabaseTools
357
 
358
  load_dotenv()
359
  import os
 
353
  from langchain_google_genai import ChatGoogleGenerativeAI
354
  # from app.tools.database_tools import DatabaseTools
355
  from app.utils.database_connection import DatabaseConnection
356
+ from app.tools.database_tools import DatabaseTools
357
 
358
  load_dotenv()
359
  import os