mirror of
https://github.com/badlogic/pi-mono.git
synced 2026-11-05 23:37:25 +00:00
8d95fd138b
Replace the single 3300-line generate-models.ts with declarative provider definitions and ordered rule packs under scripts/model-generation/. Each provider is a defineProvider() call: an upstream source (usually the generic modelsDev() mapper), static models upstream lacks, and rules scoped to that provider. Cross-provider facts live in rule packs applied in a fixed order. Every rule() carries a why, matches models with plain data, and patches them with typed set/replace/apply semantics. Generated model data is unchanged: 1443 models across 41 providers compare equal against a snapshot of the same upstream responses, apart from a now stable key order. Committed models.generated.ts and provider shards are byte-identical. Also: - Bring packages/ai/scripts under tsgo and biome; the old generator had 20+ type errors that nothing checked. - Add --upstream-snapshot <dir> to record and replay upstream catalogs and --explain <provider>:<model> to list the rules applied to a model. - Report rules that matched no model so stale corrections get removed. - Export parseRadiusGatewayConfig so the generator can replay Radius too. Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_018ExCU9zaZnN8vB5fKpTAjP
42 lines
1.3 KiB
TypeScript
42 lines
1.3 KiB
TypeScript
import { defineProvider, rule } from "../dsl.ts";
|
|
|
|
// Azure Foundry deploys these with larger context windows than OpenAI's own short-tier defaults.
|
|
// See models-sold-directly-by-azure docs.
|
|
const AZURE_CONTEXT_WINDOW_OVERRIDES: Record<string, number> = {
|
|
"gpt-5.4": 1050000,
|
|
"gpt-5.5": 1050000,
|
|
"gpt-5.6-luna": 1050000,
|
|
"gpt-5.6-sol": 1050000,
|
|
"gpt-5.6-terra": 1050000,
|
|
};
|
|
|
|
export const azureOpenaiResponses = defineProvider({
|
|
id: "azure-openai-responses",
|
|
// Mirrors the OpenAI Responses catalog after OpenAI's own corrections. The base URL is
|
|
// per deployment and long-context price tiers do not apply to Azure billing.
|
|
source: async (context) =>
|
|
context
|
|
.generated("openai")
|
|
.filter((model) => model.api === "openai-responses")
|
|
.map((model) => ({
|
|
...model,
|
|
api: "azure-openai-responses",
|
|
provider: "azure-openai-responses",
|
|
baseUrl: "",
|
|
cost: {
|
|
input: model.cost.input,
|
|
output: model.cost.output,
|
|
cacheRead: model.cost.cacheRead,
|
|
cacheWrite: model.cost.cacheWrite,
|
|
},
|
|
})),
|
|
rules: [
|
|
rule({
|
|
why: "Azure Foundry deploys these models with larger context windows than OpenAI's short-tier defaults.",
|
|
match: { id: Object.keys(AZURE_CONTEXT_WINDOW_OVERRIDES) },
|
|
apply: (model) => {
|
|
model.contextWindow = AZURE_CONTEXT_WINDOW_OVERRIDES[model.id];
|
|
},
|
|
}),
|
|
],
|
|
});
|