mirror of
https://github.com/geoffsee/open-gsio.git
synced 2025-09-08 22:56:46 +00:00
89 lines
2.3 KiB
TypeScript
89 lines
2.3 KiB
TypeScript
const SUPPORTED_MODELS_GROUPS = {
|
|
openai: [
|
|
// "o1-preview",
|
|
// "o1-mini",
|
|
// "gpt-4o",
|
|
// "gpt-3.5-turbo"
|
|
],
|
|
groq: [
|
|
// "mixtral-8x7b-32768",
|
|
// "deepseek-r1-distill-llama-70b",
|
|
'meta-llama/llama-4-scout-17b-16e-instruct',
|
|
'gemma2-9b-it',
|
|
'mistral-saba-24b',
|
|
// "qwen-2.5-32b",
|
|
'llama-3.3-70b-versatile',
|
|
// "llama-3.3-70b-versatile"
|
|
// "llama-3.1-70b-versatile",
|
|
// "llama-3.3-70b-versatile"
|
|
],
|
|
cerebras: ['llama-3.3-70b'],
|
|
claude: [
|
|
// "claude-3-5-sonnet-20241022",
|
|
// "claude-3-opus-20240229"
|
|
],
|
|
fireworks: [
|
|
// "llama-v3p1-405b-instruct",
|
|
// "llama-v3p1-70b-instruct",
|
|
// "llama-v3p2-90b-vision-instruct",
|
|
// "mixtral-8x22b-instruct",
|
|
// "mythomax-l2-13b",
|
|
// "yi-large"
|
|
],
|
|
google: [
|
|
// "gemini-2.0-flash-exp",
|
|
// "gemini-1.5-flash",
|
|
// "gemini-exp-1206",
|
|
// "gemini-1.5-pro"
|
|
],
|
|
xai: [
|
|
// "grok-beta",
|
|
// "grok-2",
|
|
// "grok-2-1212",
|
|
// "grok-2-latest",
|
|
// "grok-beta"
|
|
],
|
|
cloudflareAI: [
|
|
'llama-3.2-3b-instruct', // max_tokens
|
|
'llama-3-8b-instruct', // max_tokens
|
|
'llama-3.1-8b-instruct-fast', // max_tokens
|
|
'deepseek-math-7b-instruct',
|
|
'deepseek-coder-6.7b-instruct-awq',
|
|
'hermes-2-pro-mistral-7b',
|
|
'openhermes-2.5-mistral-7b-awq',
|
|
'mistral-7b-instruct-v0.2',
|
|
'neural-chat-7b-v3-1-awq',
|
|
'openchat-3.5-0106',
|
|
// "gemma-7b-it",
|
|
],
|
|
};
|
|
|
|
export type SupportedModel =
|
|
| keyof typeof SUPPORTED_MODELS_GROUPS
|
|
| (typeof SUPPORTED_MODELS_GROUPS)[keyof typeof SUPPORTED_MODELS_GROUPS][number];
|
|
|
|
export type ModelFamily = keyof typeof SUPPORTED_MODELS_GROUPS;
|
|
|
|
function getModelFamily(model: string): ModelFamily | undefined {
|
|
return Object.keys(SUPPORTED_MODELS_GROUPS)
|
|
.filter(family => {
|
|
return SUPPORTED_MODELS_GROUPS[family as keyof typeof SUPPORTED_MODELS_GROUPS].includes(
|
|
model.trim(),
|
|
);
|
|
})
|
|
.at(0) as ModelFamily | undefined;
|
|
}
|
|
|
|
const SUPPORTED_MODELS = [
|
|
// ...SUPPORTED_MODELS_GROUPS.xai,
|
|
// ...SUPPORTED_MODELS_GROUPS.claude,
|
|
// ...SUPPORTED_MODELS_GROUPS.google,
|
|
...SUPPORTED_MODELS_GROUPS.groq,
|
|
// ...SUPPORTED_MODELS_GROUPS.fireworks,
|
|
// ...SUPPORTED_MODELS_GROUPS.openai,
|
|
// ...SUPPORTED_MODELS_GROUPS.cerebras,
|
|
// ...SUPPORTED_MODELS_GROUPS.cloudflareAI,
|
|
];
|
|
|
|
export { SUPPORTED_MODELS, SUPPORTED_MODELS_GROUPS, getModelFamily };
|