Merge pull request #470 from Molrn/fix-required-fig

fix(flask): missing fig_json error
vanna-ai · Jun 7, 2024 · fdcb93d · fdcb93d
2 parents ee9727f + eaf6b46
commit fdcb93d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/vanna/flask/__init__.py b/src/vanna/flask/__init__.py
@@ -688,8 +688,8 @@ def generate_summary(user: any, id: str, df, question):
         @self.flask_app.route("/api/v0/load_question", methods=["GET"])
         @self.requires_auth
         @self.requires_cache(
-            ["question", "sql", "df", "fig_json"],
-            optional_fields=["summary"]
+            ["question", "sql", "df"],
+            optional_fields=["summary", "fig_json"]
         )
         def load_question(user: any, id: str, question, sql, df, fig_json, summary):
             try: