File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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 . startsWith ( "llama" ) ) {
69+ } else if ( modelName . toLowerCase ( ) . includes ( "llama" ) ) {
7070 LlmIcon = BotIconMeta ;
7171 } else if ( modelName . startsWith ( "mixtral" ) ) {
7272 LlmIcon = BotIconMistral ;
73- } else if ( modelName . startsWith ( "deepseek" ) ) {
73+ } else if ( modelName . toLowerCase ( ) . 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 . startsWith ( "glm" ) ||
88+ modelName . toLowerCase ( ) . includes ( "glm" ) ||
8989 modelName . startsWith ( "cogview-" ) ||
9090 modelName . startsWith ( "cogvideox-" )
9191 ) {
You can’t perform that action at this time.
0 commit comments