/**
 * AI Provider Selector Composable
 * 根據模型名稱自動選擇對應的 provider
 */

import type { AIProvider } from '~/types/ai'
import { GeminiClient } from '~/utils/clients/gemini'
import { GrokClient } from '~/utils/clients/grok'
import { getProviderForModel } from '~/utils/clients/all-models'

/**
 * 取得指定模型對應的 AI Provider 實例
 * @param modelName - 模型名稱，如果不提供則使用預設（Gemini）
 */
export function useAIProvider(modelName?: string): AIProvider {
  if (modelName) {
    const provider = getProviderForModel(modelName)
    return provider === 'grok' ? new GrokClient() : new GeminiClient()
  }
  
  // 預設使用 Gemini
  return new GeminiClient()
}