使用 Gemini API 根据纯文本提示生成文本


使用 Vertex AI in Firebase SDK 从应用调用 Gemini API 时,您可以提示 Gemini 模型根据纯文本输入生成文本。

准备工作

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

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

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

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

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

根据纯文本输入生成文本

您可以使用仅包含文本的输入调用 Gemini API。对于这些调用,您需要使用支持仅文本提示的模型(例如 Gemini 2.0 Flash)。

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

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

以下示例展示了如何使用 generateContentStream 从仅包含文本的提示请求中流式传输生成的文本:

import FirebaseVertexAI

// Initialize the Vertex AI service
let vertex = VertexAI.vertexAI()

// Initialize the generative model with a model that supports your use case
let model = vertex.generativeModel(modelName: "gemini-2.0-flash")

// Provide a prompt that contains text
let prompt = "Write a story about a magic backpack."

// To stream generated text output, call generateContentStream with the text input
let contentStream = try model.generateContentStream(prompt)
for try await chunk in contentStream {
  if let text = chunk.text {
    print(text)
  }
}

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

以下示例展示了如何使用 generateContent 从仅包含文本的提示请求生成文本:

import FirebaseVertexAI

// Initialize the Vertex AI service
let vertex = VertexAI.vertexAI()

// Initialize the generative model with a model that supports your use case
let model = vertex.generativeModel(modelName: "gemini-2.0-flash")

// Provide a prompt that contains text
let prompt = "Write a story about a magic backpack."

// To generate text output, call generateContent with the text input
let response = try await model.generateContent(prompt)
print(response.text ?? "No text in response.")

了解如何选择适合您的用例和应用的 Gemini 模型和(可选)位置

您还可以执行以下操作

  • 了解如何在向模型发送长提示之前计算令牌数
  • 开始考虑为正式版做好准备,包括设置 Firebase App Check,以保护 Gemini API 免遭未经授权的客户端滥用。

试用 Gemini API 的其他功能

了解如何控制内容生成

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

详细了解 Gemini 模型

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


提供有关 Vertex AI in Firebase 使用体验的反馈