|
@@ -8,12 +8,15 @@ import { querySessionInfo } from "../common"
|
|
|
import {
|
|
import {
|
|
|
IconAlibaba,
|
|
IconAlibaba,
|
|
|
IconAnthropic,
|
|
IconAnthropic,
|
|
|
|
|
+ IconArcee,
|
|
|
IconGemini,
|
|
IconGemini,
|
|
|
IconMiniMax,
|
|
IconMiniMax,
|
|
|
IconMoonshotAI,
|
|
IconMoonshotAI,
|
|
|
|
|
+ IconNvidia,
|
|
|
IconOpenAI,
|
|
IconOpenAI,
|
|
|
IconStealth,
|
|
IconStealth,
|
|
|
IconXai,
|
|
IconXai,
|
|
|
|
|
+ IconXiaomi,
|
|
|
IconZai,
|
|
IconZai,
|
|
|
} from "~/component/icon"
|
|
} from "~/component/icon"
|
|
|
import { useI18n } from "~/context/i18n"
|
|
import { useI18n } from "~/context/i18n"
|
|
@@ -29,6 +32,9 @@ const getModelLab = (modelId: string) => {
|
|
|
if (modelId.startsWith("qwen")) return "Alibaba"
|
|
if (modelId.startsWith("qwen")) return "Alibaba"
|
|
|
if (modelId.startsWith("minimax")) return "MiniMax"
|
|
if (modelId.startsWith("minimax")) return "MiniMax"
|
|
|
if (modelId.startsWith("grok")) return "xAI"
|
|
if (modelId.startsWith("grok")) return "xAI"
|
|
|
|
|
+ if (modelId.startsWith("mimo")) return "Xiaomi"
|
|
|
|
|
+ if (modelId.startsWith("nemotron")) return "NVIDIA"
|
|
|
|
|
+ if (modelId.startsWith("trinity")) return "Arcee"
|
|
|
return "Stealth"
|
|
return "Stealth"
|
|
|
}
|
|
}
|
|
|
|
|
|
|
@@ -139,6 +145,12 @@ export function ModelSection() {
|
|
|
return <IconXai width={16} height={16} />
|
|
return <IconXai width={16} height={16} />
|
|
|
case "MiniMax":
|
|
case "MiniMax":
|
|
|
return <IconMiniMax width={16} height={16} />
|
|
return <IconMiniMax width={16} height={16} />
|
|
|
|
|
+ case "Xiaomi":
|
|
|
|
|
+ return <IconXiaomi width={16} height={16} />
|
|
|
|
|
+ case "NVIDIA":
|
|
|
|
|
+ return <IconNvidia width={16} height={16} />
|
|
|
|
|
+ case "Arcee":
|
|
|
|
|
+ return <IconArcee width={16} height={16} />
|
|
|
default:
|
|
default:
|
|
|
return <IconStealth width={16} height={16} />
|
|
return <IconStealth width={16} height={16} />
|
|
|
}
|
|
}
|