Server data from the Official MCP Registry
Pre-flight query cost and result-size guardrails for AI agents on BigQuery, Snowflake, Databricks
About
Pre-flight query cost and result-size guardrails for AI agents on BigQuery, Snowflake, Databricks
Security Report
Valid MCP server (2 strong, 2 medium validity signals). 2 known CVEs in dependencies Package registry verified. Imported from the Official MCP Registry.
4 files analyzed · 3 issues found
Security scores are indicators to help you make informed decisions, not guarantees. Always review permissions before connecting any MCP server.
Permissions Required
This plugin requests these system permissions. Most are normal for its category.
What You'll Need
Set these up before or after installing:
Environment variable: GOOGLE_APPLICATION_CREDENTIALS
Environment variable: SNOWFLAKE_ACCOUNT
Environment variable: SNOWFLAKE_USER
Environment variable: SNOWFLAKE_ROLE
Environment variable: SNOWFLAKE_SECONDARY_ROLES
Environment variable: SNOWFLAKE_PRIVATE_KEY_PATH
Environment variable: SNOWFLAKE_PRIVATE_KEY_PASSPHRASE
Environment variable: SNOWFLAKE_PASSWORD
Environment variable: SNOWFLAKE_EDITION
Environment variable: DATABRICKS_SERVER_HOSTNAME
Environment variable: DATABRICKS_HTTP_PATH
Environment variable: DATABRICKS_TOKEN
Environment variable: DATABRICKS_CLIENT_ID
Environment variable: DATABRICKS_CLIENT_SECRET
Environment variable: COST_GUARD_ALLOW_MULTI_STATEMENT_SQL
Environment variable: COST_GUARD_ALLOW_WRITE_STATEMENTS
Environment variable: COST_GUARD_MAX_IN_FLIGHT_QUERIES
Environment variable: COST_GUARD_DATABRICKS_ALLOW_WAREHOUSE_START
Environment variable: COST_GUARD_SNOWFLAKE_ALLOW_WAREHOUSE_RESUME
How to Install
Add this to your MCP configuration file:
{
"mcpServers": {
"io-github-mcpsmiths-cost-guard-mcp": {
"env": {
"SNOWFLAKE_ROLE": "your-snowflake-role-here",
"SNOWFLAKE_USER": "your-snowflake-user-here",
"DATABRICKS_TOKEN": "your-databricks-token-here",
"SNOWFLAKE_ACCOUNT": "your-snowflake-account-here",
"SNOWFLAKE_EDITION": "your-snowflake-edition-here",
"SNOWFLAKE_PASSWORD": "your-snowflake-password-here",
"DATABRICKS_CLIENT_ID": "your-databricks-client-id-here",
"DATABRICKS_HTTP_PATH": "your-databricks-http-path-here",
"DATABRICKS_CLIENT_SECRET": "your-databricks-client-secret-here",
"SNOWFLAKE_SECONDARY_ROLES": "your-snowflake-secondary-roles-here",
"DATABRICKS_SERVER_HOSTNAME": "your-databricks-server-hostname-here",
"SNOWFLAKE_PRIVATE_KEY_PATH": "your-snowflake-private-key-path-here",
"GOOGLE_APPLICATION_CREDENTIALS": "your-google-application-credentials-here",
"COST_GUARD_MAX_IN_FLIGHT_QUERIES": "your-cost-guard-max-in-flight-queries-here",
"SNOWFLAKE_PRIVATE_KEY_PASSPHRASE": "your-snowflake-private-key-passphrase-here",
"COST_GUARD_ALLOW_WRITE_STATEMENTS": "your-cost-guard-allow-write-statements-here",
"COST_GUARD_ALLOW_MULTI_STATEMENT_SQL": "your-cost-guard-allow-multi-statement-sql-here",
"COST_GUARD_DATABRICKS_ALLOW_WAREHOUSE_START": "your-cost-guard-databricks-allow-warehouse-start-here",
"COST_GUARD_SNOWFLAKE_ALLOW_WAREHOUSE_RESUME": "your-cost-guard-snowflake-allow-warehouse-resume-here"
},
"args": [
"cost-guard-mcp"
],
"command": "uvx"
}
}
}Documentation
View on GitHubFrom the project's GitHub README.
cost-guard-mcp
Pre-flight query cost & result-size guardrails for AI agents, across BigQuery, Snowflake, and Databricks — before the query ever runs.
Why
An AI agent using a warehouse MCP can silently trigger a full-table scan that costs hundreds of dollars, or return millions of rows that flood its own context window. No existing warehouse MCP tells the agent "how much will this cost" or "how much data will this return" before running the query.
What makes this different
- Every cost estimate discloses its accuracy tier —
PRECISE(BigQuerydryRun),UPPER_BOUND(SnowflakeEXPLAIN), orHEURISTIC(DatabricksEXPLAIN COST) — so your agent never over-trusts a heuristic number. - Per-call bounds —
run_query_boundedtakesmax_bytes_billed/max_rows/max_estimated_cost_usdon each call; no shared session state required.max_rowsdefaults to 1000 and the serialised result is capped at 8 MiB;executedon every result says whether the query ran. - Zero infrastructure — a single local stdio process. No database, no gateway, no Docker Compose.
Tools
check_credentials(engine, warehouse?)— verifies credentials/connectivity without running a real query or a dry run; call this first after configuring a new engine. Supports BigQuery, Snowflake, and Databricks. What runs: BigQuery makes one metadata call; Snowflake logs in, runsUSE WAREHOUSEwhen a warehouse is given (SELECT CURRENT_WAREHOUSE()when none is), reads the pinned warehouse'sSHOW WAREHOUSESrow, and runs one session-context statement only when that row saysSTARTED— otherwiseok: truewith a detail built from the row (the statement could resume a suspended warehouse); Databricks reads the configured SQL warehouse through the Warehouses API (a control-planeGET) and opens a session (SELECT current_user()) only when it isRUNNING— otherwiseok: falsewith no session (the session could start a stopped warehouse). An unreadable state counts as not running on both. Opt in withCOST_GUARD_SNOWFLAKE_ALLOW_WAREHOUSE_RESUME=true/COST_GUARD_DATABRICKS_ALLOW_WAREHOUSE_START=true— see Known limitations.describe_engine_capabilities(engine)— what's exact vs. approximate for this engine.estimate_query_cost(engine, sql, warehouse?, warehouse_size?, edition?)— pre-flight cost estimate, tagged with its accuracy tier. Supports BigQuery, Snowflake, and Databricks. Accepts exactly one SQL statement per call, and only read statements by default — first keywordSELECT,WITH,VALUES,SHOW,DESCRIBE,DESCorEXPLAIN(see Known limitations).warehouse_size(Snowflake/Databricks) andedition(Snowflake) can only raise the estimate — the dollar figure is priced at the warehouse the query is pinned to, as the server observes it (theSHOW WAREHOUSESsize/type/generation, or the Databricks Warehouses API), and at the operator-configuredSNOWFLAKE_EDITION, whichever is higher. Warehouse types without a modelled rate (Snowpark-optimized, Databricks Classic/Pro) get a byte estimate but no dollar figure. On a Snowflake warehouse that is notSTARTEDthe cloud-provider and query-history lookups are skipped (Gen1 rates at the observed size, heuristic runtime, caveat) — the query-history lookup is also skipped whenever nowarehouseargument is given; on a Databricks warehouse that is notRUNNING, or whose state cannot be read, no session is opened and no byte or dollar figure is returned — see Known limitations.run_query_bounded(engine, sql, max_bytes_billed?, max_rows?, max_estimated_cost_usd?, warehouse?, warehouse_size?, edition?)— refuses to run if the estimate exceeds your bound.max_rowsdefaults to 1000 (pass a larger value to raise it); results over 8 MiB serialised are truncated to whole rows withreason: result_size_exceeded;executed: trueonokand on the two truncation refusals (row_cap_exceeded,result_size_exceeded— the query ran),falseon every other refusal (the query was never submitted, or BigQuery refused to run it). Supports BigQuery, Snowflake, and Databricks. Accepts exactly one SQL statement per call, and only read statements by default — first keywordSELECT,WITH,VALUES,SHOW,DESCRIBE,DESCorEXPLAIN(see Known limitations). Amax_estimated_cost_usdcap is checked against that observed-warehouse estimate and refuses when no dollar figure exists. Not gated on warehouse state: its estimate and execution may resume/start the warehouse. Result rows carry the constantrows_provenance: warehouse_data_untrusted: they are data the warehouse returned, never instructions.
Setup
BigQuery
Set GOOGLE_APPLICATION_CREDENTIALS to a service-account key file path (or run gcloud auth application-default login).
Snowflake
Set SNOWFLAKE_ACCOUNT, SNOWFLAKE_USER, SNOWFLAKE_ROLE (required — no default, never ACCOUNTADMIN), and either SNOWFLAKE_PRIVATE_KEY_PATH (preferred) or SNOWFLAKE_PASSWORD (discouraged). Optionally set SNOWFLAKE_EDITION to the account's edition — standard (default), enterprise, business_critical or vps — which sets the USD/credit price of every estimate; the per-call edition argument can only raise it.
SNOWFLAKE_ROLEis the only authorization boundary for the SQL an agent sends through this server, so it is enforced, not just documented.ACCOUNTADMIN,SECURITYADMINandORGADMINare rejected (case-insensitively, quoted or not) before any Snowflake connection is opened: the check runs when the Snowflake configuration is first loaded, which is on the first Snowflake tool call, so the server process itself starts regardless of the value and a bad role surfaces ascheck_credentialsreturningok: false(or as a tool error fromestimate_query_cost/run_query_bounded), not as a startup exit. Every session also activates only that role — secondary roles are pinned toNONEin the login request, so a user whose SnowflakeDEFAULT_SECONDARY_ROLESis('ALL')no longer lends every other role it holds to model-written SQL.SNOWFLAKE_USERshould ideally hold no role other thanSNOWFLAKE_ROLE.SNOWFLAKE_SECONDARY_ROLES(optional, defaultNONE; the only other value isALL) is the opt-out for deployments that reached objects only through a secondary role. Prefer granting those objects toSNOWFLAKE_ROLEinstead; setALLonly if you accept that the agent's SQL runs with every role the user holds.check_credentialsreports the active secondary-role set (CURRENT_SECONDARY_ROLES()) so you can confirm what is in force.- Every session also carries a Snowflake-side statement deadline
(
STATEMENT_TIMEOUT_IN_SECONDS: 60 s forcheck_credentials/estimate_query_cost, 120 s forrun_query_bounded, matching its watchdog) so a query cannot keep running — and billing — if this process dies mid-call. The value is applied read-then-lower: the server reads the session's effective value first and only issuesALTER SESSIONwhen that value is 0 (unbounded), unparseable, or above the target. A stricter account- or user-level policy (say 30 s) is therefore never raised; note that a session-level value would otherwise override account/user levels in Snowflake's parameter hierarchy (only the warehouse level combines lowest-wins). - Snowflake sessions are also pinned to one statement per call:
EXPLAIN USING JSONand therun_query_boundedexecution are submitted withnum_statements=1and the session logs in withMULTI_STATEMENT_COUNT=1, so Snowflake itself rejects;-separated text that the server-side scanner (see Known limitations) did not already refuse. There is no opt-out; multi-statement Snowflake text was never supported. COST_GUARD_SNOWFLAKE_ALLOW_WAREHOUSE_RESUME=true(optional, off by default; the value must be exactlytrue) letscheck_credentialsandestimate_query_costrunSELECT CURRENT_REGION(), theQUERY_HISTORYread and the session-contextSELECTon a warehouse whoseSHOW WAREHOUSESstate is notSTARTED(and theQUERY_HISTORYread without awarehouseargument), accepting that they can resume it and bill compute. See Known limitations for what is skipped by default.
Databricks
Set DATABRICKS_SERVER_HOSTNAME and DATABRICKS_HTTP_PATH (from the SQL warehouse's
Connection Details tab), and either DATABRICKS_TOKEN (a personal access token,
simplest) or DATABRICKS_CLIENT_ID + DATABRICKS_CLIENT_SECRET (OAuth machine-to-machine
via a service principal, preferred for automated use). Only Serverless SQL warehouses are
priced — a Classic/Pro warehouse gets a byte estimate but no dollar figure, see Known
Limitations.
By default check_credentials and estimate_query_cost refuse to open a session unless the
warehouse's state, read through the Warehouses API, is RUNNING (they also refuse when the
state cannot be read); set COST_GUARD_DATABRICKS_ALLOW_WAREHOUSE_START=true (exactly that
value) to let them start a stopped warehouse, billing compute. run_query_bounded is not
gated. See Known limitations.
Concurrency (optional)
COST_GUARD_MAX_IN_FLIGHT_QUERIES (default 8) caps how many run_query_bounded calls may
be in flight at once — counting calls the host has already cancelled, until their worker
thread exits (at most the 120 s watchdog). A call over the cap fails immediately with a tool
error naming this variable, before any warehouse connection is opened. Must be a positive
integer (the server refuses to start otherwise); raise it only if your host legitimately runs
more concurrent billable queries.
Read-only statements (optional opt-out)
By default estimate_query_cost and run_query_bounded accept only read statements: the first keyword after
comments and any opening parentheses must be SELECT, WITH, VALUES, SHOW, DESCRIBE, DESC or
EXPLAIN, and a WITH is admitted only when the statement after its CTE list resolves to SELECT or
VALUES (Spark SQL allows WITH ... INSERT). Anything else is refused before any dry run, EXPLAIN or
execution, because warehouse rows re-enter the model's context and can steer the next call
(DECISIONS.md #13). The gate is defence in depth, not the authorisation boundary: that is the deployed
credential's grants, so deploy a read-only principal — BigQuery: data-viewer plus job-user roles;
Snowflake: a role with USAGE/SELECT only, never an admin role; Databricks: SELECT/USE privileges on
the warehouse's catalogs. Operators who intend this server to run writes under its credential can set
COST_GUARD_ALLOW_WRITE_STATEMENTS=1 (exactly that value, read per call; anything else keeps the gate on).
With COST_GUARD_ALLOW_MULTI_STATEMENT_SQL=1 every statement in the text must be a read, so BigQuery
scripts (DECLARE/SET/BEGIN) need both opt-outs.
A note on credentials with MCP hosts
Whatever MCP client/host you use (Claude Desktop, etc.) spawns this server as its own subprocess — it does not automatically inherit your shell's environment variables, even if they're set in your .zshrc/.bashrc. Put them directly in the host's server config instead — see .mcp.json.example for the exact block, and the "Use with other AI coding tools" section below for where each specific tool wants it. This repository's .gitignore excludes the host configs named in that section (.mcp.json, .cursor/mcp.json, .vscode/mcp.json), private keys (*.p8/*.pem/*.key), credentials*.json, *service-account*.json and .env.* (a committed .env.example with placeholder values stays visible), so the documented layouts never commit real credentials by accident; if you choose another filename, add it to .gitignore yourself.
Install
uvx cost-guard-mcp
Also published on the official MCP Registry as io.github.mcpsmiths/cost-guard-mcp; the listing is published by release.yml from each tagged release, after PyPI has the version (DECISIONS.md #12).
For local development instead:
git clone https://github.com/mcpsmiths/cost-guard-mcp.git
cd cost-guard-mcp
uv sync
uv run cost-guard-mcp
Or via Docker:
docker build -t cost-guard-mcp .
docker run -i --rm --init -e GOOGLE_APPLICATION_CREDENTIALS=/creds.json -v /path/to/service-account.json:/creds.json cost-guard-mcp
Quickstart (~5 minutes to your first estimate)
This walks through the fastest path to a real tool call — no data of your own required (it uses a public BigQuery dataset), no Snowflake trial signup needed.
-
Get a GCP project with the BigQuery API enabled. Any project works, including the free-tier Sandbox mode (no billing card required to run
dryRun, which is allestimate_query_costdoes). Create one at console.cloud.google.com if you don't have one. -
Get Application Default Credentials: run
gcloud auth application-default loginlocally, or create a service-account key and pointGOOGLE_APPLICATION_CREDENTIALSat its JSON file. -
Add the server to your MCP client — see
.mcp.json.example, filling in onlyGOOGLE_APPLICATION_CREDENTIALS(leave the Snowflake vars out entirely for this quickstart). -
Restart your MCP client so it picks up the new server config, then ask your agent to call
check_credentialson bigquery. This confirms your setup without running any real query — you should get back"ok": trueand a detail line naming your project. If you get"ok": falseinstead, thedetailfield explains exactly what's missing (usuallyGOOGLE_APPLICATION_CREDENTIALSnot making it through to the server process — see the credentials note above, and double check the value is set inside the client's own server config block, not just your shell). -
Ask your agent to call
estimate_query_costagainst a public dataset — for example:Use cost-guard-mcp's estimate_query_cost tool on bigquery for this query:
SELECT name, SUM(number) AS total FROMbigquery-public-data.usa_names.usa_1910_2013GROUP BY name ORDER BY total DESC LIMIT 10 -
You'll know it worked when the response looks like this — the exact numbers will differ, but
accuracy_tiershould readPRECISE:{ "engine": "bigquery", "accuracy_tier": "PRECISE", "estimated_bytes": 320866545, "estimated_cost_usd": 0.001842, "currency": "USD", "caveats": [] }
Use with other AI coding tools
cost-guard-mcp is a standard stdio MCP server — any MCP-compatible client works, not just Claude Desktop. Every client ultimately runs the same command/args/env; only the wrapping file format differs, so there's one canonical definition — .mcp.json.example — instead of a separately maintained copy per tool below.
There is no single file every tool reads automatically (each looks in its own location), but three of the four use the exact same mcpServers wrapper .mcp.json.example already has, so those need nothing more than copying it into place. Fill in your real credential values, then:
| Client | Where it goes | Change needed from .mcp.json.example |
|---|---|---|
| Claude Code | .mcp.json (project) | None — copy as-is, or claude mcp add-json cost-guard-mcp '<the "cost-guard-mcp" object>' |
| Claude Desktop | claude_desktop_config.json | None — copy as-is |
| Cursor | .cursor/mcp.json or ~/.cursor/mcp.json | Add "type": "stdio" inside the server object |
| GitHub Copilot (VS Code) | .vscode/mcp.json | Rename top-level key mcpServers → servers, add "type": "stdio" |
| OpenAI Codex CLI | ~/.codex/config.toml | Same fields, TOML syntax instead of JSON (below) — or codex mcp add cost-guard-mcp -- uvx cost-guard-mcp |
Codex is the one genuine exception (TOML, not JSON), so it still needs its own block:
[mcp_servers.cost-guard-mcp]
command = "uvx"
args = ["cost-guard-mcp"]
[mcp_servers.cost-guard-mcp.env]
GOOGLE_APPLICATION_CREDENTIALS = "/path/to/service-account.json"
BIGQUERY_PROJECT = "your-project-id"
SNOWFLAKE_ACCOUNT = "your-account"
SNOWFLAKE_USER = "your-user"
SNOWFLAKE_ROLE = "your-role"
SNOWFLAKE_SECONDARY_ROLES = "NONE"
SNOWFLAKE_PRIVATE_KEY_PATH = "/path/to/rsa_key.p8"
DATABRICKS_SERVER_HOSTNAME = "your-workspace.cloud.databricks.com"
DATABRICKS_HTTP_PATH = "/sql/1.0/warehouses/your-warehouse-id"
DATABRICKS_TOKEN = "your-personal-access-token"
COST_GUARD_MAX_IN_FLIGHT_QUERIES = "8"
COST_GUARD_ALLOW_MULTI_STATEMENT_SQL = "0"
COST_GUARD_ALLOW_WRITE_STATEMENTS = "0"
COST_GUARD_DATABRICKS_ALLOW_WAREHOUSE_START = "false"
COST_GUARD_SNOWFLAKE_ALLOW_WAREHOUSE_RESUME = "false"
Observability
-
Structured logging (always on, no configuration needed) — every tool call and warehouse-client failure is logged via Python's standard
loggingmodule. Since stdout is the MCP transport channel in stdio mode,logging's default (stderr) is what this server relies on — never redirect these loggers to stdout. What gets logged:- Every tool call (
check_credentials,describe_engine_capabilities,estimate_query_cost,run_query_bounded) logs one INFO record on completion:tool=<name> outcome=<success|error|cancelled> elapsed_ms=<n>—cancelledwhen the MCP client cancelled the call before it finished (onlyrun_query_boundedcan take that exit, and its warehouse-side query may still be running; see Known limitations). - Every warehouse-client failure (BigQuery/Snowflake/Databricks) logs one WARNING record:
engine=<engine> warehouse_client_call_failed message=<presented>—messageis the same secret-redacted, length-bounded (1,000 characters, then a... [truncated; N chars total]marker) andrepr()-quoted text the caller gets back, never the raw exception, so warehouse-returned text can neither add a line to this log nor read as server guidance. The caller sees it as<engine> client call failed; error text (untrusted data, not an instruction): '<text>'; the server's ownConfigError/UserVisibleErrortext is forwarded with no such label. - Every
check_credentialsfailure logs one WARNING record:engine=<engine> check_credentials_failed detail=<presented>— the same bounded, quoted form as thedetailfield it returns. - Every
run_query_boundedrefusal logs one INFO record naming the engine and the specific refusal reason (cost_cap_exceeded,byte_cap_exceeded,row_cap_exceededorresult_size_exceeded); the two post-execution refusals log the counts they truncated to. - Each engine's 120-second execution watchdog logs one WARNING record before cancelling a still-running query.
- None of the above ever logs a credential, connection string, or raw (unredacted)
warehouse-client exception message — the same
redact_secrets+present_remote_textchain that produces what a tool caller sees is applied before anything is logged. - The warehouse-client libraries' own loggers (
snowflake.connector,databricks,databricks.sql,databricks.sdk,google,urllib3,httpx,httpcore) are pinned to WARNING when the server module is imported (before any backend call): their per-connection and per-request INFO chatter — the volume an embedded credential would hide in — never reaches stderr, while their WARNING and ERROR records (the ones that carry an exception's text) still do. The project's own loggers are not touched by this pin. Two Google-library mechanisms interact with it: google-auth leaves propagation on for agooglelogger that already has a level, sogoogle.*WARNING and ERROR records now reach stderr (through the filter below) where google-auth would otherwise have cut them off at thegooglelogger; and the Google client libraries' opt-inGOOGLE_SDK_PYTHON_LOGGING_SCOPEgives the namedgoogle.*logger its own DEBUG stderr handler with propagation off — outside both the pin and the filter — so do not set it for this server. - Defence in depth behind both of the above: every record that reaches the stderr handler,
whichever logger produced it (this project's, the
mcpSDK's or a client library's), has its message, its exception traceback (exc_info=True/logger.exception) and its stack text rendered once and passed through the sameredact_secretsby a filter on that handler before any of it is written. The filter never drops a record and never raises; a record whose format string or traceback cannot be rendered is written as<log record withheld: message could not be rendered>, traceback and stack text withheld with it, instead of the standard library's rawmsg/argserror dump. - The
mcpSDK writes its own records to the same stderr, and this project's one-line rule does not shape them (the redacting filter above does apply to them). Every tool error this project raises is followed by the SDK'sTool '<name>' failed: '<text>'INFO record (the SDKrepr()s the text, so it stays on one line, but it duplicates what the caller received); a call whose arguments pydantic rejects logsTool '<name>' rejected arguments: ['sql']and nothing else — the tool body never ran, so there is notool=<name> outcome=record for it — and the validation text that call returns to the caller (raw newlines, the offending input value echoed verbatim) is the SDK's, not this project's bounded, quoted presentation. An exception this project did not convert into a tool error (as_tool_errorconverts only its ownSanitizedEngineError,ConfigErrorandUserVisibleError) is logged by the SDK with its traceback instead. Operators parsing stderr should key on thetool=<name> outcome=records above and treat every other line as free text.
- Every tool call (
-
OpenTelemetry tracing (opt-in, off by default) — the underlying
mcpSDK ships anOpenTelemetryMiddlewareon by default for every server, wrapping each inbound message in a SERVER span, but that middleware is a documented no-op until a real exporter is registered — this project registers none unless you ask for it. SetOTEL_EXPORTER_OTLP_ENDPOINTto your OTel Collector's endpoint (e.g.http://localhost:4317) to turn it on: at that pointcost-guard-mcpconstructs aTracerProviderwith a gRPC OTLP exporter pointed at that endpoint and registers it as the global tracer provider before the server starts running. Leave the env var unset and nothing changes — no exporter is constructed, and the two extra dependencies below never need to be installed. Requires theotelextra:uv sync --extra otel # or: pip install "cost-guard-mcp[otel]"
Known limitations
- Snowflake cost estimates are calibrated from the caller's own recent query history
(
INFORMATION_SCHEMA.QUERY_HISTORY, no elevated privilege required) when an exact repeat of the same SQL text has run before - falling back to a coarse byte-size-tier heuristic otherwise. This only fires on an exact repeated query; a genuinely novel query always uses the heuristic. Result-cache hits are deliberately excluded from the average (a cached, near-instant repeat would otherwise corrupt calibration toward underestimating future runtime). Query history ingestion has its own latency - a query run moments ago may not yet be visible to the lookup, in which case it safely falls back to the heuristic rather than erroring. - Databricks calibration was investigated and found blocked: its Query History REST API
returns the query text as
"<REDACTED>"unconditionally on the account tested, even for the caller's own queries and even withinclude_metrics=True- confirmed server-side via a direct SDK source read, not something a client-side parameter can bypass. Not implemented for Databricks as a result; may be revisited if a future paid workspace confirms this is a toggleable setting there. - Databricks cost estimates are always
HEURISTIC(the least precise tier) - Databricks has no dry-run, andEXPLAIN COST's size statistics are frequently unavailable (noANALYZE TABLE, streaming sources, non-Delta external tables). Without them both the byte and the dollar estimate areNone- unavailable, not zero - andrun_query_boundedrefuses anymax_bytes_billedormax_estimated_cost_usdcap for that query rather than comparing it against a query-independent rate floor. Omit the cap to run such a query; there is deliberately no flag that restores the old floor (DECISIONS.md#5). - Databricks pricing only models Serverless SQL warehouses. The warehouse
DATABRICKS_HTTP_PATHbinds to is looked up through the Warehouses API (GET /api/2.0/sql/warehouses/{id},CAN_USEon the warehouse suffices): its cluster size is the floor of the estimate (warehouse_sizecan only raise it), and a Classic/Pro warehouse (different DBU rates plus a separate cloud VM cost not modeled here) gets no dollar figure, so amax_estimated_cost_usdcap refuses — omit the dollar cap to run there. If the lookup fails (e.g. a principal withoutCAN_VIEW/CAN_USE, or anhttp_paththat is not a/sql/1.0/warehouses/<id>path), the estimate keeps the caller-asserted size with a caveat — forrun_query_bounded, or for the read-only tools whenCOST_GUARD_DATABRICKS_ALLOW_WAREHOUSE_START=true; by defaultcheck_credentialsandestimate_query_costrefuse instead (see the warehouse-state bullet below). - Databricks has no per-query warehouse override - the SQL warehouse is fixed by
DATABRICKS_HTTP_PATHat connect time. - Snowflake's
UPPER_BOUNDestimate excludes Cortex AI Function ("AI Credits") cost. - Snowflake warehouse size, type and generation (Gen1 vs. the newer, pricier Gen2) are read
from the pinned warehouse's
SHOW WAREHOUSESrow, plusCURRENT_REGION()for the cloud provider (all ordinary, non-privileged SQL); when nowarehouseargument is given, the session's current warehouse (CURRENT_WAREHOUSE()) is used. The observed size is the floor of the estimate (warehouse_sizecan only raise it) and the generation picks the credit rate — Gen2 bills ~1.35x Gen1 on AWS/GCP and ~1.25x on Azure. A Snowpark-optimized or otherwise non-STANDARD warehouse, or an unrecognised size, gets a byte estimate but no dollar figure, so amax_estimated_cost_usdcap refuses — omit the dollar cap to run there. If the lookup fails for any reason (permission, timeout, unrecognized response shape, no current warehouse), the estimate falls back to the caller-asserted size at Gen1 rates with an explicit caveat rather than erroring — since Gen2 is now the default for new standard warehouses in most regions, an undetectable generation means the real cost may be higher than this estimate. Live-verified 2026-09-18 against a real trial account: its default warehouse (COMPUTE_WH) is genuinely Gen2 on AWS, and the detection applied the Gen2 rate; theSHOW WAREHOUSESsize/type text shape andCURRENT_WAREHOUSE()resolution are pinned bytests/integration/test_snowflake_live.py(run weekly and on demand byintegration.yml). The same row'sstateandauto_resumecolumns drive the warehouse-state guard described below. - Snowflake's edition is operator configuration (
SNOWFLAKE_EDITION, defaultstandard), because the server cannot observe it without ORGADMIN; the per-calleditionargument can only raise the USD/credit price, never lower it. A null or non-integerbytesAssignedin EXPLAIN's plan yields no byte or dollar estimate (unless a historical runtime for the exact query exists) rather than the information-free baseline figure. check_credentialsandestimate_query_costcarryreadOnlyHint: trueand never run your query, but they are not free of compute on Snowflake and Databricks: they run lightweight statements (a session-contextSELECT,EXPLAIN,SHOW WAREHOUSES,CURRENT_REGION(), aQUERY_HISTORYread; on DatabricksSELECT current_user()/EXPLAIN COST) on the named, default orDATABRICKS_HTTP_PATHwarehouse session, which can resume a suspendedAUTO_RESUMEwarehouse or start a stopped SQL warehouse and bill compute until it suspends or auto-stops again. By default (DECISIONS.md#20) the server therefore reads the warehouse's state first and gates the resume-prone statements on it. Snowflake:SELECT CURRENT_REGION(), theQUERY_HISTORYread andcheck_credentials' session-contextSELECTrun only when the pinned warehouse'sSHOW WAREHOUSESstate isSTARTED; they are skipped when it isSUSPENDED,RESUMING,SUSPENDINGorRESIZING, when no warehouse is pinned to the session, and when the state cannot be read (no or ambiguousSHOWrow, a name the identifier allowlist refuses, a NULLstatecolumn, a failed lookup); theQUERY_HISTORYread is additionally skipped wheneverestimate_query_costis called without awarehouseargument. The estimate then prices at Gen1 rates at the observed size with the byte-tier heuristic and says so in a caveat (a Gen2 warehouse may be underpriced by up to ~1.35x in that read-only estimate; it never feeds a cap), andcheck_credentialsreturnsok: truewith a detail built from theSHOW WAREHOUSESrow (login succeeded; session identity not probed).USE WAREHOUSE,EXPLAIN USING JSON,SHOW WAREHOUSES,SELECT CURRENT_WAREHOUSE()and the session-parameter statements always run. Databricks: the state is read through the Warehouses API (GET /api/2.0/sql/warehouses/{id}, a control-plane call that does not start the warehouse) and unless it isRUNNING— also when it cannot be read —check_credentialsreturnsok: falseandestimate_query_costreturns no byte or dollar figure, with no session opened; transitionalSTARTINGcounts as not running, retry once it is up. Principals withoutCAN_VIEW/CAN_USEon the warehouse therefore getok: false/ no figures from the read-only tools unless they opt in (run_query_boundedkeeps the caller-asserted-rate fallback). Opt-ins, read per call and accepted only as exactlytrue:COST_GUARD_SNOWFLAKE_ALLOW_WAREHOUSE_RESUME=trueandCOST_GUARD_DATABRICKS_ALLOW_WAREHOUSE_START=true.run_query_boundedis not gated: its estimate is taken with the lookups and the session regardless, because the execution wakes the warehouse anyway. Whether the statements that still run (USE WAREHOUSE,EXPLAIN USING JSON,SHOW WAREHOUSES,SELECT CURRENT_WAREHOUSE(); Snowflake documents context-function-onlySELECTs andSHOWas served without a running warehouse) and the gated ones actually resume a suspendedAUTO_RESUMEwarehouse is the owner's per-statement attribution test on a throwaway warehouse, recorded as the follow-up inDECISIONS.md#20; until it lands the gate errs on the side of skipping.- BigQuery Editions/capacity-billed projects cannot get a dollar estimate — only a byte count (capacity billing has no fixed $/byte rate). The billing-model check (Reservations
SearchAllAssignments, a per-location lookup with no all-locations form) is made for the location the dry-run job ran in — the location whose slots the query would consume — and falls back to theUSmulti-region only when the job reports no location, saying so in a caveat; before #96 it was alwaysUS, so a reservation assigned elsewhere went unseen. Each dry run logs one INFO record naming the location checked, where it came from (joborfallback) and the outcome; the capacity-billed and check-failed caveats name it too. - BigQuery dry runs always report 0 bytes processed for tables protected by row-level security, by design, to prevent a side-channel (Google's cost documentation: "dry runs always return 0 bytes to prevent side-channel attacks"). Since #96 (
DECISIONS.md#22)dry_runtreats 0 bytes on a statement that references at least one table as that shape:estimated_bytes: 0as reported,accuracy_tier: HEURISTIC(the hidden figure is no estimate of the bytes the query would read),estimated_cost_usd: nulland a caveat saying why — so amax_estimated_cost_usdcap refuses before dispatch (reason: cost_cap_exceeded,executed: false;DECISIONS.md#5) instead of admitting a $0.00 figure. The shape cannot be told from a genuinely free 0-byte dry run, so those are treated the same way: an unfilteredCOUNT(*)BigQuery answers from table metadata, aSELECTover an empty table and a partition filter that prunes every partition also yield no dollar figure, and a dollar cap refuses them too (the caveat names these causes alongside row-level security rather than asserting it) — usemax_bytes_billedfor them. The same holds when BigQuery reportstotalBytesProcessedAccuracy: LOWER_BOUND("estimate is lower bound of what the query would cost" — a floor, not a cap): the byte figure is kept, taggedHEURISTIC, with no dollar figure. Bound such runs withmax_bytes_billed, which BigQuery enforces server-side asmaximumBytesBilled(the byte gate admits the reported 0 / floor, and the provider evaluates the limit before the job runs).UNKNOWNaccuracy and values this server does not recognise keep theUPPER_BOUNDtag and the dollar figure; a statement that references no table (SELECT 1) legitimately prices at $0.00. The read-onlyestimate_query_costshares the same dry run and shows the same facts. - On an on-demand BigQuery project,
run_query_bounded'smax_estimated_cost_usdis also enforced server-side: the job carries amaximumBytesBilledderived from the dollar cap (cap ÷ $6.25/TiB, rounded up to the next MiB, floored at BigQuery's 10 MiB minimum), lowered further to a caller-suppliedmax_bytes_billedbut never raised above it. Since #96 the row-level-security andLOWER_BOUNDshapes carry no dollar figure and the dollar gate refuses them before dispatch (previous bullet); this derived cap is the second layer for the figures that do pass —UNKNOWNor unrecognised accuracy, absent statistics on a statement that references no table, and the dry-run-versus-run gap in general. A query that would bill more than that limit is rejected by BigQuery with no charge and surfaces as a normalstatus: "refused"result (reason: cost_cap_exceededwhen the limit came from the dollar cap,byte_cap_exceededwhen it was your own byte cap), not as an error. Residual: BigQuery bills a 10 MiB minimum per referenced table, so a tight cap on a many-table query can still be provider-rejected after passing the gate — it now shows up as that structured refusal rather than a provider error string. ANaNmax_estimated_cost_usdis rejected outright (it would compare as "under cap" for every estimate);+infcompares as "no dollar cap" against any dollar figure and derives no server-side byte cap, but it is still a requested dollar cap: when the estimate carries no dollar figure (capacity billing, a failed billing-model check, the row-level-security shape,LOWER_BOUND) the call is refused like any other dollar cap — omit the parameter, or usemax_bytes_billed. - BigQuery remote functions and BigQuery ML remote-model inference (e.g.
ML.GENERATE_TEXT) incur separate Cloud Run/Vertex AI billing that this byte-based dollar estimate does not include —dry_runflags this with a conservative text-based heuristic (ML.GENERATE_TEXTorCREATE FUNCTION+REMOTEin the query text) rather than the dry-run response'sreferencedRoutinesfield. Attempted to resolve this live (2026-09-23) once BigQuery credentials existed in CI: the deliberately least-privilege service account correctly lacksbigquery.datasets.create, so a throwaway routine could not be created to test dry-run population ofreferencedRoutines— and granting broader access just to answer this research question would contradict the least-privilege posture this project maintains everywhere else. Still an open follow-up for whoever next has a project where they can test this safely. estimate_query_costandrun_query_boundedaccept exactly one SQL statement per call. The cost cap is checked against one estimate of the submitted text, and the identical text is then handed to the engine — a second statement riding behind a top-level;would execute under a cap that was never checked against it. A small fail-closed scanner (src/cost_guard_mcp/sql_scope.py, no SQL-parser dependency) therefore refuses, before any dry run/EXPLAIN or execution, any text with a top-level;followed by more SQL; a single trailing;(optionally followed by whitespace/comments) is fine. It understands only the literal and comment forms each dialect actually has ('...'with\'/'',"...",--//* */everywhere; BigQuery backticks,#comments, triple quotes andr/b/rbprefixes; Databricks backticks,rraw strings andX'..'hex; Snowflake$$...$$and//comments) and refuses anything it cannot resolve with certainty: an unterminated literal/comment, an identifier glued to a quote (THEN'x',DATE'2020-01-01'— add a space), a backslash right before the closing quote of a raw/hex literal or quoted identifier, or a/*inside a block comment; and, since #104, a line comment containing one of the Unicode line-separator characters U+000B, U+000C, U+001C–U+001E, U+0085, U+2028 or U+2029 (no documented lexer ends a comment at them, and a quote placed right after such a character could hide a second statement behind the engine's;), a bare carriage return inside a line comment on BigQuery or Snowflake (their lexers' comment end could not be verified, so only LF and CRLF end a comment there; Spark SQL ends one at CR or LF, so Databricks accepts it), or a backslash right before the newline that ends a line comment (Spark continues the comment). Those refusals are deliberate false positives in the safe direction. Operators who knowingly run multi-statement scripts (e.g. BigQueryDECLARE ...; SELECT ...;) can setCOST_GUARD_ALLOW_MULTI_STATEMENT_SQL=1on the server to skip the check, accepting that the cap then prices only what the provider's dry run/EXPLAIN reported for the whole script. The opt-out only skips this server's scanner, and two limits follow from that: it does not apply on Snowflake, where the per-callMULTI_STATEMENT_COUNT=1pin (see Setup) still rejects the text (Actual statement count 2 did not match the desired statement count 1.); and the row cap does not wrap a script: only a singleSELECT/WITH/VALUESstatement (a non-recursiveWITH) is wrapped inSELECT * FROM (...) AS cost_guard_row_cap LIMIT n; for anything else the fetch stops atmax_rows + 1rows instead (max_rowsnow defaults to 1000).estimate_query_costandrun_query_boundedaccept only read statem
Documentation truncated — see the full README on GitHub.
Reviews
No reviews yet
Be the first to review this server!
More Developer Tools MCP Servers
Rea
Freeby Morluto · Developer Tools
Reverse engineer anything from your terminal or agent with one CLI and MCP server.
Paperclip
Freeby Paperclipai · Developer Tools
Trending hip-hop artist momentum scores across four cultural dimensions.
Git
Freeby Modelcontextprotocol · Developer Tools
Read, search, and manipulate Git repositories programmatically
Toleno
Freeby Toleno · Developer Tools
Toleno Network MCP Server — Manage your Toleno mining account with Claude AI using natural language.
MCP Marketplace
Freeby mcp-marketplace · Developer Tools
Search and install MCP servers from inside your AI client.
mcp-creator-python
Freeby mcp-marketplace · Developer Tools
Create, build, and publish Python MCP servers to PyPI — conversationally.
