How to beginner · 3 min read

How to use AI for SQL analysis

Q: How to use AI for SQL analysis

Use AI models like gpt-4o to analyze SQL by sending your query or database schema as input and asking for explanations, optimizations, or generation of SQL code. This can be done via API calls where the AI interprets SQL logic, suggests improvements, or generates queries based on natural language prompts.

Quick answer

Use AI models like gpt-4o to analyze SQL by sending your query or database schema as input and asking for explanations, optimizations, or generation of SQL code. This can be done via API calls where the AI interprets SQL logic, suggests improvements, or generates queries based on natural language prompts.

PREREQUISITES

Python 3.8+
OpenAI API key (free tier works)
pip install openai>=1.0

Setup

Install the OpenAI Python SDK and set your API key as an environment variable to authenticate requests.

bash

pip install openai>=1.0

Step by step

This example shows how to send a SQL query to gpt-4o for analysis and get an explanation or optimization suggestions.

python

import os
from openai import OpenAI

client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])

sql_query = "SELECT * FROM orders WHERE order_date > '2023-01-01'"

messages = [
    {"role": "user", "content": f"Explain and optimize this SQL query:\n{sql_query}"}
]

response = client.chat.completions.create(
    model="gpt-4o",
    messages=messages
)

print(response.choices[0].message.content)

output

The AI will return an explanation of the query logic and suggest optimizations such as adding indexes or rewriting the query for performance.

Common variations

You can use different prompts to generate SQL from natural language, analyze database schemas, or debug SQL errors. Also, try other models like claude-3-5-sonnet-20241022 for potentially better code understanding.

python

import os
import anthropic

client = anthropic.Anthropic(api_key=os.environ["ANTHROPIC_API_KEY"])

prompt = "Generate a SQL query to find customers with more than 5 orders in 2023."

message = client.messages.create(
    model="claude-3-5-sonnet-20241022",
    max_tokens=512,
    system="You are a helpful SQL assistant.",
    messages=[{"role": "user", "content": prompt}]
)

print(message.content[0].text)

output

The AI generates a SQL query like: SELECT customer_id, COUNT(*) FROM orders WHERE order_date BETWEEN '2023-01-01' AND '2023-12-31' GROUP BY customer_id HAVING COUNT(*) > 5;

Troubleshooting

If the AI returns irrelevant or incomplete SQL, refine your prompt to be more specific or provide schema context. Also, ensure your API key is valid and your environment variables are set correctly.

✅

Key Takeaways

Use gpt-4o or claude-3-5-sonnet-20241022 for SQL analysis and generation via natural language prompts.
Provide clear context and schema details in prompts for accurate SQL explanations or optimizations.
Set up environment variables securely and use the latest SDK patterns for API calls.
Experiment with prompt variations to generate, debug, or optimize SQL queries effectively.

Verified 2026-04 · gpt-4o, claude-3-5-sonnet-20241022

Verify ↗