adding minor changes

src/agents/basic_agents/agent1.py

@@ -1,6 +1,4 @@
-import os
 import time
-import json
 import re
 import numpy as np
 import pandas as pd
@@ -9,26 +7,6 @@
 from langchain_community.utilities import SQLDatabase
 from dotenv import load_dotenv
 
-class Config:
-    """SQL agent config"""
-    def __init__(
-        self,
-        env_fpath,
-        sql_db_fpath,
-        openai_model = 'gpt-4-1106-preview',
-        temperature  = 0.7
-        ):
-        load_dotenv(env_fpath)
-        openai_api_key = os.getenv("OPENAI_API_KEY")
-
-        self.llm = ChatOpenAI(
-            model=openai_model,
-            temperature=temperature,
-            api_key=openai_api_key
-            )
-        self.db = SQLDatabase.from_uri(f"sqlite:///{sql_db_fpath}")
-        self.engine = create_engine(f"sqlite:///{sql_db_fpath}")
-
 def get_schema_context(config):
     db = config.db
     tables = db.get_usable_table_names()
@@ -88,6 +66,7 @@ def execute_sql1(user_query, config):
     sql_query = extract_query(config.llm.invoke(sql_prompt).content)
     print(f'SQL Query: {sql_query}')
     df = pd.read_sql(sql_query, config.engine)
+    print(f'Query Results:\n{df.to_string()}')
     return df
 
 def execute_python1(user_query, df, config):

src/agents/basic_agents/agent2.py

@@ -1,4 +1,5 @@
 import os
+
 import time
 import json
 import re
@@ -117,6 +118,7 @@ def execute_sql2(sql_prompt, config):
     sql_query = extract_query(config.llm.invoke(sql_prompt).content)
     print(f'SQL Query: {sql_query}')
     df = pd.read_sql(sql_query, config.engine)
+    print(f'Query Results:\n{df.to_string()}')
     return df 
 
 def get_python_prompt2(user_query, df, config):

src/agents/basic_agents/agent3.py

@@ -154,7 +154,9 @@ def compile_sql(sql_json: dict, config) -> str:
 FROM {from_clause}'''.strip()
     print(f'=' * 30)
     print(f'SQL Query: {query}')
-    return pd.read_sql(query, config.engine)
+    df = pd.read_sql(query, config.engine)
+    print(f'Query Results:\n{df.to_string()}')
+    return df
 
 def get_python_prompt3(user_query, df, config):
     prompt = f''' 

src/agents/basic_agents/config.py

@@ -27,7 +27,7 @@ def __init__(
 def get_paths():
     """Get the paths for environment and database files."""
     SRC_PATH = os.path.dirname(os.getcwd())
-    ENV_FPATH = f'{SRC_PATH}/keys.env'
-    SQL_DB_FPATH = f'{SRC_PATH}/synthetic_data.db'
+    ENV_FPATH = os.path.join(SRC_PATH, 'agents', 'keys.env')
+    SQL_DB_FPATH = os.path.join(SRC_PATH, 'src', 'synthetic_data.db')
 
     return ENV_FPATH, SQL_DB_FPATH

src/agents/python_workflow.py

@@ -228,7 +228,7 @@ def fetch_data(self, state: State) -> Dict:
 
             if self.verbose:
                 self.format_print("Data Retrieved", f"\nShape: {df.shape}\nColumns: {df.columns.tolist()}", is_result=True)
-                self.format_print("Sample Data", f"\n{df.head()}", is_result=True)
+                self.format_print("Sample Data", f"\n{df.to_string()}", is_result=True)
 
             return {
                 "user_question": state['user_question'],