@lobehub/chat 0.91.0 → 0.92.0

package/src/chains/chat.ts

@@ -0,0 +1,77 @@
+import { chatHelpers } from '@/store/session/helpers';
+import { LanguageModel } from '@/types/llm';
+import { OpenAIChatMessage, OpenAIChatStreamPayload } from '@/types/openai/chat';
+
+export const chainSummaryTitle = async (
+  messages: OpenAIChatMessage[],
+): Promise<Partial<OpenAIChatStreamPayload>> => {
+  const finalMessages: OpenAIChatMessage[] = [
+    {
+      content:
+        '你是一名擅长会话的助理，你需要将用户的会话总结为 10 个字以内的标题，不需要包含标点符号',
+      role: 'system',
+    },
+    {
+      content: `${messages.map((message) => `${message.role}: ${message.content}`).join('\n')}
+
+请总结上述对话为10个字以内的标题，不需要包含标点符号`,
+      role: 'user',
+    },
+  ];
+  // 如果超过 4k，则使用 GPT3.5 16K 模型
+  const tokens = await chatHelpers.getMessagesTokenCount(finalMessages);
+  let model: LanguageModel | undefined = undefined;
+  if (tokens > 4000) {
+    model = LanguageModel.GPT3_5_16K;
+  }
+
+  return {
+    messages: finalMessages,
+    model,
+  };
+};
+
+export const chainTranslate = (
+  content: string,
+  targetLang: string,
+): Partial<OpenAIChatStreamPayload> => ({
+  messages: [
+    {
+      content: '你是一名擅长翻译的助理，你需要将输入的语言翻译为目标语言',
+      role: 'system',
+    },
+    {
+      content: `请将以下内容 ${content}，翻译为 ${targetLang} `,
+      role: 'user',
+    },
+  ],
+});
+export const chainLangDetect = (content: string): Partial<OpenAIChatStreamPayload> => ({
+  messages: [
+    {
+      content:
+        '你是一名精通全世界语言的语言专家，你需要识别用户输入的内容，以国际标准 locale 进行输出',
+      role: 'system',
+    },
+    {
+      content: '{你好}',
+      role: 'user',
+    },
+    {
+      content: 'zh-CN',
+      role: 'assistant',
+    },
+    {
+      content: '{hello}',
+      role: 'user',
+    },
+    {
+      content: 'en-US',
+      role: 'assistant',
+    },
+    {
+      content: `{${content}}`,
+      role: 'user',
+    },
+  ],
+});

package/src/features/AgentSetting/AgentPrompt/index.tsx

@@ -42,9 +42,9 @@ const AgentPrompt = memo(() => {
               maxValue={ModelTokens[model as LanguageModel]}
               shape={'square'}
               text={{
-                overload: t('tokenTag.overload', { ns: 'common' }),
-                remained: t('tokenTag.remained', { ns: 'common' }),
-                used: t('tokenTag.used', { ns: 'common' }),
+                overload: t('tokenTag.overload', { ns: 'chat' }),
+                remained: t('tokenTag.remained', { ns: 'chat' }),
+                used: t('tokenTag.used', { ns: 'chat' }),
               }}
               value={systemTokenCount}
             />

package/src/features/AgentSetting/store/action.ts

@@ -1,6 +1,6 @@
 import { StateCreator } from 'zustand/vanilla';
 
-import { promptPickEmoji, promptSummaryAgentName, promptSummaryDescription } from '@/prompts/agent';
+import { promptPickEmoji, promptSummaryAgentName, promptSummaryDescription } from '@/chains/agent';
 import { MetaData } from '@/types/meta';
 import { LobeAgentConfig } from '@/types/session';
 import { fetchPresetTaskResult } from '@/utils/fetch';

package/src/layout/ResponsiveLayout.client.tsx

@@ -15,10 +15,12 @@ const ResponsiveLayout = ({ children, Desktop, Mobile }: ServerResponsiveLayoutP
   const { t } = useTranslation();
   const mobile = useIsMobile();
 
-  return (
+  return mobile ? (
     <Suspense fallback={<FullscreenLoading title={t('layoutInitializing', { ns: 'common' })} />}>
-      {mobile ? <Mobile>{children}</Mobile> : <Desktop>{children}</Desktop>}
+      <Mobile>{children}</Mobile>
     </Suspense>
+  ) : (
+    <Desktop>{children}</Desktop>
   );
 };
 

package/src/locales/create.ts

@@ -4,10 +4,11 @@ import { initReactI18next } from 'react-i18next';
 
 import { getClientConfig } from '@/config/client';
 import { DEFAULT_LANG, LOBE_LOCALE_COOKIE } from '@/const/locale';
-import resources from '@/locales/resources';
 import type { Namespaces } from '@/types/locale';
 import { isOnServerSide } from '@/utils/env';
 
+import { resources } from './options';
+
 const { I18N_DEBUG, I18N_DEBUG_BROWSER, I18N_DEBUG_SERVER } = getClientConfig();
 const debugMode = I18N_DEBUG ?? isOnServerSide ? I18N_DEBUG_SERVER : I18N_DEBUG_BROWSER;
 

package/src/locales/default/chat.ts

@@ -0,0 +1,60 @@
+export default {
+  agentDefaultMessage:
+    '你好，我是 **{{name}}**，你可以立即与我开始对话，也可以前往 [助手设置](/chat/settings#session={{id}}) 完善我的信息。',
+  agentDefaultMessageWithSystemRole: '你好，我是 **{{name}}**，{{systemRole}}，让我们开始对话吧！',
+  backToBottom: '查看最新消息',
+  clearCurrentMessages: '清空当前会话消息',
+  confirmClearCurrentMessages: '即将清空当前会话消息，清空后将无法找回，请确认你的操作',
+  confirmRemoveSessionItemAlert: '即将删除该助手，删除后该将无法找回，请确认你的操作',
+  defaultAgent: '自定义助手',
+  defaultSession: '自定义助手',
+  historyRange: '历史范围',
+  inbox: {
+    defaultMessage:
+      '你好，我是你的智能助手，你可以问我任何问题，我会尽力回答你。如果需要获得更加专业或定制的助手，可以点击`+`创建自定义助手',
+    desc: '开启大脑集群，激发思维火花。你的智能助理，在这里与你交流一切',
+    title: '随便聊聊',
+  },
+  newAgent: '新建助手',
+  noDescription: '暂无描述',
+
+  regenerate: '重新生成',
+
+  roleAndArchive: '角色与记录',
+  searchAgentPlaceholder: '搜索助手和对话...',
+  send: '发送',
+  sendPlaceholder: '输入聊天内容...',
+  shareModal: {
+    download: '下载截图',
+    imageType: '图片格式',
+    screenshot: '截图',
+    settings: '导出设置',
+    shareToShareGPT: '生成 ShareGPT 分享链接',
+    withBackground: '包含背景图片',
+    withFooter: '包含页脚',
+    withPluginInfo: '包含插件信息',
+    withSystemRole: '包含助手角色设定',
+  },
+  stop: '停止',
+
+  temp: '临时',
+  tokenDetail: '角色设定: {{systemRoleToken}} · 历史消息: {{chatsToken}}',
+  tokenTag: {
+    overload: '超过限制',
+    remained: '剩余',
+    used: '使用',
+  },
+  topic: {
+    confirmRemoveTopic: '即将删除该话题，删除后将不可恢复，请谨慎操作。',
+    defaultTitle: '默认话题',
+    saveCurrentMessages: '将当前会话保存为话题',
+    searchPlaceholder: '搜索话题...',
+    title: '话题',
+  },
+  translate: {
+    clear: '删除翻译',
+  },
+  translateTo: '翻译',
+  updateAgent: '更新助理信息',
+  warp: '换行',
+};

package/src/locales/default/common.ts

@@ -1,9 +1,6 @@
 export default {
   about: '关于',
   advanceSettings: '高级设置',
-  agentDefaultMessage:
-    '你好，我是 **{{name}}**，你可以立即与我开始对话，也可以前往 [助手设置](/chat/settings#session={{id}}) 完善我的信息。',
-  agentDefaultMessageWithSystemRole: '你好，我是 **{{name}}**，{{systemRole}}，让我们开始对话吧！',
   agentMaxToken: '会话最大长度',
   agentModel: '模型',
   agentProfile: '助手信息',
@@ -11,12 +8,9 @@ export default {
   archive: '归档',
   autoGenerate: '自动补全',
   autoGenerateTooltip: '基于提示词自动补全助手描述',
-  backToBottom: '查看最新消息',
   cancel: '取消',
   changelog: '更新日志',
-  clearCurrentMessages: '清空当前会话消息',
   close: '关闭',
-  confirmClearCurrentMessages: '即将清空当前会话消息，清空后将无法找回，请确认你的操作',
   confirmRemoveSessionItemAlert: '即将删除该助手，删除后该将无法找回，请确认你的操作',
   copy: '复制',
   copySuccess: '复制成功',
@@ -37,18 +31,17 @@ export default {
   historyRange: '历史范围',
   import: '导入配置',
 
-  inbox: {
-    defaultMessage:
-      '你好，我是你的智能助手，你可以问我任何问题，我会尽力回答你。如果需要获得更加专业或定制的助手，可以点击`+`创建自定义助手',
-    desc: '开启大脑集群，激发思维火花。你的智能助理，在这里与你交流一切',
-    title: '随便聊聊',
+  lang: {
+    'en': '英语',
+    'en-US': '英语',
+    'ja-JP': '日语',
+    'ko-KR': '韩语',
+    'ru-RU': '俄语',
+    'zh': '简体中文',
+    'zh-CN': '简体中文',
+    'zh-TW': '繁体中文',
   },
   layoutInitializing: '正在加载布局...',
-  message: {
-    function_loading: '插件请求中...',
-  },
-  moreSetting: '更多设置...',
-  newAgent: '新建助手',
   noDescription: '暂无描述',
   ok: '确定',
   password: '密码',
@@ -58,24 +51,10 @@ export default {
   rename: '重命名',
   reset: '重置',
   retry: '重试',
-  roleAndArchive: '角色与记录',
-  searchAgentPlaceholder: '搜索助手和对话...',
   send: '发送',
-  sendPlaceholder: '输入聊天内容...',
   sessionList: '助手列表',
   setting: '设置',
   share: '分享',
-  shareModal: {
-    download: '下载截图',
-    imageType: '图片格式',
-    screenshot: '截图',
-    settings: '导出设置',
-    shareToShareGPT: '生成 ShareGPT 分享链接',
-    withBackground: '包含背景图片',
-    withFooter: '包含页脚',
-    withPluginInfo: '包含插件信息',
-    withSystemRole: '包含助手角色设定',
-  },
   stop: '停止',
   tab: {
     chat: '会话',
@@ -83,25 +62,11 @@ export default {
     setting: '设置',
   },
   temp: '临时',
-  tokenDetail: '角色设定: {{systemRoleToken}} · 历史消息: {{chatsToken}}',
-  tokenTag: {
-    overload: '超过限制',
-    remained: '剩余',
-    used: '使用',
-  },
-  topic: {
-    confirmRemoveTopic: '即将删除该话题，删除后将不可恢复，请谨慎操作。',
-    defaultTitle: '默认话题',
-    saveCurrentMessages: '将当前会话保存为话题',
-    searchPlaceholder: '搜索话题...',
-    title: '话题',
-  },
+
   updateAgent: '更新助理信息',
-  updatePrompt: '更新提示词',
   upgradeVersion: {
     action: '立即升级',
     hasNew: '有可用更新',
     newVersion: '有新版本可用：{{version}}',
   },
-  warp: '换行',
 };

package/src/locales/options.ts

@@ -1,5 +1,7 @@
 import type { Locales } from '@/types/locale';
 
+import _resources from './resources';
+
 type LocaleOptions = {
   label: string;
   value: Locales;
@@ -33,3 +35,11 @@ export const localeOptions: LocaleOptions = [
 ] as LocaleOptions;
 
 export const supportLangs: string[] = localeOptions.map((i) => i.value);
+
+export const resources = {
+  ..._resources,
+  en: _resources['en-US'],
+  zh: _resources['zh-CN'],
+};
+
+export const supportLocales = Object.keys(resources);

package/src/locales/resources/en_US.ts

@@ -1,3 +1,4 @@
+import chat from '../../../locales/en_US/chat.json';
 import common from '../../../locales/en_US/common.json';
 import empty from '../../../locales/en_US/empty.json';
 import error from '../../../locales/en_US/error.json';
@@ -7,6 +8,7 @@ import setting from '../../../locales/en_US/setting.json';
 import welcome from '../../../locales/en_US/welcome.json';
 
 const resources = {
+  chat,
   common,
   empty,
   error,

package/src/locales/resources/index.ts

@@ -10,10 +10,10 @@ const resources = {
   'ja-JP': ja_JP,
   'ko-KR': ko_KR,
   'ru-RU': ru_RU,
-  'zh': zh_CN,
   'zh-CN': zh_CN,
   'zh-TW': zh_TW,
 } as const;
+
 export default resources;
 export const defaultResources = zh_CN;
 export type Resources = typeof resources;

package/src/locales/resources/ja_JP.ts

@@ -1,3 +1,4 @@
+import chat from '../../../locales/ja_JP/chat.json';
 import common from '../../../locales/ja_JP/common.json';
 import empty from '../../../locales/ja_JP/empty.json';
 import error from '../../../locales/ja_JP/error.json';
@@ -7,6 +8,7 @@ import setting from '../../../locales/ja_JP/setting.json';
 import welcome from '../../../locales/ja_JP/welcome.json';
 
 const resources = {
+  chat,
   common,
   empty,
   error,

package/src/locales/resources/ko_KR.ts

@@ -1,3 +1,4 @@
+import chat from '../../../locales/ko_KR/chat.json';
 import common from '../../../locales/ko_KR/common.json';
 import empty from '../../../locales/ko_KR/empty.json';
 import error from '../../../locales/ko_KR/error.json';
@@ -7,6 +8,7 @@ import setting from '../../../locales/ko_KR/setting.json';
 import welcome from '../../../locales/ko_KR/welcome.json';
 
 const resources = {
+  chat,
   common,
   empty,
   error,

package/src/locales/resources/ru_RU.ts

@@ -1,3 +1,4 @@
+import chat from '../../../locales/ru_RU/chat.json';
 import common from '../../../locales/ru_RU/common.json';
 import empty from '../../../locales/ru_RU/empty.json';
 import error from '../../../locales/ru_RU/error.json';
@@ -7,6 +8,7 @@ import setting from '../../../locales/ru_RU/setting.json';
 import welcome from '../../../locales/ru_RU/welcome.json';
 
 const resources = {
+  chat,
   common,
   empty,
   error,

package/src/locales/resources/zh_CN.ts

@@ -1,3 +1,4 @@
+import chat from '../default/chat';
 import common from '../default/common';
 import empty from '../default/empty';
 import error from '../default/error';
@@ -7,6 +8,7 @@ import setting from '../default/setting';
 import welcome from '../default/welcome';
 
 const resources = {
+  chat,
   common,
   empty,
   error,

package/src/locales/resources/zh_TW.ts

@@ -1,3 +1,4 @@
+import chat from '../../../locales/zh_TW/chat.json';
 import common from '../../../locales/zh_TW/common.json';
 import empty from '../../../locales/zh_TW/empty.json';
 import error from '../../../locales/zh_TW/error.json';
@@ -7,6 +8,7 @@ import setting from '../../../locales/zh_TW/setting.json';
 import welcome from '../../../locales/zh_TW/welcome.json';
 
 const resources = {
+  chat,
   common,
   empty,
   error,

package/src/store/global/selectors/settings.ts

@@ -1,8 +1,10 @@
 import { DEFAULT_OPENAI_MODEL_LIST } from '@/const/llm';
+import { DEFAULT_LANG } from '@/const/locale';
 import { DEFAULT_AGENT_META } from '@/const/meta';
 import { DEFAULT_AGENT, DEFAULT_AGENT_CONFIG, DEFAULT_SETTINGS } from '@/const/settings';
 import { Locales } from '@/locales/resources';
 import { GlobalSettings } from '@/types/settings';
+import { isOnServerSide } from '@/utils/env';
 import { merge } from '@/utils/merge';
 
 import { GlobalStore } from '../store';
@@ -35,6 +37,8 @@ export const exportSettings = (s: GlobalStore) => {
 const currentLanguage = (s: GlobalStore) => {
   const locale = s.settings.language;
   if (locale === 'auto') {
+    if (isOnServerSide) return DEFAULT_LANG;
+
     return navigator.language as Locales;
   }
 

package/src/store/session/slices/chat/actions/index.ts

@@ -5,11 +5,16 @@ import { SessionStore } from '@/store/session';
 import { ChatMessageAction, chatMessage } from './message';
 import { ShareAction, chatShare } from './share';
 import { ChatTopicAction, chatTopic } from './topic';
+import { ChatTranslateAction, chatTranslate } from './translate';
 
 /**
  * 聊天操作
  */
-export interface ChatAction extends ChatTopicAction, ChatMessageAction, ShareAction {}
+export interface ChatAction
+  extends ChatTopicAction,
+    ChatMessageAction,
+    ShareAction,
+    ChatTranslateAction {}
 
 export const createChatSlice: StateCreator<
   SessionStore,
@@ -20,4 +25,5 @@ export const createChatSlice: StateCreator<
   ...chatTopic(...params),
   ...chatMessage(...params),
   ...chatShare(...params),
+  ...chatTranslate(...params),
 });

package/src/store/session/slices/chat/actions/message.ts

@@ -66,13 +66,14 @@ export interface ChatMessageAction {
    * @param text - 消息文本
    */
   sendMessage: (text: string) => Promise<void>;
-
   stopGenerateMessage: () => void;
+
   toggleChatLoading: (
     loading: boolean,
     id?: string,
     action?: string,
   ) => AbortController | undefined;
+
   triggerFunctionCall: (id: string) => Promise<void>;
 }
 
@@ -298,7 +299,6 @@ export const chatMessage: StateCreator<
 
     toggleChatLoading(false);
   },
-
   toggleChatLoading: (loading, id, action) => {
     if (loading) {
       const abortController = new AbortController();

package/src/store/session/slices/chat/actions/topic.ts

@@ -1,7 +1,7 @@
 import { StateCreator } from 'zustand/vanilla';
 
+import { chainSummaryTitle } from '@/chains/chat';
 import { LOADING_FLAT } from '@/const/message';
-import { promptSummaryTitle } from '@/prompts/chat';
 import { SessionStore } from '@/store/session';
 import { fetchPresetTaskResult } from '@/utils/fetch';
 import { setNamespace } from '@/utils/storeDebug';
@@ -116,7 +116,7 @@ export const chatTopic: StateCreator<
         output += x;
         dispatchTopic({ id: topicId, key: 'title', type: 'updateChatTopic', value: output });
       },
-      params: await promptSummaryTitle(messages),
+      params: await chainSummaryTitle(messages),
     });
 
     return topicId;

package/src/store/session/slices/chat/actions/translate.ts

@@ -0,0 +1,87 @@
+import { produce } from 'immer';
+import { StateCreator } from 'zustand/vanilla';
+
+import { chainLangDetect, chainTranslate } from '@/chains/chat';
+import { supportLocales } from '@/locales/options';
+import { SessionStore } from '@/store/session';
+import { fetchPresetTaskResult } from '@/utils/fetch';
+import { setNamespace } from '@/utils/storeDebug';
+
+import { sessionSelectors } from '../../session/selectors';
+
+const t = setNamespace('chat/translate');
+
+/**
+ * 翻译事件
+ */
+export interface ChatTranslateAction {
+  clearTranslate: (id: string) => void;
+
+  /**
+   * 翻译消息
+   * @param id
+   */
+  translateMessage: (id: string, targetLang: string) => Promise<void>;
+}
+
+export const chatTranslate: StateCreator<
+  SessionStore,
+  [['zustand/devtools', never]],
+  [],
+  ChatTranslateAction
+> = (set, get) => ({
+  clearTranslate: (id) => {
+    get().dispatchMessage({
+      id,
+      key: 'translate',
+      type: 'updateMessageExtra',
+      value: null,
+    });
+  },
+
+  translateMessage: async (id, targetLang) => {
+    const { toggleChatLoading, dispatchMessage } = get();
+    const session = sessionSelectors.currentSession(get());
+    if (!session || !id) return;
+
+    const message = session.chats[id];
+    if (!message) return;
+
+    let content = '';
+    let from = '';
+
+    dispatchMessage({
+      id,
+      key: 'translate',
+      type: 'updateMessageExtra',
+      value: { content: '', to: targetLang },
+    });
+
+    toggleChatLoading(true, id, t('translateMessage(start)', { id }) as string);
+
+    // detect from language
+    fetchPresetTaskResult({
+      params: chainLangDetect(message.content),
+    }).then((data) => {
+      if (data && supportLocales.includes(data)) from = data;
+    });
+
+    // translate to target language
+    await fetchPresetTaskResult({
+      onMessageHandle: (text) => {
+        dispatchMessage({
+          id,
+          key: 'translate',
+          type: 'updateMessageExtra',
+          value: produce({ content: '', from, to: targetLang }, (draft) => {
+            content += text;
+            draft.content += content;
+          }),
+        });
+      },
+      params: chainTranslate(message.content, targetLang),
+    });
+
+    toggleChatLoading(false);
+  },
+});

package/src/store/session/slices/chat/selectors/chat.ts

@@ -52,15 +52,21 @@ export const currentChatsWithGuideMessage = (s: SessionStore): ChatMessage[] =>
 
   const [activeId, isInbox] = [s.activeId, s.activeId === INBOX_SESSION_ID];
   const meta = agentSelectors.currentAgentMeta(s);
+
+  const inboxMsg = t('inbox.defaultMessage', { ns: 'chat' });
+  const agentSystemRoleMsg = t('agentDefaultMessageWithSystemRole', {
+    name: meta.title || t('defaultAgent'),
+    ns: 'chat',
+    systemRole: meta.description,
+  });
+  const agentMsg = t('agentDefaultMessage', {
+    id: activeId,
+    name: meta.title || t('defaultAgent'),
+    ns: 'chat',
+  });
+
   const emptyInboxGuideMessage = {
-    content: isInbox
-      ? t('inbox.defaultMessage')
-      : !!meta.description
-      ? t('agentDefaultMessageWithSystemRole', {
-          name: meta.title || t('defaultAgent'),
-          systemRole: meta.description,
-        })
-      : t('agentDefaultMessage', { id: activeId, name: meta.title || t('defaultAgent') }),
+    content: isInbox ? inboxMsg : !!meta.description ? agentSystemRoleMsg : agentMsg,
     createAt: Date.now(),
     extra: {},
     id: 'default',
@@ -86,22 +92,22 @@ export const chatsMessageString = (s: SessionStore): string => {
   return chats.map((m) => m.content).join('');
 };
 
-export const currentFunctionCallProps = (
-  s: SessionStore,
-): ((
-  props: Pick<ChatMessage, 'plugin' | 'function_call' | 'content' | 'id'>,
-) => FunctionCallProps) => {
-  const chatLoadingId = s.chatLoadingId;
-  return ({ plugin, function_call, content, id }) => {
+export const getFunctionMessageParams =
+  (
+    s: SessionStore,
+  ): ((
+    props: Pick<ChatMessage, 'plugin' | 'function_call' | 'content' | 'id'>,
+  ) => FunctionCallProps) =>
+  ({ plugin, function_call, content, id }) => {
     const itemId = plugin?.identifier || function_call?.name;
     const command = plugin ?? function_call;
     const args = command?.arguments;
+
     return {
       arguments: args,
       command,
       content,
       id: itemId,
-      loading: id === chatLoadingId,
+      loading: id === s.chatLoadingId,
     };
   };
-};

package/src/store/session/slices/chat/selectors/index.ts

@@ -3,8 +3,8 @@ import {
   currentChats,
   currentChatsWithGuideMessage,
   currentChatsWithHistoryConfig,
-  currentFunctionCallProps,
   getChatsById,
+  getFunctionMessageParams,
 } from './chat';
 import { currentTopics, getTopicMessages } from './topic';
 
@@ -13,8 +13,8 @@ export const chatSelectors = {
   currentChats,
   currentChatsWithGuideMessage,
   currentChatsWithHistoryConfig,
-  currentFunctionCallProps,
   getChatsById,
+  getFunctionMessageParams,
 };
 
 export const topicSelectors = {