import test from "node:test";
import assert from "node:assert/strict";
import { getMasterAgentModelOptions } from "../src/lib/master-agent-model-options";

test("主 Agent 模型选项会明确暴露快反和深思模型", () => {
  assert.deepEqual(
    getMasterAgentModelOptions(),
    ["gpt-5.4-mini", "gpt-5.4", "gpt-5.1", "gpt-4.1"],
  );
});

test("主 Agent 模型选项会保留当前自定义模型", () => {
  assert.deepEqual(
    getMasterAgentModelOptions("gpt-4.1-mini"),
    ["gpt-4.1-mini", "gpt-5.4-mini", "gpt-5.4", "gpt-5.1", "gpt-4.1"],
  );
});