7K star！Text2SQL还不够？试试RAG2SQL的开源工具，用自然语言查数据

!pip install vannaimport vannafrom vanna.remote import VannaDefaultvn = VannaDefault(model='chinook', api_key=vanna.get_api_key('my-email@example.com'))vn.connect_to_sqlite('<https://vanna.ai/Chinook.sqlite>')vn.ask("What are the top 10 albums by sales?")from vanna.flask import VannaFlaskAppVannaFlaskApp(vn).run()

import vanna as vnimport streamlit as st

vn.set_api_key(st.secrets[ "vanna_api_key" ]) vn.set_model( 'chinook' ) vn.connect_to_sqlite( '<https://vanna.ai/Chinook.sqlite>' )

my_question = st.text_input( "问我一个可以转化为 SQL 的问题" , key= "my_question" )

sql  = vn.generate_sql(my_question) st.code( sql , language = 'sql' )

df = vn.run_sql(sql)     st.dataframe(df, use_container_width= True )

fig = vn.get_plotly_figure(plotly_code=vn.generate_plotly_code(question=my_question, sql=sql, df=df), df=df)st.plotly_chart(fig, use_container_width=True)

import vanna as vnimport streamlit as st
vn.set_api_key(st.secrets["vanna_api_key"])vn.set_model('chinook')vn.connect_to_sqlite('<https://vanna.ai/Chinook.sqlite>')
my_question = st.session_state.get("my_question", default=None)if my_question is None:    st.image("chinook-schema.png", use_column_width=True)    my_question = st.text_input("Ask me a question that I can turn into SQL", key="my_question")else:    st.title(my_question)    sql = vn.generate_sql(my_question)    st.code(sql, language='sql')    df = vn.run_sql(sql)        st.dataframe(df, use_container_width=True)    fig = vn.get_plotly_figure(plotly_code=vn.generate_plotly_code(question=my_question, sql=sql, df=df), df=df)    st.plotly_chart(fig, use_container_width=True)    st.button("Ask another question", on_click=lambda: st.session_state.clear())