25 lines
718 B
TypeScript
25 lines
718 B
TypeScript
|
/**
|
|||
|
|
* 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()
|
|||
|
|
}
|
|||
|
|
|