@lobehub/chat 0.91.0 → 0.93.0

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

@@ -0,0 +1,88 @@
+import { produce } from 'immer';
+import { StateCreator } from 'zustand/vanilla';
+
+import { chainLangDetect } from '@/chains/langDetect';
+import { chainTranslate } from '@/chains/translate';
+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;
+}