使用 Gemini API 构建多轮对话(聊天)


使用 Gemini API,您可以跨多个回合构建自由形式的对话。Vertex AI in Firebase SDK 通过管理对话状态来简化该过程,因此与 generateContentStream()generateContent() 不同,您无需自行存储对话历史记录。

准备工作

完成 Vertex AI in Firebase SDK 入门指南(如果您尚未完成)。请确保您已完成以下所有操作:

  1. 设置新的或现有的 Firebase 项目,包括使用 Blaze 定价方案以及启用所需的 API。

  2. 将您的应用与 Firebase 相关联,包括注册应用和将 Firebase 配置添加到应用。

  3. 添加 SDK,并在应用中初始化 Vertex AI 服务和生成式模型。

将应用关联到 Firebase、添加 SDK 并初始化 Vertex AI 服务和生成模型后,您就可以调用 Gemini API 了。

发送聊天提示请求

如需构建多轮对话(如聊天),请先通过调用 startChat() 初始化聊天。然后,使用 sendMessageStream()(或 sendMessage())发送新用户消息,这也会将消息和回复附加到聊天记录中。

与对话中的内容相关联的 role 有两种可能的选项:

  • user:提供提示的角色。此值是调用 sendMessageStream()(或 sendMessage())的默认值,如果传递其他角色,该函数会抛出异常。

  • model:提供回答的角色。使用现有 history 调用 startChat() 时,可以使用此角色。

选择是流式传输回答 (sendMessageStream),还是等待系统生成完整结果 (sendMessage)。

流式传输

通过不等待模型生成的完整结果,您可以实现更快的互动,而是使用流式传输来处理部分结果。

不直播

或者,您也可以等待整个结果,而不是流式传输;只有在模型完成整个生成过程后,才会返回结果。

了解如何根据用例和应用选择 Gemini 模型和(可选)位置

您还可以执行以下操作

试用 Gemini API 的其他功能

了解如何控制内容生成

您还可以使用 Vertex AI Studio 对提示和模型配置进行实验。

详细了解 Gemini 模型

了解适用于各种用例的模型及其配额和价格


就您使用 Vertex AI in Firebase 的体验提供反馈