使用Node.js搭建OpenAI开发环境

727字
4分钟
2024-06-25
- AI

OpenAI提供了强大的API，可以用于自然语言处理、生成文本等各种应用。在本文中，我将介绍如何使用Node.js搭建OpenAI开发环境，包括环境配置、依赖安装以及示例代码讲解，帮助你快速上手OpenAI API。

环境配置

1. 安装Node.js

首先，确保你已经安装了Node.js。你可以通过以下命令检查是否安装了Node.js：

1
node -v

如果没有安装Node.js，可以从Node.js官网下载并安装最新版本。

2. 初始化项目

在项目目录下运行以下命令，初始化一个新的Node.js项目：

1
mkdir openai-nodejs-project
2
cd openai-nodejs-project
3
npm init -y

这将创建一个package.json文件，用于管理项目的依赖和配置。

依赖安装

1. 安装OpenAI SDK

OpenAI提供了官方的Node.js SDK。你可以通过以下命令安装它：

1
npm install openai

2. 安装其他依赖

为了方便后续的开发，可以安装dotenv库来管理环境变量，以及axios库来进行HTTP请求（如果需要）：

1
npm install dotenv axios

配置环境变量

在项目根目录下创建一个.env文件，添加你的OpenAI API密钥：

1
OPENAI_API_KEY=your_openai_api_key_here

然后，在项目中加载环境变量。修改或创建一个index.js文件，添加以下代码：

1
require("dotenv").config();
2
const { Configuration, OpenAIApi } = require("openai");
3

4
const configuration = new Configuration({
5
  apiKey: process.env.OPENAI_API_KEY,
6
});
7
const openai = new OpenAIApi(configuration);
8

9
async function generateText(prompt) {
10
  const response = await openai.createCompletion({
11
    model: "text-davinci-003",
12
    prompt: prompt,
13
    max_tokens: 150,
14
  });
15
  return response.data.choices[0].text.trim();
16
}
17

18
(async () => {
19
  try {
20
    const text = await generateText("解释一下机器学习的基本概念。");
21
    console.log(text);
22
  } catch (error) {
23
    console.error("Error generating text:", error);
24
  }
25
})();

示例代码讲解

1. 加载环境变量

1
require("dotenv").config();

这行代码使用dotenv库加载.env文件中的环境变量，使得process.env.OPENAI_API_KEY可以在代码中使用。

2. 配置OpenAI API

1
const { Configuration, OpenAIApi } = require("openai");
2

3
const configuration = new Configuration({
4
  apiKey: process.env.OPENAI_API_KEY,
5
});
6
const openai = new OpenAIApi(configuration);

这里我们使用从openai库中导入的Configuration和OpenAIApi类，配置API密钥并创建一个OpenAIApi实例。

3. 生成文本的函数

1
async function generateText(prompt) {
2
  const response = await openai.createCompletion({
3
    model: "text-davinci-003",
4
    prompt: prompt,
5
    max_tokens: 150,
6
  });
7
  return response.data.choices[0].text.trim();
8
}

这个函数接收一个提示词（prompt），使用OpenAI的createCompletion方法生成文本。max_tokens参数指定生成文本的最大长度。

4. 调用生成文本函数

1
(async () => {
2
  try {
3
    const text = await generateText("解释一下机器学习的基本概念。");
4
    console.log(text);
5
  } catch (error) {
6
    console.error("Error generating text:", error);
7
  }
8
})();