@lobehub/lobehub 2.0.0-next.37 → 2.0.0-next.39

package/src/store/chat/slices/plugin/actions/publicApi.ts

@@ -0,0 +1,115 @@
+/* eslint-disable sort-keys-fix/sort-keys-fix, typescript-sort-keys/interface */
+import { ChatToolPayload, UIChatMessage } from '@lobechat/types';
+import { StateCreator } from 'zustand/vanilla';
+
+import { ChatStore } from '@/store/chat/store';
+
+import { displayMessageSelectors } from '../../message/selectors';
+
+/**
+ * Public API for plugin operations
+ * These methods are called by UI components or other business scenarios
+ */
+export interface PluginPublicApiAction {
+  /**
+   * Fill plugin message content and optionally trigger AI message
+   */
+  fillPluginMessageContent: (
+    id: string,
+    content: string,
+    triggerAiMessage?: boolean,
+  ) => Promise<void>;
+
+  /**
+   * Re-invoke a tool message (retry failed plugin call)
+   */
+  reInvokeToolMessage: (id: string) => Promise<void>;
+
+  /**
+   * Summary plugin content using AI
+   */
+  summaryPluginContent: (id: string) => Promise<void>;
+
+  /**
+   * Invoke different type of plugin based on payload type
+   * This is the unified entry point for plugin invocation
+   */
+  internal_invokeDifferentTypePlugin: (id: string, payload: ChatToolPayload) => Promise<any>;
+}
+
+export const pluginPublicApi: StateCreator<
+  ChatStore,
+  [['zustand/devtools', never]],
+  [],
+  PluginPublicApiAction
+> = (set, get) => ({
+  fillPluginMessageContent: async (id, content, triggerAiMessage) => {
+    const { triggerAIMessage, optimisticUpdateMessageContent } = get();
+
+    await optimisticUpdateMessageContent(id, content);
+
+    if (triggerAiMessage) await triggerAIMessage({ parentId: id });
+  },
+
+  reInvokeToolMessage: async (id) => {
+    const message = displayMessageSelectors.getDisplayMessageById(id)(get());
+    if (!message || message.role !== 'tool' || !message.plugin) return;
+
+    // if there is error content, then clear the error
+    if (!!message.pluginError) {
+      get().optimisticUpdateMessagePluginError(id, null);
+    }
+
+    const payload: ChatToolPayload = { ...message.plugin, id: message.tool_call_id! };
+
+    await get().internal_invokeDifferentTypePlugin(id, payload);
+  },
+
+  summaryPluginContent: async (id) => {
+    const message = displayMessageSelectors.getDisplayMessageById(id)(get());
+    if (!message || message.role !== 'tool') return;
+
+    await get().internal_execAgentRuntime({
+      messages: [
+        {
+          role: 'assistant',
+          content: '作为一名总结专家，请结合以上系统提示词，将以下内容进行总结：',
+        },
+        {
+          ...message,
+          content: message.content,
+          role: 'assistant',
+          name: undefined,
+          tool_call_id: undefined,
+        },
+      ] as UIChatMessage[],
+      parentMessageId: message.id,
+      parentMessageType: 'assistant',
+    });
+  },
+
+  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);
+      }
+
+      // @ts-ignore
+      case 'mcp': {
+        return await get().invokeMCPTypePlugin(id, payload);
+      }
+
+      default: {
+        return await get().invokeDefaultTypePlugin(id, payload);
+      }
+    }
+  },
+});

package/src/store/chat/slices/plugin/actions/workflow.ts

@@ -0,0 +1,121 @@
+/* eslint-disable sort-keys-fix/sort-keys-fix, typescript-sort-keys/interface */
+import { CreateMessageParams } from '@lobechat/types';
+import { StateCreator } from 'zustand/vanilla';
+
+import { messageService } from '@/services/message';
+import { ChatStore } from '@/store/chat/store';
+
+import { dbMessageSelectors, displayMessageSelectors } from '../../message/selectors';
+import { threadSelectors } from '../../thread/selectors';
+
+/**
+ * Workflow orchestration actions
+ * Handle complex business flows involving multiple steps
+ */
+export interface PluginWorkflowAction {
+  /**
+   * Create an assistant message by plugin result
+   */
+  createAssistantMessageByPlugin: (content: string, parentId: string) => Promise<void>;
+
+  /**
+   * Trigger AI message after tool calls
+   */
+  triggerAIMessage: (params: {
+    parentId?: string;
+    traceId?: string;
+    threadId?: string;
+    inPortalThread?: boolean;
+    inSearchWorkflow?: boolean;
+  }) => Promise<void>;
+
+  /**
+   * Trigger tool calls (V1 deprecated method)
+   * @deprecated
+   */
+  triggerToolCalls: (
+    id: string,
+    params?: { threadId?: string; inPortalThread?: boolean; inSearchWorkflow?: boolean },
+  ) => Promise<void>;
+}
+
+export const pluginWorkflow: StateCreator<
+  ChatStore,
+  [['zustand/devtools', never]],
+  [],
+  PluginWorkflowAction
+> = (set, get) => ({
+  createAssistantMessageByPlugin: async (content, parentId) => {
+    const newMessage: CreateMessageParams = {
+      content,
+      parentId,
+      role: 'assistant',
+      sessionId: get().activeId,
+      topicId: get().activeTopicId, // if there is activeTopicId，then add it to topicId
+    };
+
+    const result = await messageService.createMessage(newMessage);
+    get().replaceMessages(result.messages);
+  },
+
+  triggerAIMessage: async ({ parentId, traceId, threadId, inPortalThread, inSearchWorkflow }) => {
+    const { internal_execAgentRuntime } = get();
+
+    const chats = inPortalThread
+      ? threadSelectors.portalAIChatsWithHistoryConfig(get())
+      : displayMessageSelectors.mainAIChatsWithHistoryConfig(get());
+
+    await internal_execAgentRuntime({
+      messages: chats,
+      parentMessageId: parentId ?? chats.at(-1)!.id,
+      parentMessageType: 'user',
+      traceId,
+      threadId,
+      inPortalThread,
+      inSearchWorkflow,
+    });
+  },
+
+  triggerToolCalls: async (assistantId, { threadId, inPortalThread, inSearchWorkflow } = {}) => {
+    const message = displayMessageSelectors.getDisplayMessageById(assistantId)(get());
+    if (!message || !message.tools) return;
+
+    let shouldCreateMessage = false;
+    let latestToolId = '';
+    const messagePools = message.tools.map(async (payload) => {
+      const toolMessage: CreateMessageParams = {
+        content: '',
+        parentId: assistantId,
+        plugin: payload,
+        role: 'tool',
+        sessionId: get().activeId,
+        tool_call_id: payload.id,
+        threadId,
+        topicId: get().activeTopicId, // if there is activeTopicId，then add it to topicId
+        groupId: message.groupId, // Propagate groupId from parent message for group chat
+      };
+
+      const result = await get().optimisticCreateMessage(toolMessage);
+      if (!result) return;
+
+      // trigger the plugin call
+      const data = await get().internal_invokeDifferentTypePlugin(result.id, payload);
+
+      if (data && !['markdown', 'standalone'].includes(payload.type)) {
+        shouldCreateMessage = true;
+        latestToolId = result.id;
+      }
+    });
+
+    await Promise.all(messagePools);
+
+    await get().internal_toggleMessageInToolsCalling(false, assistantId);
+
+    // only default type tool calls should trigger AI message
+    if (!shouldCreateMessage) return;
+
+    const traceId = dbMessageSelectors.getTraceIdByDbMessageId(latestToolId)(get());
+
+    await get().triggerAIMessage({ traceId, threadId, inPortalThread, inSearchWorkflow });
+  },
+});

package/src/store/chat/store.ts

@@ -10,7 +10,7 @@ import { ChatBuiltinToolAction, chatToolSlice } from './slices/builtinTool/actio
 import { ChatPortalAction, chatPortalSlice } from './slices/portal/action';
 import { ChatTranslateAction, chatTranslate } from './slices/translate/action';
 import { ChatMessageAction, chatMessage } from './slices/message/actions';
-import { ChatPluginAction, chatPlugin } from './slices/plugin/action';
+import { ChatPluginAction, chatPlugin } from './slices/plugin/actions';
 import { ChatTopicAction, chatTopic } from './slices/topic/action';
 import { ChatAIChatAction, chatAiChat } from './slices/aiChat/actions';
 import { ChatTTSAction, chatTTS } from './slices/tts/action';

package/src/store/global/initialState.ts

@@ -50,6 +50,7 @@ export enum ProfileTabs {
   Profile = 'profile',
   Security = 'security',
   Stats = 'stats',
+  Usage = 'usage',
 }
 
 export interface SystemStatus {