Daksh Chaudhary commited on
Commit
c480f4b
·
verified ·
1 Parent(s): cc5edd6

Update src/agents/nl_sql_agent.py

Browse files
Files changed (1) hide show
  1. src/agents/nl_sql_agent.py +3 -3
src/agents/nl_sql_agent.py CHANGED
@@ -3,9 +3,9 @@ import logging
3
  import sys
4
  from llama_index.llms.openai import OpenAI
5
  from llama_index.core.agent import ReActAgent
6
- from agent_tools.sql_executor_tool import get_sql_executor_tool
7
- from agent_tools.schema_retriever_tool import get_schema_retriever_tool
8
- from agent_models.models import get_finetuned_model
9
 
10
  # Configure logging for better visibility into agent's thought process
11
  logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
 
3
  import sys
4
  from llama_index.llms.openai import OpenAI
5
  from llama_index.core.agent import ReActAgent
6
+ from .agent_tools.sql_executor_tool import get_sql_executor_tool
7
+ from .agent_tools.schema_retriever_tool import get_schema_retriever_tool
8
+ from .agent_models.models import get_finetuned_model
9
 
10
  # Configure logging for better visibility into agent's thought process
11
  logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)