搭配使用 AlloyDB Omni 與 MCP、Gemini CLI 和其他代理程式

本文說明如何將 AlloyDB Omni 執行個體連結至支援 Model Context Protocol (MCP) 的各種代理。

AlloyDB Omni 與 PostgreSQL 相容,因此您可以使用專為 PostgreSQL 設計的工具和擴充功能。

如要以最簡單的方式設定,請使用 Gemini CLI 的專用 PostgreSQL 擴充功能。這項擴充功能會直接整合底層 MCP 伺服器,因此您不必另外設定伺服器。您可以設定 Gemini Code Assist 使用 Gemini CLI,在 IDE 中享有類似的設定優勢。

或者,其他支援 MCP 的 IDE 和代理,可透過 MCP Toolbox for Databases 連線。Toolbox 是開放原始碼 MCP 伺服器,可將 AI 代理連結至資料。這項工具會處理驗證和連線集區等工作,讓您直接在 IDE 中使用自然語言與資料互動。

事前準備

連線至 AlloyDB Omni 執行個體前,請務必完成下列事項:

  1. 在 Kubernetes 上安裝 AlloyDB Omni,並執行資料庫執行個體。
  2. 建立或重複使用資料庫使用者。請準備好輸入使用者名稱和密碼。

使用 Gemini CLI 的 PostgreSQL 擴充功能

Gemini CLI 是開放原始碼 AI 代理,可協助編寫程式碼、偵錯、探索資料和創作內容,輔助開發工作流程。其宗旨是提供代理介面,用於與 Data Cloud 服務和熱門開放原始碼資料庫互動。

Gemini CLI for AlloyDB Omni 是透過 PostgreSQL 擴充功能整合。這項擴充功能提供一組工具,可與 PostgreSQL 和 AlloyDB Omni 互動,讓您直接透過 Gemini CLI,使用自然語言提示詞管理資料庫、執行查詢及探索結構定義。

下列範例使用範例 ecommerce 資料庫,其中包含以下資料表:

  • products:包含產品資訊,包括 product_id、product_name、category 和 price。
  • customers:儲存客戶資料,例如 customer_id、first_name、last_name 和 email。
  • orders:保留訂單資訊,包括 order_id、customer_id 和 order_date。

PostgreSQL 擴充功能提供下列工具:

類別 工具 自然語言提示詞範例
資料庫作業 execute_sql 顯示「筆電」類別中價格最高的 10 項產品。
get_query_plan 說明查詢計畫,找出過去 6 個月內未下單的所有顧客。
list_active_queries 資料庫正在執行哪些查詢?
database_overview 請提供目前資料庫的總覽。
list_tables 顯示目前資料庫中的所有資料表。
list_views 可觀賞哪種景觀?
list_schemas 列出所有結構定義。
list_triggers 顯示 orders 資料表的觸發條件。
list_available_extensions 我可以安裝哪些擴充功能?
list_installed_extensions 列出所有已安裝的擴充功能。
list_indexes 列出所有索引。
list_sequences 列出所有序列。
資料庫健康狀態和維護 list_autovacuum_configurations 顯示目前的自動清理設定。
list_memory_configurations 目前的記憶體設定為何?
list_top_bloated_tables 列出前五個最臃腫的資料表。
list_replication_slots 顯示所有使用中的複製槽。
list_invalid_indexes 檢查 ecommerce 資料庫中是否有任何無效索引。
long_running_transactions 顯示執行時間長的交易。
list_locks 列出所有鎖。
replication_stats 顯示複製統計資料。

Gemini CLI 的 PostgreSQL 擴充功能有兩種使用方式:

  • 獨立指令列工具
  • 透過 Gemini Code Assist 整合至 IDE

Gemini CLI

  1. 安裝 Gemini CLI (0.6.0 以上版本)。
  2. 使用下列指令,從 GitHub 存放區安裝 Gemini CLI 的 PostgreSQL 擴充功能:
    gemini extensions install https://github.com/gemini-cli-extensions/postgres
  3. 設定環境變數,連線至 AlloyDB Omni 執行個體:
    export POSTGRES_HOST="DATABASE_HOST"
    export POSTGRES_PORT="DATABASE_PORT"
    export POSTGRES_DATABASE="DATABASE_NAME"
    export POSTGRES_USER="DATABASE_USER"
    export POSTGRES_PASSWORD="DATABASE_PASSWORD"
    

    您也可以從目前使用的目錄中的 .env 檔案載入這些變數。

  4. 在互動模式下啟動 Gemini CLI:
    gemini
    CLI 會自動載入 PostgreSQL 擴充功能和相關工具,方便您與資料庫互動。

Gemini Code Assist

如要避免手動設定 MCP 伺服器,請將 Gemini Code Assist 設為使用 Gemini CLI。

  1. 按照「Gemini CLI」分頁中的說明,安裝及設定 Gemini CLI 和 postgres 擴充功能。
  2. 設定 Gemini Code Assist,以便使用 Gemini CLI。
  3. 直接在 Gemini Code Assist 對話中,使用自然語言與 AlloyDB Omni 執行個體互動。

使用 MCP Toolbox for Databases 連結其他 IDE

本節說明如何使用 MCP Toolbox for Databases,從各種代理連線至 AlloyDB Omni 執行個體。

安裝 MCP Toolbox for Databases

如要將 IDE 連結至 AlloyDB Omni,您必須安裝 MCP Toolbox for Databases,這是一個開放原始碼伺服器,可將 AI 代理連結至您的資料。

  1. 以二進位檔形式下載最新版本 Toolbox。選取與作業系統 (OS) 和 CPU 架構對應的二進位檔。

    linux/amd64

    curl -L -O https://storage.googleapis.com/genai-toolbox/v0.21.0/linux/amd64/toolbox

    darwin/arm64

    curl -L -O https://storage.googleapis.com/genai-toolbox/v0.21.0/darwin/arm64/toolbox

    darwin/amd64

    curl -L -O https://storage.googleapis.com/genai-toolbox/v0.21.0/darwin/amd64/toolbox

    windows/amd64

    curl -L -O https://storage.googleapis.com/genai-toolbox/v0.21.0/windows/amd64/toolbox.exe
  2. 將二進位檔設為可執行檔。

    chmod +x toolbox
    
  3. 驗證安裝。

    ./toolbox --version
    

設定用戶端

從下列選項中選取代理工具:

Cursor

  1. 如果專案根目錄中沒有 .cursor 目錄,請建立該目錄。
  2. 如果 .cursor/mcp.json 檔案不存在,請建立並開啟該檔案。
  3. 新增下列設定,將環境變數替換為您的值,然後儲存。
{
  "mcpServers": {
    "alloydbomni": {
      "command": "PATH_TO_TOOLBOX",
      "args": ["--prebuilt","postgres","--stdio"],
      "env": {
        "POSTGRES_HOST": "DATABASE_HOST",
        "POSTGRES_PORT": "DATABASE_PORT",
        "POSTGRES_DATABASE": "DATABASE_NAME",
        "POSTGRES_USER": "DATABASE_USER",
        "POSTGRES_PASSWORD": "DATABASE_PASSWORD"
      }
    }
  }
}
  1. 開啟 Cursor,然後依序前往「Settings」>「Cursor Settings」>「MCP」。伺服器連線後,會顯示綠色的「有效」狀態。

風浪板

  1. 開啟 Windsurf,然後前往 Cascade 助理。
  2. 依序點按 MCP 圖示和「設定」,開啟設定檔。
  3. 新增下列設定,將環境變數替換為您的值,然後儲存。
{
  "mcpServers": {
    "alloydbomni": {
      "command": "PATH_TO_TOOLBOX",
      "args": ["--prebuilt","postgres","--stdio"],
      "env": {
        "POSTGRES_HOST": "DATABASE_HOST",
        "POSTGRES_PORT": "DATABASE_PORT",
        "POSTGRES_DATABASE": "DATABASE_NAME",
        "POSTGRES_USER": "DATABASE_USER",
        "POSTGRES_PASSWORD": "DATABASE_PASSWORD"
      }
    }
  }
}

Visual Studio Code (Copilot)

  1. 開啟 VS Code,並在專案根目錄中建立 .vscode 目錄 (如果尚未建立)。
  2. 如果 .vscode/mcp.json 檔案不存在,請建立並開啟該檔案。
  3. 新增下列設定,將環境變數替換為您的值,然後儲存。
{
  "servers": {
    "alloydbomni": {
      "command": "PATH_TO_TOOLBOX",
      "args": ["--prebuilt","postgres","--stdio"],
      "env": {
        "POSTGRES_HOST": "DATABASE_HOST",
        "POSTGRES_PORT": "DATABASE_PORT",
        "POSTGRES_DATABASE": "DATABASE_NAME",
        "POSTGRES_USER": "DATABASE_USER",
        "POSTGRES_PASSWORD": "DATABASE_PASSWORD"
      }
    }
  }
}

Cline

  1. 在 VS Code 中開啟 Cline 擴充功能,然後點按「MCP 伺服器」圖示。
  2. 按一下「設定 MCP 伺服器」,開啟設定檔。
  3. 新增下列設定,將環境變數替換為您的值,然後儲存。
{
  "mcpServers": {
    "alloydbomni": {
      "command": "PATH_TO_TOOLBOX",
      "args": ["--prebuilt","postgres","--stdio"],
      "env": {
        "POSTGRES_HOST": "DATABASE_HOST",
        "POSTGRES_PORT": "DATABASE_PORT",
        "POSTGRES_DATABASE": "DATABASE_NAME",
        "POSTGRES_USER": "DATABASE_USER",
        "POSTGRES_PASSWORD": "DATABASE_PASSWORD"
      }
    }
  }
}
      
  1. 伺服器連線成功後,會顯示綠色的「有效」狀態。

Claude 電腦版

  1. 開啟 Claude Desktop,然後前往「設定」。
  2. 在「Developer」分頁中,按一下「Edit Config」開啟設定檔。
  3. 新增設定,將環境變數替換為您的值,然後儲存。
{
  "mcpServers": {
    "alloydbomni": {
      "command": "PATH_TO_TOOLBOX",
      "args": ["--prebuilt","postgres","--stdio"],
      "env": {
        "POSTGRES_HOST": "DATABASE_HOST",
        "POSTGRES_PORT": "DATABASE_PORT",
        "POSTGRES_DATABASE": "DATABASE_NAME",
        "POSTGRES_USER": "DATABASE_USER",
        "POSTGRES_PASSWORD": "DATABASE_PASSWORD"
      }
    }
  }
}
  1. 重新啟動 Claude Desktop。
  2. 新的對話畫面會顯示鎚子 (MCP) 圖示和新的 MCP 伺服器。

Claude Code

  1. 安裝 Claude Code。
  2. 如果專案根目錄中沒有 .mcp.json 檔案,請建立該檔案。
  3. 新增設定,將環境變數替換為您的值,然後儲存。
{
  "mcpServers": {
    "alloydbomni": {
      "command": "PATH_TO_TOOLBOX",
      "args": ["--prebuilt","postgres","--stdio"],
      "env": {
        "POSTGRES_HOST": "DATABASE_HOST",
        "POSTGRES_PORT": "DATABASE_PORT",
        "POSTGRES_DATABASE": "DATABASE_NAME",
        "POSTGRES_USER": "DATABASE_USER",
        "POSTGRES_PASSWORD": "DATABASE_PASSWORD"
      }
    }
  }
}
  1. 重新啟動 Claude Code,套用新設定。