test/model-available-in-server.test.ts

Rama main · réplica de lectura del repositorio original.

import { isModelAvailableInServer } from "../app/utils/model";

describe("isModelAvailableInServer", () => {
  // NOTE: the helper returns true when a model has been *explicitly disabled*
  // in the server's custom-model string (available === false), and false
  // otherwise (including when the model is enabled or unknown).

  test("returns false for a model that is available by default", () => {
    expect(isModelAvailableInServer("", "gpt-4", "openai")).toBe(false);
  });

  test("returns true when '-all' disables every model", () => {
    expect(isModelAvailableInServer("-all", "gpt-4", "openai")).toBe(true);
  });

  test("returns true when the specific model is disabled", () => {
    expect(isModelAvailableInServer("-gpt-4@openai", "gpt-4", "openai")).toBe(
      true,
    );
  });

  test("returns false for an unknown model that is not in the table", () => {
    expect(
      isModelAvailableInServer("", "totally-unknown-model", "openai"),
    ).toBe(false);
  });

  test("a model re-enabled after '-all' is not reported as disabled", () => {
    expect(
      isModelAvailableInServer("-all,+gpt-4@openai", "gpt-4", "openai"),
    ).toBe(false);
  });
});