47 lines
1.3 KiB
TypeScript
47 lines
1.3 KiB
TypeScript
import { vi } from "vitest";
|
|
import { discoverModels } from "../pi-model-discovery.js";
|
|
|
|
export const makeModel = (id: string) => ({
|
|
id,
|
|
name: id,
|
|
reasoning: false,
|
|
input: ["text"] as const,
|
|
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
|
|
contextWindow: 1,
|
|
maxTokens: 1,
|
|
});
|
|
|
|
export const OPENAI_CODEX_TEMPLATE_MODEL = {
|
|
id: "gpt-5.2-codex",
|
|
name: "GPT-5.2 Codex",
|
|
provider: "openai-codex",
|
|
api: "openai-codex-responses",
|
|
baseUrl: "https://chatgpt.com/backend-api",
|
|
reasoning: true,
|
|
input: ["text", "image"] as const,
|
|
cost: { input: 1.75, output: 14, cacheRead: 0.175, cacheWrite: 0 },
|
|
contextWindow: 272000,
|
|
maxTokens: 128000,
|
|
};
|
|
|
|
export function resetMockDiscoverModels(): void {
|
|
vi.mocked(discoverModels).mockReturnValue({
|
|
find: vi.fn(() => null),
|
|
} as unknown as ReturnType<typeof discoverModels>);
|
|
}
|
|
|
|
export function mockDiscoveredModel(params: {
|
|
provider: string;
|
|
modelId: string;
|
|
templateModel: unknown;
|
|
}): void {
|
|
vi.mocked(discoverModels).mockReturnValue({
|
|
find: vi.fn((provider: string, modelId: string) => {
|
|
if (provider === params.provider && modelId === params.modelId) {
|
|
return params.templateModel;
|
|
}
|
|
return null;
|
|
}),
|
|
} as unknown as ReturnType<typeof discoverModels>);
|
|
}
|