提示词优化,迟到,请假

This commit is contained in:
yujj128
2025-11-01 10:16:06 +08:00
parent e72b24e7f7
commit 78fb9f4d54
7 changed files with 498 additions and 111 deletions

View File

@@ -57,7 +57,7 @@ def create_vana():
def init_vn(vn):
logger.info("--------------init vana-----connect to datasouce db----")
connect_database(vn)
# connect_database(vn)
load_ddl_doc.add_ddl(vn)
load_ddl_doc.add_documentation(vn)
if config('IS_FIRST_LOAD', default=False, cast=bool):
@@ -98,7 +98,6 @@ def generate_sql_2():
"""
logger.info("Start to generate sql in main")
question = flask.request.args.get("question")
if question is None:
return jsonify({"type": "error", "error": "No question provided"})
try:
@@ -185,6 +184,7 @@ def run_sql_2(id: str, sql: str):
"""
logger.info("Start to run sql in main")
try:
user_id = request.args.get("user_id")
if not vn.run_sql_is_set:
return jsonify(
{
@@ -198,7 +198,7 @@ def run_sql_2(id: str, sql: str):
# print(df_count,"is type",type(df_count))
# total_count = df_count.to_dict(orient="records")[0]["total_count"]
# logger.info("Total count is {0}".format(total_count))
df = vn.run_sql(sql=sql)
df = vn.run_sql_2(sql=sql)
result = df.to_dict(orient='records')
logger.info("df ---------------{0} {1}".format(result,type(result)))
return jsonify(