SQL Agent

1. Uses Agentic RAG to search for table metadata, sample queries and rules for writing better SQL queries.
2. Uses dynamic few-shot examples and rules to improve query construction.
3. Provides an interactive Streamlit UI for users to query the database.

​

Key capabilities

• Natural language to SQL conversion
• Retrieve table metadata, sample queries and rules using Agentic RAG
• Better query construction with the help of dynamic few-shot examples and rules
• Interactive Streamlit UI

​

Simple queries to try

• “Who are the top 5 drivers with the most race wins?”
• “Compare Mercedes vs Ferrari performance in constructors championships”
• “Show me the progression of fastest lap times at Monza”
• “Which drivers have won championships with multiple teams?”
• “What tracks have hosted the most races?”
• “Show me Lewis Hamilton’s win percentage by season”

​

Advanced queries with table joins

• “How many races did the championship winners win each year?”
• “Compare the number of race wins vs championship positions for constructors in 2019”
• “Show me Lewis Hamilton’s race wins and championship positions by year”
• “Which drivers have both won races and set fastest laps at Monaco?”
• “Show me Ferrari’s race wins and constructor championship positions from 2015-2020”

​

Code

​

Usage

git clone https://github.com/agno-agi/agno.git
cd agno

python3 -m venv .venv
source .venv/bin/activate

pip install -r cookbook/examples/streamlit_apps/sql_agent/requirements.txt

./cookbook/scripts/run_pgvector.sh

docker run -d \
  -e POSTGRES_DB=ai \
  -e POSTGRES_USER=ai \
  -e POSTGRES_PASSWORD=ai \
  -e PGDATA=/var/lib/postgresql/data/pgdata \
  -v pgvolume:/var/lib/postgresql/data \
  -p 5532:5432 \
  --name pgvector \
  agnohq/pgvector:16

python cookbook/examples/streamlit_apps/sql_agent/load_f1_data.py

python cookbook/examples/streamlit_apps/sql_agent/load_knowledge.py

# Required
export OPENAI_API_KEY=***

# Optional
export ANTHROPIC_API_KEY=***
export GOOGLE_API_KEY=***
export GROQ_API_KEY=***

streamlit run cookbook/examples/streamlit_apps/sql_agent/app.py