GitHub user matebenyovszky created a discussion: Vambery AI Agent — AI-powered SQL assistant extension for SQL Lab (public beta)
## What is it? **Vambery AI Agent** is an AI-powered data assistant that lives inside SQL Lab as a sidebar panel. It uses Superset's native Extension System (Module Federation) — no core modifications needed. Ask questions in natural language, and the AI inspects your schema, writes SQL, executes queries, and creates charts — all without leaving SQL Lab. > Named after [Ármin > Vámbéry](https://en.wikipedia.org/wiki/%C3%81rmin_V%C3%A1mb%C3%A9ry) — the > Hungarian orientalist and polyglot who explored unknown territories. Like its > namesake, Vambery navigates your data landscape so you don't have to. ## Features - **Natural language → SQL** — describe what you need, the AI writes and executes the query - **Schema-aware** — automatically inspects databases, schemas, tables, columns, comments, descriptions, and predefined metrics - **Interactive chart creation** — creates bar, line, pie, and table charts from query results (preview or save) - **Auto-fill editor** — sets the final query in SQL Lab and auto-executes - **Markdown responses** — syntax-highlighted code blocks, tables, formatted text - **Streaming** — real-time SSE with visible tool-use steps - **Multiple LLM providers** — Azure OpenAI, OpenAI, OpenRouter - **Database support** — MSSQL, PostgreSQL, and any SQL database Superset connects to ## How it works The extension registers a right sidebar panel in SQL Lab. The AI agent has access to 8 tools: | Tool | What it does | |------|-------------| | `list_schemas` / `list_tables` | Browse database structure | | `get_table_columns` | Get columns with types, comments, descriptions, metrics | | `sample_table_data` | Preview rows | | `get_distinct_values` | Explore column values | | `execute_sql` | Run queries safely | | `set_editor_sql` | Set SQL in the editor + auto-execute | | `create_chart` | Create visualizations in Explore | ## Installation ```bash # Add as git submodule git submodule add https://github.com/integrityauthority/superset-extensions.git extensions bash extensions/build-extensions.sh ``` Add to `superset_config.py`: ```python FEATURE_FLAGS = {"ENABLE_EXTENSIONS": True} LOCAL_EXTENSIONS = ["/app/extensions/ai_assistant"] ``` Full docs: **https://github.com/integrityauthority/superset-extensions** ## Status **Public Beta** — functional and actively developed. We use it in production, but expect rough edges. This is (to our knowledge) the first community AI extension built using Superset's native Extension System. We'd love feedback, feature requests, and contributions. **GitHub:** https://github.com/integrityauthority/superset-extensions **Feature requests:** https://github.com/integrityauthority/superset-extensions/issues/new?labels=enhancement&template=feature_request.md Built by [Integrity Authority](https://github.com/integrityauthority) GitHub link: https://github.com/apache/superset/discussions/38356 ---- This is an automatically sent email for [email protected]. To unsubscribe, please send an email to: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
