mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-05 01:07:56 +08:00
dayuan.jiang
6236124338
The "Fetch models" button asks the provider for its models (OpenAI-style /models, Anthropic, Google, Ollama, OpenRouter, Vercel Gateway, AIHubMix) and shows them in a searchable picker. This replaces the route that only worked for AIHubMix. A snapshot of models.dev (MIT) says which models support tool calls. Models without them get a "no tool calls" badge in the picker and a hint in the model list, since drawing needs tool calls. Refresh the snapshot with scripts/update-model-catalog.mjs.
46 lines
1.3 KiB
TypeScript
46 lines
1.3 KiB
TypeScript
import catalog from "@/lib/model-catalog.json"
|
|
import type { ProviderName } from "@/lib/types/model-config"
|
|
|
|
/**
|
|
* What models.dev knows about a model (scripts/update-model-catalog.mjs).
|
|
* Only used for hints: requests are sent the same way either way, since
|
|
* the data can be wrong or out of date.
|
|
*/
|
|
export interface ModelInfo {
|
|
tools: boolean
|
|
images: boolean
|
|
reasoning: boolean
|
|
context?: number
|
|
output?: number
|
|
}
|
|
|
|
const CATALOG = catalog as Record<string, Record<string, ModelInfo>>
|
|
|
|
/**
|
|
* The entry for a model: an exact match ignoring case, else the longest id
|
|
* the model id starts with, followed by "-", ":" or ".". So
|
|
* claude-sonnet-4-5-20250929 finds claude-sonnet-4-5, but gpt-4 does not
|
|
* find gpt-4o.
|
|
*/
|
|
export function getModelInfo(
|
|
provider: ProviderName,
|
|
modelId: string,
|
|
): ModelInfo | undefined {
|
|
const models = CATALOG[provider]
|
|
if (!models) return undefined
|
|
const wanted = modelId.trim().toLowerCase()
|
|
let best: string | undefined
|
|
for (const id of Object.keys(models)) {
|
|
const lower = id.toLowerCase()
|
|
if (lower === wanted) return models[id]
|
|
if (
|
|
wanted.startsWith(lower) &&
|
|
"-:.".includes(wanted[lower.length]) &&
|
|
lower.length > (best?.length ?? 0)
|
|
) {
|
|
best = id
|
|
}
|
|
}
|
|
return best ? models[best] : undefined
|
|
}
|