nodejs调用openai代码生成内容保存到txt
要调用OpenAI API,你需要先申请一个API Key,然后使用Node.js的HTTP请求模块向API发送请求,获取生成的内容并将其保存到txt文件中。
下面是一个示例代码:
const https = require('https');
const fs = require('fs');
const API_KEY = 'YOUR_API_KEY';
const MODEL_ID = 'YOUR_MODEL_ID';
const PROMPT = 'YOUR_PROMPT';
const MAX_TOKENS = 100;
const options = {
hostname: 'api.openai.com',
path: '/v1/engines/' + MODEL_ID + '/completions',
method: 'POST',
headers: {
'Content-Type': 'application/json',
'Authorization': 'Bearer ' + API_KEY
}
};
const data = JSON.stringify({
prompt: PROMPT,
max_tokens: MAX_TOKENS
});
const req = https.request(options, res => {
let chunks = [];
res.on('data', chunk => {
chunks.push(chunk);
});
res.on('end', () => {
const result = Buffer.concat(chunks);
const content = JSON.parse(result.toString()).choices[0].text;
fs.writeFile('output.txt', content, err => {
if (err) {
console.error(err);
} else {
console.log('Content saved to output.txt');
}
});
});
});
req.on('error', error => {
console.error(error);
});
req.write(data);
req.end();
在这个示例中,我们使用了一个名为https的Node.js模块来发送HTTP请求。我们首先定义了一些常量,包括API Key、模型ID、提示语和最大标记数。然后,我们设置了HTTP请求选项,包括请求的方法、头文件和请求体数据。我们使用JSON.stringify函数将请求体数据从JavaScript对象转换为JSON格式字符串。
接下来,我们创建了一个HTTP请求,并使用https.request方法发送它。我们定义了res.on('data')和res.on('end')事件处理程序来处理API的响应。我们将响应数据存储在一个缓冲区中,并使用JSON.parse函数将其从JSON格式字符串转换回JavaScript对象。我们从结果中提取生成的文本内容,并使用fs.writeFile将其保存到名为output.txt的文件中。
最后,我们定义了一个错误处理程序来处理HTTP请求中发生的任何错误。
你可以根据自己的需要更改常量和选项来使用此代码。
原文地址: http://www.cveoy.top/t/topic/bERY 著作权归作者所有。请勿转载和采集!