Node.js SDK

环境要求

• Node.js 18 或更高版本

安装

npm install openai

快速开始

配置客户端

import OpenAI from 'openai';

const client = new OpenAI({
    baseURL: 'https://ai-tokenhub.com/api/v1',
    apiKey: '$YOUR_API_KEY'
});

基本调用

async function main() {
    const response = await client.chat.completions.create({
        model: 'gpt-4o',
        messages: [
            { role: 'system', content: 'You are a helpful assistant' },
            { role: 'user', content: 'Hello, please introduce yourself' }
        ]
    });

    console.log(response.choices[0].message.content);
}

main();

流式调用

async function main() {
    const stream = await client.chat.completions.create({
        model: 'gpt-4o',
        messages: [
            { role: 'user', content: 'Tell a joke about artificial intelligence' }
        ],
        stream: true
    });

    for await (const chunk of stream) {
        if (chunk.choices[0].delta.content) {
            process.stdout.write(chunk.choices[0].delta.content);
        }
    }
}

main();

使用 Claude 模型

async function main() {
    const response = await client.chat.completions.create({
        model: 'claude-3-5-sonnet-20241022',
        messages: [
            { role: 'user', content: 'Hello' }
        ]
    });

    console.log(response.choices[0].message.content);
}

main();

错误处理

import OpenAI from 'openai';

const client = new OpenAI({
    baseURL: 'https://ai-tokenhub.com/api/v1',
    apiKey: '$YOUR_API_KEY'
});

async function main() {
    try {
        const response = await client.chat.completions.create({
            model: 'gpt-4o',
            messages: [{ role: 'user', content: 'Hello' }]
        });
        console.log(response.choices[0].message.content);
    } catch (error) {
        if (error.status === 401) {
            console.error('Invalid API key');
        } else if (error.status === 429) {
            console.error('Rate limit exceeded, please try again later');
        } else {
            console.error('An error occurred:', error.message);
        }
    }
}

main();

Next.js 集成示例

环境变量

# .env.local
OPENAI_API_KEY=$YOUR_API_KEY
OPENAI_BASE_URL=https://ai-tokenhub.com/api/v1

API 路由

// app/api/chat/route.js
import OpenAI from 'openai';

const client = new OpenAI({
    apiKey: process.env.OPENAI_API_KEY,
    baseURL: process.env.OPENAI_BASE_URL
});

export async function POST(request) {
    const { messages } = await request.json();

    const response = await client.chat.completions.create({
        model: 'gpt-4o',
        messages
    });

    return Response.json(response);
}