cc-switch/tests/config/piProviderPresets.test.ts

import { describe, expect, it } from "vitest";
import {
  getPiModelCatalogReference,
  piModelCatalog,
} from "@/config/piModelCatalog";
import { piProviderPresets } from "@/config/piProviderPresets";

describe("Pi provider presets", () => {
  it("owns a broad provider catalog without OpenCode-only templates", () => {
    const names = piProviderPresets.map((preset) => preset.name);

    expect(piProviderPresets.length).toBeGreaterThanOrEqual(50);
    expect(names).toEqual(
      expect.arrayContaining(["Kimi", "DeepSeek", "OpenRouter", "AWS Bedrock"]),
    );
    expect(names).not.toContain("Oh My OpenCode");
    expect(names).not.toContain("Oh My OpenCode Slim");
  });

  it("uses distinct managed keys instead of shadowing Pi-native providers", () => {
    const keys = piProviderPresets.map((preset) => preset.providerKey);

    expect(new Set(keys).size).toBe(keys.length);
    expect(keys.every((key) => key.startsWith("cc-switch-"))).toBe(true);
  });

  it("only supplies configuration defaults, never a second gateway decision", () => {
    for (const preset of piProviderPresets) {
      expect("allowGateway" in preset).toBe(false);
      expect(preset.settingsConfig.baseUrl).toMatch(/^https?:\/\//);
      expect(preset.settingsConfig.api).not.toBe("");
      expect(preset.settingsConfig.apiKey).toBe("");
      expect(preset.settingsConfig.models.length).toBeGreaterThan(0);
      const modelIds = preset.settingsConfig.models.map((model) => model.id);
      expect(new Set(modelIds).size).toBe(modelIds.length);
      for (const model of preset.settingsConfig.models) {
        expect(model.id).not.toBe("");
        expect(model.name).not.toBe("");
        expect(typeof model.reasoning).toBe("boolean");
        expect(model.input.length).toBeGreaterThan(0);
        expect(model.input.every((input) => ["text", "image"].includes(input)));
        expect(model.contextWindow).toBeGreaterThan(0);
        expect(model.maxTokens).toBeGreaterThan(0);
        const reference = getPiModelCatalogReference(model);
        if (!reference) {
          throw new Error(`${preset.name}/${model.id} has no catalog profile`);
        }
        expect(piModelCatalog).toHaveProperty(reference.catalogKey);
        expect(JSON.parse(JSON.stringify(model))).not.toHaveProperty(
          "catalogKey",
        );
      }
    }
  });

  it("uses Anthropic roots that produce Pi request paths", () => {
    const requestUrls = Object.fromEntries(
      piProviderPresets
        .filter((preset) => preset.settingsConfig.api === "anthropic-messages")
        .map((preset) => {
          const base = new URL(preset.settingsConfig.baseUrl);
          const basePath = base.pathname.replace(/\/+$/, "");
          base.pathname = `${basePath}/v1/messages`;
          return [preset.name, base.toString()];
        }),
    );

    expect(requestUrls).toMatchObject({
      "Kimi For Coding": "https://api.kimi.com/coding/v1/messages",
      PackyCode: "https://www.packyapi.ai/v1/messages",
      AICodeMirror: "https://api.aicodemirror.ai/api/claudecode/v1/messages",
      OpenRouter: "https://openrouter.ai/api/v1/messages",
    });
  });

  it("stores Pi-native API formats in its own catalog", () => {
    expect(
      Object.fromEntries(
        piProviderPresets.map((preset) => [
          preset.name,
          preset.settingsConfig.api,
        ]),
      ),
    ).toMatchObject({
      Kimi: "openai-completions",
      "Kimi For Coding": "anthropic-messages",
      RightCode: "openai-responses",
      "AWS Bedrock": "bedrock-converse-stream",
    });
  });

  it("uses Pi's 272K context value for every GPT-5.6 Sol preset", () => {
    const models = piProviderPresets.flatMap((preset) =>
      preset.settingsConfig.models.filter(
        (model) => model.id === "gpt-5.6-sol",
      ),
    );

    expect(models.length).toBeGreaterThan(0);
    expect(models.every((model) => model.contextWindow === 272_000)).toBe(true);
  });

  // #7859: 官方 V4.1 Flash 识图，V4 Pro 仍是纯文本
  it("declares image input for official DeepSeek V4.1 Flash only", () => {
    const models =
      piProviderPresets.find((preset) => preset.name === "DeepSeek")
        ?.settingsConfig.models ?? [];
    const input = (id: string) =>
      models.find((model) => model.id === id)?.input;

    expect(input("deepseek-flash")).toEqual(["text", "image"]);
    expect(input("deepseek-v4-pro")).toEqual(["text"]);
  });

  // #7956: 火山各套餐端点不同，Agent Plan 曾误用 Coding Plan 的端点
  it("points each Volcengine plan at its own Ark endpoint", () => {
    const endpoints = Object.fromEntries(
      piProviderPresets
        .filter((preset) => preset.family === "volcengine")
        .map((preset) => [preset.planKey, preset.settingsConfig.baseUrl]),
    );

    expect(endpoints).toEqual({
      agentPlan: "https://ark.cn-beijing.volces.com/api/plan/v3",
      codingPlan: "https://ark.cn-beijing.volces.com/api/coding/v3",
      payg: "https://ark.cn-beijing.volces.com/api/v3",
    });
  });

  it("keeps provider-specific OpenAI compatibility metadata", () => {
    const preset = (name: string) => {
      const found = piProviderPresets.find((item) => item.name === name);
      if (!found) throw new Error(`Missing Pi preset: ${name}`);
      return found;
    };
    const model = (presetName: string, id: string) => {
      const found = preset(presetName).settingsConfig.models.find(
        (item) => item.id === id,
      );
      if (!found) throw new Error(`Missing Pi model: ${presetName}/${id}`);
      return found;
    };

    expect(model("Kimi", "kimi-k3").compat).toEqual({
      supportsStore: false,
      supportsDeveloperRole: false,
      supportsReasoningEffort: true,
      maxTokensField: "max_tokens",
      supportsStrictMode: false,
      thinkingFormat: "openai",
      requiresReasoningContentOnAssistantMessages: true,
      deferredToolsMode: "kimi",
    });

    const openCodeBase = {
      supportsStore: false,
      supportsDeveloperRole: false,
      maxTokensField: "max_tokens",
    };
    expect(model("OpenCode Go", "glm-5.3").compat).toEqual(openCodeBase);
    expect(model("OpenCode Go", "kimi-k3.1-pro").compat).toEqual(openCodeBase);
    expect(model("OpenCode Go", "mimo-v2.6-pro").compat).toEqual(openCodeBase);
    for (const id of ["deepseek-v4-pro", "deepseek-v4.1-flash"]) {
      expect(model("OpenCode Go", id).compat).toEqual({
        ...openCodeBase,
        requiresReasoningContentOnAssistantMessages: true,
        thinkingFormat: "deepseek",
      });
    }

    for (const id of ["mimo-v2.6-pro", "mimo-v2.6-flash"]) {
      expect(model("Xiaomi MiMo", id).compat).toEqual({
        requiresReasoningContentOnAssistantMessages: true,
        thinkingFormat: "deepseek",
      });
    }

    const qwenCompat = {
      thinkingFormat: "qwen",
      supportsDeveloperRole: false,
    };
    for (const name of [
      "千问AI平台",
      "千问AI平台 Token Plan",
      "QwenCloud",
      "QwenCloud Token Plan",
    ]) {
      const models = preset(name).settingsConfig.models;
      expect(models.every((item) => item.reasoning)).toBe(true);
      for (const item of models) {
        expect(item.compat).toEqual(qwenCompat);
      }
    }

    for (const item of preset("QwenCloud For Coding").settingsConfig.models) {
      expect(item.compat).toBeUndefined();
    }
  });
});