Merge pull request #19 from CADLabs/fix_st_print_return

Attempt at fixing st_print_return args

Author: BenSchZA

analytics_bot_langchain/agents/agent_toolkits/bigquery/base.py

@@ -47,10 +47,10 @@ def query_post_processing(query: str) -> str:
         import plotly.graph_objects as go
         import pandas as pd
 
-        def st_print_return(value):
+        def st_print_return(*args):
             import streamlit as st
-            st.write(value)
-            return value
+            st.write(*args)
+            return args
         """)
         query = imports + "\n" + query
         query = re.sub(".*client =.*\n?", "client = bigquery_client", query)