bq_mcp_server 0.1.0

Python-based MCP (Model Context Protocol) server that retrieves dataset, table, and schema information from Google Cloud BigQuery

BigQuery MCP Server

This is a Python-based MCP (Model Context Protocol) server that retrieves dataset, table, and schema information from Google Cloud BigQuery, caches it locally, and serves it via MCP. Its primary purpose is to enable generative AI systems to quickly understand BigQuery's structure and execute queries securely.

Key Features

• Metadata Management: Retrieves and caches information about BigQuery datasets, tables, and columns
• Keyword Search: Supports keyword search of cached metadata
• Secure Query Execution: Provides SQL execution capabilities with automatic LIMIT clause insertion and cost control
• MCP Compliance: Offers tools via the Model Context Protocol

MCP Server Tools

Available tools:

1. get_datasets - Retrieves a list of all datasets
2. get_tables - Retrieves all tables within a specified dataset (requires dataset_id, optionally accepts project_id)
3. search_metadata - Searches metadata for datasets, tables, and columns
4. execute_query - Safely executes BigQuery SQL queries with automatic LIMIT clause insertion and cost control
5. check_query_scan_amount - Retrieves the scan amount for BigQuery SQL queries

Installation and Environment Setup

Prerequisites

• Python 3.11 or later
• Google Cloud Platform account
• GCP project with BigQuery API enabled

Installing Dependencies

This project uses uv for package management:

# Install uv if not already installed
curl -LsSf https://astral.sh/uv/install.sh | sh

# Install dependencies
uv sync

Configuring Option

For a list of configuration values, see:

docs/settings.md

MCP Setting

{
    "mcpServers": {
        "bq_mcp_server": {
            "command": "uv",
            "args": [
                "run",
                "--directory",
                "<your install directory>",
                "mcp_server"
            ],
            "env": {
                "PYTHONPATH": "<your install directory>",
                "PROJECT_IDS": "<your project id>"
            }
        }
    }
}

Running Tests

Running All Tests

pytest

Running Specific Test Files

pytest tests/test_logic.py

Running Specific Test Functions

pytest -k test_function_name

Checking Test Coverage

pytest --cov=bq_mcp_server

Local Development

Starting the MCP Server

uv run mcp_server

Starting the FastAPI REST API Server

uvicorn bq_mcp_server.adapters.web:app --reload

Development Commands

Code Formatting and Linting

# Code formatting
ruff format

# Linting checks
ruff check

# Automatic fixes
ruff check --fix

Dependency Management

# Adding new dependencies
uv add <package>

# Adding development dependencies
uv add --dev <package>

# Updating dependencies
uv sync