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

export function getAIProvider(config?: Record<string, string>): AIProvider {
  const providerType = config?.AI_PROVIDER || process.env.AI_PROVIDER || 'ollama';

  switch (providerType.toLowerCase()) {
    case 'ollama':
      let baseUrl = config?.OLLAMA_BASE_URL || process.env.OLLAMA_BASE_URL || 'http://localhost:11434';
      const model = config?.AI_MODEL_TAGS || process.env.OLLAMA_MODEL || 'granite4:latest';
      const embedModel = config?.AI_MODEL_EMBEDDING || process.env.OLLAMA_EMBEDDING_MODEL || 'embeddinggemma:latest';

      // Ensure baseUrl doesn't end with /api, we'll add it in OllamaProvider
      if (baseUrl.endsWith('/api')) {
        baseUrl = baseUrl.slice(0, -4); // Remove /api
      }

      return new OllamaProvider(baseUrl, model, embedModel);
    case 'openai':
    default:
      const apiKey = config?.OPENAI_API_KEY || process.env.OPENAI_API_KEY || '';
      const aiModel = config?.AI_MODEL_TAGS || process.env.OPENAI_MODEL || 'gpt-4o-mini';

      if (!apiKey && providerType.toLowerCase() === 'openai') {
        console.warn('OPENAI_API_KEY non configurée.');
      }
      return new OpenAIProvider(apiKey, aiModel);
  }
}