创建代码补全提示

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:
  

后续步骤