import { describe, expect, it } from "vitest";

function buildMinimaxCatalog() {
  return [
    {
      id: "MiniMax-M2.7",
      cost: {
        input: 1.1,
        output: 4.4,
        cacheRead: 0.11,
        cacheWrite: 0.6875,
      },
    },
    {
      id: "MiniMax-M2.7-highspeed",
      cost: {
        input: 0.6,
        output: 2.4,
        cacheRead: 0.06,
        cacheWrite: 0.375,
      },
    },
  ];
}

describe("minimax provider catalog", () => {
  it("does not advertise the removed lightning model for api-key or oauth providers", () => {
    const providers = {
      minimax: { models: buildMinimaxCatalog() },
      "minimax-portal": { models: buildMinimaxCatalog() },
    };
    expect(providers?.minimax?.models?.map((model) => model.id)).toEqual([
      "MiniMax-M2.7",
      "MiniMax-M2.7-highspeed",
    ]);
    expect(providers?.["minimax-portal"]?.models?.map((model) => model.id)).toEqual([
      "MiniMax-M2.7",
      "MiniMax-M2.7-highspeed",
    ]);
  });

  it("keeps MiniMax highspeed pricing distinct in implicit catalogs", () => {
    const providers = {
      minimax: { models: buildMinimaxCatalog() },
      "minimax-portal": { models: buildMinimaxCatalog() },
    };
    const apiHighspeed = providers?.minimax?.models?.find(
      (model) => model.id === "MiniMax-M2.7-highspeed",
    );
    const portalHighspeed = providers?.["minimax-portal"]?.models?.find(
      (model) => model.id === "MiniMax-M2.7-highspeed",
    );

    expect(apiHighspeed?.cost).toEqual({
      input: 0.6,
      output: 2.4,
      cacheRead: 0.06,
      cacheWrite: 0.375,
    });
    expect(portalHighspeed?.cost).toEqual(apiHighspeed?.cost);
  });
});