创建代码补全提示

Vertex AI Codey API 包含代码补全 API，该 API 支持基于最近编写的代码提供代码建议。使用生成式 AI 基础模型 code-gecko 与 code completion API 交互。本主题可帮助您了解如何创建提示，以便使用 code-gecko 模型创建代码补全建议。

使用场景

使用 code completion API 将 code-gecko 模型及其代码补全功能集成到 IDE 中。以下是代码补全的一些常见应用场景：

  • 更快地编写代码：利用 code-gecko 模型和为您建议的代码更快地编写代码。

  • 最大限度地减少代码中的 bug：使用语法上正确的代码建议来避免错误。代码补全功能可帮助您最大限度地降低在编写代码时意外引入 bug 的风险。

支持的模型

以下模型支持代码补全任务：

  • Code Gecko

代码补全提示示例

请参阅以下示例，了解如何设计代码补全提示。

补全代码函数的提示

您可以使用代码补全模型来补全代码定义。以下提示会生成代码，用于补全部分类型化的代码定义：

def reverse_string(s):
  
"""
   :type s: str
   :rtype: str
   """

补全测试函数的提示

您可以使用代码补全模型来补全测试函数。以下提示会生成用于补全测试函数的代码。回复中的函数 test_empty_input_string 会测试 reverse_string 函数。

def reverse_string(s):
    return s[::-1]
def test_empty_input_string()
  
-> None:
    assert reverse_string("") == ""
def test_one_character_string() -> None:
    assert reverse_string("a") == "a"
def test_two_character_string() -> None:

后续步骤