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

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

了解 Gemini for 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 角色:

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

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

启用代码补全和代码生成

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

  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 帮我写代码按钮,然后选择代码补全(如果尚未选择)。
  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. 在工具栏中,点击  帮我写代码按钮,然后选择代码生成(如果尚未选择)。

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

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

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

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

关闭代码补全和代码生成

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

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

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

问题排查

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

Gemini 功能无法正常运行

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

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

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

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

后续步骤