開始使用免費開始

使用 SQL 代理查詢烘焙銷售資料

Databricks 中 samples catalog 裡的「bakehouse」schema 包含為教學目的而產製的烘焙銷售資料。我們可以建立一個 LangChain 的 Databricks SQL 代理來連線到這個資料集，然後向代理提出與資料相關的問題。代理會用 SQL 取得答案，並且「展示解題過程」，說明為了得到解答所使用的 SQL 查詢與推理邏輯。

本練習屬於課程

使用 Python SDK 的 Databricks

檢視課程

練習說明

  • 從 samples catalog 中的 bakehouse 資料集建立一個 LangChain SQL 資料庫。
  • 建立一個連線到該 LangChain SQL 資料庫與 llm 的 SQLDatabaseToolkit 物件。
  • 建立一個同時連線到 llm 與 toolkit 的 LangChain SQL Agent。
  • 詢問 Databricks SQL 代理：哪一個產品的營收最高。

動手互動練習

試著完成這個範例程式碼，體驗一下這個練習。

# Connect to the bakehouse schema
bakehouse_db = SQLDatabase.from_databricks(
    ____="samples", 
    schema="bakehouse",  
  	warehouse_id=warehouse_id)

# Build the SQL toolkit
toolkit = SQLDatabaseToolkit(____=bakehouse_db, llm=llm)
# Create the SQL agent
agent = ____(llm=llm, toolkit=toolkit, verbose=True, handle_parsing_errors=True)

# Run the revenue query
prompt = "Which product generates the most revenue?"
result = agent.run(____)
print(result)
編輯並執行程式碼