Databricks Query Bot

Summary

A chatbot that lets you ask questions about Databricks tables in plain English. A LangChain SQL agent reads the schema, writes the SQL with an OpenAI model (GPT-4o by default), runs it on a Databricks SQL warehouse, and answers in natural language.

Highlights

• Natural-language questions answered with generated SQL
• LangChain SQLDatabase toolkit exposes the table schema to the agent
• SQLAlchemy and the Databricks SQL connector, set up in one shared connection module
• Simple command-line chat loop