@lobehub/chat 0.90.3 → 0.92.0

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 = {

package/src/types/chatMessage.ts

@@ -1,6 +1,7 @@
 import { PluginRequestPayload } from '@lobehub/chat-plugin-sdk';
 
 import { ErrorType } from '@/types/fetch';
+import { Translate } from '@/types/translate';
 
 import { LLMRoleType } from './llm';
 import { BaseDataModel } from './meta';
@@ -18,6 +19,9 @@ export interface OpenAIFunctionCall {
   name: string;
 }
 
+export interface ChatTranslate extends Translate {
+  content?: string;
+}
 export interface ChatMessage extends BaseDataModel {
   /**
    * @title 内容
@@ -29,10 +33,7 @@ export interface ChatMessage extends BaseDataModel {
   extra?: {
     fromModel?: string;
     // 翻译
-    translate?: {
-      target: string;
-      to: string;
-    };
+    translate?: ChatTranslate;
   } & Record<string, any>;
 
   /**

package/src/types/translate.ts

@@ -0,0 +1,4 @@
+export interface Translate {
+  from: string;
+  to: string;
+}

package/src/prompts/chat.ts

@@ -1,32 +0,0 @@
-import { chatHelpers } from '@/store/session/helpers';
-import { LanguageModel } from '@/types/llm';
-import { OpenAIChatMessage, OpenAIChatStreamPayload } from '@/types/openai/chat';
-
-export const promptSummaryTitle = 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,
-  };
-};

package/src/utils/genSiteHeadTitle.ts

@@ -1 +0,0 @@
-export const genSiteHeadTitle = (title?: string) => (title ? `${title} · LobeChat` : 'LobeChat');