28 lines
797 B
TypeScript
28 lines
797 B
TypeScript
import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
|
|
import { generateText } from 'ai';
|
|
import 'dotenv/config';
|
|
|
|
// 使用 custom-cnb provider (聚合了多个模型)
|
|
const cnb = createOpenAICompatible({
|
|
baseURL: 'https://api.cnb.cool/kevisual/kevisual/-/ai',
|
|
name: 'custom-cnb',
|
|
apiKey: process.env.CNB_API_KEY!,
|
|
});
|
|
|
|
// 测试所有可用模型
|
|
// const models = ['qwen3-coder-plus', 'ark-code-latest', 'GLM-4.7', 'MiniMax-M2.1'];
|
|
const models = ['CNB_Models'];
|
|
|
|
for (const model of models) {
|
|
console.log(`\n=== Testing model: ${model} ===`);
|
|
try {
|
|
const { text } = await generateText({
|
|
model: cnb(model),
|
|
prompt: 'Say hello in one sentence.',
|
|
});
|
|
console.log(`Response: ${text}`);
|
|
} catch (error) {
|
|
console.error(`Error: ${error}`);
|
|
}
|
|
}
|