@lobehub/chat 0.146.2 → 0.147.1

package/src/components/ModelSelect/index.tsx

@@ -6,11 +6,12 @@ import { rgba } from 'polished';
 import { memo } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Center, Flexbox } from 'react-layout-kit';
-import ModelIcon from 'src/components/ModelIcon';
-import ModelProviderIcon from 'src/components/ModelProviderIcon';
 
 import { ChatModelCard } from '@/types/llm';
 
+import ModelIcon from '../ModelIcon';
+import ModelProviderIcon from '../ModelProviderIcon';
+
 const useStyles = createStyles(({ css, token }) => ({
   custom: css`
     width: 36px;
@@ -56,13 +57,71 @@ const useStyles = createStyles(({ css, token }) => ({
   `,
 }));
 
+interface ModelInfoTagsProps extends ChatModelCard {
+  directionReverse?: boolean;
+  placement?: 'top' | 'right';
+}
+export const ModelInfoTags = memo<ModelInfoTagsProps>(
+  ({ directionReverse, placement = 'right', ...model }) => {
+    const { t } = useTranslation('components');
+    const { styles, cx } = useStyles();
+
+    return (
+      <Flexbox direction={directionReverse ? 'horizontal-reverse' : 'horizontal'} gap={4}>
+        {model.files && (
+          <Tooltip placement={placement} title={t('ModelSelect.featureTag.file')}>
+            <div className={cx(styles.tag, styles.tagGreen)}>
+              <Icon icon={LucidePaperclip} />
+            </div>
+          </Tooltip>
+        )}
+        {model.vision && (
+          <Tooltip placement={placement} title={t('ModelSelect.featureTag.vision')}>
+            <div className={cx(styles.tag, styles.tagGreen)}>
+              <Icon icon={LucideEye} />
+            </div>
+          </Tooltip>
+        )}
+        {model.functionCall && (
+          <Tooltip
+            overlayStyle={{ maxWidth: 'unset' }}
+            placement={placement}
+            title={t('ModelSelect.featureTag.functionCall')}
+          >
+            <div className={cx(styles.tag, styles.tagBlue)}>
+              <Icon icon={ToyBrick} />
+            </div>
+          </Tooltip>
+        )}
+        {model.tokens && (
+          <Tooltip
+            overlayStyle={{ maxWidth: 'unset' }}
+            placement={placement}
+            title={t('ModelSelect.featureTag.tokens', {
+              tokens: numeral(model.tokens).format('0,0'),
+            })}
+          >
+            <Center className={styles.token}>{Math.floor(model.tokens / 1000)}K</Center>
+          </Tooltip>
+        )}
+        {/*{model.isCustom && (*/}
+        {/*  <Tooltip*/}
+        {/*    overlayStyle={{ maxWidth: 300 }}*/}
+        {/*    placement={placement}*/}
+        {/*    title={t('ModelSelect.featureTag.custom')}*/}
+        {/*  >*/}
+        {/*    <Center className={styles.custom}>DIY</Center>*/}
+        {/*  </Tooltip>*/}
+        {/*)}*/}
+      </Flexbox>
+    );
+  },
+);
+
 interface ModelItemRenderProps extends ChatModelCard {
   showInfoTag?: boolean;
 }
 export const ModelItemRender = memo<ModelItemRenderProps>(({ showInfoTag = true, ...model }) => {
-  const { styles, cx } = useStyles();
-  const { t } = useTranslation('common');
-
   return (
     <Flexbox align={'center'} gap={32} horizontal justify={'space-between'}>
       <Flexbox align={'center'} gap={8} horizontal>
@@ -70,55 +129,7 @@ export const ModelItemRender = memo<ModelItemRenderProps>(({ showInfoTag = true,
         {model.displayName || model.id}
       </Flexbox>
 
-      {showInfoTag && (
-        <Flexbox gap={4} horizontal>
-          {model.files && (
-            <Tooltip placement={'right'} title={t('ModelSelect.featureTag.file')}>
-              <div className={cx(styles.tag, styles.tagGreen)}>
-                <Icon icon={LucidePaperclip} />
-              </div>
-            </Tooltip>
-          )}
-          {model.vision && (
-            <Tooltip placement={'right'} title={t('ModelSelect.featureTag.vision')}>
-              <div className={cx(styles.tag, styles.tagGreen)}>
-                <Icon icon={LucideEye} />
-              </div>
-            </Tooltip>
-          )}
-          {model.functionCall && (
-            <Tooltip
-              overlayStyle={{ maxWidth: 'unset' }}
-              placement={'right'}
-              title={t('ModelSelect.featureTag.functionCall')}
-            >
-              <div className={cx(styles.tag, styles.tagBlue)}>
-                <Icon icon={ToyBrick} />
-              </div>
-            </Tooltip>
-          )}
-          {model.tokens && (
-            <Tooltip
-              overlayStyle={{ maxWidth: 'unset' }}
-              placement={'right'}
-              title={t('ModelSelect.featureTag.tokens', {
-                tokens: numeral(model.tokens).format('0,0'),
-              })}
-            >
-              <Center className={styles.token}>{Math.floor(model.tokens / 1000)}K</Center>
-            </Tooltip>
-          )}
-          {model.isCustom && (
-            <Tooltip
-              overlayStyle={{ maxWidth: 300 }}
-              placement={'right'}
-              title={t('ModelSelect.featureTag.custom')}
-            >
-              <Center className={styles.custom}>DIY</Center>
-            </Tooltip>
-          )}
-        </Flexbox>
-      )}
+      {showInfoTag && <ModelInfoTags {...model} />}
     </Flexbox>
   );
 });
@@ -128,12 +139,12 @@ interface ProviderItemRenderProps {
 }
 
 export const ProviderItemRender = memo<ProviderItemRenderProps>(({ provider }) => {
-  const { t } = useTranslation('common');
+  const { t } = useTranslation('modelProvider');
 
   return (
     <Flexbox align={'center'} gap={4} horizontal>
       <ModelProviderIcon provider={provider} />
-      {t(`modelProvider.${provider}` as any)}
+      {t(`${provider}.title` as any)}
     </Flexbox>
   );
 });

package/src/config/modelProviders/anthropic.ts

@@ -6,6 +6,7 @@ const Anthropic: ModelProviderCard = {
       description:
         'Ideal balance of intelligence and speed for enterprise workloads. Maximum utility at a lower price, dependable, balanced for scaled deployments',
       displayName: 'Claude 3 Sonnet',
+      enabled: true,
       id: 'claude-3-sonnet-20240229',
       maxOutput: 4096,
       tokens: 200_000,
@@ -15,6 +16,7 @@ const Anthropic: ModelProviderCard = {
       description:
         'Most powerful model for highly complex tasks. Top-level performance, intelligence, fluency, and understanding',
       displayName: 'Claude 3 Opus',
+      enabled: true,
       id: 'claude-3-opus-20240229',
       maxOutput: 4096,
       tokens: 200_000,
@@ -24,6 +26,7 @@ const Anthropic: ModelProviderCard = {
       description:
         'Fastest and most compact model for near-instant responsiveness. Quick and accurate targeted performance',
       displayName: 'Claude 3 Haiku',
+      enabled: true,
       id: 'claude-3-haiku-20240307',
       maxOutput: 4096,
       tokens: 200_000,
@@ -31,19 +34,21 @@ const Anthropic: ModelProviderCard = {
     },
     {
       displayName: 'Claude 2.1',
+      enabled: false,
       id: 'claude-2.1',
       maxOutput: 4096,
       tokens: 200_000,
     },
     {
       displayName: 'Claude 2.0',
+      enabled: false,
       id: 'claude-2.0',
       maxOutput: 4096,
       tokens: 100_000,
     },
     {
       displayName: 'Claude Instant 1.2',
-      hidden: true,
+      enabled: false,
       id: 'claude-instant-1.2',
       maxOutput: 4096,
       tokens: 100_000,

package/src/config/modelProviders/azure.ts

@@ -0,0 +1,42 @@
+import { ModelProviderCard } from '@/types/llm';
+
+const Azure: ModelProviderCard = {
+  chatModels: [
+    {
+      deploymentName: 'gpt-35-turbo',
+      description: 'GPT 3.5 Turbo，适用于各种文本生成和理解任务',
+      displayName: 'GPT-3.5 Turbo',
+      enabled: true,
+      functionCall: true,
+      id: 'gpt-35-turbo',
+      maxOutput: 4096,
+      tokens: 16_385,
+    },
+    {
+      deploymentName: 'gpt-35-turbo-16k',
+      displayName: 'GPT-3.5 Turbo',
+      functionCall: true,
+      id: 'gpt-35-turbo-16k',
+      tokens: 16_384,
+    },
+    {
+      deploymentName: 'gpt-4-turbo',
+      displayName: 'GPT-4 Turbo',
+      enabled: true,
+      functionCall: true,
+      id: 'gpt-4',
+      tokens: 128_000,
+    },
+    {
+      deploymentName: 'gpt-4-vision',
+      description: 'GPT-4 视觉预览版，支持视觉任务',
+      displayName: 'GPT-4 Turbo with Vision Preview',
+      id: 'gpt-4-vision-preview',
+      tokens: 128_000,
+      vision: true,
+    },
+  ],
+  id: 'azure',
+};
+
+export default Azure;

package/src/config/modelProviders/bedrock.ts

@@ -6,7 +6,6 @@ const Bedrock: ModelProviderCard = {
       description:
         'Amazon Titan Text G1 - Express v1，上下文长度可达 8000 个 token，适合广泛的用途。',
       displayName: 'Titan Text G1 - Express',
-      hidden: true,
       id: 'amazon.titan-text-express-v1:0:8k',
       tokens: 8000,
     },
@@ -14,6 +13,7 @@ const Bedrock: ModelProviderCard = {
       description:
         'Anthropic 推出的 Claude 3 Sonnet 模型在智能和速度之间取得理想的平衡，尤其是在处理企业工作负载方面。该模型提供最大的效用，同时价格低于竞争产品，并且其经过精心设计，是大规模部署人工智能的可信赖、高耐久性骨干模型。 Claude 3 Sonnet 可以处理图像和返回文本输出，并且提供 200K 上下文窗口。',
       displayName: 'Claude 3 Sonnet',
+      enabled: true,
       id: 'anthropic.claude-3-sonnet-20240229-v1:0',
       tokens: 200_000,
       vision: true,
@@ -22,6 +22,7 @@ const Bedrock: ModelProviderCard = {
       description:
         'Claude 3 Haiku 是 Anthropic 最快速、最紧凑的模型，具有近乎即时的响应能力。该模型可以快速回答简单的查询和请求。客户将能够构建模仿人类交互的无缝人工智能体验。 Claude 3 Haiku 可以处理图像和返回文本输出，并且提供 200K 上下文窗口。',
       displayName: 'Claude 3 Haiku',
+      enabled: true,
       id: 'anthropic.claude-3-haiku-20240307-v1:0',
       tokens: 200_000,
       vision: true,
@@ -43,12 +44,14 @@ const Bedrock: ModelProviderCard = {
     {
       description: 'Llama 2 Chat 13B v1，上下文大小为 4k，Llama 2 模型的对话用例优化变体。',
       displayName: 'Llama 2 Chat 13B',
+      enabled: true,
       id: 'meta.llama2-13b-chat-v1',
       tokens: 4000,
     },
     {
       description: 'Llama 2 Chat 70B v1，上下文大小为 4k，Llama 2 模型的对话用例优化变体。',
       displayName: 'Llama 2 Chat 70B',
+      enabled: true,
       id: 'meta.llama2-70b-chat-v1',
       tokens: 4000,
     },

package/src/config/modelProviders/google.ts

@@ -5,15 +5,12 @@ const Google: ModelProviderCard = {
     {
       description: 'A legacy text-only model optimized for chat conversations',
       displayName: 'PaLM 2 Chat (Legacy)',
-      hidden: true,
       id: 'chat-bison-001',
       maxOutput: 1024,
-      tokens: 5120,
     },
     {
       description: 'A legacy model that understands text and generates text as an output',
       displayName: 'PaLM 2 (Legacy)',
-      hidden: true,
       id: 'text-bison-001',
       maxOutput: 1024,
       tokens: 9220,
@@ -21,6 +18,7 @@ const Google: ModelProviderCard = {
     {
       description: 'The best model for scaling across a wide range of tasks',
       displayName: 'Gemini 1.0 Pro',
+      enabled: true,
       id: 'gemini-pro',
       maxOutput: 2048,
       tokens: 32_768,
@@ -28,6 +26,7 @@ const Google: ModelProviderCard = {
     {
       description: 'The best image understanding model to handle a broad range of applications',
       displayName: 'Gemini 1.0 Pro Vision',
+      enabled: true,
       id: 'gemini-1.0-pro-vision-latest',
       maxOutput: 4096,
       tokens: 16_384,
@@ -36,25 +35,15 @@ const Google: ModelProviderCard = {
     {
       description: 'The best image understanding model to handle a broad range of applications',
       displayName: 'Gemini 1.0 Pro Vision',
-      hidden: true,
       id: 'gemini-pro-vision',
       maxOutput: 4096,
       tokens: 16_384,
       vision: true,
     },
-    {
-      description: 'The best model for scaling across a wide range of tasks',
-      displayName: 'Gemini 1.0 Pro',
-      hidden: true,
-      id: '1.0-pro',
-      maxOutput: 2048,
-      tokens: 32_768,
-    },
     {
       description:
         'The best model for scaling across a wide range of tasks. This is a stable model that supports tuning.',
       displayName: 'Gemini 1.0 Pro 001 (Tuning)',
-      hidden: true,
       id: 'gemini-1.0-pro-001',
       maxOutput: 2048,
       tokens: 32_768,
@@ -63,7 +52,6 @@ const Google: ModelProviderCard = {
       description:
         'The best model for scaling across a wide range of tasks. This is the latest model.',
       displayName: 'Gemini 1.0 Pro Latest',
-      hidden: true,
       id: 'gemini-1.0-pro-latest',
       maxOutput: 2048,
       tokens: 32_768,

package/src/config/modelProviders/groq.ts

@@ -4,16 +4,19 @@ const Groq: ModelProviderCard = {
   chatModels: [
     {
       displayName: 'Mixtral-8x7b-Instruct-v0.1',
+      enabled: true,
       id: 'mixtral-8x7b-32768',
       tokens: 32_768,
     },
     {
       displayName: 'Gemma-7b-it',
+      enabled: true,
       id: 'gemma-7b-it',
       tokens: 8192,
     },
     {
       displayName: 'LLaMA2-70b-chat',
+      enabled: true,
       id: 'llama2-70b-4096',
       tokens: 4096,
     },

package/src/config/modelProviders/index.ts

@@ -1,4 +1,4 @@
-import { ChatModelCard } from '@/types/llm';
+import { ChatModelCard, ModelProviderCard } from '@/types/llm';
 
 import AnthropicProvider from './anthropic';
 import BedrockProvider from './bedrock';
@@ -30,16 +30,21 @@ export const LOBE_DEFAULT_MODEL_LIST: ChatModelCard[] = [
   ZeroOneProvider.chatModels,
 ].flat();
 
-export { default as AnthropicProvider } from './anthropic';
-export { default as BedrockProvider } from './bedrock';
-export { default as GoogleProvider } from './google';
-export { default as GroqProvider } from './groq';
-export { default as MistralProvider } from './mistral';
-export { default as MoonshotProvider } from './moonshot';
-export { default as OllamaProvider } from './ollama';
-export { default as OpenAIProvider } from './openai';
-export { default as OpenRouterProvider } from './openrouter';
-export { default as PerplexityProvider } from './perplexity';
-export { default as TogetherAIProvider } from './togetherai';
-export { default as ZeroOneProvider } from './zeroone';
-export { default as ZhiPuProvider } from './zhipu';
+export const filterEnabledModels = (provider: ModelProviderCard) => {
+  return provider.chatModels.filter((v) => v.enabled).map((m) => m.id);
+};
+
+export { default as AnthropicProviderCard } from './anthropic';
+export { default as AzureProviderCard } from './azure';
+export { default as BedrockProviderCard } from './bedrock';
+export { default as GoogleProviderCard } from './google';
+export { default as GroqProviderCard } from './groq';
+export { default as MistralProviderCard } from './mistral';
+export { default as MoonshotProviderCard } from './moonshot';
+export { default as OllamaProviderCard } from './ollama';
+export { default as OpenAIProviderCard } from './openai';
+export { default as OpenRouterProviderCard } from './openrouter';
+export { default as PerplexityProviderCard } from './perplexity';
+export { default as TogetherAIProviderCard } from './togetherai';
+export { default as ZeroOneProviderCard } from './zeroone';
+export { default as ZhiPuProviderCard } from './zhipu';

package/src/config/modelProviders/mistral.ts

@@ -4,26 +4,31 @@ const Mistral: ModelProviderCard = {
   chatModels: [
     {
       displayName: 'Mistral 7B',
+      enabled: true,
       id: 'open-mistral-7b',
       tokens: 32_768,
     },
     {
       displayName: 'Mixtral 8x7B',
+      enabled: true,
       id: 'open-mixtral-8x7b',
       tokens: 32_768,
     },
     {
       displayName: 'Mistral Small (2402)',
+      enabled: true,
       id: 'mistral-small-2402',
       tokens: 32_768,
     },
     {
       displayName: 'Mistral Medium (2312)',
+      enabled: true,
       id: 'mistral-medium-2312',
       tokens: 32_768,
     },
     {
       displayName: 'Mistral Large (2402)',
+      enabled: true,
       id: 'mistral-large-2402',
       tokens: 32_768,
     },

package/src/config/modelProviders/moonshot.ts

@@ -4,23 +4,25 @@ const Moonshot: ModelProviderCard = {
   chatModels: [
     {
       displayName: 'Moonshot V1 8K',
+      enabled: true,
       id: 'moonshot-v1-8k',
       tokens: 8192,
     },
     {
       displayName: 'Moonshot V1 32K',
+      enabled: true,
       id: 'moonshot-v1-32k',
       tokens: 32_768,
     },
     {
       displayName: 'Moonshot V1 128K',
+      enabled: true,
       id: 'moonshot-v1-128k',
       tokens: 128_000,
     },
     {
       displayName: 'Moonshot Kimi Reverse',
       files: true,
-      hidden: true,
       id: 'moonshot-v1',
       tokens: 200_000,
       vision: true,

package/src/config/modelProviders/ollama.ts

@@ -4,6 +4,7 @@ const Ollama: ModelProviderCard = {
   chatModels: [
     {
       displayName: 'Gemma 7B',
+      enabled: true,
       functionCall: false,
       id: 'gemma',
       tokens: 4000,
@@ -11,6 +12,7 @@ const Ollama: ModelProviderCard = {
     },
     {
       displayName: 'Gemma 2B',
+      enabled: true,
       functionCall: false,
       id: 'gemma:2b',
       tokens: 4000,
@@ -19,13 +21,13 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Llama2 Chat 13B',
       functionCall: false,
-      hidden: true,
       id: 'llama2:13b',
       tokens: 4000,
       vision: false,
     },
     {
       displayName: 'Llama2 Chat 7B',
+      enabled: true,
       functionCall: false,
       id: 'llama2',
       tokens: 4000,
@@ -34,7 +36,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Llama2 Chat 70B',
       functionCall: false,
-      hidden: true,
       id: 'llama2:70b',
       tokens: 4000,
       vision: false,
@@ -42,7 +43,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Llama2 CN 13B',
       functionCall: false,
-      hidden: true,
       id: 'llama2-chinese:13b',
       tokens: 4000,
       vision: false,
@@ -64,7 +64,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Code Llama 34B',
       functionCall: false,
-      hidden: true,
       id: 'codellama:34b',
       tokens: 16_000,
       vision: false,
@@ -72,7 +71,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Code Llama 70B',
       functionCall: false,
-      hidden: true,
       id: 'codellama:70b',
       tokens: 16_000,
       vision: false,
@@ -80,13 +78,13 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Code Llama 7B (Python)',
       functionCall: false,
-      hidden: true,
       id: 'codellama:python',
       tokens: 16_000,
       vision: false,
     },
     {
       displayName: 'Mistral',
+      enabled: true,
       functionCall: false,
       id: 'mistral',
       tokens: 4800,
@@ -94,6 +92,7 @@ const Ollama: ModelProviderCard = {
     },
     {
       displayName: 'Mixtral 8x7B',
+      enabled: true,
       functionCall: false,
       id: 'mixtral',
       tokens: 32_000,
@@ -108,6 +107,7 @@ const Ollama: ModelProviderCard = {
     },
     {
       displayName: 'Qwen Chat 7B',
+      enabled: true,
       functionCall: false,
       id: 'qwen:7b',
       tokens: 32_768,
@@ -116,7 +116,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Qwen Chat 14B',
       functionCall: false,
-      hidden: true,
       id: 'qwen:14b',
       tokens: 32_768,
       vision: false,
@@ -124,7 +123,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Qwen Chat 72B',
       functionCall: false,
-      hidden: true,
       id: 'qwen:72b',
       tokens: 32_768,
       vision: false,
@@ -132,7 +130,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'LLaVA 7B',
       functionCall: false,
-      hidden: true,
       id: 'llava',
       tokens: 4000,
       vision: true,
@@ -140,7 +137,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'LLaVA 13B',
       functionCall: false,
-      hidden: true,
       id: 'llava:13b',
       tokens: 4000,
       vision: true,
@@ -148,7 +144,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'LLaVA 34B',
       functionCall: false,
-      hidden: true,
       id: 'llava:34b',
       tokens: 4000,
       vision: true,
@@ -157,7 +152,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Qwen Plus',
       functionCall: true,
-      hidden: true,
       id: 'qwen-plus',
       tokens: 30_000,
       vision: false,
@@ -165,7 +159,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Qwen Turbo',
       functionCall: true,
-      hidden: true,
       id: 'qwen-turbo',
       tokens: 6000,
       vision: false,
@@ -173,7 +166,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Qwen Max',
       functionCall: true,
-      hidden: true,
       id: 'qwen-max',
       tokens: 6000,
       vision: false,
@@ -181,7 +173,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Qwen Max Long',
       functionCall: true,
-      hidden: true,
       id: 'qwen-max-longcontext',
       tokens: 28_000,
       vision: false,
@@ -189,7 +180,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Qwen VL Max',
       functionCall: false,
-      hidden: true,
       id: 'qwen-vl-max',
       tokens: 6000,
       vision: true,
@@ -197,7 +187,6 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Qwen VL Plus',
       functionCall: false,
-      hidden: true,
       id: 'qwen-vl-plus',
       tokens: 30_000,
       vision: true,