Merge pull request vanna-ai#3 from sthakur369/main

Addressing Missing 'sql' Parameter in generate_followup_questions Method
avinashkurup · Mar 5, 2024 · c2e6e49 · c2e6e49
2 parents 1984f65 + e3c659f
commit c2e6e49
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/app.py b/app.py
@@ -169,9 +169,9 @@ def add_training_data():
         return jsonify({"type": "error", "error": str(e)})
 
 @app.route('/api/v0/generate_followup_questions', methods=['GET'])
-@requires_cache(['df', 'question'])
-def generate_followup_questions(id: str, df, question):
-    followup_questions = vn.generate_followup_questions(question=question, df=df)
+@requires_cache(['df', 'question', 'sql'])
+def generate_followup_questions(id: str, df, question, sql):
+    followup_questions = vn.generate_followup_questions(question=question, sql=sql, df=df)
 
     cache.set(id=id, field='followup_questions', value=followup_questions)