This commit is contained in:
ryanhex53
2024-12-24 08:41:17 +08:00
committed by GitHub
13 changed files with 424 additions and 23 deletions

View File

@@ -1,4 +1,4 @@
import { getModelProvider } from "../app/utils/model";
import { getModelProvider, isModelAvailableInServer } from "../app/utils/model";
describe("getModelProvider", () => {
test("should return model and provider when input contains '@'", () => {
@@ -29,3 +29,15 @@ describe("getModelProvider", () => {
expect(provider).toBeUndefined();
});
});
describe("isModelAvailableInServer", () => {
test("works when model null", () => {
const jsonBody = JSON.parse("{}") as { model?: string };
const result = isModelAvailableInServer(
"gpt-3.5-turbo@OpenAI",
jsonBody.model as string,
"OpenAI",
);
expect(result).toBe(false);
});
});