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.
52 lines
1.7 KiB
TypeScript
52 lines
1.7 KiB
TypeScript
import { writeFileSync } from "node:fs";
|
|
import type { ModelsDevV2Catalog } from "../catalog/sources/models-dev.ts";
|
|
|
|
/**
|
|
* Write a `node --import` preload that serves every upstream catalog the
|
|
* generator fetches: the given models.dev v2 catalog, and minimal responses for
|
|
* the other sources. Any other request fails.
|
|
*/
|
|
export function writeUpstreamMock(preloadPath: string, modelsDev: ModelsDevV2Catalog): void {
|
|
const responses = {
|
|
"https://models.dev/experimental/v2.0/api.json": modelsDev,
|
|
"https://models.dev/models.json?type=decision": {
|
|
"typesafe/jev-latest": { id: "typesafe/jev-latest", name: "Jev", type: "decision", limit: { context: 64000 } },
|
|
},
|
|
"https://openrouter.ai/api/v1/models": { data: [] },
|
|
"https://openrouter.ai/api/v1/models?output_modalities=image": {
|
|
data: [
|
|
{
|
|
id: "test/image",
|
|
name: "Test Image",
|
|
architecture: { input_modalities: ["text"], output_modalities: ["image"] },
|
|
pricing: {},
|
|
},
|
|
],
|
|
},
|
|
"https://openrouter.ai/api/v1/models?output_modalities=decisions": { data: [] },
|
|
"https://ai-gateway.vercel.sh/v1/models": { data: [] },
|
|
"https://radius.pi.dev/v1/config": {
|
|
baseUrl: "https://radius.pi.dev",
|
|
models: [
|
|
{
|
|
id: "test",
|
|
name: "Test",
|
|
reasoning: false,
|
|
input: ["text"],
|
|
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
|
|
contextWindow: 4096,
|
|
maxTokens: 4096,
|
|
},
|
|
],
|
|
},
|
|
};
|
|
writeFileSync(
|
|
preloadPath,
|
|
`const responses = ${JSON.stringify(responses)};\n` +
|
|
`globalThis.fetch = async (input) => {\n` +
|
|
` const url = String(input);\n` +
|
|
` if (url in responses) return Response.json(responses[url]);\n` +
|
|
` throw new Error(\`Unexpected fetch: \${url}\`);\n` +
|
|
`};\n`,
|
|
);
|
|
}
|