nodejs_example.js

/**
 * BlueEagle AI Gateway - Node.js 调用示例
 * 官方文档: https://ahg.codes
 */

import OpenAI from 'openai';

// 初始化客户端 | Initialize client
// 仅需修改 baseUrl，其余与官方 OpenAI SDK 完全一致
const client = new OpenAI({
  apiKey: 'YOUR_API_KEY',       // 替换为您的蓝鹰API密钥
  baseURL: 'https://ahg.codes/v1'
});

/**
 * 非流式对话 | Non-streaming chat
 */
async function chatNonStreaming() {
  try {
    const response = await client.chat.completions.create({
      model: 'gpt-4o',
      messages: [
        { role: 'system', content: 'You are a helpful assistant.' },
        { role: 'user', content: 'Hello, BlueEagle AI Gateway!' }
      ],
      temperature: 0.7,
      max_tokens: 500
    });

    console.log('='.repeat(50));
    console.log('非流式响应 | Non-streaming Response:');
    console.log('='.repeat(50));
    console.log(response.choices[0].message.content);
    console.log(`\n消耗tokens: ${response.usage.total_tokens}`);
  } catch (error) {
    console.error('Error:', error.message);
  }
}

/**
 * 流式对话 | Streaming chat
 */
async function chatStreaming() {
  try {
    const stream = await client.chat.completions.create({
      model: 'gpt-4o',
      messages: [
        { role: 'user', content: '请用三句话介绍蓝鹰AI网关的优势' }
      ],
      stream: true,
      temperature: 0.7
    });

    console.log('='.repeat(50));
    console.log('流式响应 | Streaming Response:');
    console.log('='.repeat(50));

    for await (const chunk of stream) {
      if (chunk.choices[0]?.delta?.content) {
        process.stdout.write(chunk.choices[0].delta.content);
      }
    }
    console.log('\n');
  } catch (error) {
    console.error('Error:', error.message);
  }
}

/**
 * 调用 Claude 模型 | Claude model
 */
async function chatWithClaude() {
  try {
    const response = await client.chat.completions.create({
      model: 'claude-3-5-sonnet-20241022',
      messages: [
        { role: 'user', content: 'Explain quantum computing in simple terms.' }
      ],
      max_tokens: 1000
    });

    console.log('='.repeat(50));
    console.log('Claude 响应 | Claude Response:');
    console.log('='.repeat(50));
    console.log(response.choices[0].message.content);
  } catch (error) {
    console.error('Error:', error.message);
  }
}

/**
 * 调用 Gemini 模型 | Gemini model
 */
async function chatWithGemini() {
  try {
    const response = await client.chat.completions.create({
      model: 'gemini-1.5-pro-latest',
      messages: [
        { role: 'user', content: 'What are the benefits of AI gateways?' }
      ],
      max_tokens: 1000
    });

    console.log('='.repeat(50));
    console.log('Gemini 响应 | Gemini Response:');
    console.log('='.repeat(50));
    console.log(response.choices[0].message.content);
  } catch (error) {
    console.error('Error:', error.message);
  }
}

/**
 * 函数调用 | Function calling
 */
async function functionCalling() {
  try {
    const tools = [
      {
        type: 'function',
        function: {
          name: 'get_weather',
          description: 'Get current weather for a location',
          parameters: {
            type: 'object',
            properties: {
              location: {
                type: 'string',
                description: 'City name, e.g. Beijing'
              }
            },
            required: ['location']
          }
        }
      }
    ];

    const response = await client.chat.completions.create({
      model: 'gpt-4o',
      messages: [
        { role: 'user', content: "What's the weather like in Shanghai?" }
      ],
      tools: tools,
      tool_choice: 'auto'
    });

    console.log('='.repeat(50));
    console.log('函数调用 | Function Calling:');
    console.log('='.repeat(50));
    console.log(JSON.stringify(response.choices[0].message, null, 2));
  } catch (error) {
    console.error('Error:', error.message);
  }
}

// 运行示例 | Run examples
console.log('🦅 BlueEagle AI Gateway - Node.js Examples\n');

// 取消注释以运行对应示例
chatNonStreaming();
// chatStreaming();
// chatWithClaude();
// chatWithGemini();
// functionCalling();