nodejs_example.js

/**
 * 蓝鹰AI网关 - Node.js 调用示例
 * BlueEagle AI Gateway - Node.js Usage Example
 *
 * 官方文档: https://ahg.codes
 *
 * 安装依赖: npm install openai
 */

import OpenAI from 'openai';

const client = new OpenAI({
  baseURL: 'https://ahg.codes/v1',
  apiKey: 'YOUR_API_KEY', // 从控制台获取 | Get from dashboard
});

/**
 * 使用 GPT-4o 进行对话
 * Chat with GPT-4o
 */
async function chatWithGPT4o() {
  const response = await client.chat.completions.create({
    model: 'gpt-4o',
    messages: [
      { role: 'system', content: 'You are a helpful assistant.' },
      { role: 'user', content: 'Hello, BlueEagle!' },
    ],
    temperature: 0.7,
    max_tokens: 2048,
  });

  console.log('GPT-4o Response:');
  console.log(response.choices[0].message.content);
}

/**
 * 使用 Claude 3.5 Sonnet 进行对话
 * Chat with Claude 3.5 Sonnet
 */
async function chatWithClaude() {
  const response = await client.chat.completions.create({
    model: 'claude-3-5-sonnet-20241022',
    messages: [
      { role: 'user', content: 'Explain quantum computing in simple terms.' },
    ],
    temperature: 0.5,
    max_tokens: 4096,
  });

  console.log('Claude 3.5 Sonnet Response:');
  console.log(response.choices[0].message.content);
}

/**
 * 使用 Gemini 1.5 Pro 进行对话
 * Chat with Gemini 1.5 Pro
 */
async function chatWithGemini() {
  const response = await client.chat.completions.create({
    model: 'gemini-1.5-pro-latest',
    messages: [
      { role: 'user', content: 'Write a JavaScript function to calculate Fibonacci numbers.' },
    ],
    temperature: 0.3,
    max_tokens: 2048,
  });

  console.log('Gemini 1.5 Pro Response:');
  console.log(response.choices[0].message.content);
}

/**
 * 流式输出示例
 * Streaming response example
 */
async function streamChat() {
  const response = await client.chat.completions.create({
    model: 'gpt-4o-mini',
    messages: [{ role: 'user', content: 'Tell me a short story.' }],
    stream: true,
    max_tokens: 1024,
  });

  console.log('Streaming Response:');
  for await (const chunk of response) {
    if (chunk.choices[0]?.delta?.content) {
      process.stdout.write(chunk.choices[0].delta.content);
    }
  }
  console.log();
}

/**
 * 列出可用模型
 * List available models
 */
async function listModels() {
  const models = await client.models.list();
  console.log('Available Models:');
  for (const model of models.data) {
    console.log(`  - ${model.id}`);
  }
}

// 运行示例 | Run example
(async () => {
  try {
    await chatWithGPT4o();
    // await chatWithClaude();
    // await chatWithGemini();
    // await streamChat();
    // await listModels();
  } catch (error) {
    console.error('Error:', error.message);
  }
})();