generate-validation-notebook
Generate SQL validation notebooks for dbt changes. Pass a GitHub PR URL or local dbt repo path.
Install
npx skills add https://github.com/monte-carlo-data/mc-agent-toolkit/tree/main/skills/generate-validation-notebook
claude plugin marketplace add https://llmmart.ai/marketplace.json && claude plugin install monte-carlo-data-mc-agent-toolkit@llmmart
git clone https://github.com/monte-carlo-data/mc-agent-toolkit.git
The skills CLI installs just this skill, for any of its supported agents. Claude Code installs the whole monte-carlo-data/mc-agent-toolkit collection as a plugin from our marketplace. Git is the plain clone.
README
Generate Validation Notebook Skill
Automatically generate SQL validation notebooks for dbt model changes. Given a GitHub PR or local dbt repository, this skill identifies modified models and produces a Monte Carlo SQL Notebook with targeted validation queries comparing baseline and development data.
What it does
- Identifies changed dbt models from a PR diff or local branch
- Analyzes each model's schema, config, segmentation fields, and time axis
- Generates SQL validation queries (row counts, distribution checks, NULL rates, before/after comparisons, uniqueness checks)
- Packages everything into a Monte Carlo SQL Notebook with parameterized database references
- Outputs an import URL that opens the notebook directly in Monte Carlo's notebook interface
Prerequisites
- Claude Code or any MCP-capable editor
- GitHub CLI (
gh) — required for PR mode, must be authenticated - Python 3 with
pyyamlinstalled (pip install pyyaml) - MC Bridge running and connected to your warehouse
Setup
Via the mc-agent-toolkit plugin (recommended)
Install the plugin for your editor — see the main README for instructions. The skill is bundled automatically.
Standalone
Copy the skill to your local skills directory:
cp -r skills/generate-validation-notebook ~/.claude/skills/generate-validation-notebook
Usage
PR mode
/mc-generate-validation-notebook https://github.com/your-org/dbt/pull/123
Fetches the PR diff from GitHub, identifies changed models, and generates validation queries.
Local mode
/mc-generate-validation-notebook .
Uses git diff against the base branch to find changed models in the current repository.
Options
--mc-base-url <URL>— Monte Carlo base URL (defaults tohttps://getmontecarlo.com)--models <model1,model2,...>— only generate for specific models (by filename, without.sql)
What gets generated
The notebook includes:
- Parameter cells —
prod_dbanddev_dbfor selecting databases - Markdown summary — PR metadata, changed models, usage instructions
- SQL validation queries organized by pattern:
- Row counts (single and comparison)
- Segmentation distribution
- Changed field distribution
- NULL rate checks
- Uniqueness checks
- Time-axis continuity
- Before/after comparisons
- Sample data previews
Up to 10 changed models are processed per invocation.
Supported warehouses
Generated SQL uses ANSI-compatible syntax that works across Snowflake, BigQuery, Redshift, and Athena. Minor adjustments may be needed for specific warehouse quirks.
Skill manifest
Tip: This skill works well with Sonnet. Run
/model sonnetbefore invoking for faster generation.
Generate a SQL Notebook with validation queries for dbt changes.
Arguments: $ARGUMENTS
Parse the arguments:
- Target (required): first argument — a GitHub PR URL or local dbt repo path
- MC Base URL (optional):
--mc-base-url <URL>— defaults tohttps://getmontecarlo.com - Models (optional):
--models <model1,model2,...>— comma-separated list of model filenames (without.sqlextension) to generate queries for. Only these models will be included. By default, all changed models are included up to a maximum of 10.
Setup
Prerequisites:
gh(GitHub CLI) — required for PR mode. Must be authenticated (gh auth status).python3— required for helper scripts.pyyaml— install withpip3 install pyyaml(orpip install pyyaml,uv pip install pyyaml, etc.)
Note: Generated SQL uses ANSI-compatible syntax that works across Snowflake, BigQuery, Redshift, and Athena. Minor adjustments may be needed for specific warehouse quirks.
This skill includes two helper scripts in ${CLAUDE_PLUGIN_ROOT}/skills/generate-validation-notebook/scripts/:
resolve_dbt_schema.py- Resolves dbt model output schemas fromdbt_project.ymlrouting rules and model config overrides.generate_notebook_url.py- Encodes notebook YAML into a base64 import URL and opens it in the browser.
Mode Detection
Auto-detect mode from the target argument:
- If target looks like a URL (contains
://orgithub.com) -> PR mode - If target is a path (
.,/path/to/repo, relative path) -> Local mode
Context
This command generates a SQL Notebook containing validation queries for dbt changes. The notebook can be opened in the MC Bridge SQL Notebook interface for interactive validation.
The output is an import URL that opens directly in the notebook interface:
<MC_BASE_URL>/notebooks/import#<base64-encoded-yaml>
Key Features:
- Database Parameters: Two
textparameters (prod_dbanddev_db) for selecting databases - Schema Inference: Automatically infers schema per model from
dbt_project.ymland model configs - Single-table queries: Basic validation queries using
{{prod_db}}.<SCHEMA>.<TABLE> - Comparison queries: Before/after queries comparing
{{prod_db}}vs{{dev_db}} - Flexible usage: Users can set both parameters to the same database for single-database analysis
Notebook YAML Spec Reference
Key structure:
version: 1
metadata:
id: string # kebab-case + random suffix
name: string # display name
created_at: string # ISO 8601
updated_at: string # ISO 8601
default_context: # optional database/schema context
database: string
schema: string
cells:
- id: string
type: sql | markdown | parameter
content: string # SQL, markdown, or parameter config (JSON)
display_type: table | bar | timeseries
Parameter Cell Spec
Parameter cells allow defining variables referenced in SQL via {{param_name}} syntax:
- id: param-prod-db
type: parameter
content:
name: prod_db # variable name
config:
type: text # free-form text input
default_value: "ANALYTICS"
placeholder: "Prod database"
display_type: table
Parameter types:
text: Free-form text input (used for database names)schema_selector: Two dropdowns (database -> schema), value stored asDATABASE.SCHEMAdropdown: Select from predefined options
Task
Generate a SQL Notebook with validation queries based on the mode and target.
Phase 1: Get Changed Files
The approach differs based on mode:
If PR mode (GitHub PR):
Extract the PR number and repo from the target URL.
- Example:
https://github.com/monte-carlo-data/dbt/pull/3386-> owner=monte-carlo-data, repo=dbt, PR=3386
- Example:
Fetch PR metadata using
gh:
gh pr view <PR#> --repo <owner>/<repo> --json number,title,author,mergedAt,headRefOid
- Fetch the list of changed files:
gh pr view <PR#> --repo <owner>/<repo> --json files --jq '.files[].path'
- Fetch the diff:
gh pr diff <PR#> --repo <owner>/<repo>
Filter the changed files list to only
.sqlfiles undermodels/orsnapshots/directories (at any depth — e.g.,models/,analytics/models/,dbt/models/). These are the dbt models to analyze. If no model SQL files were changed, report that and stop.For each changed model file, fetch the full file content at the head SHA:
gh api repos/<owner>/<repo>/contents/<file_path>?ref=<head_sha> --jq '.content' | python3 -c "import sys,base64; sys.stdout.write(base64.b64decode(sys.stdin.read()).decode())"
- Fetch dbt_project.yml for schema resolution. Detect the dbt project root by looking at the changed file paths — find the common parent directory that contains
dbt_project.yml. Try these paths in order until one succeeds:
gh api repos/<owner>/<repo>/contents/<dbt_root>/dbt_project.yml?ref=<head_sha> --jq '.content' | python3 -c "import sys,base64; sys.stdout.write(base64.b64decode(sys.stdin.read()).decode())"
Common <dbt_root> locations: analytics, . (repo root), dbt, transform. Try each until found.
Save dbt_project.yml to /tmp/validation_notebook_working/<PR#>/dbt_project.yml.
If Local mode (Local Directory):
Change to the target directory.
Get current branch info:
git rev-parse --abbrev-ref HEAD
Detect base branch - try
main,master,developin order, or use upstream tracking branch.Get the list of changed SQL files compared to base branch:
git diff --name-only <base_branch>...HEAD -- '*.sql'
Filter to only
.sqlfiles undermodels/orsnapshots/directories (at any depth — e.g.,models/,analytics/models/,dbt/models/). If no model SQL files were changed, report that and stop.Get the diff for each changed file:
git diff <base_branch>...HEAD -- <file_path>
Read model files directly from the filesystem.
Find dbt_project.yml:
find . -name "dbt_project.yml" -type f | head -1
- For notebook metadata in local mode, use:
- ID:
local-<branch-name>-<timestamp> - Title:
Local: <branch-name> - Author: Output of
git config user.name - Merged: "N/A (local)"
- ID:
Model Selection (applies to both modes)
After filtering to .sql files under models/ or snapshots/:
If
--modelswas specified: Filter the changed files list to only include models whose filename (without.sqlextension, case-insensitive) matches one of the specified model names. If any specified model is not found in the changed files, warn the user but continue with the models that were found. If none match, report that and stop.Model cap: If more than 10 models remain after filtering, select the first 10 (by file path order) and warn the user:
⚠️ <total_count> models changed — generating validation queries for the first 10 only. To generate for specific models, re-run with: --models <model1,model2,...> Skipped models: <list of skipped model filenames>
Phase 2: Parse Changed Models
For EACH changed dbt model .sql file, parse and extract:
2a. Model Metadata
Output table name -- Derive from file name:
<any_path>/models/<subdir>/<model_name>.sql-> table is<MODEL_NAME>(uppercase, taken from the filename)
Output schema -- Use the schema resolution script:
Setup: Save
dbt_project.ymland model files to/tmp/validation_notebook_working/<id>/preserving paths:/tmp/validation_notebook_working/<id>/ +-- dbt_project.yml +-- models/ +-- <path>/<model>.sqlRun the script for each model:
python3 ${CLAUDE_PLUGIN_ROOT}/skills/generate-validation-notebook/scripts/resolve_dbt_schema.py /tmp/validation_notebook_working/<id>/dbt_project.yml /tmp/validation_notebook_working/<id>/models/<path>/<model>.sqlError handling: If the script fails, STOP immediately and report the error. Do NOT proceed with notebook generation if schema resolution fails.
Output: The script prints the resolved schema (e.g.,
PROD,PROD_STAGE,PROD_LINEAGE)
Note: Do NOT manually parse dbt_project.yml or model configs for schema -- always use the script. It handles model config overrides, dbt_project.yml routing rules, PROD_ prefix for custom schemas, and defaults to PROD.
Config block -- Look for {{ config(...) }} and extract:
materialized-- 'table', 'view', 'incremental', 'ephemeral'unique_key-- the dedup key (may be a string or list)cluster_by-- clustering fields (may contain the time axis)
Core segmentation fields -- Scan the entire model SQL for fields likely to be business keys:
- Fields named
*_id(e.g.,account_id,resource_id,monitor_id) that appear in JOIN ON, GROUP BY, PARTITION BY, orunique_key - Deduplicate and rank by frequency. Take the top 3.
Time axis field -- Detect the model's time dimension (in priority order):
is_incremental()block: field used in the WHERE comparisoncluster_byconfig: timestamp/date fields- Field name conventions:
ingest_ts,created_time,date_part,timestamp,run_start_time,export_ts,event_created_time - ORDER BY DESC in QUALIFY/ROW_NUMBER
If no time axis is found, skip time-axis queries for this model.
2b. Diff Analysis
Parse the diff hunks for this file. Classify each changed line:
- Changed fields -- Lines added/modified in SELECT clauses or CTE definitions. Extract the output column name.
- Changed filters -- Lines added/modified in WHERE clauses.
- Changed joins -- Lines added/modified in JOIN ON conditions.
- Changed unique_key -- If
unique_keyin config was modified, note both old and new values. - New columns -- Columns in "after" SELECT that don't appear in "before" (pure additions).
2c. Model Classification
Classify each model as new or modified based on the diff:
- If the diff for this file contains
new file mode→ classify as new - Otherwise → classify as modified
This classification determines which query patterns are generated in Phase 3.
Note: For new models, Phase 2b diff analysis is skipped (there is no "before" to compare against). Phase 2a metadata extraction still applies.
Phase 3: Generate Validation Queries
For each changed model, generate the applicable queries based on its classification (new vs modified).
CRITICAL: Parameter Placeholder Syntax
Use double curly braces {{...}} for parameter placeholders. Do NOT use ${...} or any other syntax.
Correct: {{prod_db}}.PROD.AGENT_RUNS
Wrong: ${prod_db}.PROD.AGENT_RUNS
Table Reference Format:
- Use
{{prod_db}}.<SCHEMA>.<TABLE_NAME>for prod queries - Use
{{dev_db}}.<SCHEMA>.<TABLE_NAME>for dev queries <SCHEMA>is hardcoded per-model using the output from the schema resolution script
Query Patterns for NEW Models
For new models, all queries target {{dev_db}} only. No comparison queries are generated since no prod table exists.
Pattern 7-new: Total Row Count
Trigger: Always.
SELECT COUNT(*) AS total_rows
FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
Pattern 9: Sample Data Preview
Trigger: Always.
SELECT *
FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
LIMIT 20
Pattern 2-new: Core Segmentation Counts
Trigger: Always.
SELECT
<segmentation_field>,
COUNT(*) AS row_count
FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
GROUP BY <segmentation_field>
ORDER BY row_count DESC
LIMIT 100
Pattern 5: Uniqueness Check
Trigger: Always for new models (verify unique_key constraint from the start).
SELECT
COUNT(*) AS total_rows,
COUNT(DISTINCT <key_fields>) AS distinct_keys,
COUNT(*) - COUNT(DISTINCT <key_fields>) AS duplicate_count
FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
SELECT <key_fields>, COUNT(*) AS n
FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
GROUP BY <key_fields>
HAVING COUNT(*) > 1
ORDER BY n DESC
LIMIT 100
Pattern 6-new: NULL Rate Check (all columns)
Trigger: Always. Checks all output columns since everything is new.
SELECT
COUNT(*) AS total_rows,
SUM(CASE WHEN <col1> IS NULL THEN 1 ELSE 0 END) AS <col1>_null_count,
ROUND(100.0 * SUM(CASE WHEN <col1> IS NULL THEN 1 ELSE 0 END) / NULLIF(COUNT(*), 0), 2) AS <col1>_null_pct,
SUM(CASE WHEN <col2> IS NULL THEN 1 ELSE 0 END) AS <col2>_null_count,
ROUND(100.0 * SUM(CASE WHEN <col2> IS NULL THEN 1 ELSE 0 END) / NULLIF(COUNT(*), 0), 2) AS <col2>_null_pct
-- repeat for each output column
FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
Pattern 8: Time-Axis Continuity
Trigger: Model is materialized='incremental' OR a time axis field was identified.
SELECT
CAST(<time_axis> AS DATE) AS day,
COUNT(*) AS row_count
FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
WHERE <time_axis> >= CURRENT_TIMESTAMP - INTERVAL '14' DAY
GROUP BY day
ORDER BY day DESC
LIMIT 30
Query Patterns for MODIFIED Models
For modified models, single-table queries use {{prod_db}} and comparison queries use both.
Pattern 7: Total Row Count
Trigger: Always.
SELECT COUNT(*) AS total_rows
FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME>
Pattern 9: Sample Data Preview
Trigger: Always.
SELECT *
FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME>
LIMIT 20
Pattern 2: Core Segmentation Counts
Trigger: Always.
SELECT
<segmentation_field>,
COUNT(*) AS row_count
FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME>
GROUP BY <segmentation_field>
ORDER BY row_count DESC
LIMIT 100
Pattern 1: Changed Field Distribution
Trigger: Changed fields found in Phase 2b. Exclude added columns (from "New columns" in Phase 2b) — only include fields that exist in prod.
SELECT
<changed_field>,
COUNT(*) AS row_count,
ROUND(COUNT(*) * 100.0 / SUM(COUNT(*)) OVER(), 2) AS pct
FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME>
GROUP BY <changed_field>
ORDER BY row_count DESC
LIMIT 100
Pattern 5: Uniqueness Check
Trigger: JOIN condition changed, unique_key changed, or model is incremental.
SELECT
COUNT(*) AS total_rows,
COUNT(DISTINCT <key_fields>) AS distinct_keys,
COUNT(*) - COUNT(DISTINCT <key_fields>) AS duplicate_count
FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
SELECT <key_fields>, COUNT(*) AS n
FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
GROUP BY <key_fields>
HAVING COUNT(*) > 1
ORDER BY n DESC
LIMIT 100
Pattern 6: NULL Rate Check
Trigger: New column added, or column wrapped in COALESCE/NULLIF.
Important: Added columns (from "New columns" in Phase 2b) do NOT exist in prod yet. For added columns, query {{dev_db}} only. For modified columns (COALESCE/NULLIF changes), compare both databases.
For added columns (dev only):
SELECT
COUNT(*) AS total_rows,
SUM(CASE WHEN <column> IS NULL THEN 1 ELSE 0 END) AS null_count,
ROUND(100.0 * SUM(CASE WHEN <column> IS NULL THEN 1 ELSE 0 END) / NULLIF(COUNT(*), 0), 2) AS null_pct
FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
For modified columns (prod vs dev):
SELECT
'prod' AS source,
COUNT(*) AS total_rows,
SUM(CASE WHEN <column> IS NULL THEN 1 ELSE 0 END) AS null_count,
ROUND(100.0 * SUM(CASE WHEN <column> IS NULL THEN 1 ELSE 0 END) / NULLIF(COUNT(*), 0), 2) AS null_pct
FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME>
UNION ALL
SELECT
'dev' AS source,
COUNT(*) AS total_rows,
SUM(CASE WHEN <column> IS NULL THEN 1 ELSE 0 END) AS null_count,
ROUND(100.0 * SUM(CASE WHEN <column> IS NULL THEN 1 ELSE 0 END) / NULLIF(COUNT(*), 0), 2) AS null_pct
FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
Pattern 8: Time-Axis Continuity
Trigger: Model is materialized='incremental' OR a time axis field was identified.
SELECT
CAST(<time_axis> AS DATE) AS day,
COUNT(*) AS row_count
FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME>
WHERE <time_axis> >= CURRENT_TIMESTAMP - INTERVAL '14' DAY
GROUP BY day
ORDER BY day DESC
LIMIT 30
Pattern 3: Before/After Comparison
Trigger: Always (for changed fields + top segmentation field). Modified models only.
Important: Exclude added columns (from "New columns" in Phase 2b) from <group_fields>. Only use fields that exist in BOTH prod and dev. Added columns don't exist in prod and will cause query errors.
WITH prod AS (
SELECT <group_fields>, COUNT(*) AS cnt
FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME>
GROUP BY <group_fields>
),
dev AS (
SELECT <group_fields>, COUNT(*) AS cnt
FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
GROUP BY <group_fields>
)
SELECT
COALESCE(b.<field>, d.<field>) AS <field>,
COALESCE(b.cnt, 0) AS cnt_prod,
COALESCE(d.cnt, 0) AS cnt_dev,
COALESCE(d.cnt, 0) - COALESCE(b.cnt, 0) AS diff
FROM prod b
FULL OUTER JOIN dev d ON b.<field> = d.<field>
ORDER BY ABS(diff) DESC
LIMIT 100
Pattern 7b: Row Count Comparison
Trigger: Always. Modified models only.
SELECT 'prod' AS source, COUNT(*) AS row_count FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME>
UNION ALL
SELECT 'dev' AS source, COUNT(*) AS row_count FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME>
Phase 4: Build Notebook YAML
4a. Metadata
version: 1
metadata:
id: validation-pr-<PR_NUMBER>-<random_suffix>
name: "Validation: PR #<PR_NUMBER> - <PR_TITLE_TRUNCATED>"
created_at: "<current_iso_timestamp>"
updated_at: "<current_iso_timestamp>"
4b. Parameter Cells
Only include prod_db if there are modified models. If all models are new, only include dev_db.
# Include ONLY if there are modified models:
- id: param-prod-db
type: parameter
content:
name: prod_db
config:
type: text
default_value: "ANALYTICS"
placeholder: "Prod database (e.g., ANALYTICS)"
display_type: table
# Always include:
- id: param-dev-db
type: parameter
content:
name: dev_db
config:
type: text
default_value: "PERSONAL_<USER>"
placeholder: "Dev database (e.g., PERSONAL_JSMITH)"
display_type: table
4c. Markdown Summary Cell
- id: cell-summary
type: markdown
content: |
# Validation Queries for <PR or Local Branch>
## Summary
- **Title:** <title>
- **Author:** <author>
- **Source:** <PR URL or "Local branch: <branch>">
- **Status:** <merge_timestamp or "Not yet merged" or "N/A (local)">
## Changes
<brief description based on diff analysis>
## Changed Models
- `<SCHEMA>.<TABLE_NAME>` (from `<file_path>`)
## How to Use
1. Select your Snowflake connector above
2. Set **dev_db** to your dev database (e.g., `PERSONAL_JSMITH`)
3. If modified models are present, set **prod_db** to your prod database (e.g., `ANALYTICS`)
4. Run single-table queries first, then comparison queries
display_type: table
4d. SQL Cell Format
- id: cell-<pattern>-<model>-<index>
type: sql
content: |
/*
========================================
<Pattern Name (human-readable, e.g. "Total Row Count" — do NOT include pattern numbers like "Pattern 7:")>
========================================
Model: <SCHEMA>.<TABLE_NAME>
Triggered by: <why this pattern was generated>
What to look for: <interpretation guidance>
----------------------------------------
*/
<actual_sql_query>
display_type: table
4e. Cell Organization
Cells are ordered consistently for both model types, following this sequence:
New models:
- Summary markdown cell (note that model is new)
- Parameter cells (dev_db only — no prod_db if all models are new)
- Total row count (Pattern 7-new)
- Sample data preview (Pattern 9)
- Core segmentation counts (Pattern 2-new)
- Uniqueness check (Pattern 5), NULL rate check (Pattern 6-new), Time-axis continuity (Pattern 8)
Modified models:
- Summary markdown cell
- Parameter cells (prod_db, dev_db)
- Total row count (Pattern 7)
- Sample data preview (Pattern 9)
- Core segmentation counts (Pattern 2)
- Changed field distribution (Pattern 1)
- Uniqueness check (Pattern 5), NULL rate check (Pattern 6), Time-axis continuity (Pattern 8)
- Before/after comparisons (Pattern 3), Row count comparison (Pattern 7b)
Phase 5: Generate Import URL
- Write notebook YAML to
/tmp/validation_notebook_working/<id>/notebook.yaml - Run the URL generation script:
python3 ${CLAUDE_PLUGIN_ROOT}/skills/generate-validation-notebook/scripts/generate_notebook_url.py /tmp/validation_notebook_working/<id>/notebook.yaml --mc-base-url <MC_BASE_URL>
- The script validates both YAML syntax and notebook schema (required fields on metadata and cells). If validation fails, read the error messages carefully, fix the YAML to match the spec in Phase 4, and re-run.
Phase 6: Output
Present:
# Validation Notebook Generated
## Summary
- **Source:** PR #<number> - <title> OR Local: <branch>
- **Author:** <author>
- **Changed Models:** <count> models (of <total_count> changed)
- **Generated Queries:** <count> queries
> ⚠️ If models were capped: "Only the first 10 of <total_count> changed models were included. Re-run with `--models` to select specific models."
## Notebook Opened
The notebook has been opened directly in your browser.
Select your Snowflake connector in the notebook interface to begin running queries.
*Make sure MC Bridge is running. Let me know if you want tips on how to install this locally*
Important Guidelines
- Do NOT execute queries -- only generate the notebook
- Keep SQL readable -- proper formatting and meaningful aliases
- Include LIMIT 100 on queries that could return many rows
- Use double curly braces --
{{prod_db}}NOT${prod_db} - Use correct table format --
{{prod_db}}.<SCHEMA>.<TABLE>and{{dev_db}}.<SCHEMA>.<TABLE> - Always use the schema resolution script -- do NOT manually parse dbt_project.yml
- Schema is NOT a parameter -- only
prod_dbanddev_dbare parameters - Skip ephemeral models -- they have no physical table
- Truncate notebook name -- keep under 50 chars
- Generate unique cell IDs -- use pattern like
cell-p3-model-1 - YAML multiline content -- use
|block scalar for SQL with comments - ASCII-only YAML -- the script sanitizes and validates before encoding
Query Pattern Reference
| Pattern | Name | Trigger | Model Type | Database | Order |
|---|---|---|---|---|---|
| 7 / 7-new | Total Row Count | Always | Both | {{prod_db}} (modified) / {{dev_db}} (new) |
1 |
| 9 | Sample Data Preview | Always | Both | {{prod_db}} (modified) / {{dev_db}} (new) |
2 |
| 2 / 2-new | Core Segmentation Counts | Always | Both | {{prod_db}} (modified) / {{dev_db}} (new) |
3 |
| 1 | Changed Field Distribution | Column modified in diff (not added) | Modified only | {{prod_db}} |
4 |
| 5 | Uniqueness Check | JOIN/unique_key changed (modified) / Always (new) | Both | {{dev_db}} |
5 |
| 6 / 6-new | NULL Rate Check | New column or COALESCE (modified) / Always (new) | Both | Added col: {{dev_db}} only; COALESCE: Both (modified) / {{dev_db}} (new) |
5 |
| 8 | Time-Axis Continuity | Incremental or time field | Both | {{prod_db}} (modified) / {{dev_db}} (new) |
5 |
| 3 | Before/After Comparison | Changed fields (not added) | Modified only | Both | 6 |
| 7b | Row Count Comparison | Always | Modified only | Both | 6 |
MC Bridge Setup Help
If the user asks how to install or set up MC Bridge, fetch the README from the mc-bridge repo and show the relevant quick start / setup instructions:
gh api repos/monte-carlo-data/mc-bridge/readme --jq '.content' | base64 --decode
Focus on: how to install, configure connections, and run MC Bridge. Don't dump the entire README — extract just the setup-relevant sections.
Files (mc-agent-toolkit)
-
scripts
-
generate_notebook_url.py 4.7 KB
#!/usr/bin/env python3 """ Encode a notebook YAML file into a base64 import URL and open it in the browser. Usage: python3 generate_notebook_url.py <notebook_yaml_path> [--mc-base-url URL] """ import argparse import base64 import os import re import subprocess import sys import yaml def sanitize_yaml(content: str) -> str: """Replace non-ASCII characters with ASCII equivalents.""" replacements = { "\u2014": "-", "\u2013": "-", "\u2018": "'", "\u2019": "'", "\u201c": '"', "\u201d": '"', "\u2026": "...", "\u00a0": " ", } for char, replacement in replacements.items(): content = content.replace(char, replacement) content = re.sub(r"[^\x00-\x7F]", "?", content) return content def validate_yaml(content: str) -> None: """Parse YAML, validate notebook schema, and exit with context on failure.""" try: doc = yaml.safe_load(content) except yaml.YAMLError as e: print(f"YAML validation failed: {e}", file=sys.stderr) sys.exit(1) errors: list[str] = [] # Top-level structure if not isinstance(doc, dict): errors.append("Root must be a mapping") else: if "version" not in doc: errors.append("Missing top-level 'version'") metadata = doc.get("metadata") if not isinstance(metadata, dict): errors.append("Missing or invalid 'metadata' mapping") else: for field in ("id", "name", "created_at", "updated_at"): if field not in metadata: errors.append(f"metadata.{field}: missing required field") for bad_field in ("title", "description", "pr_number", "generated_by"): if bad_field in metadata: errors.append( f"metadata.{bad_field}: unexpected field (use 'name' for the notebook title)" ) cells = doc.get("cells") if not isinstance(cells, list): errors.append("Missing or invalid 'cells' list") else: for i, cell in enumerate(cells): prefix = f"cells[{i}]" if not isinstance(cell, dict): errors.append(f"{prefix}: must be a mapping") continue if "id" not in cell: errors.append(f"{prefix}: missing 'id'") if "type" not in cell: errors.append(f"{prefix}: missing 'type'") cell_type = cell.get("type") if cell_type not in ("sql", "markdown", "parameter"): errors.append( f"{prefix}: invalid type '{cell_type}' (must be sql, markdown, or parameter)" ) if "display_type" not in cell: errors.append(f"{prefix}: missing 'display_type'") if cell_type == "parameter": content_val = cell.get("content") if not isinstance(content_val, dict): errors.append(f"{prefix}: parameter cell 'content' must be a mapping with 'name' and 'config'") else: if "name" not in content_val: errors.append(f"{prefix}: parameter content missing 'name'") if "config" not in content_val: errors.append(f"{prefix}: parameter content missing 'config'") if errors: print("Invalid notebook:", file=sys.stderr) for err in errors: print(f" - {err}", file=sys.stderr) sys.exit(1) def main() -> None: parser = argparse.ArgumentParser(description="Encode notebook YAML to import URL") parser.add_argument("yaml_path", help="Path to notebook YAML file") parser.add_argument( "--mc-base-url", default="https://getmontecarlo.com", help="MC Bridge base URL", ) args = parser.parse_args() with open(args.yaml_path) as f: notebook_yaml = f.read() yaml_content = sanitize_yaml(notebook_yaml.strip()) validate_yaml(yaml_content) encoded = base64.b64encode(yaml_content.encode()).decode() url = f"{args.mc_base_url}/notebooks/import#{encoded}" print(f"URL length: {len(url)} chars") # Save URL to file alongside the YAML url_file = os.path.join(os.path.dirname(os.path.abspath(args.yaml_path)), "notebook_url.txt") with open(url_file, "w") as f: f.write(url) print(f"URL saved to: {url_file}") print("\n" + "=" * 60) print("NOTEBOOK URL:") print("=" * 60) print(url) print("=" * 60 + "\n") print("Opening notebook in browser...") subprocess.run(["open", url]) if __name__ == "__main__": main() -
resolve_dbt_schema.py 5.4 KB
#!/usr/bin/env python3 """ Resolve the output schema for a dbt model. Usage: python3 resolve_dbt_schema.py <dbt_project_yml_path> <model_sql_path> Returns the resolved schema name (uppercase), e.g., "PROD", "PROD_STAGE", "PROD_LINEAGE" """ import argparse import re import sys from pathlib import Path from typing import Dict, List, Optional, Tuple, Union import yaml def parse_model_config_schema(model_content: str) -> Optional[str]: """Extract schema from model's config block.""" pattern = r"\{\{\s*config\s*\([^)]*\bschema\s*=\s*['\"]([^'\"]+)['\"][^)]*\)\s*\}\}" match = re.search(pattern, model_content, re.IGNORECASE | re.DOTALL) if match: return match.group(1).upper() snapshot_pattern = r"target_schema\s*=\s*generate_schema_name\s*\(\s*['\"]([^'\"]+)['\"]" match = re.search(snapshot_pattern, model_content, re.IGNORECASE | re.DOTALL) if match: return match.group(1).upper() return None def parse_dbt_project_routing( dbt_project: dict, project_name: str ) -> Tuple[Dict[str, str], Dict[str, str]]: """Extract schema and database routing rules from dbt_project.yml.""" schema_routing = {} # type: Dict[str, str] database_routing = {} # type: Dict[str, str] models_config = dbt_project.get("models", {}) project_config = models_config.get(project_name, {}) def extract_routing(config: dict, current_path: str = "") -> None: for key, value in config.items(): if key.startswith("+"): continue if not isinstance(value, dict): continue new_path = f"{current_path}/{key}" if current_path else key schema = value.get("schema") or value.get("+schema") if schema: if "{{" not in schema: schema_routing[new_path] = schema.upper() database = value.get("database") or value.get("+database") if database: if "{{" not in database: database_routing[new_path] = database.upper() extract_routing(value, new_path) extract_routing(project_config) return schema_routing, database_routing def parse_dbt_project_schema_routing(dbt_project: dict, project_name: str) -> Dict[str, str]: schema_routing, _ = parse_dbt_project_routing(dbt_project, project_name) return schema_routing def get_model_relative_path(dbt_project_path: Path, model_path: Path) -> str: dbt_project_dir = dbt_project_path.parent model_relative = model_path.relative_to(dbt_project_dir) parts = model_relative.parts if parts and parts[0] == "models": return str(Path(*parts[1:])) return str(model_relative) def find_matching_schema( model_relative_path: str, routing: Dict[str, str] ) -> Optional[str]: model_dir = str(Path(model_relative_path).parent) matches = [] # type: List[Tuple[str, str]] for route_path, schema in routing.items(): if model_dir == route_path or model_dir.startswith(route_path + "/"): matches.append((route_path, schema)) if not matches: return None matches.sort(key=lambda x: len(x[0]), reverse=True) return matches[0][1] def apply_schema_prefix(schema: str, target_schema: str = "PROD") -> str: if not schema or schema.upper() == target_schema.upper(): return target_schema.upper() return f"{target_schema.upper()}_{schema.upper()}" def resolve_schema( dbt_project_path: Union[str, Path], model_path: Union[str, Path], default_schema: str = "PROD", apply_prefix: bool = True, ) -> str: dbt_project_path = Path(dbt_project_path) model_path = Path(model_path) model_content = model_path.read_text() config_schema = parse_model_config_schema(model_content) if config_schema: if apply_prefix: return apply_schema_prefix(config_schema, default_schema) return config_schema with open(dbt_project_path) as f: dbt_project = yaml.safe_load(f) project_name = dbt_project.get("name", "") routing = parse_dbt_project_schema_routing(dbt_project, project_name) model_relative = get_model_relative_path(dbt_project_path, model_path) matched_schema = find_matching_schema(model_relative, routing) if matched_schema: if apply_prefix: return apply_schema_prefix(matched_schema, default_schema) return matched_schema return default_schema.upper() def main() -> None: parser = argparse.ArgumentParser( description="Resolve the output schema for a dbt model" ) parser.add_argument("dbt_project_path", help="Path to dbt_project.yml") parser.add_argument("model_path", help="Path to the model SQL file") parser.add_argument("--default", default="PROD", help="Default schema (default: PROD)") parser.add_argument("--no-prefix", action="store_true", help="Don't apply PROD_ prefix") args = parser.parse_args() dbt_project_path = Path(args.dbt_project_path) model_path = Path(args.model_path) if not dbt_project_path.exists(): print(f"Error: dbt_project.yml not found: {dbt_project_path}", file=sys.stderr) sys.exit(1) if not model_path.exists(): print(f"Error: Model file not found: {model_path}", file=sys.stderr) sys.exit(1) apply_prefix = not args.no_prefix schema = resolve_schema(dbt_project_path, model_path, args.default, apply_prefix) print(schema) if __name__ == "__main__": main()
-
-
README.md 2.7 KB
# Generate Validation Notebook Skill Automatically generate SQL validation notebooks for dbt model changes. Given a GitHub PR or local dbt repository, this skill identifies modified models and produces a Monte Carlo SQL Notebook with targeted validation queries comparing baseline and development data. ## What it does 1. Identifies changed dbt models from a PR diff or local branch 2. Analyzes each model's schema, config, segmentation fields, and time axis 3. Generates SQL validation queries (row counts, distribution checks, NULL rates, before/after comparisons, uniqueness checks) 4. Packages everything into a Monte Carlo SQL Notebook with parameterized database references 5. Outputs an import URL that opens the notebook directly in Monte Carlo's notebook interface ## Prerequisites - Claude Code or any MCP-capable editor - [GitHub CLI](https://cli.github.com/) (`gh`) — required for PR mode, must be authenticated - Python 3 with `pyyaml` installed (`pip install pyyaml`) - [MC Bridge](https://docs.getmontecarlo.com/docs/mc-bridge) running and connected to your warehouse ## Setup ### Via the mc-agent-toolkit plugin (recommended) Install the plugin for your editor — see the [main README](../../README.md) for instructions. The skill is bundled automatically. ### Standalone Copy the skill to your local skills directory: ```bash cp -r skills/generate-validation-notebook ~/.claude/skills/generate-validation-notebook ``` ## Usage ### PR mode ``` /mc-generate-validation-notebook https://github.com/your-org/dbt/pull/123 ``` Fetches the PR diff from GitHub, identifies changed models, and generates validation queries. ### Local mode ``` /mc-generate-validation-notebook . ``` Uses `git diff` against the base branch to find changed models in the current repository. ### Options - `--mc-base-url <URL>` — Monte Carlo base URL (defaults to `https://getmontecarlo.com`) - `--models <model1,model2,...>` — only generate for specific models (by filename, without `.sql`) ## What gets generated The notebook includes: - **Parameter cells** — `prod_db` and `dev_db` for selecting databases - **Markdown summary** — PR metadata, changed models, usage instructions - **SQL validation queries** organized by pattern: - Row counts (single and comparison) - Segmentation distribution - Changed field distribution - NULL rate checks - Uniqueness checks - Time-axis continuity - Before/after comparisons - Sample data previews Up to 10 changed models are processed per invocation. ## Supported warehouses Generated SQL uses ANSI-compatible syntax that works across Snowflake, BigQuery, Redshift, and Athena. Minor adjustments may be needed for specific warehouse quirks. -
SKILL.md 23.8 KB
--- name: generate-validation-notebook description: Generate SQL validation notebooks for dbt changes. Pass a GitHub PR URL or local dbt repo path. bucket: Prevent --- > **Tip:** This skill works well with Sonnet. Run `/model sonnet` before invoking for faster generation. Generate a SQL Notebook with validation queries for dbt changes. **Arguments:** $ARGUMENTS Parse the arguments: - **Target** (required): first argument — a GitHub PR URL or local dbt repo path - **MC Base URL** (optional): `--mc-base-url <URL>` — defaults to `https://getmontecarlo.com` - **Models** (optional): `--models <model1,model2,...>` — comma-separated list of model filenames (without `.sql` extension) to generate queries for. Only these models will be included. By default, all changed models are included up to a maximum of 10. --- # Setup **Prerequisites:** - **`gh`** (GitHub CLI) — required for PR mode. Must be authenticated (`gh auth status`). - **`python3`** — required for helper scripts. - **`pyyaml`** — install with `pip3 install pyyaml` (or `pip install pyyaml`, `uv pip install pyyaml`, etc.) **Note:** Generated SQL uses ANSI-compatible syntax that works across Snowflake, BigQuery, Redshift, and Athena. Minor adjustments may be needed for specific warehouse quirks. This skill includes two helper scripts in `${CLAUDE_PLUGIN_ROOT}/skills/generate-validation-notebook/scripts/`: - **`resolve_dbt_schema.py`** - Resolves dbt model output schemas from `dbt_project.yml` routing rules and model config overrides. - **`generate_notebook_url.py`** - Encodes notebook YAML into a base64 import URL and opens it in the browser. # Mode Detection Auto-detect mode from the target argument: - If target looks like a URL (contains `://` or `github.com`) -> **PR mode** - If target is a path (`.`, `/path/to/repo`, relative path) -> **Local mode** --- # Context This command generates a SQL Notebook containing validation queries for dbt changes. The notebook can be opened in the MC Bridge SQL Notebook interface for interactive validation. The output is an import URL that opens directly in the notebook interface: ``` <MC_BASE_URL>/notebooks/import#<base64-encoded-yaml> ``` **Key Features:** - **Database Parameters**: Two `text` parameters (`prod_db` and `dev_db`) for selecting databases - **Schema Inference**: Automatically infers schema per model from `dbt_project.yml` and model configs - **Single-table queries**: Basic validation queries using `{{prod_db}}.<SCHEMA>.<TABLE>` - **Comparison queries**: Before/after queries comparing `{{prod_db}}` vs `{{dev_db}}` - **Flexible usage**: Users can set both parameters to the same database for single-database analysis # Notebook YAML Spec Reference Key structure: ```yaml version: 1 metadata: id: string # kebab-case + random suffix name: string # display name created_at: string # ISO 8601 updated_at: string # ISO 8601 default_context: # optional database/schema context database: string schema: string cells: - id: string type: sql | markdown | parameter content: string # SQL, markdown, or parameter config (JSON) display_type: table | bar | timeseries ``` ## Parameter Cell Spec Parameter cells allow defining variables referenced in SQL via `{{param_name}}` syntax: ```yaml - id: param-prod-db type: parameter content: name: prod_db # variable name config: type: text # free-form text input default_value: "ANALYTICS" placeholder: "Prod database" display_type: table ``` Parameter types: - `text`: Free-form text input (used for database names) - `schema_selector`: Two dropdowns (database -> schema), value stored as `DATABASE.SCHEMA` - `dropdown`: Select from predefined options # Task Generate a SQL Notebook with validation queries based on the mode and target. ## Phase 1: Get Changed Files The approach differs based on mode: ### If PR mode (GitHub PR): 1. Extract the PR number and repo from the target URL. - Example: `https://github.com/monte-carlo-data/dbt/pull/3386` -> owner=`monte-carlo-data`, repo=`dbt`, PR=`3386` 2. Fetch PR metadata using `gh`: ```bash gh pr view <PR#> --repo <owner>/<repo> --json number,title,author,mergedAt,headRefOid ``` 3. Fetch the list of changed files: ```bash gh pr view <PR#> --repo <owner>/<repo> --json files --jq '.files[].path' ``` 4. Fetch the diff: ```bash gh pr diff <PR#> --repo <owner>/<repo> ``` 5. Filter the changed files list to only `.sql` files under `models/` or `snapshots/` directories (at any depth — e.g., `models/`, `analytics/models/`, `dbt/models/`). These are the dbt models to analyze. If no model SQL files were changed, report that and stop. 6. For each changed model file, fetch the full file content at the head SHA: ```bash gh api repos/<owner>/<repo>/contents/<file_path>?ref=<head_sha> --jq '.content' | python3 -c "import sys,base64; sys.stdout.write(base64.b64decode(sys.stdin.read()).decode())" ``` 7. **Fetch dbt_project.yml** for schema resolution. Detect the dbt project root by looking at the changed file paths — find the common parent directory that contains `dbt_project.yml`. Try these paths in order until one succeeds: ```bash gh api repos/<owner>/<repo>/contents/<dbt_root>/dbt_project.yml?ref=<head_sha> --jq '.content' | python3 -c "import sys,base64; sys.stdout.write(base64.b64decode(sys.stdin.read()).decode())" ``` Common `<dbt_root>` locations: `analytics`, `.` (repo root), `dbt`, `transform`. Try each until found. Save `dbt_project.yml` to `/tmp/validation_notebook_working/<PR#>/dbt_project.yml`. ### If Local mode (Local Directory): 1. Change to the target directory. 2. Get current branch info: ```bash git rev-parse --abbrev-ref HEAD ``` 3. Detect base branch - try `main`, `master`, `develop` in order, or use upstream tracking branch. 4. Get the list of changed SQL files compared to base branch: ```bash git diff --name-only <base_branch>...HEAD -- '*.sql' ``` 5. Filter to only `.sql` files under `models/` or `snapshots/` directories (at any depth — e.g., `models/`, `analytics/models/`, `dbt/models/`). If no model SQL files were changed, report that and stop. 6. Get the diff for each changed file: ```bash git diff <base_branch>...HEAD -- <file_path> ``` 7. Read model files directly from the filesystem. 8. **Find dbt_project.yml**: ```bash find . -name "dbt_project.yml" -type f | head -1 ``` 9. For notebook metadata in local mode, use: - **ID**: `local-<branch-name>-<timestamp>` - **Title**: `Local: <branch-name>` - **Author**: Output of `git config user.name` - **Merged**: "N/A (local)" ### Model Selection (applies to both modes) After filtering to `.sql` files under `models/` or `snapshots/`: 1. **If `--models` was specified:** Filter the changed files list to only include models whose filename (without `.sql` extension, case-insensitive) matches one of the specified model names. If any specified model is not found in the changed files, warn the user but continue with the models that were found. If none match, report that and stop. 2. **Model cap:** If more than 10 models remain after filtering, select the first 10 (by file path order) and warn the user: ``` ⚠️ <total_count> models changed — generating validation queries for the first 10 only. To generate for specific models, re-run with: --models <model1,model2,...> Skipped models: <list of skipped model filenames> ``` ## Phase 2: Parse Changed Models For EACH changed dbt model `.sql` file, parse and extract: ### 2a. Model Metadata **Output table name** -- Derive from file name: - `<any_path>/models/<subdir>/<model_name>.sql` -> table is `<MODEL_NAME>` (uppercase, taken from the filename) **Output schema** -- Use the schema resolution script: 1. **Setup**: Save `dbt_project.yml` and model files to `/tmp/validation_notebook_working/<id>/` preserving paths: ``` /tmp/validation_notebook_working/<id>/ +-- dbt_project.yml +-- models/ +-- <path>/<model>.sql ``` 2. **Run the script** for each model: ```bash python3 ${CLAUDE_PLUGIN_ROOT}/skills/generate-validation-notebook/scripts/resolve_dbt_schema.py /tmp/validation_notebook_working/<id>/dbt_project.yml /tmp/validation_notebook_working/<id>/models/<path>/<model>.sql ``` 3. **Error handling**: If the script fails, **STOP immediately** and report the error. Do NOT proceed with notebook generation if schema resolution fails. 4. **Output**: The script prints the resolved schema (e.g., `PROD`, `PROD_STAGE`, `PROD_LINEAGE`) **Note**: Do NOT manually parse dbt_project.yml or model configs for schema -- always use the script. It handles model config overrides, dbt_project.yml routing rules, PROD_ prefix for custom schemas, and defaults to `PROD`. **Config block** -- Look for `{{ config(...) }}` and extract: - `materialized` -- 'table', 'view', 'incremental', 'ephemeral' - `unique_key` -- the dedup key (may be a string or list) - `cluster_by` -- clustering fields (may contain the time axis) **Core segmentation fields** -- Scan the entire model SQL for fields likely to be business keys: - Fields named `*_id` (e.g., `account_id`, `resource_id`, `monitor_id`) that appear in JOIN ON, GROUP BY, PARTITION BY, or `unique_key` - Deduplicate and rank by frequency. Take the top 3. **Time axis field** -- Detect the model's time dimension (in priority order): 1. `is_incremental()` block: field used in the WHERE comparison 2. `cluster_by` config: timestamp/date fields 3. Field name conventions: `ingest_ts`, `created_time`, `date_part`, `timestamp`, `run_start_time`, `export_ts`, `event_created_time` 4. ORDER BY DESC in QUALIFY/ROW_NUMBER If no time axis is found, skip time-axis queries for this model. ### 2b. Diff Analysis Parse the diff hunks for this file. Classify each changed line: - **Changed fields** -- Lines added/modified in SELECT clauses or CTE definitions. Extract the output column name. - **Changed filters** -- Lines added/modified in WHERE clauses. - **Changed joins** -- Lines added/modified in JOIN ON conditions. - **Changed unique_key** -- If `unique_key` in config was modified, note both old and new values. - **New columns** -- Columns in "after" SELECT that don't appear in "before" (pure additions). ### 2c. Model Classification Classify each model as **new** or **modified** based on the diff: - If the diff for this file contains `new file mode` → classify as **new** - Otherwise → classify as **modified** This classification determines which query patterns are generated in Phase 3. **Note:** For **new models**, Phase 2b diff analysis is skipped (there is no "before" to compare against). Phase 2a metadata extraction still applies. ## Phase 3: Generate Validation Queries For each changed model, generate the applicable queries based on its classification (new vs modified). **CRITICAL: Parameter Placeholder Syntax** Use **double curly braces** `{{...}}` for parameter placeholders. Do NOT use `${...}` or any other syntax. Correct: `{{prod_db}}.PROD.AGENT_RUNS` Wrong: `${prod_db}.PROD.AGENT_RUNS` **Table Reference Format:** - Use `{{prod_db}}.<SCHEMA>.<TABLE_NAME>` for prod queries - Use `{{dev_db}}.<SCHEMA>.<TABLE_NAME>` for dev queries - `<SCHEMA>` is **hardcoded per-model** using the output from the schema resolution script --- ### Query Patterns for NEW Models For new models, all queries target `{{dev_db}}` only. No comparison queries are generated since no prod table exists. #### Pattern 7-new: Total Row Count **Trigger:** Always. ```sql SELECT COUNT(*) AS total_rows FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> ``` #### Pattern 9: Sample Data Preview **Trigger:** Always. ```sql SELECT * FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> LIMIT 20 ``` #### Pattern 2-new: Core Segmentation Counts **Trigger:** Always. ```sql SELECT <segmentation_field>, COUNT(*) AS row_count FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> GROUP BY <segmentation_field> ORDER BY row_count DESC LIMIT 100 ``` #### Pattern 5: Uniqueness Check **Trigger:** Always for new models (verify unique_key constraint from the start). ```sql SELECT COUNT(*) AS total_rows, COUNT(DISTINCT <key_fields>) AS distinct_keys, COUNT(*) - COUNT(DISTINCT <key_fields>) AS duplicate_count FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> ``` ```sql SELECT <key_fields>, COUNT(*) AS n FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> GROUP BY <key_fields> HAVING COUNT(*) > 1 ORDER BY n DESC LIMIT 100 ``` #### Pattern 6-new: NULL Rate Check (all columns) **Trigger:** Always. Checks all output columns since everything is new. ```sql SELECT COUNT(*) AS total_rows, SUM(CASE WHEN <col1> IS NULL THEN 1 ELSE 0 END) AS <col1>_null_count, ROUND(100.0 * SUM(CASE WHEN <col1> IS NULL THEN 1 ELSE 0 END) / NULLIF(COUNT(*), 0), 2) AS <col1>_null_pct, SUM(CASE WHEN <col2> IS NULL THEN 1 ELSE 0 END) AS <col2>_null_count, ROUND(100.0 * SUM(CASE WHEN <col2> IS NULL THEN 1 ELSE 0 END) / NULLIF(COUNT(*), 0), 2) AS <col2>_null_pct -- repeat for each output column FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> ``` #### Pattern 8: Time-Axis Continuity **Trigger:** Model is `materialized='incremental'` OR a time axis field was identified. ```sql SELECT CAST(<time_axis> AS DATE) AS day, COUNT(*) AS row_count FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> WHERE <time_axis> >= CURRENT_TIMESTAMP - INTERVAL '14' DAY GROUP BY day ORDER BY day DESC LIMIT 30 ``` --- ### Query Patterns for MODIFIED Models For modified models, single-table queries use `{{prod_db}}` and comparison queries use both. #### Pattern 7: Total Row Count **Trigger:** Always. ```sql SELECT COUNT(*) AS total_rows FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME> ``` #### Pattern 9: Sample Data Preview **Trigger:** Always. ```sql SELECT * FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME> LIMIT 20 ``` #### Pattern 2: Core Segmentation Counts **Trigger:** Always. ```sql SELECT <segmentation_field>, COUNT(*) AS row_count FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME> GROUP BY <segmentation_field> ORDER BY row_count DESC LIMIT 100 ``` #### Pattern 1: Changed Field Distribution **Trigger:** Changed fields found in Phase 2b. **Exclude added columns** (from "New columns" in Phase 2b) — only include fields that exist in prod. ```sql SELECT <changed_field>, COUNT(*) AS row_count, ROUND(COUNT(*) * 100.0 / SUM(COUNT(*)) OVER(), 2) AS pct FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME> GROUP BY <changed_field> ORDER BY row_count DESC LIMIT 100 ``` #### Pattern 5: Uniqueness Check **Trigger:** JOIN condition changed, `unique_key` changed, or model is incremental. ```sql SELECT COUNT(*) AS total_rows, COUNT(DISTINCT <key_fields>) AS distinct_keys, COUNT(*) - COUNT(DISTINCT <key_fields>) AS duplicate_count FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> ``` ```sql SELECT <key_fields>, COUNT(*) AS n FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> GROUP BY <key_fields> HAVING COUNT(*) > 1 ORDER BY n DESC LIMIT 100 ``` #### Pattern 6: NULL Rate Check **Trigger:** New column added, or column wrapped in COALESCE/NULLIF. **Important:** Added columns (from "New columns" in Phase 2b) do NOT exist in prod yet. For added columns, query `{{dev_db}}` only. For modified columns (COALESCE/NULLIF changes), compare both databases. **For added columns** (dev only): ```sql SELECT COUNT(*) AS total_rows, SUM(CASE WHEN <column> IS NULL THEN 1 ELSE 0 END) AS null_count, ROUND(100.0 * SUM(CASE WHEN <column> IS NULL THEN 1 ELSE 0 END) / NULLIF(COUNT(*), 0), 2) AS null_pct FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> ``` **For modified columns** (prod vs dev): ```sql SELECT 'prod' AS source, COUNT(*) AS total_rows, SUM(CASE WHEN <column> IS NULL THEN 1 ELSE 0 END) AS null_count, ROUND(100.0 * SUM(CASE WHEN <column> IS NULL THEN 1 ELSE 0 END) / NULLIF(COUNT(*), 0), 2) AS null_pct FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME> UNION ALL SELECT 'dev' AS source, COUNT(*) AS total_rows, SUM(CASE WHEN <column> IS NULL THEN 1 ELSE 0 END) AS null_count, ROUND(100.0 * SUM(CASE WHEN <column> IS NULL THEN 1 ELSE 0 END) / NULLIF(COUNT(*), 0), 2) AS null_pct FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> ``` #### Pattern 8: Time-Axis Continuity **Trigger:** Model is `materialized='incremental'` OR a time axis field was identified. ```sql SELECT CAST(<time_axis> AS DATE) AS day, COUNT(*) AS row_count FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME> WHERE <time_axis> >= CURRENT_TIMESTAMP - INTERVAL '14' DAY GROUP BY day ORDER BY day DESC LIMIT 30 ``` #### Pattern 3: Before/After Comparison **Trigger:** Always (for changed fields + top segmentation field). **Modified models only.** **Important:** Exclude added columns (from "New columns" in Phase 2b) from `<group_fields>`. Only use fields that exist in BOTH prod and dev. Added columns don't exist in prod and will cause query errors. ```sql WITH prod AS ( SELECT <group_fields>, COUNT(*) AS cnt FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME> GROUP BY <group_fields> ), dev AS ( SELECT <group_fields>, COUNT(*) AS cnt FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> GROUP BY <group_fields> ) SELECT COALESCE(b.<field>, d.<field>) AS <field>, COALESCE(b.cnt, 0) AS cnt_prod, COALESCE(d.cnt, 0) AS cnt_dev, COALESCE(d.cnt, 0) - COALESCE(b.cnt, 0) AS diff FROM prod b FULL OUTER JOIN dev d ON b.<field> = d.<field> ORDER BY ABS(diff) DESC LIMIT 100 ``` #### Pattern 7b: Row Count Comparison **Trigger:** Always. **Modified models only.** ```sql SELECT 'prod' AS source, COUNT(*) AS row_count FROM {{prod_db}}.<SCHEMA>.<TABLE_NAME> UNION ALL SELECT 'dev' AS source, COUNT(*) AS row_count FROM {{dev_db}}.<SCHEMA>.<TABLE_NAME> ``` ## Phase 4: Build Notebook YAML ### 4a. Metadata ```yaml version: 1 metadata: id: validation-pr-<PR_NUMBER>-<random_suffix> name: "Validation: PR #<PR_NUMBER> - <PR_TITLE_TRUNCATED>" created_at: "<current_iso_timestamp>" updated_at: "<current_iso_timestamp>" ``` ### 4b. Parameter Cells **Only include `prod_db` if there are modified models.** If all models are new, only include `dev_db`. ```yaml # Include ONLY if there are modified models: - id: param-prod-db type: parameter content: name: prod_db config: type: text default_value: "ANALYTICS" placeholder: "Prod database (e.g., ANALYTICS)" display_type: table # Always include: - id: param-dev-db type: parameter content: name: dev_db config: type: text default_value: "PERSONAL_<USER>" placeholder: "Dev database (e.g., PERSONAL_JSMITH)" display_type: table ``` ### 4c. Markdown Summary Cell ```yaml - id: cell-summary type: markdown content: | # Validation Queries for <PR or Local Branch> ## Summary - **Title:** <title> - **Author:** <author> - **Source:** <PR URL or "Local branch: <branch>"> - **Status:** <merge_timestamp or "Not yet merged" or "N/A (local)"> ## Changes <brief description based on diff analysis> ## Changed Models - `<SCHEMA>.<TABLE_NAME>` (from `<file_path>`) ## How to Use 1. Select your Snowflake connector above 2. Set **dev_db** to your dev database (e.g., `PERSONAL_JSMITH`) 3. If modified models are present, set **prod_db** to your prod database (e.g., `ANALYTICS`) 4. Run single-table queries first, then comparison queries display_type: table ``` ### 4d. SQL Cell Format ```yaml - id: cell-<pattern>-<model>-<index> type: sql content: | /* ======================================== <Pattern Name (human-readable, e.g. "Total Row Count" — do NOT include pattern numbers like "Pattern 7:")> ======================================== Model: <SCHEMA>.<TABLE_NAME> Triggered by: <why this pattern was generated> What to look for: <interpretation guidance> ---------------------------------------- */ <actual_sql_query> display_type: table ``` ### 4e. Cell Organization Cells are ordered consistently for both model types, following this sequence: **New models:** 1. Summary markdown cell (note that model is new) 2. Parameter cells (dev_db only — no prod_db if all models are new) 3. Total row count (Pattern 7-new) 4. Sample data preview (Pattern 9) 5. Core segmentation counts (Pattern 2-new) 6. Uniqueness check (Pattern 5), NULL rate check (Pattern 6-new), Time-axis continuity (Pattern 8) **Modified models:** 1. Summary markdown cell 2. Parameter cells (prod_db, dev_db) 3. Total row count (Pattern 7) 4. Sample data preview (Pattern 9) 5. Core segmentation counts (Pattern 2) 6. Changed field distribution (Pattern 1) 7. Uniqueness check (Pattern 5), NULL rate check (Pattern 6), Time-axis continuity (Pattern 8) 8. Before/after comparisons (Pattern 3), Row count comparison (Pattern 7b) ## Phase 5: Generate Import URL 1. Write notebook YAML to `/tmp/validation_notebook_working/<id>/notebook.yaml` 2. Run the URL generation script: ```bash python3 ${CLAUDE_PLUGIN_ROOT}/skills/generate-validation-notebook/scripts/generate_notebook_url.py /tmp/validation_notebook_working/<id>/notebook.yaml --mc-base-url <MC_BASE_URL> ``` 3. The script validates both YAML syntax and notebook schema (required fields on metadata and cells). If validation fails, read the error messages carefully, fix the YAML to match the spec in Phase 4, and re-run. ## Phase 6: Output Present: ```markdown # Validation Notebook Generated ## Summary - **Source:** PR #<number> - <title> OR Local: <branch> - **Author:** <author> - **Changed Models:** <count> models (of <total_count> changed) - **Generated Queries:** <count> queries > ⚠️ If models were capped: "Only the first 10 of <total_count> changed models were included. Re-run with `--models` to select specific models." ## Notebook Opened The notebook has been opened directly in your browser. Select your Snowflake connector in the notebook interface to begin running queries. *Make sure MC Bridge is running. Let me know if you want tips on how to install this locally* ``` ## Important Guidelines 1. **Do NOT execute queries** -- only generate the notebook 2. **Keep SQL readable** -- proper formatting and meaningful aliases 3. **Include LIMIT 100** on queries that could return many rows 4. **Use double curly braces** -- `{{prod_db}}` NOT `${prod_db}` 5. **Use correct table format** -- `{{prod_db}}.<SCHEMA>.<TABLE>` and `{{dev_db}}.<SCHEMA>.<TABLE>` 6. **Always use the schema resolution script** -- do NOT manually parse dbt_project.yml 7. **Schema is NOT a parameter** -- only `prod_db` and `dev_db` are parameters 8. **Skip ephemeral models** -- they have no physical table 9. **Truncate notebook name** -- keep under 50 chars 10. **Generate unique cell IDs** -- use pattern like `cell-p3-model-1` 11. **YAML multiline content** -- use `|` block scalar for SQL with comments 12. **ASCII-only YAML** -- the script sanitizes and validates before encoding ## Query Pattern Reference | Pattern | Name | Trigger | Model Type | Database | Order | |---------|------|---------|------------|----------|-------| | 7 / 7-new | Total Row Count | Always | Both | `{{prod_db}}` (modified) / `{{dev_db}}` (new) | 1 | | 9 | Sample Data Preview | Always | Both | `{{prod_db}}` (modified) / `{{dev_db}}` (new) | 2 | | 2 / 2-new | Core Segmentation Counts | Always | Both | `{{prod_db}}` (modified) / `{{dev_db}}` (new) | 3 | | 1 | Changed Field Distribution | Column modified in diff (not added) | Modified only | `{{prod_db}}` | 4 | | 5 | Uniqueness Check | JOIN/unique_key changed (modified) / Always (new) | Both | `{{dev_db}}` | 5 | | 6 / 6-new | NULL Rate Check | New column or COALESCE (modified) / Always (new) | Both | Added col: `{{dev_db}}` only; COALESCE: Both (modified) / `{{dev_db}}` (new) | 5 | | 8 | Time-Axis Continuity | Incremental or time field | Both | `{{prod_db}}` (modified) / `{{dev_db}}` (new) | 5 | | 3 | Before/After Comparison | Changed fields (not added) | Modified only | Both | 6 | | 7b | Row Count Comparison | Always | Modified only | Both | 6 | ## MC Bridge Setup Help If the user asks how to install or set up MC Bridge, fetch the README from the mc-bridge repo and show the relevant quick start / setup instructions: ```bash gh api repos/monte-carlo-data/mc-bridge/readme --jq '.content' | base64 --decode ``` Focus on: how to install, configure connections, and run MC Bridge. Don't dump the entire README — extract just the setup-relevant sections.
Comments (0)
Sign in to join the conversation.
Reviews (0)
No reviews yet.
No comments yet.