Caching Snowflake queries in Streamlit

caching.py

def get_pandas_from_sql(sql: str) -> pd.DataFrame:
    @st.cache_data(ttl=TTL)
    def get_df(sql):
        dataframe = session.sql(sql).to_pandas()
        return dataframe

    dataframe = get_df(sql)

    return dataframe
    
def get_pandas_from_snowpark(df: snowpark.DataFrame) -> pd.DataFrame:
    query = df._plan.queries[0].sql # Get the SQL that will be run when .to_pandas() is called

    @st.cache_data(ttl=TTL)
    def get_df(
        _df: snowpark.DataFrame,
        query_str: str,
    ) -> pd.DataFrame:
        return _df.to_pandas()

    pd_df = get_df(_df=df, query_str=query)

    return pd_df