使用代码补全和代码生成功能

本文档介绍了如何借助 Gemini in Colab Enterprise(Gemini for Google Cloud 产品组合中的一款产品) 在 Colab Enterprise 笔记本中编写和生成代码。

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

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

限制

Colab Enterprise 不支持 Colab 的“转换代码”功能。 如需了解在 Colab Enterprise 中使用 AI 赋能的功能来帮助您修改代码的其他方法, 请参阅使用数据科学智能体

准备工作

  1. 确保已为您的 用户账号和项目设置了 Google Cloud Gemini in Colab Enterprise。
  2. 登录您的 Google Cloud 账号。如果您是 Google Cloud新手, 请创建一个账号来评估我们的产品在 实际场景中的表现。新客户还可获享 $300 赠金,用于 运行、测试和部署工作负载。
  3. Enable the Vertex AI, Gemini for Google Cloud, Dataform, and Compute Engine APIs.

    Roles required to enable APIs

    To enable APIs, you need the Service Usage Admin IAM role (roles/serviceusage.serviceUsageAdmin), which contains the serviceusage.services.enable permission. Learn how to grant roles.

    Enable the APIs

  4. Enable the Vertex AI, Gemini for Google Cloud, Dataform, and Compute Engine APIs.

    Roles required to enable APIs

    To enable APIs, you need the Service Usage Admin IAM role (roles/serviceusage.serviceUsageAdmin), which contains the serviceusage.services.enable permission. Learn how to grant roles.

    Enable the APIs

所需的角色

如需获得使用 Colab Enterprise 中的 Gemini 所需的权限,请让您的管理员为您授予项目的以下 IAM 角色:

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

您也可以通过自定义角色或其他预定义角色来获取所需的权限。

启用代码补全和代码生成功能

如需启用代码补全和代码生成功能,请执行以下操作:

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

    前往我的笔记本

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

    “帮我编码”按钮下方的 Gemini in Colab Enterprise 功能列表,包括代码补全和代码生成。
    • 代码补全 功能会在您输入时建议代码。
    • 代码生成 功能会根据您对 需求的说明生成代码。

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

使用代码补全功能

如需在 Colab Enterprise 笔记本中使用代码补全功能,请执行以下操作:
  1. 在 Google Cloud 控制台中,前往 Colab Enterprise 我的笔记本 页面。

    前往我的笔记本

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

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

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

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

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

使用代码生成功能

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

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

    前往我的笔记本

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

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

  4. 在工具栏中,点击  Help me code 按钮,然后选择 Code generation (如果尚未选择)。

  5. 在笔记本中,添加一个空白代码单元,或点击一个不包含任何代码的代码单元。

  6. 点击  可用的 AI 功能,然后点击 生成代码

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

  8. 如需关闭生成 面板,请点击关闭

关闭代码补全和代码生成功能

如需关闭代码补全和代码生成功能,请执行以下操作:

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

  2. 取消选中要关闭的功能。

问题排查

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

Gemini 功能无法正常运行

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

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

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

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

后续步骤