Snowflake MCP Server

🚀 Snowflake MCP 服務器

Snowflake MCP 服務器是一個基於 Model Context Protocol (MCP) 的工具，它允許 Claude 安全地對 Snowflake 數據庫執行只讀查詢，而不會對數據進行任何修改。

🚀 快速開始

Snowflake MCP 服務器為 Claude 提供了安全查詢 Snowflake 數據庫的途徑。通過簡單的配置和安裝，你可以快速搭建並運行該服務器。

✨ 主要特性

• 靈活的身份驗證：支持通過服務賬戶和私有密鑰進行身份驗證，也可使用外部瀏覽器進行交互式會話身份驗證，連接到 Snowflake。
• 連接池管理：帶有自動背景刷新的連接池，確保與 Snowflake 的持久連接。
• 多視圖和數據庫查詢：支持在單個會話中查詢多個視圖和數據庫。
• 多 SQL 語句類型支持：支持多種 SQL 語句類型，如 SELECT、SHOW、DESCRIBE、EXPLAIN、WITH。
• MCP 兼容處理器：提供 MCP 兼容的處理器，用於查詢 Snowflake 數據。
• 只讀操作與安全檢查：僅支持只讀操作，並帶有安全檢查，防止數據修改。
• Python 3.12+ 支持：支持 Python 3.12 及以上版本。
• 標準輸入輸出集成：基於標準輸入輸出 (stdio) 的 MCP 服務器，方便與 Claude Desktop 集成。

📦 安裝指南

安裝要求

python-dotenv
mypy
ruff
pytest
pydantic
snowflake-connector-python

安裝步驟

pip install python-dotenv mypy ruff pytest pydantic snowflake-connector-python

💻 使用示例

環境變量配置

在項目根目錄創建一個 .env 文件，並添加以下內容：

SNOWFLAKE_AUTH_TYPE=private_key 或 external_browser
SNOWFLAKE_PRIVATE_KEY_PATH=私鑰路徑（如使用 private_key 身份驗證）
SNOWFLAKE_ACCOUNT_NAME=賬戶名稱

運行服務器

python mcp_server_snowflake/main.py

📚 詳細文檔

功能模塊

• mcp_server_snowflake.mcp_server：MCP 服務器主邏輯。
• mcp_server_snowflake.models：數據模型和驗證。
• mcp_server_snowflake.auth：身份驗證處理。
• mcp_server_snowflake.utils：工具函數和配置管理。

安全注意事項

• 查詢限制：所有查詢都會自動添加 LIMIT 子句，防止大數據集的風險。
• 寫入操作禁止：禁止執行任何寫入操作（如 INSERT、UPDATE、DELETE）。
• 輸入驗證：輸入驗證嚴格過濾 SQL 注入風險。
• 憑據安全：私鑰和憑據不會持久化到磁盤。
• 通信安全：使用 HTTPS 進行與 Snowflake 的通信。

開發信息

代碼檢查工具

# 靜態類型檢查
mypy mcp_server_snowflake/

# 代碼格式檢查
ruff check .

# 代碼自動格式化
ruff format .

# 單元測試
pytest

貢獻指南

歡迎貢獻！請隨意提交 Pull Request。

🔧 技術細節

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