MulaiMulai sekarang secara gratis

Query bakery sales data using SQL agent

The "bakehouse" schema in the samples catalog in Databricks contains fabricated data about bakery sales. We can create a LangChain Databricks SQL agent to connect to this dataset and then ask the agent queries about the data. The agent can use SQL to get the answers and can show it's work, describing the SQL queries and logic used to get to the solution.

Latihan ini adalah bagian dari kursus

Databricks with the Python SDK

Lihat Kursus

Petunjuk latihan

  • Create a LangChain SQL database from the bakehouse dataset in the samples catalog.
  • Create a SQLDatabaseToolkit object that connects to the LangChain SQL database and llm.
  • Create a LangChain SQL Agent that connects to the llm and toolkit.
  • Ask the Databricks SQL agent which product generates the most revenue.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Create a LangChain SQL database from the `bakehouse` dataset in the `samples` catalog
bakehouse_db = SQLDatabase.from_databricks(
    ____="samples", 
    schema="bakehouse",  
  	warehouse_id=warehouse_id)

# Create a SQLDatabaseToolkit object that connects to the LangChain SQL database and llm
toolkit = SQLDatabaseToolkit(____=bakehouse_db, llm=llm)
# Create a LangChain SQL Agent that connects to the llm and toolkit
agent = ____(llm=llm, toolkit=toolkit, verbose=True, handle_parsing_errors=True)

# Ask the Databricks SQL agent which product generates the most revenue
prompt = "Which product generates the most revenue?"
result = agent.run(____)
print(result)
Edit dan Jalankan Kode