Merge pull request #208 from danielcorin/dc/return-results

Return sql and dataframe if call is successful
vanna-ai · Jan 31, 2024 · 8c5e05a · 8c5e05a
2 parents aa432ab + ef02500
commit 8c5e05a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py
@@ -607,6 +607,7 @@ def ask(
                 return None
             else:
                 return sql, None, None
+        return sql, df, None
 
     def train(
         self,