node.js调用豆包AI

最新推荐文章于 2024-10-07 06:31:57 发布

AriesSui

最新推荐文章于 2024-10-07 06:31:57 发布

阅读量451

点赞数 8

文章标签： node.js 人工智能

本文链接：https://blog.csdn.net/m0_46248994/article/details/141393634

版权

安装 openai 库
```
  npm install openai
```

请参考如下示例代码进行调用

import OpenAI from 'openai';

const openai = new OpenAI({
  apiKey: process.env['ARK_API_KEY'], // 豆包AI的key
  baseURL: 'https://ark.cn-beijing.volces.com/api/v3',
});

async function main() {
  // Non-streaming:
  console.log('----- standard request -----')
  const completion = await openai.chat.completions.create({
    messages: [
      { role: 'system', content: '你是豆包，是由字节跳动开发的 AI 人工智能助手' },
      { role: 'user', content: '常见的十字花科植物有哪些？' },
    ],
    model: 'ep-20240628043416-x9skm',
  });
  console.log(completion.choices[0]?.message?.content);

  // Streaming:
  console.log('----- streaming request -----')
  const stream = await openai.chat.completions.create({
    messages: [
      { role: 'system', content: '你是豆包，是由字节跳动开发的 AI 人工智能助手' },
      { role: 'user', content: '常见的十字花科植物有哪些？' },
    ],
    model: 'ep-20240628043416-x9skm',
    stream: true, // 用来控制是否打开流式传输
  });
  for await (const part of stream) {
    // 监听流式传输，当done时结束监听
    process.stdout.write(part.choices[0]?.delta?.content || '');
  }
  process.stdout.write('\n');
}

main();