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.
50 lines
2 KiB
TypeScript
50 lines
2 KiB
TypeScript
import type { Model } from "../../src/types.ts";
|
|
import { defineProvider, modelsDevModels, type ProviderContext } from "../define.ts";
|
|
import { getEffortThinkingLevelMap } from "../reasoning.ts";
|
|
import { modelsDevChatModels, modelsDevCost } from "../sources/models-dev.ts";
|
|
|
|
// Tool-call streaming (`tool_stream`) is rejected by the GLM-4.5 generation.
|
|
const TOOL_STREAM_UNSUPPORTED_MODEL_IDS = new Set(["glm-4.5", "glm-4.5-air", "glm-4.5-flash", "glm-4.5v"]);
|
|
|
|
/** The Z.AI coding plans. The international and China plans list the same models. */
|
|
export function zaiCodingPlanModels(ctx: ProviderContext, source: string, baseUrl: string) {
|
|
// Coding-plan entries carry plan prices; show the pay-as-you-go API rates.
|
|
const apiModels = new Map(modelsDevChatModels(ctx.sources.modelsDev, "zai").map((model) => [model.id, model]));
|
|
return modelsDevModels(ctx, source, {
|
|
api: "openai-completions",
|
|
baseUrl,
|
|
map: (model, source): Model<"openai-completions"> => {
|
|
const thinkingLevelMap = getEffortThinkingLevelMap(source.reasoningOptions ?? []);
|
|
if (thinkingLevelMap && (model.id === "glm-5.2" || model.id === "glm-5.2-highspeed")) {
|
|
thinkingLevelMap.off = "none";
|
|
}
|
|
const apiModel = apiModels.get(model.id);
|
|
return {
|
|
...model,
|
|
...(thinkingLevelMap ? { thinkingLevelMap } : {}),
|
|
cost: modelsDevCost(apiModel?.source.cost ? apiModel.cost : source.cost),
|
|
compat: {
|
|
supportsDeveloperRole: false,
|
|
thinkingFormat: "zai",
|
|
...(thinkingLevelMap ? { supportsReasoningEffort: true } : {}),
|
|
...(!TOOL_STREAM_UNSUPPORTED_MODEL_IDS.has(model.id) ? { zaiToolStream: true } : {}),
|
|
},
|
|
};
|
|
},
|
|
});
|
|
}
|
|
|
|
export const ZAI_COMPLETIONS_COMPAT = {
|
|
supportsStore: false,
|
|
supportsDeveloperRole: false,
|
|
supportsReasoningEffort: false,
|
|
maxTokensField: "max_tokens",
|
|
thinkingFormat: "zai",
|
|
supportsStrictMode: true,
|
|
} as const;
|
|
|
|
export default defineProvider({
|
|
id: "zai",
|
|
models: (ctx) => zaiCodingPlanModels(ctx, "zai-coding-plan", "https://api.z.ai/api/coding/paas/v4"),
|
|
compat: { "openai-completions": ZAI_COMPLETIONS_COMPAT },
|
|
});
|