{ "cells": [ { "cell_type": "markdown", "id": "4bf05298-718f-5dfe-a938-0ee07c63b4dd", "metadata": { "id": "4bf05298-718f-5dfe-a938-0ee07c63b4dd" }, "source": [ "# Generating SQL for SQLite using OpenAI, ChromaDB\n", "\n", "This notebook runs through the process of using the `vanna` Python package to generate SQL using AI (RAG + LLMs) including connecting to a database and training. If you're not ready to train on your own database, you can still try it using a sample [SQLite database](app.md).\n" ] }, { "cell_type": "markdown", "id": "453b54b4-c4b0-53f1-9f69-7e0199beb21b", "metadata": { "id": "453b54b4-c4b0-53f1-9f69-7e0199beb21b" }, "source": [ "
\n", " | id | \n", "question | \n", "content | \n", "training_data_type | \n", "
---|---|---|---|---|
0 | \n", "4dfd73a9-4462-5a23-8b28-b9d5c7125b49-sql | \n", "What are the details for the person named John... | \n", "SELECT * FROM my-table WHERE name = 'John Doe' | \n", "sql | \n", "
0 | \n", "6493e730-8fd5-5c9b-ba80-d991dcf0c4e9-ddl | \n", "None | \n", "CREATE TABLE Drivers (\\n driver... | \n", "ddl | \n", "
1 | \n", "894692ac-1600-54b6-af13-4a9ff62c6e32-ddl | \n", "None | \n", "CREATE TABLE Tracks (\\n track_i... | \n", "ddl | \n", "
2 | \n", "7258dd79-d9f7-5a33-86c9-c8cfe10e7967-ddl | \n", "None | \n", "CREATE TABLE Event (\\n event_id... | \n", "ddl | \n", "
3 | \n", "64a28e8b-3faf-5ea0-8867-3e1e92958607-ddl | \n", "None | \n", "CREATE TABLE Sessions (\\n sessi... | \n", "ddl | \n", "
4 | \n", "feb0cd7c-e372-50a8-a10a-736c42386360-ddl | \n", "None | \n", "CREATE TABLE Weather (\\n weathe... | \n", "ddl | \n", "
5 | \n", "0f9d7dec-f228-53a0-879b-26a7918f177b-ddl | \n", "None | \n", "CREATE TABLE Laps (\\n lap_id IN... | \n", "ddl | \n", "
6 | \n", "ae41976c-72dd-549c-a4d5-327db48beadb-ddl | \n", "None | \n", "CREATE TABLE Telemetry (\\n tele... | \n", "ddl | \n", "
7 | \n", "bb13e8f0-6934-52fd-bbb6-62ab4a40b8b0-ddl | \n", "None | \n", "CREATE INDEX idx_laps_driver_name ON Laps(driv... | \n", "ddl | \n", "
8 | \n", "d0efbc31-d1e0-54f0-a19c-5905d905d9d3-ddl | \n", "None | \n", "CREATE INDEX idx_laps_session_id ON Laps(sessi... | \n", "ddl | \n", "
9 | \n", "db5f7760-144e-5d1d-ae5d-8f699e2d60a4-ddl | \n", "None | \n", "CREATE INDEX idx_telemetry_lap_id ON Telemetry... | \n", "ddl | \n", "
10 | \n", "892655e2-8e10-5a00-b97f-c5916038f464-ddl | \n", "None | \n", "CREATE INDEX idx_telemetry_datetime ON Telemet... | \n", "ddl | \n", "
11 | \n", "9caa16be-95d6-5695-8091-b2bee5bd103b-ddl | \n", "None | \n", "CREATE INDEX idx_weather_session_id ON Weather... | \n", "ddl | \n", "
12 | \n", "3c897036-893c-5975-9e75-fa9858906478-ddl | \n", "None | \n", "CREATE INDEX idx_weather_datetime ON Weather(d... | \n", "ddl | \n", "
13 | \n", "5dacfa83-1fbd-578b-afc0-11571ca5b52f-ddl | \n", "None | \n", "CREATE INDEX idx_event_date ON Event(event_date) | \n", "ddl | \n", "
14 | \n", "b8e57921-9fe3-5d95-b480-67776aaebe07-ddl | \n", "None | \n", "CREATE VIEW DriverPerformanceSummaryWithWeathe... | \n", "ddl | \n", "
15 | \n", "16e24e76-db3f-5495-9943-6bd4d08ba03a-ddl | \n", "None | \n", "CREATE VIEW TyrePerformanceAnalysisWithWeather... | \n", "ddl | \n", "
16 | \n", "874e4eac-9c02-5047-8657-d148b125b702-ddl | \n", "None | \n", "CREATE VIEW WeatherImpactAnalysis AS\\n ... | \n", "ddl | \n", "
17 | \n", "fd528edd-cdd1-5f3d-b7e8-eae03f073560-ddl | \n", "None | \n", "CREATE VIEW EventPerformanceOverview AS\\n ... | \n", "ddl | \n", "
18 | \n", "ce0a1604-e717-5bb2-921a-abedc453f13c-ddl | \n", "None | \n", "CREATE VIEW TelemetryAnalysisWithWeather AS\\n ... | \n", "ddl | \n", "
19 | \n", "ac7cedb2-8062-5ba7-b596-dea4c331fb9a-ddl | \n", "None | \n", "\\n CREATE VIEW IF NOT EXISTS DriverPerforma... | \n", "ddl | \n", "
0 | \n", "51cf1d6d-7637-5b87-b9e7-31c577fbde59-doc | \n", "None | \n", "Our business defines OTIF score as the percent... | \n", "documentation | \n", "
\n", " | id | \n", "question | \n", "content | \n", "training_data_type | \n", "
---|---|---|---|---|
0 | \n", "4dfd73a9-4462-5a23-8b28-b9d5c7125b49-sql | \n", "What are the details for the person named John... | \n", "SELECT * FROM my-table WHERE name = 'John Doe' | \n", "sql | \n", "
0 | \n", "6493e730-8fd5-5c9b-ba80-d991dcf0c4e9-ddl | \n", "None | \n", "CREATE TABLE Drivers (\\n driver... | \n", "ddl | \n", "
1 | \n", "894692ac-1600-54b6-af13-4a9ff62c6e32-ddl | \n", "None | \n", "CREATE TABLE Tracks (\\n track_i... | \n", "ddl | \n", "
2 | \n", "7258dd79-d9f7-5a33-86c9-c8cfe10e7967-ddl | \n", "None | \n", "CREATE TABLE Event (\\n event_id... | \n", "ddl | \n", "
3 | \n", "64a28e8b-3faf-5ea0-8867-3e1e92958607-ddl | \n", "None | \n", "CREATE TABLE Sessions (\\n sessi... | \n", "ddl | \n", "
4 | \n", "feb0cd7c-e372-50a8-a10a-736c42386360-ddl | \n", "None | \n", "CREATE TABLE Weather (\\n weathe... | \n", "ddl | \n", "
5 | \n", "0f9d7dec-f228-53a0-879b-26a7918f177b-ddl | \n", "None | \n", "CREATE TABLE Laps (\\n lap_id IN... | \n", "ddl | \n", "
6 | \n", "ae41976c-72dd-549c-a4d5-327db48beadb-ddl | \n", "None | \n", "CREATE TABLE Telemetry (\\n tele... | \n", "ddl | \n", "
7 | \n", "bb13e8f0-6934-52fd-bbb6-62ab4a40b8b0-ddl | \n", "None | \n", "CREATE INDEX idx_laps_driver_name ON Laps(driv... | \n", "ddl | \n", "
8 | \n", "d0efbc31-d1e0-54f0-a19c-5905d905d9d3-ddl | \n", "None | \n", "CREATE INDEX idx_laps_session_id ON Laps(sessi... | \n", "ddl | \n", "
9 | \n", "db5f7760-144e-5d1d-ae5d-8f699e2d60a4-ddl | \n", "None | \n", "CREATE INDEX idx_telemetry_lap_id ON Telemetry... | \n", "ddl | \n", "
10 | \n", "892655e2-8e10-5a00-b97f-c5916038f464-ddl | \n", "None | \n", "CREATE INDEX idx_telemetry_datetime ON Telemet... | \n", "ddl | \n", "
11 | \n", "9caa16be-95d6-5695-8091-b2bee5bd103b-ddl | \n", "None | \n", "CREATE INDEX idx_weather_session_id ON Weather... | \n", "ddl | \n", "
12 | \n", "3c897036-893c-5975-9e75-fa9858906478-ddl | \n", "None | \n", "CREATE INDEX idx_weather_datetime ON Weather(d... | \n", "ddl | \n", "
13 | \n", "5dacfa83-1fbd-578b-afc0-11571ca5b52f-ddl | \n", "None | \n", "CREATE INDEX idx_event_date ON Event(event_date) | \n", "ddl | \n", "
14 | \n", "b8e57921-9fe3-5d95-b480-67776aaebe07-ddl | \n", "None | \n", "CREATE VIEW DriverPerformanceSummaryWithWeathe... | \n", "ddl | \n", "
15 | \n", "16e24e76-db3f-5495-9943-6bd4d08ba03a-ddl | \n", "None | \n", "CREATE VIEW TyrePerformanceAnalysisWithWeather... | \n", "ddl | \n", "
16 | \n", "874e4eac-9c02-5047-8657-d148b125b702-ddl | \n", "None | \n", "CREATE VIEW WeatherImpactAnalysis AS\\n ... | \n", "ddl | \n", "
17 | \n", "fd528edd-cdd1-5f3d-b7e8-eae03f073560-ddl | \n", "None | \n", "CREATE VIEW EventPerformanceOverview AS\\n ... | \n", "ddl | \n", "
18 | \n", "ce0a1604-e717-5bb2-921a-abedc453f13c-ddl | \n", "None | \n", "CREATE VIEW TelemetryAnalysisWithWeather AS\\n ... | \n", "ddl | \n", "
19 | \n", "ac7cedb2-8062-5ba7-b596-dea4c331fb9a-ddl | \n", "None | \n", "\\n CREATE VIEW IF NOT EXISTS DriverPerforma... | \n", "ddl | \n", "