vanna 0.6.1__tar.gz → 0.6.2__tar.gz

{vanna-0.6.1 → vanna-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vanna
-Version: 0.6.1
+Version: 0.6.2
 Summary: Generate SQL queries from natural language
 Author-email: Zain Hoda <zain@vanna.ai>
 Requires-Python: >=3.9

{vanna-0.6.1 → vanna-0.6.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "vanna"
-version = "0.6.1"
+version = "0.6.2"
 authors = [
   { name="Zain Hoda", email="zain@vanna.ai" },
 ]

{vanna-0.6.1 → vanna-0.6.2}/src/vanna/flask/__init__.py

@@ -494,15 +494,12 @@ class VannaFlaskApp:
         def generate_plotly_figure(user: any, id: str, df, question, sql):
             chart_instructions = flask.request.args.get('chart_instructions')
 
-            if chart_instructions is not None:
-                question = f"{question}. When generating the chart, use these special instructions: {chart_instructions}"
-
             try:
                 # If chart_instructions is not set then attempt to retrieve the code from the cache
                 if chart_instructions is None or len(chart_instructions) == 0:
                     code = self.cache.get(id=id, field="plotly_code")
-
-                if code is None:
+                else:
+                    question = f"{question}. When generating the chart, use these special instructions: {chart_instructions}"
                     code = vn.generate_plotly_code(
                         question=question,
                         sql=sql,