@lobehub/chat 1.48.3 → 1.49.0

package/src/config/aiModels/doubao.ts

@@ -0,0 +1,65 @@
+import { AIChatModelCard } from '@/types/aiModel';
+
+const doubaoChatModels: AIChatModelCard[] = [
+    {
+        contextWindowTokens: 4096,
+        description: 
+        '拥有极致的响应速度，更好的性价比，为客户不同场景提供更灵活的选择。支持 4k 上下文窗口的推理和精调。',
+        displayName: 'Doubao Lite 4k',
+        enabled: true,
+        id: 'Doubao-lite-4k',
+        type: 'chat',
+    },
+    {
+        contextWindowTokens: 32_768,
+        description: 
+            '拥有极致的响应速度，更好的性价比，为客户不同场景提供更灵活的选择。支持 32k 上下文窗口的推理和精调。',
+        displayName: 'Doubao Lite 32k',
+        enabled: true,
+        id: 'Doubao-lite-32k',
+        type: 'chat',
+    },
+    {
+        contextWindowTokens: 128_000,
+        description: 
+            '拥有极致的响应速度，更好的性价比，为客户不同场景提供更灵活的选择。支持 128k 上下文窗口的推理和精调。',
+        displayName: 'Doubao Lite 128k',
+        enabled: true,
+        id: 'Doubao-lite-128k',
+        type: 'chat',
+    },
+    {
+        contextWindowTokens: 4096,
+        description: 
+            '效果最好的主力模型，适合处理复杂任务，在参考问答、总结摘要、创作、文本分类、角色扮演等场景都有很好的效果。支持 4k 上下文窗口的推理和精调。',
+        displayName: 'Doubao Pro 4k',
+        enabled: true,
+        id: 'Doubao-pro-4k',
+        type: 'chat',
+    },
+    {
+        config: {
+          deploymentName: 'Doubao-pro-test',
+        },
+        contextWindowTokens: 32_768,
+        description: 
+            '效果最好的主力模型，适合处理复杂任务，在参考问答、总结摘要、创作、文本分类、角色扮演等场景都有很好的效果。支持 32k 上下文窗口的推理和精调。',
+        displayName: 'Doubao Pro 32k',
+        enabled: true,
+        id: 'Doubao-pro-32k',
+        type: 'chat',
+    },
+    {
+        contextWindowTokens: 128_000,
+        description: 
+            '效果最好的主力模型，适合处理复杂任务，在参考问答、总结摘要、创作、文本分类、角色扮演等场景都有很好的效果。支持 128k 上下文窗口的推理和精调。',
+        displayName: 'Doubao Pro 128k',
+        enabled: true,
+        id: 'Doubao-pro-128k',
+        type: 'chat',
+    },
+];
+
+export const allModels = [...doubaoChatModels];
+
+export default allModels;

package/src/config/aiModels/index.ts

@@ -8,6 +8,7 @@ import { default as baichuan } from './baichuan';
 import { default as bedrock } from './bedrock';
 import { default as cloudflare } from './cloudflare';
 import { default as deepseek } from './deepseek';
+import { default as doubao } from './doubao';
 import { default as fireworksai } from './fireworksai';
 import { default as giteeai } from './giteeai';
 import { default as github } from './github';
@@ -67,6 +68,7 @@ export const LOBE_DEFAULT_MODEL_LIST = buildDefaultModelList({
   bedrock,
   cloudflare,
   deepseek,
+  doubao,
   fireworksai,
   giteeai,
   github,
@@ -107,6 +109,7 @@ export { default as baichuan } from './baichuan';
 export { default as bedrock } from './bedrock';
 export { default as cloudflare } from './cloudflare';
 export { default as deepseek } from './deepseek';
+export { default as doubao } from './doubao';
 export { default as fireworksai } from './fireworksai';
 export { default as giteeai } from './giteeai';
 export { default as github } from './github';

package/src/config/llm.ts

@@ -123,6 +123,9 @@ export const getLLMConfig = () => {
 
       ENABLED_HIGRESS: z.boolean(),
       HIGRESS_API_KEY: z.string().optional(),
+
+      ENABLED_DOUBAO: z.boolean(),
+      DOUBAO_API_KEY: z.string().optional(),
     },
     runtimeEnv: {
       API_KEY_SELECT_MODE: process.env.API_KEY_SELECT_MODE,
@@ -244,6 +247,9 @@ export const getLLMConfig = () => {
 
       ENABLED_HIGRESS: !!process.env.HIGRESS_API_KEY,
       HIGRESS_API_KEY: process.env.HIGRESS_API_KEY,
+
+      ENABLED_DOUBAO: !!process.env.DOUBAO_API_KEY,
+      DOUBAO_API_KEY: process.env.DOUBAO_API_KEY,
     },
   });
 };

package/src/config/modelProviders/doubao.ts

@@ -0,0 +1,23 @@
+import { ModelProviderCard } from '@/types/llm';
+
+// ref https://www.volcengine.com/docs/82379/1330310
+const Doubao: ModelProviderCard = {
+  chatModels: [],
+  // checkModel: 'Doubao-lite-4k',
+  description: '字节跳动推出的自研大模型。通过字节跳动内部50+业务场景实践验证，每日万亿级tokens大使用量持续打磨，提供多种模态能力，以优质模型效果为企业打造丰富的业务体验。',
+  id: 'doubao',
+  modelsUrl: 'https://www.volcengine.com/product/doubao',
+  name: '豆包',
+  settings: {
+    disableBrowserRequest: true, // CORS error
+    sdkType: 'doubao',
+    // showModelFetcher: false,
+    smoothing: {
+      speed: 2,
+      text: true,
+    },
+  },
+  url: 'https://www.volcengine.com/product/doubao',
+};
+
+export default Doubao;

package/src/config/modelProviders/index.ts

@@ -8,6 +8,7 @@ import BaichuanProvider from './baichuan';
 import BedrockProvider from './bedrock';
 import CloudflareProvider from './cloudflare';
 import DeepSeekProvider from './deepseek';
+import DoubaoProvider from './doubao';
 import FireworksAIProvider from './fireworksai';
 import GiteeAIProvider from './giteeai';
 import GithubProvider from './github';
@@ -120,6 +121,7 @@ export const DEFAULT_MODEL_PROVIDER_LIST = [
   GiteeAIProvider,
   TaichuProvider,
   Ai360Provider,
+  DoubaoProvider,
 ];
 
 export const filterEnabledModels = (provider: ModelProviderCard) => {
@@ -139,6 +141,7 @@ export { default as BaichuanProviderCard } from './baichuan';
 export { default as BedrockProviderCard } from './bedrock';
 export { default as CloudflareProviderCard } from './cloudflare';
 export { default as DeepSeekProviderCard } from './deepseek';
+export { default as DoubaoProviderCard } from './doubao';
 export { default as FireworksAIProviderCard } from './fireworksai';
 export { default as GiteeAIProviderCard } from './giteeai';
 export { default as GithubProviderCard } from './github';

package/src/features/Conversation/Messages/Assistant/Reasoning/index.tsx

@@ -1,123 +1,19 @@
-import { Icon, Markdown } from '@lobehub/ui';
-import { createStyles } from 'antd-style';
-import { AtomIcon, ChevronDown, ChevronRight } from 'lucide-react';
-import { rgba } from 'polished';
-import { memo, useEffect, useState } from 'react';
-import { useTranslation } from 'react-i18next';
-import { Flexbox } from 'react-layout-kit';
+import { memo } from 'react';
 
+import Thinking from '@/components/Thinking';
 import { useChatStore } from '@/store/chat';
 import { aiChatSelectors } from '@/store/chat/selectors';
 
-const useStyles = createStyles(({ css, token, isDarkMode }) => ({
-  container: css`
-    cursor: pointer;
-
-    width: fit-content;
-    padding-block: 4px;
-    padding-inline: 8px;
-    border-radius: 6px;
-
-    color: ${token.colorTextTertiary};
-
-    &:hover {
-      background: ${isDarkMode ? token.colorFillQuaternary : token.colorFillTertiary};
-    }
-  `,
-  expand: css`
-    background: ${isDarkMode ? token.colorFillQuaternary : token.colorFillTertiary} !important;
-  `,
-  shinyText: css`
-    color: ${rgba(token.colorText, 0.45)};
-
-    background: linear-gradient(
-      120deg,
-      ${rgba(token.colorTextBase, 0)} 40%,
-      ${token.colorTextSecondary} 50%,
-      ${rgba(token.colorTextBase, 0)} 60%
-    );
-    background-clip: text;
-    background-size: 200% 100%;
-
-    animation: shine 1.5s linear infinite;
-
-    @keyframes shine {
-      0% {
-        background-position: 100%;
-      }
-
-      100% {
-        background-position: -100%;
-      }
-    }
-  `,
-  title: css`
-    overflow: hidden;
-    display: -webkit-box;
-    -webkit-box-orient: vertical;
-    -webkit-line-clamp: 1;
-
-    font-size: 12px;
-    text-overflow: ellipsis;
-  `,
-}));
-
-interface ThinkingProps {
+interface ReasoningProps {
   content?: string;
   duration?: number;
   id: string;
 }
 
-const Thinking = memo<ThinkingProps>(({ content = '', duration, id }) => {
-  const { t } = useTranslation('chat');
-  const { styles, cx } = useStyles();
-
-  const [showDetail, setShowDetail] = useState(false);
-
+const Reasoning = memo<ReasoningProps>(({ content = '', duration, id }) => {
   const isReasoning = useChatStore(aiChatSelectors.isMessageInReasoning(id));
 
-  useEffect(() => {
-    if (isReasoning && !content) {
-      setShowDetail(true);
-    }
-
-    if (!isReasoning) {
-      setShowDetail(false);
-    }
-  }, [isReasoning, content]);
-
-  return (
-    <Flexbox
-      className={cx(styles.container, showDetail && styles.expand)}
-      gap={16}
-      onClick={() => {
-        setShowDetail(!showDetail);
-      }}
-    >
-      <Flexbox distribution={'space-between'} flex={1} horizontal>
-        {isReasoning ? (
-          <Flexbox gap={8} horizontal>
-            <Icon icon={AtomIcon} />
-            <Flexbox className={styles.shinyText} horizontal>
-              {t('reasoning.thinking')}
-            </Flexbox>
-          </Flexbox>
-        ) : (
-          <Flexbox gap={8} horizontal>
-            <Icon icon={AtomIcon} />
-            {t('reasoning.thought', { duration: ((duration || 0) / 1000).toFixed(1) })}
-          </Flexbox>
-        )}
-        <Icon icon={showDetail ? ChevronDown : ChevronRight} />
-      </Flexbox>
-
-      {showDetail && (
-        <Flexbox>
-          <Markdown variant={'chat'}>{content}</Markdown>
-        </Flexbox>
-      )}
-    </Flexbox>
-  );
+  return <Thinking content={content} duration={duration} thinking={isReasoning} />;
 });
 
-export default Thinking;
+export default Reasoning;

package/src/libs/agent-runtime/AgentRuntime.ts

@@ -11,6 +11,7 @@ import { LobeBaichuanAI } from './baichuan';
 import { LobeBedrockAI, LobeBedrockAIParams } from './bedrock';
 import { LobeCloudflareAI, LobeCloudflareParams } from './cloudflare';
 import { LobeDeepSeekAI } from './deepseek';
+import { LobeDoubaoAI } from './doubao';
 import { LobeFireworksAI } from './fireworksai';
 import { LobeGiteeAI } from './giteeai';
 import { LobeGithubAI } from './github';
@@ -139,6 +140,7 @@ class AgentRuntime {
       bedrock: Partial<LobeBedrockAIParams>;
       cloudflare: Partial<LobeCloudflareParams>;
       deepseek: Partial<ClientOptions>;
+      doubao: Partial<ClientOptions>;
       fireworksai: Partial<ClientOptions>;
       giteeai: Partial<ClientOptions>;
       github: Partial<ClientOptions>;
@@ -363,6 +365,11 @@ class AgentRuntime {
         runtimeModel = new LobeHigressAI(params.higress);
         break;
       }
+
+      case ModelProvider.Doubao: {
+        runtimeModel = new LobeDoubaoAI(params.doubao);
+        break;
+      }
     }
     return new AgentRuntime(runtimeModel);
   }

package/src/libs/agent-runtime/doubao/index.ts

@@ -0,0 +1,10 @@
+import { ModelProvider } from '../types';
+import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+
+export const LobeDoubaoAI = LobeOpenAICompatibleFactory({
+  baseURL: 'https://ark.cn-beijing.volces.com/api/v3',
+  debug: {
+    chatCompletion: () => process.env.DEBUG_DOUBAO_CHAT_COMPLETION === '1',
+  },
+  provider: ModelProvider.Doubao,
+});

package/src/libs/agent-runtime/index.ts

@@ -4,6 +4,7 @@ export { LobeAzureOpenAI } from './azureOpenai';
 export * from './BaseAI';
 export { LobeBedrockAI } from './bedrock';
 export { LobeDeepSeekAI } from './deepseek';
+export { LobeDoubaoAI } from './doubao';
 export * from './error';
 export { LobeGoogleAI } from './google';
 export { LobeGroq } from './groq';

package/src/libs/agent-runtime/types/type.ts

@@ -30,6 +30,7 @@ export enum ModelProvider {
   Bedrock = 'bedrock',
   Cloudflare = 'cloudflare',
   DeepSeek = 'deepseek',
+  Doubao = 'doubao',
   FireworksAI = 'fireworksai',
   GiteeAI = 'giteeai',
   Github = 'github',

package/src/locales/default/chat.ts

@@ -84,10 +84,6 @@ export default {
       },
     },
   },
-  reasoning: {
-    thinking: '深度思考中',
-    thought: '已深度思考（用时 {{duration}} 秒）',
-  },
   regenerate: '重新生成',
   roleAndArchive: '角色与记录',
   searchAgentPlaceholder: '搜索助手...',

package/src/locales/default/components.ts

@@ -88,4 +88,9 @@ export default {
     emptyModel: '没有启用的模型，请前往设置开启',
     provider: '提供商',
   },
+  Thinking: {
+    thinking: '深度思考中...',
+    thought: '已深度思考',
+    thoughtWithDuration: '已深度思考（用时 {{duration}} 秒）',
+  },
 };

package/src/server/globalConfig/index.ts

@@ -25,6 +25,9 @@ export const getServerGlobalConfig = () => {
         enabledKey: 'ENABLED_AWS_BEDROCK',
         modelListKey: 'AWS_BEDROCK_MODEL_LIST',
       },
+      doubao: {
+        withDeploymentName: true,
+      },
       giteeai: {
         enabledKey: 'ENABLED_GITEE_AI',
         modelListKey: 'GITEE_AI_MODEL_LIST',

package/src/services/chat.ts

@@ -221,7 +221,7 @@ class ChatService {
     let model = res.model || DEFAULT_AGENT_CONFIG.model;
 
     // if the provider is Azure, get the deployment name as the request model
-    if (provider === ModelProvider.Azure) {
+    if (provider === ModelProvider.Azure || provider === ModelProvider.Doubao) {
       model = findAzureDeploymentName(model);
     }
 

package/src/types/aiProvider.ts

@@ -18,6 +18,7 @@ export const AiProviderSDKEnum = {
   Azure: 'azure',
   Bedrock: 'bedrock',
   Cloudflare: 'cloudflare',
+  Doubao: 'doubao',
   Google: 'google',
   Huggingface: 'huggingface',
   Ollama: 'ollama',

package/src/types/user/settings/keyVaults.ts

@@ -39,6 +39,7 @@ export interface UserKeyVaults {
   bedrock?: AWSBedrockKeyVault;
   cloudflare?: CloudflareKeyVault;
   deepseek?: OpenAICompatibleKeyVault;
+  doubao?: OpenAICompatibleKeyVault;
   fireworksai?: OpenAICompatibleKeyVault;
   giteeai?: OpenAICompatibleKeyVault;
   github?: OpenAICompatibleKeyVault;