import axios from 'axios';

const OPENROUTER_API_URL = 'https://openrouter.ai/api/v1/chat/completions';

const TARGET_MODELS = [
  'mistralai/mistral-small',
  'mistralai/mistral-medium',
  'mistralai/mistral-nemo' // fallback free-tier compatible model
];

let currentModelIndex = 0;

export async function getAiCompletion(prompt: string) {
  let attempts = 0;
  
  while (attempts < TARGET_MODELS.length) {
    const model = TARGET_MODELS[currentModelIndex];
    try {
      const response = await axios.post(
        OPENROUTER_API_URL,
        {
          model: model,
          messages: [{ role: 'user', content: prompt }],
        },
        {
          headers: {
            'Authorization': `Bearer ${process.env.OPENROUTER_API_KEY}`,
            'HTTP-Referer': process.env.APP_URL || 'http://localhost:3000',
            'X-Title': 'BRIDGE ModulS Dashboard',
          },
        }
      );
      
      // If successful, we keep this model as the active one for next time
      return response.data.choices[0].message.content;
    } catch (error: any) {
      console.error(`Error with model ${model}:`, error?.response?.data || error.message);
      attempts++;
      // Rotate to the next model for the next attempt
      currentModelIndex = (currentModelIndex + 1) % TARGET_MODELS.length;
    }
  }
  
  throw new Error('All OpenRouter Mistral models failed. Please check your API key and quotas.');
}