Skip to content

Commit e123076

Browse files
authored
Merge pull request ChatGPTNextWeb#6295 from rexkyng/patch-1
Fix: Improve Mistral icon detection and remove redundant code.
2 parents f5f3ce9 + ebcb4db commit e123076

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

app/components/emoji.tsx

+4-4
Original file line numberDiff line numberDiff line change
@@ -66,11 +66,11 @@ export function Avatar(props: { model?: ModelType; avatar?: string }) {
6666
LlmIcon = BotIconGemma;
6767
} else if (modelName.startsWith("claude")) {
6868
LlmIcon = BotIconClaude;
69-
} else if (modelName.toLowerCase().includes("llama")) {
69+
} else if (modelName.includes("llama")) {
7070
LlmIcon = BotIconMeta;
71-
} else if (modelName.startsWith("mixtral")) {
71+
} else if (modelName.startsWith("mixtral") || modelName.startsWith("codestral")) {
7272
LlmIcon = BotIconMistral;
73-
} else if (modelName.toLowerCase().includes("deepseek")) {
73+
} else if (modelName.includes("deepseek")) {
7474
LlmIcon = BotIconDeepseek;
7575
} else if (modelName.startsWith("moonshot")) {
7676
LlmIcon = BotIconMoonshot;
@@ -85,7 +85,7 @@ export function Avatar(props: { model?: ModelType; avatar?: string }) {
8585
} else if (modelName.startsWith("doubao") || modelName.startsWith("ep-")) {
8686
LlmIcon = BotIconDoubao;
8787
} else if (
88-
modelName.toLowerCase().includes("glm") ||
88+
modelName.includes("glm") ||
8989
modelName.startsWith("cogview-") ||
9090
modelName.startsWith("cogvideox-")
9191
) {

0 commit comments

Comments
 (0)