@lobehub/chat 1.0.11 → 1.0.13

package/src/store/chat/slices/plugin/action.ts

@@ -1,3 +1,4 @@
+/* eslint-disable sort-keys-fix/sort-keys-fix, typescript-sort-keys/interface */
 import { PluginErrorType } from '@lobehub/chat-plugin-sdk';
 import { t } from 'i18next';
 import { Md5 } from 'ts-md5';
@@ -25,11 +26,6 @@ export interface ChatPluginAction {
     triggerAiMessage?: boolean,
   ) => Promise<void>;
 
-  internal_callPluginApi: (id: string, payload: ChatToolPayload) => Promise<string | undefined>;
-  internal_invokeDifferentTypePlugin: (id: string, payload: ChatToolPayload) => Promise<any>;
-  internal_transformToolCalls: (toolCalls: MessageToolCall[]) => ChatToolPayload[];
-  internal_updatePluginError: (id: string, error: any) => Promise<void>;
-
   invokeBuiltinTool: (id: string, payload: ChatToolPayload) => Promise<void>;
   invokeDefaultTypePlugin: (id: string, payload: any) => Promise<string | undefined>;
   invokeMarkdownTypePlugin: (id: string, payload: ChatToolPayload) => Promise<void>;
@@ -41,6 +37,16 @@ export interface ChatPluginAction {
 
   triggerToolCalls: (id: string) => Promise<void>;
   updatePluginState: (id: string, key: string, value: any) => Promise<void>;
+
+  internal_callPluginApi: (id: string, payload: ChatToolPayload) => Promise<string | undefined>;
+  internal_invokeDifferentTypePlugin: (id: string, payload: ChatToolPayload) => Promise<any>;
+  internal_togglePluginApiCalling: (
+    loading: boolean,
+    id?: string,
+    action?: string,
+  ) => AbortController | undefined;
+  internal_transformToolCalls: (toolCalls: MessageToolCall[]) => ChatToolPayload[];
+  internal_updatePluginError: (id: string, error: any) => Promise<void>;
 }
 
 export const chatPlugin: StateCreator<
@@ -69,124 +75,17 @@ export const chatPlugin: StateCreator<
 
     if (triggerAiMessage) await triggerAIMessage({ parentId: id });
   },
-  internal_callPluginApi: async (id, payload) => {
-    const { internal_updateMessageContent, refreshMessages, internal_toggleChatLoading } = get();
-    let data: string;
-
-    try {
-      const abortController = internal_toggleChatLoading(
-        true,
-        id,
-        n('fetchPlugin/start') as string,
-      );
-
-      const message = chatSelectors.getMessageById(id)(get());
-
-      const res = await chatService.runPluginApi(payload, {
-        signal: abortController?.signal,
-        trace: { observationId: message?.observationId, traceId: message?.traceId },
-      });
-      data = res.text;
-
-      // save traceId
-      if (res.traceId) {
-        await messageService.updateMessage(id, { traceId: res.traceId });
-      }
-    } catch (error) {
-      console.log(error);
-      const err = error as Error;
-
-      // ignore the aborted request error
-      if (!err.message.includes('The user aborted a request.')) {
-        await messageService.updateMessageError(id, error as any);
-        await refreshMessages();
-      }
-
-      data = '';
-    }
-
-    internal_toggleChatLoading(false, id, n('fetchPlugin/end') as string);
-    // 如果报错则结束了
-    if (!data) return;
-
-    await internal_updateMessageContent(id, data);
-
-    return data;
-  },
-
-  internal_invokeDifferentTypePlugin: async (id, payload) => {
-    switch (payload.type) {
-      case 'standalone': {
-        return await get().invokeStandaloneTypePlugin(id, payload);
-      }
-
-      case 'markdown': {
-        return await get().invokeMarkdownTypePlugin(id, payload);
-      }
-
-      case 'builtin': {
-        return await get().invokeBuiltinTool(id, payload);
-      }
-
-      default: {
-        return await get().invokeDefaultTypePlugin(id, payload);
-      }
-    }
-  },
-
-  internal_transformToolCalls: (toolCalls) => {
-    return toolCalls
-      .map((toolCall): ChatToolPayload | null => {
-        let payload: ChatToolPayload;
-
-        const [identifier, apiName, type] = toolCall.function.name.split(PLUGIN_SCHEMA_SEPARATOR);
-
-        if (!apiName) return null;
-
-        payload = {
-          apiName,
-          arguments: toolCall.function.arguments,
-          id: toolCall.id,
-          identifier,
-          type: (type ?? 'default') as any,
-        };
-
-        // if the apiName is md5, try to find the correct apiName in the plugins
-        if (apiName.startsWith(PLUGIN_SCHEMA_API_MD5_PREFIX)) {
-          const md5 = apiName.replace(PLUGIN_SCHEMA_API_MD5_PREFIX, '');
-          const manifest = pluginSelectors.getPluginManifestById(identifier)(
-            useToolStore.getState(),
-          );
-
-          const api = manifest?.api.find((api) => Md5.hashStr(api.name).toString() === md5);
-          if (api) {
-            payload.apiName = api.name;
-          }
-        }
-
-        return payload;
-      })
-      .filter(Boolean) as ChatToolPayload[];
-  },
-
-  internal_updatePluginError: async (id, error) => {
-    const { refreshMessages } = get();
-
-    await messageService.updateMessage(id, { pluginError: error });
-    await refreshMessages();
-  },
-
   invokeBuiltinTool: async (id, payload) => {
-    const { internal_toggleChatLoading, internal_updateMessageContent } = get();
+    const { internal_togglePluginApiCalling, internal_updateMessageContent } = get();
     const params = JSON.parse(payload.arguments);
-    internal_toggleChatLoading(true, id, n('invokeBuiltinTool') as string);
+    internal_togglePluginApiCalling(true, id, n('invokeBuiltinTool') as string);
     let data;
     try {
       data = await useToolStore.getState().invokeBuiltinTool(payload.apiName, params);
     } catch (error) {
       console.log(error);
     }
-    internal_toggleChatLoading(false);
+    internal_togglePluginApiCalling(false);
 
     if (!data) return;
 
@@ -248,6 +147,8 @@ export const chatPlugin: StateCreator<
     const message = chatSelectors.getMessageById(id)(get());
     if (!message || message.role !== 'tool' || !message.plugin) return;
 
+    get().internal_updateMessageError(id, null);
+
     const payload: ChatToolPayload = { ...message.plugin, id: message.tool_call_id! };
 
     await get().internal_invokeDifferentTypePlugin(id, payload);
@@ -301,4 +202,116 @@ export const chatPlugin: StateCreator<
     await messageService.updateMessagePluginState(id, { [key]: value });
     await refreshMessages();
   },
+
+  internal_callPluginApi: async (id, payload) => {
+    const { internal_updateMessageContent, refreshMessages, internal_togglePluginApiCalling } =
+      get();
+    let data: string;
+
+    try {
+      const abortController = internal_togglePluginApiCalling(
+        true,
+        id,
+        n('fetchPlugin/start') as string,
+      );
+
+      const message = chatSelectors.getMessageById(id)(get());
+
+      const res = await chatService.runPluginApi(payload, {
+        signal: abortController?.signal,
+        trace: { observationId: message?.observationId, traceId: message?.traceId },
+      });
+      data = res.text;
+
+      // save traceId
+      if (res.traceId) {
+        await messageService.updateMessage(id, { traceId: res.traceId });
+      }
+    } catch (error) {
+      console.log(error);
+      const err = error as Error;
+
+      // ignore the aborted request error
+      if (!err.message.includes('The user aborted a request.')) {
+        await messageService.updateMessageError(id, error as any);
+        await refreshMessages();
+      }
+
+      data = '';
+    }
+
+    internal_togglePluginApiCalling(false, id, n('fetchPlugin/end') as string);
+    // 如果报错则结束了
+    if (!data) return;
+
+    await internal_updateMessageContent(id, data);
+
+    return data;
+  },
+
+  internal_invokeDifferentTypePlugin: async (id, payload) => {
+    switch (payload.type) {
+      case 'standalone': {
+        return await get().invokeStandaloneTypePlugin(id, payload);
+      }
+
+      case 'markdown': {
+        return await get().invokeMarkdownTypePlugin(id, payload);
+      }
+
+      case 'builtin': {
+        return await get().invokeBuiltinTool(id, payload);
+      }
+
+      default: {
+        return await get().invokeDefaultTypePlugin(id, payload);
+      }
+    }
+  },
+
+  internal_togglePluginApiCalling: (loading, id, action) => {
+    return get().internal_toggleLoadingArrays('pluginApiLoadingIds', loading, id, action);
+  },
+
+  internal_transformToolCalls: (toolCalls) => {
+    return toolCalls
+      .map((toolCall): ChatToolPayload | null => {
+        let payload: ChatToolPayload;
+
+        const [identifier, apiName, type] = toolCall.function.name.split(PLUGIN_SCHEMA_SEPARATOR);
+
+        if (!apiName) return null;
+
+        payload = {
+          apiName,
+          arguments: toolCall.function.arguments,
+          id: toolCall.id,
+          identifier,
+          type: (type ?? 'default') as any,
+        };
+
+        // if the apiName is md5, try to find the correct apiName in the plugins
+        if (apiName.startsWith(PLUGIN_SCHEMA_API_MD5_PREFIX)) {
+          const md5 = apiName.replace(PLUGIN_SCHEMA_API_MD5_PREFIX, '');
+          const manifest = pluginSelectors.getPluginManifestById(identifier)(
+            useToolStore.getState(),
+          );
+
+          const api = manifest?.api.find((api) => Md5.hashStr(api.name).toString() === md5);
+          if (api) {
+            payload.apiName = api.name;
+          }
+        }
+
+        return payload;
+      })
+      .filter(Boolean) as ChatToolPayload[];
+  },
+  internal_updatePluginError: async (id, error) => {
+    const { refreshMessages } = get();
+
+    get().internal_dispatchMessage({ id, type: 'updateMessages', value: { error } });
+    await messageService.updateMessage(id, { pluginError: error });
+    await refreshMessages();
+  },
 });

package/src/store/chat/utils/index.ts

@@ -0,0 +1,19 @@
+import { produce } from 'immer';
+
+export const preventLeavingFn = (e: BeforeUnloadEvent) => {
+  // set returnValue to trigger alert modal
+  // Note: No matter what value is set, the browser will display the standard text
+  e.returnValue = '你有正在生成中的请求，确定要离开吗？';
+};
+
+export const toggleBooleanList = (ids: string[], id: string, loading: boolean) => {
+  return produce(ids, (draft) => {
+    if (loading) {
+      draft.push(id);
+    } else {
+      const index = draft.indexOf(id);
+
+      if (index >= 0) draft.splice(index, 1);
+    }
+  });
+};

package/src/store/user/slices/settings/action.ts

@@ -8,7 +8,13 @@ import { userService } from '@/services/user';
 import type { UserStore } from '@/store/user';
 import { LocaleMode } from '@/types/locale';
 import { LobeAgentSettings } from '@/types/session';
-import { UserGeneralConfig, UserKeyVaults, UserSettings } from '@/types/user/settings';
+import {
+  SystemAgentItem,
+  UserGeneralConfig,
+  UserKeyVaults,
+  UserSettings,
+  UserSystemAgentConfigKey,
+} from '@/types/user/settings';
 import { switchLang } from '@/utils/client/switchLang';
 import { difference } from '@/utils/difference';
 import { merge } from '@/utils/merge';
@@ -25,7 +31,7 @@ export interface UserSettingsAction {
   updateGeneralConfig: (settings: Partial<UserGeneralConfig>) => Promise<void>;
   updateKeyVaults: (settings: Partial<UserKeyVaults>) => Promise<void>;
 
-  updateSystemAgent: (key: string, value: { model: string; provider: string }) => Promise<void>;
+  updateSystemAgent: (key: UserSystemAgentConfigKey, value: SystemAgentItem) => Promise<void>;
 }
 
 export const createSettingsSlice: StateCreator<

package/src/store/user/slices/settings/selectors/systemAgent.ts

@@ -9,8 +9,10 @@ const currentSystemAgent = (s: UserStore) =>
 
 const translation = (s: UserStore) => currentSystemAgent(s).translation;
 const topic = (s: UserStore) => currentSystemAgent(s).topic;
+const agentMeta = (s: UserStore) => currentSystemAgent(s).agentMeta;
 
 export const systemAgentSelectors = {
+  agentMeta,
   topic,
   translation,
 };

package/src/types/message/index.ts

@@ -32,7 +32,7 @@ export * from './tools';
 
 export interface ChatMessage extends BaseDataModel {
   content: string;
-  error?: ChatMessageError;
+  error?: ChatMessageError | null;
   // 扩展字段
   extra?: {
     fromModel?: string;

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

@@ -4,6 +4,9 @@ export interface SystemAgentItem {
 }
 
 export interface UserSystemAgentConfig {
+  agentMeta: SystemAgentItem;
   topic: SystemAgentItem;
   translation: SystemAgentItem;
 }
+
+export type UserSystemAgentConfigKey = keyof UserSystemAgentConfig;