批量添加ddl到向量数据库,添加documention,重写generate_rewritten_question

添加id,sql,question到上下文
This commit is contained in:
yujj128
2025-09-23 20:06:32 +08:00
parent 1393b021eb
commit 23916fc328
3 changed files with 16 additions and 16 deletions

View File

@@ -55,8 +55,7 @@ from vanna.flask import VannaFlaskApp
vn = create_vana()
app = VannaFlaskApp(vn,chart=False)
init_vn(vn)
cache = app.cache
@app.flask_app.route("/api/v0/generate_sql_2", methods=["GET"])
def generate_sql_2():
"""
@@ -87,10 +86,14 @@ def generate_sql_2():
if question is None:
return jsonify({"type": "error", "error": "No question provided"})
#id = self.cache.generate_id(question=question)
id = cache.generate_id(question=question)
data = vn.generate_sql_2(question=question)
data['id'] =id
sql = data["resp"]["sql"]
print("sql:",sql)
cache.set(id=id, field="question", value=question)
cache.set(id=id, field="sql", value=sql)
print("data---------------------------",data)
return jsonify(data)