Daksh Chaudhary
commited on
Update src/agents/nl_sql_agent.py
Browse files
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)
|