diff --git a/src/shared/model-capabilities/bundled-snapshot.ts b/src/shared/model-capabilities/bundled-snapshot.ts index 65644a8cf..18ffec737 100644 --- a/src/shared/model-capabilities/bundled-snapshot.ts +++ b/src/shared/model-capabilities/bundled-snapshot.ts @@ -1,5 +1,6 @@ import bundledModelCapabilitiesSnapshotJson from "../../generated/model-capabilities.generated.json" +import { SUPPLEMENTAL_MODEL_CAPABILITIES } from "./supplemental-entries" import type { ModelCapabilitiesSnapshot } from "./types" function normalizeSnapshot( @@ -8,7 +9,15 @@ function normalizeSnapshot( return snapshot as ModelCapabilitiesSnapshot } -const bundledModelCapabilitiesSnapshot = normalizeSnapshot(bundledModelCapabilitiesSnapshotJson) +const normalizedBundledSnapshot = normalizeSnapshot(bundledModelCapabilitiesSnapshotJson) + +const bundledModelCapabilitiesSnapshot: ModelCapabilitiesSnapshot = { + ...normalizedBundledSnapshot, + models: { + ...normalizedBundledSnapshot.models, + ...SUPPLEMENTAL_MODEL_CAPABILITIES, + }, +} export function getBundledModelCapabilitiesSnapshot(): ModelCapabilitiesSnapshot { return bundledModelCapabilitiesSnapshot diff --git a/src/shared/model-capabilities/supplemental-entries.ts b/src/shared/model-capabilities/supplemental-entries.ts new file mode 100644 index 000000000..197d88001 --- /dev/null +++ b/src/shared/model-capabilities/supplemental-entries.ts @@ -0,0 +1,20 @@ +import type { ModelCapabilitiesSnapshotEntry } from "./types" + +export const SUPPLEMENTAL_MODEL_CAPABILITIES: Record = { + "gpt-5.4-mini-fast": { + id: "gpt-5.4-mini-fast", + family: "gpt-mini", + reasoning: true, + temperature: false, + toolCall: true, + modalities: { + input: ["text", "image", "pdf"], + output: ["text"], + }, + limit: { + context: 400000, + input: 272000, + output: 128000, + }, + }, +}