@blueking/chat-helper 0.0.1-beta.3 → 0.0.1-beta.31

package/dist/event/ag-ui.ts.js

@@ -35,11 +35,9 @@
     }
     return obj;
 }
-import { transferMessageApi2Message } from '../http/transform/message.ts.js';
-import { MessageContentStatus, MessageContentType, MessageRole, MessageStatus } from '../message/index.ts.js';
-import { CustomEventType, EventType } from './type.ts.js';
-// 临时存储时间戳
-let tempTimestamp = 0;
+import { transferMessageApi2Message } from '../http/transform/index.ts.js';
+import { ActivityType, MessageRole, MessageStatus, MessageType } from '../message/index.ts.js';
+import { CustomEventName, EventType, FlowTaskState } from './type.ts.js';
 /**
  * AGUI 协议
  * @param message - 消息模块
@@ -47,85 +45,171 @@ let tempTimestamp = 0;
     /**
    * 处理活动增量更新事件
    * 使用 JSON Patch 格式增量更新活动数据
-   */ handleActivityDeltaEvent(_event) {
-    // 可以在这里实现 JSON Patch 逻辑
-    // console.log('ActivityDelta:', event);
-    }
+   */ handleActivityDeltaEvent(_event) {}
     /**
    * 处理活动快照事件
    * 记录活动的完整状态
-   */ handleActivitySnapshotEvent(_event) {
-    // 可以在这里存储活动快照
-    // console.log('ActivitySnapshot:', event);
-    }
+   */ handleActivitySnapshotEvent(_event) {}
     /**
    * 处理自定义事件
    */ handleCustomEvent(event) {
-        switch(event.value.type){
-            case CustomEventType.ReferenceDocument:
+        switch(event.name){
+            case CustomEventName.FlowAgentStart:
+                this.handleFlowAgentStartCustomEvent(event);
+                break;
+            case CustomEventName.FlowAgentResult:
+                this.handleFlowAgentResultCustomEvent(event);
+                break;
+            case CustomEventName.FlowAgentEnd:
+                this.handleFlowAgentEndCustomEvent(event);
+                break;
+            case CustomEventName.ReferenceDocument:
                 this.handleReferenceDocumentCustomEvent(event);
                 break;
+            case CustomEventName.KnowledgeRagStart:
+                this.handleKnowledgeRagStartCustomEvent(event);
+                break;
+            case CustomEventName.KnowledgeRagTextContent:
+                this.handleKnowledgeRagTextContentCustomEvent(event);
+                break;
+            case CustomEventName.KnowledgeRagResult:
+                this.handleKnowledgeRagResultCustomEvent(event);
+                break;
+            case CustomEventName.KnowledgeRagEnd:
+                this.handleKnowledgeRagEndCustomEvent(event);
+                break;
+            case CustomEventName.TempMessage:
+                this.handleTempMessageCustomEvent(event);
+                break;
             default:
                 break;
         }
     }
     /**
+   * 自定义事件 处理流程编排任务结束
+   */ handleFlowAgentEndCustomEvent(_event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Activity && message.activityType === ActivityType.FlowAgent) {
+            const value = _event.value;
+            message.content.task_outputs = value.task_outputs;
+            message.status = MessageStatus.Complete;
+        }
+    }
+    /**
+   * 自定义事件 处理流程编排任务结果（增量更新节点状态）
+   */ handleFlowAgentResultCustomEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Activity && message.activityType === ActivityType.FlowAgent) {
+            message.content = event.value;
+        }
+    }
+    /**
+   * 自定义事件 处理流程编排任务开始
+   */ handleFlowAgentStartCustomEvent(event) {
+        const value = event.value;
+        this.messageModule.plusMessage({
+            role: MessageRole.Activity,
+            activityType: ActivityType.FlowAgent,
+            content: {
+                nodes: {},
+                task_id: Number(value.task_id),
+                task_name: '',
+                task_outputs: [],
+                task_state: FlowTaskState.Created
+            },
+            status: MessageStatus.Streaming
+        });
+    }
+    /**
+   * 自定义事件 处理意图识别结束
+   */ handleKnowledgeRagEndCustomEvent(_event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Activity && message.activityType === ActivityType.KnowledgeRag) {
+            message.status = MessageStatus.Complete;
+        }
+    }
+    /**
+   * 自定义事件 处理意图识别结果
+   */ handleKnowledgeRagResultCustomEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Activity && message.activityType === ActivityType.KnowledgeRag) {
+            message.content.referenceDocument = event.value;
+        }
+    }
+    /**
+   * 自定义事件 处理意图识别开始
+   */ handleKnowledgeRagStartCustomEvent(_event) {
+        this.messageModule.plusMessage({
+            role: MessageRole.Activity,
+            activityType: ActivityType.KnowledgeRag,
+            content: {
+                content: '',
+                referenceDocument: []
+            },
+            status: MessageStatus.Streaming
+        });
+    }
+    /**
+   *  自定义事件 处理意图识别文本
+   */ handleKnowledgeRagTextContentCustomEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Activity && message.activityType === ActivityType.KnowledgeRag) {
+            const value = event.value;
+            message.content.content = value.cover ? value.chunk.content : message.content.content + value.chunk.content;
+        }
+    }
+    /**
    * 处理消息快照事件
    * 用于同步多端消息状态
    */ handleMessagesSnapshotEvent(event) {
-        this.messageModule.list.value = event.messages.map(transferMessageApi2Message);
+        if (event.messages && event.messages.length > 0) {
+            this.messageModule.list.value = event.messages.map(transferMessageApi2Message);
+        }
     }
     /**
    * 处理原始事件
    * 透传底层 AI 服务的原始事件
    */ handleRawEvent(_event) {
-    // console.log('Raw:', event);
+    // console.log('Raw:', _event);
     }
     /**
    * 自定义事件 处理参考文档事件
    */ handleReferenceDocumentCustomEvent(event) {
-        const message = this.messageModule.getCurrentLoadingMessage();
-        if (message && message.role === MessageRole.Assistant) {
-            const referenceDocumentMessageContent = {
-                type: MessageContentType.ReferenceDocument,
-                data: event.value.data,
-                status: MessageContentStatus.Streaming
-            };
-            message.content.push(referenceDocumentMessageContent);
-        }
+        // 可以在这里处理参考文档事件
+        this.messageModule.plusMessage({
+            role: MessageRole.Activity,
+            activityType: ActivityType.ReferenceDocument,
+            content: event.value,
+            status: MessageStatus.Complete
+        });
     }
     /**
    * 处理运行错误事件
    */ handleRunErrorEvent(event) {
+        // 先把最后一个loading消息标记为complete
         const message = this.messageModule.getCurrentLoadingMessage();
         if (message) {
-            message.status = MessageStatus.Error;
-            message.content = [
-                {
-                    type: MessageContentType.Text,
-                    data: event.message,
-                    status: MessageContentStatus.Error
-                }
-            ];
+            message.status = MessageStatus.Complete;
         }
+        // 然后创建一个错误消息
+        this.messageModule.plusMessage({
+            role: MessageRole.Assistant,
+            content: event.message,
+            status: MessageStatus.Error
+        });
     }
     /**
    * 处理运行完成事件
-   */ handleRunFinishedEvent(event) {
-        const message = this.messageModule.getMessageById(event.runId);
+   */ handleRunFinishedEvent(_event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
         if (message) {
             message.status = MessageStatus.Complete;
         }
     }
     /**
    * 处理运行开始事件
-   */ handleRunStartedEvent(event) {
-        this.messageModule.plusMessage({
-            id: event.runId,
-            role: MessageRole.Assistant,
-            content: [],
-            status: MessageStatus.Streaming
-        });
+   */ handleRunStartedEvent(_event) {
+    // 运行开始，可以在这里初始化相关状态
     }
     /**
    * 处理状态增量更新事件
@@ -141,190 +225,180 @@ let tempTimestamp = 0;
     /**
    * 处理步骤完成事件
    */ handleStepFinishedEvent(_event) {
-    // console.log('StepFinished:', event);
+    // console.log('StepFinished:', _event);
     }
     /**
    * 处理步骤开始事件
    */ handleStepStartedEvent(_event) {
-    // console.log('StepStarted:', event);
+    // console.log('StepStarted:', _event);
     }
     /**
-   * 处理文本消息块事件（流式输出）
+   * 自定义事件 处理临时消息，转换为 AI 文本消息
+   */ handleTempMessageCustomEvent(event) {
+        const value = event.value;
+        this.messageModule.plusMessage({
+            role: MessageRole.Assistant,
+            content: value.message,
+            status: value.status
+        });
+    }
+    /**
+   * 处理文本消息块事件
    */ handleTextMessageChunkEvent(event) {
         const message = this.messageModule.getCurrentLoadingMessage();
         if (message) {
-            const messageContent = message.content.find((item)=>item.type === MessageContentType.Text && item.id === event.messageId);
-            if (messageContent) {
-                // 增量
-                messageContent.data += event.delta;
-            } else {
-                // 新增
-                message.content.push({
-                    id: event.messageId,
-                    status: MessageContentStatus.Complete,
-                    type: MessageContentType.Text,
-                    data: event.delta
-                });
-            }
+            message.content += event.delta || '';
+        } else {
+            this.messageModule.plusMessage({
+                role: event.role,
+                content: event.delta || '',
+                status: MessageStatus.Complete,
+                messageId: event.messageId
+            });
         }
     }
     /**
-   * 处理文本消息内容事件
+   * 处理文本消息内容事件（流式输出）
    */ handleTextMessageContentEvent(event) {
-        const message = this.messageModule.getCurrentLoadingMessage();
+        const message = this.messageModule.getMessageByMessageId(event.messageId);
         if (message) {
-            const messageContent = message.content.find((item)=>item.type === MessageContentType.Text && item.id === event.messageId);
-            if (messageContent) {
-                messageContent.data += event.delta;
-            }
+            message.content += event.delta;
         }
     }
     /**
    * 处理文本消息结束事件
    */ handleTextMessageEndEvent(event) {
-        const message = this.messageModule.getCurrentLoadingMessage();
+        const message = this.messageModule.getMessageByMessageId(event.messageId);
         if (message) {
-            const messageContent = message.content.find((item)=>item.type === MessageContentType.Text && item.id === event.messageId);
-            if (messageContent) {
-                messageContent.status = MessageContentStatus.Complete;
-            }
+            message.status = MessageStatus.Complete;
         }
     }
     /**
    * 处理文本消息开始事件
    */ handleTextMessageStartEvent(event) {
-        const message = this.messageModule.getCurrentLoadingMessage();
-        if (message) {
-            message.content.push({
-                id: event.messageId,
-                status: MessageContentStatus.Streaming,
-                type: MessageContentType.Text,
-                data: ''
-            });
-        }
+        this.messageModule.plusMessage({
+            role: event.role,
+            content: '',
+            status: MessageStatus.Streaming,
+            messageId: event.messageId
+        });
     }
     /**
    * 处理思考结束事件
    */ handleThinkingEndEvent(event) {
         const message = this.messageModule.getCurrentLoadingMessage();
-        if (message && message.role === MessageRole.Assistant) {
-            const lastThinkingMessageContent = message.content.findLast((item)=>item.type === MessageContentType.Thinking);
-            if (lastThinkingMessageContent) {
-                lastThinkingMessageContent.status = MessageContentStatus.Complete;
-                lastThinkingMessageContent.data.duration = event.timestamp - tempTimestamp;
-            }
+        if (message && message.role === MessageRole.Reasoning) {
+            message.duration = event.duration;
+            message.status = MessageStatus.Complete;
         }
     }
     /**
    * 处理思考开始事件
-   */ handleThinkingStartEvent(event) {
-        const message = this.messageModule.getCurrentLoadingMessage();
-        if (message && message.role === MessageRole.Assistant) {
-            // 临时存储时间戳
-            tempTimestamp = event.timestamp;
-            // 创建思考消息内容
-            const thinkingMessageContent = {
-                status: MessageContentStatus.Streaming,
-                type: MessageContentType.Thinking,
-                data: {
-                    text: '',
-                    title: event.title
-                }
-            };
-            message.content.push(thinkingMessageContent);
-        }
+   */ handleThinkingStartEvent(_event) {
+        this.messageModule.plusMessage({
+            role: MessageRole.Reasoning,
+            content: [],
+            status: MessageStatus.Streaming
+        });
     }
     /**
    * 处理思考文本消息内容事件
    */ handleThinkingTextMessageContentEvent(event) {
         const message = this.messageModule.getCurrentLoadingMessage();
-        if (message && message.role === MessageRole.Assistant) {
-            const lastThinkingMessageContent = message.content.findLast((item)=>item.type === MessageContentType.Thinking);
-            if (lastThinkingMessageContent) {
-                lastThinkingMessageContent.data.text += event.delta;
-            }
+        if (message && message.role === MessageRole.Reasoning) {
+            message.content[message.content.length - 1] += event.delta;
         }
     }
     /**
    * 处理思考文本消息结束事件
    */ handleThinkingTextMessageEndEvent(_event) {
-    // console.log(event);
+    // console.log(_event);
     }
     /**
    * 处理思考文本消息开始事件
    */ handleThinkingTextMessageStartEvent(_event) {
-    // console.log(event);
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Reasoning) {
+            message.content.push('');
+        }
     }
     /**
    * 处理工具调用参数事件
    */ handleToolCallArgsEvent(event) {
-        const message = this.messageModule.getCurrentLoadingMessage();
-        if (message && message.role === MessageRole.Assistant) {
-            const toolCallMessageContent = message.content.find((item)=>item.type === MessageContentType.ToolCall && item.data.toolCallId === event.toolCallId);
-            if ((toolCallMessageContent === null || toolCallMessageContent === void 0 ? void 0 : toolCallMessageContent.type) === MessageContentType.ToolCall) {
-                toolCallMessageContent.data.args += event.delta;
+        this.messageModule.list.value.forEach((item)=>{
+            if (item.role === MessageRole.Assistant && item.toolCalls) {
+                item.toolCalls.forEach((toolCall)=>{
+                    if (toolCall.id === event.toolCallId) {
+                        toolCall['function'].arguments += event.delta;
+                    }
+                });
             }
-        }
+        });
     }
     /**
    * 处理工具调用块事件
-   */ handleToolCallChunkEvent(event) {
-        const message = this.messageModule.getCurrentLoadingMessage();
-        if (message && message.role === MessageRole.Assistant) {
-            const toolCallMessageContent = message.content.find((item)=>item.type === MessageContentType.ToolCall && item.data.toolCallId === event.toolCallId);
-            if ((toolCallMessageContent === null || toolCallMessageContent === void 0 ? void 0 : toolCallMessageContent.type) === MessageContentType.ToolCall) {
-                toolCallMessageContent.data.chunk += event.delta;
-            }
-        }
-    }
+   */ handleToolCallChunkEvent(_event) {}
     /**
    * 处理工具调用结束事件
    */ handleToolCallEndEvent(event) {
-        const message = this.messageModule.getCurrentLoadingMessage();
-        if (message && message.role === MessageRole.Assistant) {
-            const toolCallMessageContent = message.content.find((item)=>item.type === MessageContentType.ToolCall && item.data.toolCallId === event.toolCallId);
-            if ((toolCallMessageContent === null || toolCallMessageContent === void 0 ? void 0 : toolCallMessageContent.type) === MessageContentType.ToolCall) {
-                toolCallMessageContent.status = MessageContentStatus.Complete;
+        this.messageModule.list.value.forEach((item)=>{
+            if (item.role === MessageRole.Assistant && item.toolCalls) {
+                if (item.toolCalls.some((toolCall)=>toolCall.id === event.toolCallId)) {
+                    item.status = MessageStatus.Complete;
+                }
             }
-        }
+        });
     }
     /**
    * 处理工具调用结果事件
    */ handleToolCallResultEvent(event) {
-        const message = this.messageModule.getCurrentLoadingMessage();
-        if (message && message.role === MessageRole.Assistant) {
-            const toolCallMessageContent = message.content.find((item)=>item.type === MessageContentType.ToolCall && item.data.toolCallId === event.toolCallId);
-            if ((toolCallMessageContent === null || toolCallMessageContent === void 0 ? void 0 : toolCallMessageContent.type) === MessageContentType.ToolCall) {
-                toolCallMessageContent.data.result += event.content;
-            }
-        }
+        this.messageModule.plusMessage({
+            role: MessageRole.Tool,
+            content: event.content,
+            duration: event.duration,
+            status: MessageStatus.Complete,
+            toolCallId: event.toolCallId,
+            messageId: event.messageId
+        });
     }
     /**
    * 处理工具调用开始事件
    */ handleToolCallStartEvent(event) {
-        const message = this.messageModule.getCurrentLoadingMessage();
-        if (message && message.role === MessageRole.Assistant) {
-            const toolCallMessageContent = {
-                status: MessageContentStatus.Streaming,
-                type: MessageContentType.ToolCall,
-                data: {
-                    toolCallId: event.toolCallId,
-                    toolCallName: event.toolCallName,
-                    parentMessageId: event.parentMessageId
+        this.messageModule.plusMessage({
+            role: MessageRole.Assistant,
+            content: '',
+            status: MessageStatus.Streaming,
+            toolCalls: [
+                {
+                    id: event.toolCallId,
+                    type: MessageType.Function,
+                    function: {
+                        mcpName: event.mcpName,
+                        name: event.toolCallName,
+                        arguments: '',
+                        description: event.description
+                    }
                 }
-            };
-            message.content.push(toolCallMessageContent);
-        }
+            ]
+        });
     }
     injectMessageModule(message) {
         this.messageModule = message;
     }
     onDone() {
-        var _this_onDoneCallback, _this;
+        var // 回调给上层
+        _this_onDoneCallback, _this;
         (_this_onDoneCallback = (_this = this).onDoneCallback) === null || _this_onDoneCallback === void 0 ? void 0 : _this_onDoneCallback.call(_this);
     }
     onError(error) {
-        var _this_onErrorCallback, _this;
+        var // 回调给上层
+        _this_onErrorCallback, _this;
+        // 创建一个错误消息
+        this.messageModule.plusMessage({
+            role: MessageRole.Assistant,
+            content: error.message,
+            status: MessageStatus.Error
+        });
         (_this_onErrorCallback = (_this = this).onErrorCallback) === null || _this_onErrorCallback === void 0 ? void 0 : _this_onErrorCallback.call(_this, error);
     }
     onMessage(message) {