Update utility.py

utility.py

@@ -12,6 +12,7 @@ from pandasai.llm import OpenAI
 #from langchain.llms.sambanova import Sambaverse
 #from langchain_community.llms.sambanova import Sambaverse
 from langchain_community.chat_models.sambanova import ChatSambaNovaCloud
+import ast
 
 
 # Make API connection
@@ -282,7 +283,7 @@ def read_datalake(user_phone, user_question):
     sales_df = pd.DataFrame(sales_list)
     # return f"{inventory_df} \n\n {sales_df}"
 
-    lake = SmartDatalake([inventory_df, sales_df], config={"llm": llm, "response_parser": FlaskResponse, "enable_cache": False, "save_logs": False})
+    lake = SmartDatalake([inventory_df, sales_df], config={"llm": llm, "custom_whitelisted_dependencies":["ast"], "response_parser": FlaskResponse, "enable_cache": False, "save_logs": False})
     response = lake.chat(user_question)
 
     return response