mirror of
https://github.com/coaidev/coai.git
synced 2025-05-30 02:10:25 +09:00
138 lines
3.5 KiB
TypeScript
138 lines
3.5 KiB
TypeScript
export const modelColorMapper: Record<string, string> = {
|
|
"gpt-3.5-turbo": "green-500",
|
|
"gpt-3.5-turbo-instruct": "green-500",
|
|
"gpt-3.5-turbo-0613": "green-500",
|
|
"gpt-3.5-turbo-0301": "green-500",
|
|
"gpt-3.5-turbo-1106": "green-500",
|
|
"gpt-3.5-turbo-0125": "green-500",
|
|
dalle: "green-600",
|
|
"dall-e-2": "green-600",
|
|
|
|
midjourney: "indigo-600",
|
|
"midjourney-fast": "indigo-600",
|
|
"midjourney-turbo": "indigo-600",
|
|
"stable-diffusion": "gray-400",
|
|
|
|
"gpt-3.5-turbo-16k": "green-500",
|
|
"gpt-3.5-turbo-16k-0613": "green-500",
|
|
"gpt-3.5-turbo-16k-0301": "green-500",
|
|
|
|
"gpt-4": "purple-600",
|
|
"gpt-4-1106-preview": "purple-600",
|
|
"gpt-4-0125-preview": "purple-600",
|
|
"gpt-4-turbo-preview": "purple-600",
|
|
"gpt-4-1106-vision-preview": "purple-600",
|
|
"gpt-4-vision-preview": "purple-600",
|
|
"gpt-4-turbo": "purple-600",
|
|
"gpt-4-turbo-2024-04-09": "purple-600",
|
|
"gpt-4-0613": "purple-600",
|
|
"gpt-4-0314": "purple-600",
|
|
"gpt-4-all": "purple-600",
|
|
"gpt-4-v": "purple-600",
|
|
"gpt-4-dalle": "purple-600",
|
|
"gpt-4-32k": "purple-600",
|
|
"gpt-4-32k-0613": "purple-600",
|
|
"gpt-4-32k-0314": "purple-600",
|
|
"gpt-4o": "purple-600",
|
|
"gpt-4o-2024-05-13": "purple-600",
|
|
"gpt-4o-mini": "purple-600",
|
|
"gpt-4o-mini-2024-07-18": "purple-600",
|
|
|
|
"dall-e-3": "purple-700",
|
|
|
|
"claude-1": "orange-400",
|
|
"claude-1-100k": "orange-400",
|
|
"claude-slack": "orange-400",
|
|
"claude-2": "orange-400",
|
|
"claude-2.1": "orange-400",
|
|
"claude-2-100k": "orange-400",
|
|
"claude-instant-1.2": "orange-400",
|
|
"claude-3-opus-20240229": "orange-400",
|
|
"claude-3-sonnet-20240229": "orange-400",
|
|
"claude-3-haiku-20240307": "orange-400",
|
|
|
|
"spark-desk-v1.5": "blue-400",
|
|
"spark-desk-v2": "blue-400",
|
|
"spark-desk-v3": "blue-400",
|
|
"spark-desk-v3.5": "blue-400",
|
|
|
|
"moonshot-v1-8k": "black-500",
|
|
"moonshot-v1-32k": "black-500",
|
|
"moonshot-v1-128k": "black-500",
|
|
|
|
"llama2-70b-4096": "red-500",
|
|
"mixtral-8x7b-32768": "red-500",
|
|
"gemma-7b-it": "red-500",
|
|
|
|
"chat-bison-001": "red-500",
|
|
"gemini-pro": "red-500",
|
|
"gemini-pro-vision": "red-500",
|
|
"gemini-1.5-pro-latest": "red-500",
|
|
"gemini-1.5-flash-latest": "red-500",
|
|
|
|
"deepseek-coder": "blue-700",
|
|
"deepseek-chat": "blue-700",
|
|
|
|
"bing-creative": "blue-700",
|
|
"bing-balanced": "blue-700",
|
|
"bing-precise": "blue-700",
|
|
|
|
"zhipu-chatglm-turbo": "lime-500",
|
|
"zhipu-chatglm-pro": "lime-500",
|
|
"zhipu-chatglm-std": "lime-500",
|
|
"zhipu-chatglm-lite": "lime-500",
|
|
"glm-4": "lime-500",
|
|
"glm-4v": "lime-500",
|
|
"glm-3-turbo": "lime-500",
|
|
|
|
"qwen-plus": "indigo-600",
|
|
"qwen-plus-net": "indigo-600",
|
|
"qwen-turbo": "indigo-600",
|
|
"qwen-turbo-net": "indigo-600",
|
|
|
|
"llama-2-70b": "sky-400",
|
|
"llama-2-13b": "sky-400",
|
|
"llama-2-7b": "sky-400",
|
|
"code-llama-34b": "sky-400",
|
|
"code-llama-13b": "sky-400",
|
|
"code-llama-7b": "sky-400",
|
|
|
|
hunyuan: "blue-500",
|
|
"360-gpt-v9": "stone-500",
|
|
"baichuan-53b": "orange-700",
|
|
|
|
"skylark-lite-public": "sky-300",
|
|
"skylark-plus-public": "sky-300",
|
|
"skylark-pro-public": "sky-300",
|
|
"skylark-chat": "sky-300",
|
|
};
|
|
|
|
const unknownColors = [
|
|
"gray-700",
|
|
"indigo-600",
|
|
"green-500",
|
|
"green-600",
|
|
"purple-600",
|
|
"purple-700",
|
|
"orange-400",
|
|
"blue-400",
|
|
"red-500",
|
|
"blue-700",
|
|
"lime-500",
|
|
"sky-400",
|
|
"stone-500",
|
|
"orange-700",
|
|
"sky-300",
|
|
];
|
|
|
|
export function getUnknownModelColor(model: string): string {
|
|
const char = model.length > 0 ? model[0] : "A";
|
|
const code = char.charCodeAt(0);
|
|
|
|
return unknownColors[code % unknownColors.length];
|
|
}
|
|
|
|
export function getModelColor(model: string): string {
|
|
return modelColorMapper[model] || getUnknownModelColor(model);
|
|
}
|