text2sql-demo / src /app /executor.py
minimew's picture
Upload src/app/executor.py with huggingface_hub
e3492a8 verified
Raw
History Blame Contribute Delete
305 Bytes
"""Executor -- chay SQL (da qua sql_validator) tren SQLite, tra ve pandas DataFrame."""
import sqlite3
import pandas as pd
def run_query(db_path: str, sql: str) -> pd.DataFrame:
conn = sqlite3.connect(db_path)
try:
return pd.read_sql_query(sql, conn)
finally:
conn.close()