Pre-flight query cost and result-size guardrails for AI agents across BigQuery, Snowflake, and Databricks
Setup
Choose your client, copy the config, then fill in any required environment variables.
claude mcp add cost-guard --env GOOGLE_APPLICATION_CREDENTIALS='<GOOGLE_APPLICATION_CREDENTIALS>' --env SNOWFLAKE_PRIVATE_KEY_PATH='<SNOWFLAKE_PRIVATE_KEY_PATH>' --env SNOWFLAKE_PRIVATE_KEY_PASSPHRASE='<SNOWFLAKE_PRIVATE_KEY_PASSPHRASE>' --env SNOWFLAKE_PASSWORD='<SNOWFLAKE_PASSWORD>' --env DATABRICKS_TOKEN='<DATABRICKS_TOKEN>' --env DATABRICKS_CLIENT_SECRET='<DATABRICKS_CLIENT_SECRET>' -- uvx cost-guard-mcp~/.codex/config.toml
[mcp_servers.cost-guard]
command = "uvx"
args = ["cost-guard-mcp"]
env = { GOOGLE_APPLICATION_CREDENTIALS = "<GOOGLE_APPLICATION_CREDENTIALS>", SNOWFLAKE_PRIVATE_KEY_PATH = "<SNOWFLAKE_PRIVATE_KEY_PATH>", SNOWFLAKE_PRIVATE_KEY_PASSPHRASE = "<SNOWFLAKE_PRIVATE_KEY_PASSPHRASE>", SNOWFLAKE_PASSWORD = "<SNOWFLAKE_PASSWORD>", DATABRICKS_TOKEN = "<DATABRICKS_TOKEN>", DATABRICKS_CLIENT_SECRET = "<DATABRICKS_CLIENT_SECRET>" }
# codex mcp add cost-guard --env GOOGLE_APPLICATION_CREDENTIALS='<GOOGLE_APPLICATION_CREDENTIALS>' --env SNOWFLAKE_PRIVATE_KEY_PATH='<SNOWFLAKE_PRIVATE_KEY_PATH>' --env SNOWFLAKE_PRIVATE_KEY_PASSPHRASE='<SNOWFLAKE_PRIVATE_KEY_PASSPHRASE>' --env SNOWFLAKE_PASSWORD='<SNOWFLAKE_PASSWORD>' --env DATABRICKS_TOKEN='<DATABRICKS_TOKEN>' --env DATABRICKS_CLIENT_SECRET='<DATABRICKS_CLIENT_SECRET>' -- uvx cost-guard-mcp~/.cursor/mcp.json
{
"mcpServers": {
"cost-guard": {
"command": "uvx",
"args": [
"cost-guard-mcp"
],
"env": {
"GOOGLE_APPLICATION_CREDENTIALS": "<GOOGLE_APPLICATION_CREDENTIALS>",
"SNOWFLAKE_PRIVATE_KEY_PATH": "<SNOWFLAKE_PRIVATE_KEY_PATH>",
"SNOWFLAKE_PRIVATE_KEY_PASSPHRASE": "<SNOWFLAKE_PRIVATE_KEY_PASSPHRASE>",
"SNOWFLAKE_PASSWORD": "<SNOWFLAKE_PASSWORD>",
"DATABRICKS_TOKEN": "<DATABRICKS_TOKEN>",
"DATABRICKS_CLIENT_SECRET": "<DATABRICKS_CLIENT_SECRET>"
}
}
}
}.vscode/mcp.json
{
"servers": {
"cost-guard": {
"type": "stdio",
"command": "uvx",
"args": [
"cost-guard-mcp"
],
"env": {
"GOOGLE_APPLICATION_CREDENTIALS": "<GOOGLE_APPLICATION_CREDENTIALS>",
"SNOWFLAKE_PRIVATE_KEY_PATH": "<SNOWFLAKE_PRIVATE_KEY_PATH>",
"SNOWFLAKE_PRIVATE_KEY_PASSPHRASE": "<SNOWFLAKE_PRIVATE_KEY_PASSPHRASE>",
"SNOWFLAKE_PASSWORD": "<SNOWFLAKE_PASSWORD>",
"DATABRICKS_TOKEN": "<DATABRICKS_TOKEN>",
"DATABRICKS_CLIENT_SECRET": "<DATABRICKS_CLIENT_SECRET>"
}
}
}
}~/.gemini/settings.json
{
"mcpServers": {
"cost-guard": {
"command": "uvx",
"args": [
"cost-guard-mcp"
],
"env": {
"GOOGLE_APPLICATION_CREDENTIALS": "<GOOGLE_APPLICATION_CREDENTIALS>",
"SNOWFLAKE_PRIVATE_KEY_PATH": "<SNOWFLAKE_PRIVATE_KEY_PATH>",
"SNOWFLAKE_PRIVATE_KEY_PASSPHRASE": "<SNOWFLAKE_PRIVATE_KEY_PASSPHRASE>",
"SNOWFLAKE_PASSWORD": "<SNOWFLAKE_PASSWORD>",
"DATABRICKS_TOKEN": "<DATABRICKS_TOKEN>",
"DATABRICKS_CLIENT_SECRET": "<DATABRICKS_CLIENT_SECRET>"
}
}
}
}claude_desktop_config.json
{
"mcpServers": {
"cost-guard": {
"command": "uvx",
"args": [
"cost-guard-mcp"
],
"env": {
"GOOGLE_APPLICATION_CREDENTIALS": "<GOOGLE_APPLICATION_CREDENTIALS>",
"SNOWFLAKE_PRIVATE_KEY_PATH": "<SNOWFLAKE_PRIVATE_KEY_PATH>",
"SNOWFLAKE_PRIVATE_KEY_PASSPHRASE": "<SNOWFLAKE_PRIVATE_KEY_PASSPHRASE>",
"SNOWFLAKE_PASSWORD": "<SNOWFLAKE_PASSWORD>",
"DATABRICKS_TOKEN": "<DATABRICKS_TOKEN>",
"DATABRICKS_CLIENT_SECRET": "<DATABRICKS_CLIENT_SECRET>"
}
}
}
}.trae/mcp.json
{
"mcpServers": {
"cost-guard": {
"command": "uvx",
"args": [
"cost-guard-mcp"
],
"env": {
"GOOGLE_APPLICATION_CREDENTIALS": "<GOOGLE_APPLICATION_CREDENTIALS>",
"SNOWFLAKE_PRIVATE_KEY_PATH": "<SNOWFLAKE_PRIVATE_KEY_PATH>",
"SNOWFLAKE_PRIVATE_KEY_PASSPHRASE": "<SNOWFLAKE_PRIVATE_KEY_PASSPHRASE>",
"SNOWFLAKE_PASSWORD": "<SNOWFLAKE_PASSWORD>",
"DATABRICKS_TOKEN": "<DATABRICKS_TOKEN>",
"DATABRICKS_CLIENT_SECRET": "<DATABRICKS_CLIENT_SECRET>"
}
}
}
}Environment variables
GOOGLE_APPLICATION_CREDENTIALSSNOWFLAKE_PRIVATE_KEY_PATHSNOWFLAKE_PRIVATE_KEY_PASSPHRASESNOWFLAKE_PASSWORDDATABRICKS_TOKENDATABRICKS_CLIENT_SECRET
Replace the placeholders with your own credentials and keep them out of version control.