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

本文說明如何將 AlloyDB Omni 執行個體連線至支援模型上下文協定 (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容器適用的 AlloyDB Omni,並執行資料庫執行個體。
  2. 建立或重複使用資料庫使用者。準備好輸入使用者名稱和密碼。

使用 Gemini CLI 的 PostgreSQL 擴充功能

Gemini CLI 是開放原始碼的 AI 代理,可協助程式設計、偵錯、資料探索和內容建立,進而輔助開發工作流程。其任務是提供代理介面,與 Data Cloud 服務和熱門的開放原始碼資料庫互動。

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

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

  • products:包含產品資訊,包括 product_idproduct_namecategoryprice
  • customers:儲存顧客資料,例如 customer_idfirst_namelast_nameemail
  • orders:包含訂單資訊,包括 order_idcustomer_idorder_date

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

類別 工具 自然語言提示範例
資料庫作業 execute_sql 顯示「筆記型電腦」類別中價格最高的 10 項產品。
get_query_plan 說明查詢計畫,找出過去 6 個月內未下單的所有顧客。
list_active_queries 資料庫正在執行哪些查詢?
database_overview 請簡要說明目前的資料庫。
list_tables 顯示目前資料庫中的所有資料表。
list_views 可觀賞哪種景觀?
list_schemas 列出所有結構定義。
list_triggers Show me the triggers on the orders table.
list_available_extensions 我可以安裝哪些擴充功能?
list_installed_extensions 列出所有已安裝的擴充功能。
list_indexes 列出所有索引。
list_sequences 列出所有序列。
資料庫健康狀態和維護 list_autovacuum_configurations 顯示目前的 autovacuum 設定。
list_memory_configurations 目前的記憶體設定為何?
list_top_bloated_tables 列出前五個最臃腫的資料表。
list_replication_slots 顯示所有有效的複寫位置。
list_invalid_indexes 檢查 ecommerce 資料庫中是否有任何無效的索引。
long_running_transactions Show me the 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 CLIpostgres 擴充功能
  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 Servers」(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. 在「開發人員」分頁中,按一下「編輯設定」開啟設定檔。
  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 代碼

  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 程式碼,套用新設定。