text2sql-rag / utils.py
fahmiaziz's picture
Update utils.py
61e83c9 verified
raw
history blame contribute delete
489 Bytes
import sqlite3
import pandas as pd
def execute_query_and_return_df(query):
"""
Executes a SQL query and returns the results as a Pandas DataFrame.
Args:
query: The SQL query to execute.
Returns:
A Pandas DataFrame containing the query results.
"""
conn = sqlite3.connect("Chinook_Sqlite.sqlite")
try:
df = pd.read_sql_query(query, conn)
return df
except Exception as e:
print(f"Error executing query: {e}")
return None
finally:
conn.close()