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


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

准备工作

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

  1. 设置一个新的 Firebase 项目或一个现有的 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 的体验提供反馈