如何在 Node.js 项目中集成 Google Gemini AI？

生成式 AI 技术正在彻底改变技术世界的面貌，Google 推出的 Gemini 就是这场变革中的佼佼者。Google 通过发布 Gemini 的 API，为开发者们开辟了前所未有的新天地。

在本文中，小编要带大家一起了解如何使用 Google Gemini SDK 并将其集成到 Node.js 应用中去。

1. 关于 Google Gemini

Google Gemini 是由 Google AI 打造的一款强大的多模态大语言模型，它能够理解并生成多种格式的内容，包括文本、代码、音频、图像以及视频。这种多才多艺的能力，让它成为了那些想要在自己的 Node.js 项目中融入 AI 技术的开发者们的得力助手。

2. 项目配置

2.1 Node.js 项目初始化

请在终端中执行以下命令来新建一个 Node.js 项目，该项目将采用默认配置：

npm init -y

2.2 安装依赖包

在终端中使用以下命令安装必须的依赖包：

npm install express @google/generative-ai dotenv

express: 一个用于 Node.js 的网页应用程序框架。
@google/generative-ai: 这是官方提供的 SDK，用于接入 Google Gemini。
dotenv: 这个模块用于管理环境变量。

2.3 配置环境变量

在项目根目录创建一个 .env 文件，并在文件中填写以下配置信息：

API_KEY=YOUR_API_KEY PORT=3000

请注意，需要将 YOUR_API_KEY 替换成您从 Google 获取的 API 密钥。

2.4 获取 Google Gemini API KEY

想要使用 Google Gemini，首先要申请 API 密钥。方法如下：

先登录您的 Google 账户，然后前往 Google API 控制台。
根据页面指引，便可生成一个全新的 API 密钥。

2.5 设置 Express 服务器

创建以下 index.js 文件来设置 Express Server：

const express = require("express");
require("dotenv").config();

const app = express();
app.use(express.json());

const port = process.env.PORT || 3000;

app.get("/", (req, res) => {
  res.send("Node.js and Google Gemini integration example");
});

app.listen(port, () => {
  console.log(`Server listening on port ${port}`);
});

2.6 运行项目

将以下配置追加到 package.json 文件：

"scripts": {
  "start": "node index.js"
}

随后，在终端中执行以下命令运行服务：

npm start

最后，打开浏览器输入以下地址来访问服务：

http://localhost:3000/

3. 集成 Google Gemini

3.1 配置路由及中间件

在 index.js 文件中添加以下路由代码来集成 Gemini：

const { handleGenerateRequest } = require("./handlers/generateContent");

app.post("/generate", handleGenerateRequest);

3.2 谷歌 Generative AI 设置

在 handlers 目录中创建以下名为 generateContent.js 的文件：

const { GoogleGenerativeAI } = require("@google/generative-ai");

const apiKey = process.env.API_KEY;
const config = new GoogleGenerativeAI(apiKey);
const modelId = "gemini-pro";
const model = config.getGenerativeModel({ model: modelId });

exports.handleGenerateRequest = async (req, res) => {
  try {
    const { prompt } = req.body;
    const result = await model.generateContent({ prompt });
    res.json({ response: result });
  } catch (error) {
    console.error(error);
    res.status(500).send("An error occurred while generating content.");
  }
};

3.3 集成测试

想要进行测试，可以用 Postman 或者其他 HTTP 客户端，向 http://localhost:3000/generate 地址发送一个 POST 请求，请求体是一个包含提示词的 JSON 格式信息。然后，根据响应内容来确保集成的正确性。

如果一切顺利，那么您已经成功地把 Google Gemini 集成到了 Node.js 应用程序中，相信借助 Gemini 强大的 AI 生成能力，您的应用功能一定更加丰富多彩。