黑马天启 · 多模型 Token 平台

Node.js 调用

使用 OpenAI 官方 SDK

npm i openai
import OpenAI from 'openai';

const client = new OpenAI({
  apiKey: process.env.HEIMA_API_KEY,
  baseURL: 'https://api.token.heimatianqi.com/v1',
});

const resp = await client.chat.completions.create({
  model: 'deepseek-flash',
  messages: [{ role: 'user', content: '你好' }],
});
console.log(resp.choices[0].message.content, resp.usage);

流式输出

const stream = await client.chat.completions.create({
  model: 'deepseek-flash',
  messages: [{ role: 'user', content: '用三句话介绍你自己' }],
  stream: true,
});
for await (const chunk of stream) {
  process.stdout.write(chunk.choices[0]?.delta?.content ?? '');
}

不依赖 SDK(fetch)

const base = process.env.HEIMA_BASE_URL || 'https://api.token.heimatianqi.com/v1';
const res = await fetch(`${base}/chat/completions`, {
  method: 'POST',
  headers: {
    Authorization: `Bearer ${process.env.HEIMA_API_KEY}`,
    'Content-Type': 'application/json',
  },
  body: JSON.stringify({ model: 'mock-chat', messages: [{ role: 'user', content: '你好' }] }),
});
const data = await res.json();
console.log(data.choices[0].message.content.slice(0, 50));
console.log('usage:', data.usage);