111 lines
2.6 KiB
TypeScript
111 lines
2.6 KiB
TypeScript
|
/**
|
|||
|
|
* 統一的模型列表
|
|||
|
|
* 包含 Gemini 和 Grok 所有可用模型
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
import { GEMINI_MODELS, type GeminiModel } from './gemini-models'
|
|||
|
|
import { GROK_MODELS, type GrokModel } from './grok-models'
|
|||
|
|
|
|||
|
|
export interface UnifiedModel {
|
|||
|
|
name: string
|
|||
|
|
displayName: string
|
|||
|
|
description: string
|
|||
|
|
provider: 'gemini' | 'grok'
|
|||
|
|
capabilities: {
|
|||
|
|
text: boolean
|
|||
|
|
vision: boolean
|
|||
|
|
audio: boolean
|
|||
|
|
video: boolean
|
|||
|
|
imageGeneration: boolean
|
|||
|
|
tts: boolean
|
|||
|
|
}
|
|||
|
|
category: 'text' | 'vision' | 'multimodal' | 'image-generation' | 'video-generation' | 'audio' | 'tts' | 'robotics'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 將 Gemini 模型轉換為統一格式
|
|||
|
|
*/
|
|||
|
|
function convertGeminiModel(model: GeminiModel): UnifiedModel {
|
|||
|
|
return {
|
|||
|
|
name: model.name,
|
|||
|
|
displayName: model.displayName,
|
|||
|
|
description: model.description,
|
|||
|
|
provider: 'gemini',
|
|||
|
|
capabilities: model.capabilities,
|
|||
|
|
category: model.category
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 將 Grok 模型轉換為統一格式
|
|||
|
|
*/
|
|||
|
|
function convertGrokModel(model: GrokModel): UnifiedModel {
|
|||
|
|
return {
|
|||
|
|
name: model.name,
|
|||
|
|
displayName: model.displayName,
|
|||
|
|
description: model.description,
|
|||
|
|
provider: 'grok',
|
|||
|
|
capabilities: model.capabilities,
|
|||
|
|
category: model.category
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 所有可用模型列表（Gemini + Grok）
|
|||
|
|
*/
|
|||
|
|
export const ALL_MODELS: UnifiedModel[] = [
|
|||
|
|
...GEMINI_MODELS.map(convertGeminiModel),
|
|||
|
|
...GROK_MODELS.map(convertGrokModel)
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 根據功能過濾模型
|
|||
|
|
*/
|
|||
|
|
export function getModelsByCapability(capability: keyof UnifiedModel['capabilities']): UnifiedModel[] {
|
|||
|
|
return ALL_MODELS.filter(m => m.capabilities[capability] === true)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 根據 Provider 過濾模型
|
|||
|
|
*/
|
|||
|
|
export function getModelsByProvider(provider: 'gemini' | 'grok'): UnifiedModel[] {
|
|||
|
|
return ALL_MODELS.filter(m => m.provider === provider)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 根據類別過濾模型
|
|||
|
|
*/
|
|||
|
|
export function getModelsByCategory(category: UnifiedModel['category']): UnifiedModel[] {
|
|||
|
|
return ALL_MODELS.filter(m => m.category === category)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 取得模型資訊
|
|||
|
|
*/
|
|||
|
|
export function getModelInfo(modelName: string): UnifiedModel | undefined {
|
|||
|
|
return ALL_MODELS.find(m => m.name === modelName)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 根據模型名稱判斷 Provider
|
|||
|
|
*/
|
|||
|
|
export function getProviderForModel(modelName: string): 'gemini' | 'grok' {
|
|||
|
|
const model = getModelInfo(modelName)
|
|||
|
|
if (model) {
|
|||
|
|
return model.provider
|
|||
|
|
}
|
|||
|
|
// 如果找不到模型，根據名稱判斷
|
|||
|
|
if (modelName.toLowerCase().includes('grok')) {
|
|||
|
|
return 'grok'
|
|||
|
|
}
|
|||
|
|
return 'gemini'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 根據模型名稱判斷 Provider（別名，為了向後兼容）
|
|||
|
|
*/
|
|||
|
|
export function getModelProvider(modelName: string): 'gemini' | 'grok' {
|
|||
|
|
return getProviderForModel(modelName)
|
|||
|
|
}
|
|||
|
|
|