Was macht vanna?
Open-Source KI-SQL-Agent. Trainieren Sie ein RAG-Modell auf Ihrem Schema und stellen Sie natürliche Sprachfragen, um sofortige Datenausgaben zu erhalten.
Beschreibung
Vanna AI ist ein Open-Source-Tool, das es Benutzern ermöglicht, mit Datenbanken über natürliche Sprache zu interagieren. Es verwendet ein Retrieval-Augmented Generation (RAG)-Modell, das auf Ihrem Datenbankschema trainiert wird. Nach dem Training können Sie Fragen in Alltagssprache stellen, und Vanna generiert die entsprechenden SQL-Abfragen, führt sie aus und gibt die Ergebnisse zurück.
Hauptmerkmale
- Natürliche Sprachschnittstelle: Stellen Sie Fragen wie „Wie viele Kunden haben im letzten Monat gekauft?“ und erhalten Sie sofortige Antworten.
- RAG-Modelltraining: Trainieren Sie das Modell auf Ihrem Schema, um kontextbezogene und genaue SQL-Abfragen zu generieren.
- Breite Datenbankunterstützung: Funktioniert mit PostgreSQL, MySQL, Snowflake, BigQuery, SQLite und mehr.
- Open Source: Frei verfügbar und anpassbar auf GitHub.
- Einfache Integration: Bietet Python-API und Jupyter-Notebook-Unterstützung.
Installation
pip install vanna
Schnellstart
import vanna
# Verbindung zur Datenbank herstellen
vanna.connect(
host='localhost',
database='meine_datenbank',
user='benutzer',
password='passwort'
)
# Modell trainieren
vanna.train(schema='mein_schema')
# Frage stellen
antwort = vanna.ask('Wie viele Bestellungen gab es im Januar 2023?')
print(antwort)
Verwendung in Jupyter Notebook
from vanna import VannaNotebook
nb = VannaNotebook()
nb.run()
Konfiguration
| Variable | Beschreibung | Standardwert |
|---|---|---|
VANNA_MODEL | Name des RAG-Modells | 'default' |
VANNA_DATABASE | Datenbanktyp | 'postgresql' |
VANNA_API_KEY | API-Schlüssel (optional) | None |
Lizenz
Dieses Projekt ist unter der MIT-Lizenz lizenziert – siehe die LICENSE-Datei für Details.