Поговорите с Sql: я хочу показать таблицу результатов запроса в виде фрейма данных в моем веб-приложении, а также постро ⇐ Python

[Anonymous]

Код генерирует запрос SQL, выполняет его и сохраняет результат в переменной с именем result. Я хочу отобразить этот результат в виде фрейма данных, включая имена столбцов из исходного запроса. Я использую последнюю версию каждой библиотеки.

Код: Выделить всё

#To load the database
os.environ["OPENAI_API_KEY"]=os.getenv("OPENAI_API_KEY")

def load_db(db_name):
db=SQLDatabase.from_uri(f"sqlite:///Chinook.db")
return db

# to create sql query chain to execute queries
def chain_create(db):
llm = ChatOpenAI(model="gpt-3.5-turbo")
chain = create_sql_query_chain(llm, db)
return chain

# To create another chain to answer the queries based on the data retrieved by querying

def sql_infer(db,chain,user_question):
sql_query=chain.invoke({"question":user_question})
result=db.run(sql_query)
print(result)
print(type(result))

st.code(sql_query)
st.write(result)

answer_prompt=PromptTemplate.from_template(
"""Given the following user question, corresponding SQL query, and SQL result,
generate a proper reply to give to user

Question:{question}
SQL Query:{query}
SQL Result:{result}
Answer:"""
)
llm_model=ChatOpenAI(model="gpt-3.5-turbo")
llm=LLMChain(llm=llm_model,prompt=answer_prompt)
ans=llm(inputs={"question":user_question,"query":sql_query,"result":result})
return ans["text"]

Может ли кто-нибудь подсказать мне, как мне следует к этому подойти?

Подробнее здесь: https://stackoverflow.com/questions/792 ... rame-in-my