批量添加ddl到向量数据库,添加documention,重写generate_rewritten_question
添加id,sql,question到上下文
This commit is contained in:
		| @@ -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) | ||||
|  | ||||
|   | ||||
		Reference in New Issue
	
	Block a user
	 yujj128
					yujj128