feat:更新智能体架构
This commit is contained in:
31
config/db.py
31
config/db.py
@@ -1,31 +0,0 @@
|
||||
import os
|
||||
|
||||
from dotenv import load_dotenv
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy import text
|
||||
|
||||
load_dotenv()
|
||||
|
||||
engine = create_engine(
|
||||
os.getenv("DB_URL"),
|
||||
pool_pre_ping=True,
|
||||
)
|
||||
|
||||
conn = engine.connect()
|
||||
|
||||
|
||||
def is_select(sql: str) -> bool:
|
||||
return sql.lower().lstrip().startswith("select")
|
||||
|
||||
|
||||
def execute_sql(sql: str):
|
||||
if not is_select(sql):
|
||||
raise ValueError("只允许 SELECT 查询")
|
||||
|
||||
result = conn.execute(text(sql))
|
||||
rows = result.fetchall()
|
||||
columns = result.keys()
|
||||
return {
|
||||
"columns": list(columns),
|
||||
"rows": rows
|
||||
}
|
||||
Reference in New Issue
Block a user