index.js

const { loadConfig, CONFIG_FILE } = require('./lib/config');
const { OpenAICompatibleClient } = require('./lib/ai');

async function main() {
  const command = process.argv[2] || 'chat';
  const prompt = process.argv.slice(3).join(' ') || 'Say hello in English and Chinese.';

  const config = loadConfig();
  if (!config) {
    console.error(`Missing config file: ${CONFIG_FILE}`);
    console.error('Copy config.example.json to config.json and fill in baseURL/apiKey/model first.');
    process.exit(1);
  }

  const client = new OpenAICompatibleClient(config);
  const messages = [
    {
      role: 'system',
      content: 'You are a helpful assistant.'
    },
    {
      role: 'user',
      content: prompt
    }
  ];

  if (command === 'stream') {
    await client.streamChat(messages, {
      maxTokens: 1000,
      onDelta: async (delta) => {
        process.stdout.write(delta);
      },
      onComplete: async () => {
        process.stdout.write('\n');
      }
    });
    return;
  }

  if (command === 'serve') {
    const { startServer } = require('./lib/server');
    startServer(config);
    return;
  }

  const reply = await client.chat(messages, { maxTokens: 1000 });
  console.log(reply);
}

main().catch((err) => {
  console.error(err.message);
  process.exit(1);
});