DEV Community

Vivek Kumar
Vivek Kumar

Posted on

How to Prompt AI Tools to Write Accurate SQL Queries (And Why Most Developers Get This Wrong)

If you've tried asking ChatGPT, Claude, or any AI SQL assistant to generate a query and gotten back something that looked plausible but was subtly wrong — you're not alone. The frustrating part is it often runs. The database returns rows, the numbers look reasonable, and you ship it. Three days later, someone points out the totals are off by 20%.

The problem isn't the AI. The problem is the prompt.

Text-to-SQL works remarkably well when you give the model what it actually needs. According to AWS's benchmarking, GPT-4-class models achieve a 94% first-try success rate on ad-hoc analytics queries when the schema and foreign key constraints are properly provided. Without that context? You're closer to 60%. The difference is entirely in how you prompt.

This guide covers the practical techniques — schema context, few-shot examples, business term definitions, and chain-of-thought decomposition — that separate accurate AI-generated SQL from the kind that silently lies to you.


Why AI Gets SQL Wrong (It's Not the Model's Fault)

When you ask an AI "give me last month's revenue by plan tier," the model has to make a series of guesses:

  • Which table holds revenue? orders? subscriptions? invoices?
  • What column tracks the amount? total? amount_cents? mrr?
  • What does "last month" mean in your timezone?
  • Is "revenue" recognized revenue, gross, or net of refunds?
  • What column stores "plan tier"?

Without your schema, the model invents plausible-sounding answers to all of these. It will write syntactically valid SQL that is semantically wrong for your specific database.

The key insight: From the model's perspective, the schema is the problem space. You wouldn't ask a contractor to remodel your kitchen without giving them the floor plan.


Technique 1: Always Include Your Schema (But Not All of It)

The most impactful single change you can make is providing your table definitions in the prompt.

The naive approach — dumping your entire database schema — backfires. Enterprise databases with 100+ tables drown the model in noise and push important tables out of its effective context window. The right move is to include only the tables relevant to the question.

Here's a solid schema context block to include in your prompt:

You are a SQL expert working with a PostgreSQL database.

Relevant tables:

CREATE TABLE subscriptions (
  id          SERIAL PRIMARY KEY,
  user_id     INTEGER REFERENCES users(id),
  plan        TEXT,         -- 'starter', 'pro', 'enterprise'
  status      TEXT,         -- 'active', 'cancelled', 'trialing'
  mrr_cents   INTEGER,      -- monthly recurring revenue in cents
  started_at  TIMESTAMPTZ,
  cancelled_at TIMESTAMPTZ
);

CREATE TABLE users (
  id          SERIAL PRIMARY KEY,
  email       TEXT,
  created_at  TIMESTAMPTZ,
  country     TEXT
);

Write a SQL query to: [your question here]
Enter fullscreen mode Exit fullscreen mode

Notice the inline comments on columns like plan and status. These are crucial — they tell the model the exact set of values to filter on. Without them, the model might write WHERE plan = 'professional' instead of WHERE plan = 'pro'.

Practical tip: If you don't know which tables are relevant upfront, ask the model first: "Given this question, which tables from the following list would you need?" Then provide only those.


Technique 2: Define Your Business Terms

Business vocabulary rarely maps cleanly to column names. "Revenue" might mean mrr_cents. "Active users" might mean users who logged in within 30 days, not users with status = 'active'. "Churn" could be calculated six different ways.

Add a glossary block to your prompt for any domain-specific terms:

Business term definitions:
- "Active subscriber": a user with subscriptions.status = 'active'
- "MRR": SUM(mrr_cents) / 100.0, expressed in dollars
- "Last month": the full calendar month before the current month
  (e.g., if today is May 15 2026, last month = April 1–30 2026)
- "Churned": status changed from 'active' to 'cancelled' in the period
Enter fullscreen mode Exit fullscreen mode

This is the step most developers skip. And it's the reason queries that look right return numbers that don't match your finance team's spreadsheet.


Technique 3: Use Few-Shot Examples

Few-shot prompting — showing the model one or two example question/query pairs — dramatically improves accuracy on complex queries. It teaches the model your style, your naming conventions, and the kinds of JOINs your schema requires.

Here are two example queries to guide your output:

-- Question: How many new subscribers signed up in March 2026?
-- Query:
SELECT COUNT(*)
FROM subscriptions
WHERE started_at >= '2026-03-01'
  AND started_at < '2026-04-01';

-- Question: What is total MRR by plan for active subscriptions?
-- Query:
SELECT
  plan,
  SUM(mrr_cents) / 100.0 AS mrr_dollars
FROM subscriptions
WHERE status = 'active'
GROUP BY plan
ORDER BY mrr_dollars DESC;

-- Now answer this question: [your new question]
Enter fullscreen mode Exit fullscreen mode

The examples serve as a template. The model learns that you use >= / < for date ranges (not BETWEEN), that you divide cents by 100.0, and that you prefer ORDER BY descending. You'll get output that fits into your codebase without needing cleanup.


Technique 4: Break Complex Questions into Steps (Chain-of-Thought)

For multi-step analytical queries — cohort analysis, funnel calculations, retention — don't ask for the whole query at once. Ask the model to reason through it first.

I need to calculate 30-day retention for users who signed up in April 2026.
Retention means: they had at least one active session in the 30-day window
after signup (tracked in the `events` table with event_type = 'session_start').

Before writing the SQL:
1. What intermediate steps are needed?
2. Which tables will you join?
3. How will you define the 30-day window?

Then write the final query.
Enter fullscreen mode Exit fullscreen mode

When you ask the model to think through the steps explicitly, it catches its own mistakes before producing the final SQL. This mirrors what an experienced developer does mentally before writing a complex query. The chain-of-thought approach consistently outperforms direct generation on multi-table, multi-condition queries.

Here's what the output might look like for that retention query:

WITH april_signups AS (
  SELECT id, created_at
  FROM users
  WHERE created_at >= '2026-04-01'
    AND created_at < '2026-05-01'
),
retained AS (
  SELECT DISTINCT u.id
  FROM april_signups u
  JOIN events e ON e.user_id = u.id
    AND e.event_type = 'session_start'
    AND e.occurred_at >= u.created_at + INTERVAL '1 day'
    AND e.occurred_at <= u.created_at + INTERVAL '30 days'
)
SELECT
  COUNT(DISTINCT a.id)                        AS total_signups,
  COUNT(DISTINCT r.id)                        AS retained_users,
  ROUND(
    COUNT(DISTINCT r.id)::NUMERIC /
    NULLIF(COUNT(DISTINCT a.id), 0) * 100, 1
  )                                           AS retention_pct
FROM april_signups a
LEFT JOIN retained r ON r.id = a.id;
Enter fullscreen mode Exit fullscreen mode

A direct one-shot prompt rarely produces something this structurally sound. Decomposing the problem does.


Technique 5: Ask for Explanation and Validation

After receiving a query, always ask the model to explain it:

Explain what this query does, step by step.
Are there any edge cases (nulls, division by zero, time zone assumptions)
that could return incorrect results?
Enter fullscreen mode Exit fullscreen mode

This serves two purposes: it helps you catch logical errors before running the query in production, and it forces the model to re-examine its own output. Models often catch their own mistakes during this review pass.

For example, the model might flag:

  • "This query uses CURRENT_DATE which assumes UTC. If your database runs in a different timezone, you may want CURRENT_DATE AT TIME ZONE 'America/New_York'."
  • "If mrr_cents is NULL for any rows, SUM() will silently exclude them. You may want COALESCE(mrr_cents, 0)."

These are exactly the gotchas that cause silent data quality issues.


Common Mistakes to Avoid

Vague questions. "Show me user activity" will produce a vague, probably useless query. "Show me the count of distinct users who triggered at least one purchase event in the last 7 days, grouped by their plan" will produce something accurate and useful.

No schema, but expecting column-level accuracy. The model will guess, and its guesses will be syntactically valid. That's what makes them dangerous.

Ignoring dialect differences. If you're on BigQuery, asking a generic model for date functions might get you DATE_TRUNC when you need DATE_TRUNC(date, MONTH). Always tell the model your database flavor (PostgreSQL, MySQL, BigQuery, Snowflake, etc.).

Trusting output that executes without errors. A query that runs is not a query that's correct. Always sanity-check with a small sample (LIMIT 100), compare a few rows to known-good data, and check totals against a source of truth.

Not iterating. The first query you get back is a first draft. Paste it back in with corrections: "This is close, but I need the results grouped by week, not month, and I only want users in the 'pro' plan." Iteration gets you to accuracy faster than trying to write the perfect prompt on the first try.


Key Takeaways

The gap between 60% and 94% AI SQL accuracy comes down to what you put into the prompt:

  1. Include your schema — just the relevant tables, with column comments for values
  2. Define business terms — don't let the model guess what "revenue" or "active" means
  3. Add 1–2 example queries — establish your style and naming conventions
  4. Decompose complex questions — ask for reasoning steps before the final query
  5. Ask for explanation and edge-case review — let the model catch its own mistakes
  6. Always specify your database flavor — PostgreSQL, MySQL, BigQuery, etc.

AI-powered SQL generation is genuinely useful once you treat prompting as a craft, not an afterthought. The developers who get the most out of these tools aren't the ones using the fanciest models — they're the ones who've learned to give the model what it actually needs to succeed.


What techniques have worked well for you when prompting AI for SQL? Have you found ways to handle large, complex schemas effectively? Drop your approach in the comments — I'd love to compare notes.

Top comments (0)