import { OpenAIProvider } from './providers/openai';
import { OllamaProvider } from './providers/ollama';
import { AIProvider } from './types';

export function getAIProvider(): AIProvider {
  const providerType = process.env.AI_PROVIDER || 'ollama'; // Default to ollama for local dev

  switch (providerType.toLowerCase()) {
    case 'ollama':
      console.log('Using Ollama Provider with model:', process.env.OLLAMA_MODEL || 'granite4:latest');
      return new OllamaProvider(
        process.env.OLLAMA_BASE_URL || 'http://localhost:11434/api',
        process.env.OLLAMA_MODEL || 'granite4:latest'
      );
    case 'openai':
    default:
      if (!process.env.OPENAI_API_KEY) {
        console.warn('OPENAI_API_KEY non configurée. Les fonctions IA pourraient échouer.');
      }
      return new OpenAIProvider(
        process.env.OPENAI_API_KEY || '',
        process.env.OPENAI_MODEL || 'gpt-4o-mini'
      );
  }
}