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

· 3分钟阅读
Google Gemini
Google Gemini

生成式 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 密钥。方法如下:

  1. 先登录您的 Google 账户,然后前往 Google API 控制台
  2. 根据页面指引,便可生成一个全新的 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 生成能力,您的应用功能一定更加丰富多彩。