💄 style: fix AI21 modelTag display in Bedrock & add providerIcon (lob…

…ehub#36)

* 💄 style: fix AI21 modelTag display in Bedrock

* 💄 style: support Ai21Labs provider icon

src/ModelIcon/const.ts

@@ -119,7 +119,7 @@ export const modelMappings: ModelMapping[] = [
   { Icon: Suno, keywords: ['suno'] },
   { Icon: Azure, keywords: ['wizardlm', 'phi3', 'phi-3'] },
   { Icon: Adobe, keywords: ['firefly'] },
-  { Icon: Ai21, keywords: ['jamba', '^j2-'] },
+  { Icon: Ai21, keywords: ['jamba', '^j2-', 'ai21'] },
   { Icon: InternLM, keywords: ['internlm'] },
   { Icon: Upstage, keywords: ['^solar-'] },
   { Icon: PaLM, keywords: ['palm'] },

src/ProviderCombine/const.tsx

@@ -3,6 +3,7 @@ import { Divider } from 'antd';
 import { FC, ReactNode, memo } from 'react';
 import { Flexbox } from 'react-layout-kit';
 
+import Ai21 from '@/Ai21'
 import Ai360 from '@/Ai360';
 import AiMass from '@/AiMass';
 import Anthropic from '@/Anthropic';
@@ -139,4 +140,5 @@ export const providerMappings: ProviderMapping[] = [
   { Icon: Novita.Combine, keywords: [ModelProvider.Novita], multiple: 0.95 },
   { Icon: SiliconCloud.Combine, keywords: [ModelProvider.SiliconCloud], multiple: 0.75 },
   { Icon: Upstage.Combine, keywords: [ModelProvider.Upstage], multiple: 0.9 },
+  { Icon: Ai21.Combine, keywords: [ModelProvider.Ai21] },
 ];

src/ProviderIcon/const.ts

@@ -1,5 +1,6 @@
 import { FC } from 'react';
 
+import Ai21 from '@/Ai21';
 import Ai360 from '@/Ai360';
 import AiMass from '@/AiMass';
 import Anthropic from '@/Anthropic';
@@ -47,6 +48,7 @@ export interface ProviderMapping {
 }
 
 export enum ModelProvider {
+  Ai21 = 'ai21',
   Ai360 = 'ai360',
   Anthropic = 'anthropic',
   Azure = 'azure',
@@ -106,4 +108,5 @@ export const providerMappings: ProviderMapping[] = [
   { Icon: Ai360, keywords: [ModelProvider.Ai360] },
   { Icon: SiliconCloud, keywords: [ModelProvider.SiliconCloud] },
   { Icon: Upstage, keywords: [ModelProvider.Upstage] },
+  { Icon: Ai21, keywords: [ModelProvider.Ai21] },
 ];