藉助 Gemini 編寫 SQL 程式碼

本文說明如何使用 Gemini Code Assist,在 Bigtable 中透過自然語言提示生成 SQL 查詢,取得 AI 輔助。

瞭解 Gemini for Google Cloud 如何使用您的資料

本文適用於熟悉 Bigtable、SQL 和資料分析的資料庫管理員和資料工程師。如果您是 Bigtable 新手,請參閱快速入門:使用 Google Cloud 控制台建立 Bigtable 執行個體。

限制

  • 您無法使用 Gemini in Bigtable,以自然語言說明 SQL 查詢。
  • Gemini in Bigtable 可能會生成在 Bigtable 的 GoogleSQL 中無效的查詢,或使用下列不支援的陳述式或語法:

    • SELECT 以外的資料操縱語言 (DML) 陳述式,例如 INSERTUPDATEDELETE
    • 資料定義語言 (DDL) 陳述式,例如 CREATEALTERDROP
    • 資料存取控管陳述式。
    • 子查詢、JOINUNION 和 CTE 的查詢語法。

    詳情請參閱 Bigtable 適用的 GoogleSQL 總覽

事前準備

  1. 在 Google Cloud 專案中啟用 Gemini for Google Cloud API

  2. 如要完成本文中的工作,請確認您具備必要的 Identity and Access Management (IAM) 權限

  3. 選用:如要按照本文中的範例操作,請建立並填入 test-table,如「建立測試資料表」一文所述。

必要的角色

如要取得完成本文中工作所需的權限,請要求管理員授予您專案的「Gemini for Google Cloud 使用者」 (roles/cloudaicompanion.user) IAM 角色。如要進一步瞭解如何授予角色,請參閱「管理專案、資料夾和機構的存取權」。

您或許也能透過自訂角色或其他預先定義的角色,取得必要權限。

使用自然語言提示詞生成 SQL 查詢

您可以提供自然語言註解 (或提示) 給 Gemini,要求 Gemini 根據結構定義產生查詢。舉例來說,您可以提示 Gemini 根據下列提示生成 SQL:

  • 「計算 1 GB 數據方案的裝置數量。」
  • 「列出 2019 年 5 月 1 日連上 Wi-Fi 的所有手機。」
  • 「哪些平板電腦已啟用 10 GB 數據方案?」

如要使用 Gemini 生成 Bigtable 的 SQL 程式碼,請按照下列步驟操作:

  1. 前往 Google Cloud 控制台的「Bigtable」頁面。

    前往 Bigtable

  2. 從清單中選取執行個體。

  3. 在導覽窗格中,按一下「Bigtable Studio」

  4. 點選 開啟新分頁,然後選擇「編輯器」

  5. 按一下「Generate SQL」(生成 SQL 查詢)

  6. 在「幫我寫程式碼」對話方塊中輸入提示詞,然後按一下「生成」

    舉例來說,如果您輸入提示詞 Count the number of devices that have a 1GB data plan.,Gemini 會生成類似下列內容的 SQL:

    SELECT
        count(*)
    FROM
        `test_table`
    WHERE
        cell_plan['data_plan_01gb'] = 'true'
    
  7. 查看生成的 SQL,然後執行下列其中一項操作:

    • 如要執行建議的 SQL,請依序點選「插入」和「執行」
    • 如要編輯提示,請依序點選「編輯」和「更新」
    • 如要編輯建議的 SQL,請按一下「插入」。在查詢產生器中手動編輯查詢,然後點選「執行」
  8. 選用步驟:如要繼續 test_table 範例,請輸入提示 Show all data for phones from May 2019.,然後 Gemini 會生成類似下列內容的 SQL:

    SELECT
        *
    FROM
        `test_table`
    WHERE
        _key LIKE 'phone#%#201905%'
    

後續步驟