File size: 1,740 Bytes
246d201 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
import { expect, test } from "vitest";
import { organizeModelsAndProviders } from "../../src/utils/organize-models-and-providers";
test("organizeModelsAndProviders", () => {
const models = [
"azure/ada",
"azure/gpt-35-turbo",
"azure/gpt-3-turbo",
"azure/standard/1024-x-1024/dall-e-2",
"vertex_ai_beta/chat-bison",
"vertex_ai_beta/chat-bison-32k",
"sagemaker/meta-textgeneration-llama-2-13b",
"cohere.command-r-v1:0",
"cloudflare/@cf/mistral/mistral-7b-instruct-v0.1",
"gpt-4o",
"together-ai-21.1b-41b",
"gpt-4o-mini",
"anthropic/claude-3-5-sonnet-20241022",
"claude-3-haiku-20240307",
"claude-2",
"claude-2.1",
"anthropic.unsafe-claude-2.1",
];
const object = organizeModelsAndProviders(models);
expect(object).toEqual({
azure: {
separator: "/",
models: [
"ada",
"gpt-35-turbo",
"gpt-3-turbo",
"standard/1024-x-1024/dall-e-2",
],
},
vertex_ai_beta: {
separator: "/",
models: ["chat-bison", "chat-bison-32k"],
},
sagemaker: { separator: "/", models: ["meta-textgeneration-llama-2-13b"] },
cohere: { separator: ".", models: ["command-r-v1:0"] },
cloudflare: {
separator: "/",
models: ["@cf/mistral/mistral-7b-instruct-v0.1"],
},
openai: {
separator: "/",
models: ["gpt-4o", "gpt-4o-mini"],
},
anthropic: {
separator: "/",
models: [
"claude-3-5-sonnet-20241022",
"claude-3-haiku-20240307",
"claude-2",
"claude-2.1",
],
},
other: {
separator: "",
models: ["together-ai-21.1b-41b"],
},
});
});
|