MCP Server Snowflake

🚀 Snowflake MCP 服務器

Snowflake MCP 服務器是一款用於對 Snowflake 數據庫執行只讀操作的模型上下文協議（MCP）服務器。它能讓 Claude 在不修改任何信息的情況下，安全地查詢 Snowflake 數據。

🚀 快速開始

Snowflake MCP 服務器可讓 Claude 安全地對 Snowflake 數據庫進行只讀查詢。以下為你介紹如何快速開啟使用之旅：

安裝

先決條件

• Python 3.12+
• Snowflake 賬戶（需具備只讀權限）

安裝步驟

1. 複製 .env.example 文件並重命名為 .env。
2. 配置環境變量：

cp .env.example .env
# 配置以下內容到.env文件中
SNOWFLAKE_AUTH_TYPE=private_key  # 或 external_browser
SNOWFLAKE_ACCOUNT_NAME=your_account_name
SNOWFLAKE_USER_NAME=your_user_name
SNOWFLAKE_PRIVATE_KEY_PATH=path_to_your_private_key  # 如果使用 private_key 認證

3. 安裝依賴：

pip install -r requirements.txt

運行服務器

python mcp_server_snowflake.py

與 Claude 桌面集成

1. 在終端中運行服務器。
2. 打開 Claude 桌面並連接到運行中的 MCP 服務器即可開始交互。

✨ 主要特性

• 靈活的身份驗證：可通過以下方式連接到 Snowflake：
  • 使用服務賬戶和私有密鑰進行身份驗證。
  • 使用外部瀏覽器進行交互式會話身份驗證。
• 連接池管理：帶自動後臺刷新，以保持長時間運行的連接。
• 豐富的操作支持：支持查詢數據、顯示數據庫結構、描述查詢計劃、解釋執行計劃以及限制結果集大小等操作。
• 自動添加限制：自動添加 LIMIT 子句以防止大數據集的返回。
• 輸入驗證：使用 Pydantic 進行輸入驗證以防止 SQL 注入。

📦 安裝指南

先決條件

• Python 3.12+
• Snowflake 賬戶（需具備只讀權限）

安裝步驟

1. 複製 .env.example 文件並重命名為 .env。
2. 配置環境變量如下：

cp .env.example .env
# 配置以下內容到.env文件中
SNOWFLAKE_AUTH_TYPE=private_key  # 或 external_browser
SNOWFLAKE_ACCOUNT_NAME=your_account_name
SNOWFLAKE_USER_NAME=your_user_name
SNOWFLAKE_PRIVATE_KEY_PATH=path_to_your_private_key  # 如果使用 private_key 認證

3. 安裝依賴：

pip install -r requirements.txt

💻 使用示例

基礎用法

運行服務器

python mcp_server_snowflake.py

與 Claude 桌面集成

1. 在終端中運行服務器。
2. 打開 Claude 桌面並連接到運行中的 MCP 服務器即可開始交互。

📚 詳細文檔

配置選項

環境變量配置

• SNOWFLAKE_CONN_REFRESH_HOURS：設置連接的自動刷新間隔（默認為 24）。

SNOWFLAKE_CONN_REFRESH_HOURS=4  # 設置為4小時

認證方式

私有密鑰認證

1. 在 Snowflake 中創建用戶並生成密鑰對，具體操作請參考 Snowflake 文檔。
2. 配置環境變量：

SNOWFLAKE_AUTH_TYPE=private_key
SNOWFLAKE_PRIVATE_KEY_PATH=/path/to/your/private-key.pem

外部瀏覽器認證

1. 配置環境變量：

SNOWFLAKE_AUTH_TYPE=external_browser

2. 啟動服務器後，瀏覽器會自動打開進行身份驗證。

安全注意事項

• 僅允許執行讀取操作（SELECT、SHOW、DESCRIBE、EXPLAIN 和 WITH 語句）。
• 輸入經過嚴格驗證以防止 SQL 注入攻擊。
• 確保 .env 文件的安全性，避免洩露敏感信息。
• 使用 HTTPS 進行通信以確保數據傳輸安全。

開發細節

靜態類型檢查

mypy mcp_server_snowflake/

代碼格式化

ruff format .

執行測試

pytest

🔧 技術細節

• 使用 Snowflake Connector Python 進行數據庫連接。
• 基於 MCP 協議 實現與 Claude 的交互。
• 使用 Pydantic 進行數據驗證。
• 使用 python-dotenv 管理環境變量。