在 Colab Enterprise 笔记本中使用 Gemini 编写代码

本文档介绍了如何使用 Gemini for Google Cloud 产品组合中的 Gemini in Colab Enterprise 帮助您在 Colab Enterprise 笔记本中编写代码。

了解 Google Cloud 专用 Gemini Google Cloud 如何以及何时使用您的数据

本文档适用于使用 Colab Enterprise 的数据分析师、数据科学家和数据开发者。本文档假定您已了解如何在笔记本环境中编写代码。

准备工作

  1. 确保已为您的 Google Cloud 用户账号和项目设置了 Gemini in Colab Enterprise。
  2. Sign in to your Google Cloud account. If you're new to Google Cloud, create an account to evaluate how our products perform in real-world scenarios. New customers also get $300 in free credits to run, test, and deploy workloads.
  3. Enable the Vertex AI, Cloud AI Companion, Dataform, and Compute Engine APIs.

    Enable the APIs

  4. Enable the Vertex AI, Cloud AI Companion, Dataform, and Compute Engine APIs.

    Enable the APIs

所需的角色

如需确保您的用户账号拥有在 Colab Enterprise 笔记本中在 Gemini 的协助下编写代码所需的权限,请让您的管理员为您的用户账号授予项目的以下 IAM 角色:

如需详细了解如何授予角色,请参阅管理对项目、文件夹和组织的访问权限

您的管理员还可以通过自定义角色或其他预定义角色向您的用户账号授予所需的权限。

设置 Gemini in Colab Enterprise

如需使用 Gemini 功能,请执行以下操作:

  1. 在 Google Cloud 控制台中,进入 Colab Enterprise 我的笔记本页面。

    前往“我的笔记本”

  2. 在工具栏中,点击 pen_spark 帮我写代码按钮,然后选择要使用的功能。

    “帮我写代码”按钮下方列出了 Gemini in Colab Enterprise 的功能,包括代码补全和代码生成。
    • 代码补全功能会在您输入时提供代码建议。
    • 代码生成功能可根据您对所需功能的说明生成代码,并建议修复代码中的错误。

    如果 pen_spark 帮我编写代码按钮处于非活动状态,或者您点击该按钮后没有看到功能列表,请参阅为项目在 Colab Enterprise 中设置 Gemini

使用代码补全功能

如需在 Colab Enterprise 笔记本中使用代码补全功能,请执行以下操作:

  1. 在 Google Cloud 控制台中,进入 Colab Enterprise 我的笔记本页面。

    前往“我的笔记本”

  2. 区域菜单中,选择包含您的笔记本的区域。

  3. 点击要打开的笔记本。

  4. 在工具栏中,点击 pen_spark 帮我写代码按钮,然后选择代码补全(如果尚未选择)。
  5. 在笔记本的代码单元中,开始输入代码。 Colab Enterprise 会以灰色斜体文本的形式建议可能的代码补全方式。

    例如,如果您输入 import p,Colab Enterprise 可能会建议您输入导入 pandas as pd 的代码补全,后跟一行用于读取 CSV 文件并将内容存储在 pandas 数据框中的代码。

  6. 如需接受建议的代码补全,请按 Tab 键。

使用代码生成功能

如需使用 Gemini 为 Google Cloud生成代码,请执行以下操作:

  1. 在 Google Cloud 控制台中,进入 Colab Enterprise 我的笔记本页面。

    前往“我的笔记本”

  2. 区域菜单中,选择包含您的笔记本的区域。

  3. 点击要打开的笔记本。

  4. 在工具栏中,点击 pen_spark 帮我写代码按钮,然后选择代码生成(如果尚未选择)。

  5. 在笔记本的空白代码单元中,点击生成

  6. pen_spark Generate 字段中,输入要生成的代码的说明,然后按 Enter 键。

  7. 如要关闭 Generate 面板,请点击 Close

修正错误

如需使用 Gemini Google Cloud 帮助您修复代码中的错误,请执行以下操作:

  1. 在 Google Cloud 控制台中,进入 Colab Enterprise 我的笔记本页面。

    前往“我的笔记本”

  2. 区域菜单中,选择包含您的笔记本的区域。

  3. 点击要打开的笔记本。

  4. 在工具栏中,点击 pen_spark 帮我写代码按钮,然后选择代码生成(如果尚未选择)。

  5. 在笔记本的代码单元中,输入包含错误的代码,然后运行该单元。例如,您可能会输入 print(1,其中缺少右括号。

    代码单元运行后,Colab Enterprise 会在代码单元下方输出错误消息。如果 Gemini in Colab Enterprise 提供了可能修复代码的建议,系统会显示修正错误按钮。

  6. 点击修正错误按钮。

    Colab Enterprise 会显示针对代码的建议修复方法。

  7. 评估建议,然后执行以下操作之一:

    • 如要接受建议,请点击  接受建议
    • 如要拒绝建议,请点击  拒绝建议

关闭 Gemini 功能

如需在 Colab Enterprise 中停用 Gemini 功能,请执行以下操作:

  1. 在 Colab Enterprise 工具栏中,点击 pen_spark 帮我写代码按钮。

  2. 清除您要停用的 Gemini 功能。

问题排查

本部分介绍了如何解决在 Colab Enterprise 中使用 Gemini 时遇到的问题。

Gemini 功能无法正常运行

在 Colab Enterprise 笔记本中输入内容时,系统不会显示代码补全建议。 或者,pen_spark 帮我写代码按钮未显示在代码单元格上方。

您可能在 Colab Enterprise 设置中关闭了显示基于上下文的代码补全结果设置。

如需解决此问题,请执行以下操作:

  1. 点击  显示/隐藏标题
  2. 工具菜单中,点击设置
  3. 点击编辑器
  4. 选择显示基于上下文的代码补全结果
  5. 点击保存

后续步骤