chatbot-template/lib/ai/models.ts

79 lines
1.9 KiB
TypeScript

// Curated list of top models from Vercel AI Gateway
export const DEFAULT_CHAT_MODEL = "openai/gpt-4.1-mini";
export type ChatModel = {
id: string;
name: string;
provider: string;
description: string;
};
export const chatModels: ChatModel[] = [
// Anthropic
{
id: "anthropic/claude-haiku-4.5",
name: "Claude Haiku 4.5",
provider: "anthropic",
description: "Fast and affordable, great for everyday tasks",
},
// OpenAI
{
id: "openai/gpt-4.1-mini",
name: "GPT-4.1 Mini",
provider: "openai",
description: "Fast and cost-effective for simple tasks",
},
{
id: "openai/gpt-5-mini",
name: "GPT-5 Mini",
provider: "openai",
description: "Most capable OpenAI model",
},
// Google
{
id: "google/gemini-2.5-flash-lite",
name: "Gemini 2.5 Flash Lite",
provider: "google",
description: "Ultra fast and affordable",
},
{
id: "google/gemini-3-pro-preview",
name: "Gemini 3 Pro",
provider: "google",
description: "Most capable Google model",
},
// xAI
{
id: "xai/grok-4.1-fast-non-reasoning",
name: "Grok 4.1 Fast",
provider: "xai",
description: "Fast with 30K context",
},
// Reasoning models (extended thinking)
{
id: "anthropic/claude-3.7-sonnet-thinking",
name: "Claude 3.7 Sonnet",
provider: "reasoning",
description: "Extended thinking for complex problems",
},
{
id: "xai/grok-code-fast-1-thinking",
name: "Grok Code Fast",
provider: "reasoning",
description: "Reasoning optimized for code",
},
];
// Group models by provider for UI
export const allowedModelIds = new Set(chatModels.map((m) => m.id));
export const modelsByProvider = chatModels.reduce(
(acc, model) => {
if (!acc[model.provider]) {
acc[model.provider] = [];
}
acc[model.provider].push(model);
return acc;
},
{} as Record<string, ChatModel[]>
);