mirror of
https://github.com/badlogic/pi-mono.git
synced 2026-11-05 23:37:25 +00:00
2e0f175bd5
This builds the model catalog from per-provider definitions and models.dev v2 data. It eplacer `generate-models.ts` with one `defineProvider()` file per built-in provider, shared model-family rules (Claude, GPT, DeepSeek, Kimi, Google), and source adapters. models.dev is read through its v2 API. Outputs and CLI flags are unchanged.
68 lines
2.4 KiB
TypeScript
68 lines
2.4 KiB
TypeScript
import type { Model } from "../../src/types.ts";
|
|
import { defineProvider, mergeCompat, mergeThinking } from "../define.ts";
|
|
import {
|
|
enableOpenAiGrammarTools,
|
|
enableOpenAiMidConvoSystemMessages,
|
|
OPENAI_TOOL_SEARCH_MODEL_IDS,
|
|
openAiStandardCost,
|
|
supportsOpenAiXhigh,
|
|
} from "../families/openai.ts";
|
|
|
|
// ChatGPT OAuth models. models.dev does not list them; keep a small, explicit
|
|
// list without aliases. Only list models the Codex backend accepts for ChatGPT
|
|
// accounts: GPT-5.5 and GPT-5.3 Codex Spark are rejected with "not supported
|
|
// when using Codex with a ChatGPT account".
|
|
const MODELS = [
|
|
// GPT-6.1 uses the 500k default returned by /backend-api/codex/models.
|
|
{ id: "gpt-6.1-sol", name: "GPT-6.1 Sol", contextWindow: 500000 },
|
|
{ id: "gpt-6.1-terra", name: "GPT-6.1 Terra", contextWindow: 500000 },
|
|
// GPT-5.6 and GPT-6 use Codex's 272k default catalog limit.
|
|
{ id: "gpt-6-astra", name: "GPT-6 Astra", contextWindow: 272000 },
|
|
{ id: "gpt-6-sol", name: "GPT-6 Sol", contextWindow: 272000 },
|
|
{ id: "gpt-6-luna", name: "GPT-6 Luna", contextWindow: 272000 },
|
|
{ id: "gpt-5.6-luna", name: "GPT-5.6 Luna", contextWindow: 272000 },
|
|
{ id: "gpt-5.6-sol", name: "GPT-5.6 Sol", contextWindow: 272000 },
|
|
{ id: "gpt-5.6-terra", name: "GPT-5.6 Terra", contextWindow: 272000 },
|
|
];
|
|
|
|
// Codex CLI keeps additional_tools behind its own model list.
|
|
const ADDITIONAL_TOOLS_MODEL_IDS = new Set([
|
|
"gpt-5.6-sol",
|
|
"gpt-5.6-terra",
|
|
"gpt-5.6-luna",
|
|
"gpt-6-astra",
|
|
"gpt-6-sol",
|
|
"gpt-6-luna",
|
|
"gpt-6.1-sol",
|
|
"gpt-6.1-terra",
|
|
]);
|
|
|
|
export default defineProvider({
|
|
id: "openai-codex",
|
|
models: () =>
|
|
MODELS.map(
|
|
({ id, name, contextWindow }): Model<"openai-codex-responses"> => ({
|
|
id,
|
|
name,
|
|
api: "openai-codex-responses",
|
|
provider: "openai-codex",
|
|
baseUrl: "https://chatgpt.com/backend-api",
|
|
reasoning: true,
|
|
input: ["text", "image"],
|
|
cost: openAiStandardCost(id),
|
|
contextWindow,
|
|
maxTokens: 128000,
|
|
}),
|
|
),
|
|
finalize: (model) => {
|
|
if (supportsOpenAiXhigh(model.id)) mergeThinking(model, { minimal: "low" });
|
|
enableOpenAiGrammarTools(model);
|
|
if (OPENAI_TOOL_SEARCH_MODEL_IDS.has(model.id)) {
|
|
mergeCompat(model as Model<"openai-codex-responses">, {
|
|
...(ADDITIONAL_TOOLS_MODEL_IDS.has(model.id) ? { supportsAdditionalTools: true } : {}),
|
|
supportsToolSearch: true,
|
|
});
|
|
}
|
|
enableOpenAiMidConvoSystemMessages(model, { proxy: false });
|
|
},
|
|
});
|