使用JS调用OpenAI API

使用 JS 调用 OpenAI API

分为 await 和 stream 两种方式

const { Configuration, OpenAIApi } = require('openai');

const configuration = new Configuration({
  apiKey: 'sk-123456789876543421',
});

const openai = new OpenAIApi(configuration);

async function main() {
  const response = await openai.createCompletion(
    {
      model: 'text-davinci-003',
      prompt: '你好,今天你的心情好吗？',
      temperature: 0.7,
      max_tokens: 1000,
      // stream: true
    },
    {
      timeout: 120000,
      timeoutErrorMessage: 'open ai timeout',
      // responseType: 'stream'
    }
  );

  console.log(response.data?.choices?.[0]?.text);

  //   Stream
  //   response.data.on("data", (data) => {
  //     const lines = data
  //         ?.toString()
  //         ?.split("\n")
  //         .filter((line) => line.trim() !== "");
  //     for (const line of lines) {
  //         const message = line.replace(/^data: /, "");
  //         if (message == "[DONE]") {
  //            console.log(message)
  //         } else {
  //             let token;
  //             try {
  //                 token = JSON.parse(message)?.choices?.[0]?.text;
  //                 result.data += token
  //                 console.log(token)
  //             } catch {
  //                 console.log(token);
  //             }
  //         }
  //     }
  //   });
}

main();