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}`);
  }
}