From 974ecedff596dd63c764335ba38e62402dee5e60 Mon Sep 17 00:00:00 2001
From: huchengyi <xtuhcy@163.com>
Date: Sun, 12 May 2024 19:02:22 +0800
Subject: [PATCH] Fixed a bug where the ask function did not return fig

---
 src/vanna/base/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py
index c13489b7..d7e1f487 100644
--- a/src/vanna/base/base.py
+++ b/src/vanna/base/base.py
@@ -1701,7 +1701,7 @@ def ask(
                 return None
             else:
                 return sql, None, None
-        return sql, df, None
+        return sql, df, fig
 
     def train(
         self,