diff --git a/app.py b/app.py index 8bde50e..6a4094f 100644 --- a/app.py +++ b/app.py @@ -170,7 +170,7 @@ def add_training_data(): @app.route('/api/v0/generate_followup_questions', methods=['GET']) @requires_cache(['df', 'question', 'sql']) -def generate_followup_questions(id: str, df, sql, question): +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)