From cc5e16b0454481fab48b1115eda9b8fb11ce0054 Mon Sep 17 00:00:00 2001 From: code-october <148516338+code-october@users.noreply.github.com> Date: Sat, 30 Nov 2024 07:30:52 +0000 Subject: [PATCH] update unit test --- test/model-available.test.ts | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/test/model-available.test.ts b/test/model-available.test.ts index 2d222e052..2ceda56f0 100644 --- a/test/model-available.test.ts +++ b/test/model-available.test.ts @@ -16,6 +16,7 @@ describe("isModelNotavailableInServer", () => { const result = isModelNotavailableInServer(customModels, modelName, providerNames); expect(result).toBe(true); }); + test("should respect DISABLE_GPT4 setting", () => { process.env.DISABLE_GPT4 = "1"; const result = isModelNotavailableInServer("", "gpt-4", "OpenAI"); @@ -27,6 +28,11 @@ describe("isModelNotavailableInServer", () => { expect(result).toBe(true); }); + test("should be case insensitive for model names", () => { + const result = isModelNotavailableInServer("-all,GPT-4", "gpt-4", "OpenAI"); + expect(result).toBe(true); + }); + test("support passing multiple providers, model unavailable on one of the providers will return true", () => { const customModels = "-all,gpt-4@Google"; const modelName = "gpt-4";