@blueking/chat-helper 0.0.12-beta.1 → 0.0.12-beta.11

package/dist/agent/use-agent.ts.js

@@ -80,9 +80,38 @@ function _object_spread(target) {
     return target;
 }
 import { ref } from 'vue';
-import { AGUIProtocol } from '../event/index.ts.js';
-import { MessageRole, MessageStatus } from '../message/index.ts.js';
+import { AGUIProtocol, ApprovalInterruptTicketStatus, EventType, ResumeStatus, RunFinishedOutcomeType } from '../event/index.ts.js';
+import { MessageRole, MessageStatus, UserOperation } from '../message/index.ts.js';
 import { SessionStatus } from '../session/type.ts.js';
+/** SSE 静默重连最大次数（退避总时长约 23s，落在后端 orphan grace ~30s 内） */ const MAX_RECONNECT_ATTEMPTS = 5;
+const RECONNECT_BASE_DELAY_MS = 1000;
+const RECONNECT_MAX_DELAY_MS = 8000;
+const getReconnectDelayMs = (attempt)=>Math.min(RECONNECT_BASE_DELAY_MS * Math.pow(2, attempt - 1), RECONNECT_MAX_DELAY_MS);
+/**
+ * 判断流式错误是否可静默重连。
+ * 不重试：Abort、401/403 等业务 4xx（408/429 除外）；可重试：网络错误、5xx、无状态码的中断。
+ */ const isRecoverableStreamError = (error)=>{
+    var _requestError_response;
+    if (error.name === 'AbortError') {
+        return false;
+    }
+    const requestError = error;
+    const statusFromResponse = (_requestError_response = requestError.response) === null || _requestError_response === void 0 ? void 0 : _requestError_response.status;
+    const statusFromMessage = error.message.match(/status code (\d+)/i);
+    const status = statusFromResponse !== null && statusFromResponse !== void 0 ? statusFromResponse : statusFromMessage ? Number(statusFromMessage[1]) : undefined;
+    if (status !== undefined) {
+        if (status === 408 || status === 429) {
+            return true;
+        }
+        if (status >= 400 && status < 500) {
+            return false;
+        }
+        if (status >= 500) {
+            return true;
+        }
+    }
+    return true;
+};
 /**
  * Agent 模块
  * @param options - 配置选项
@@ -92,7 +121,15 @@ import { SessionStatus } from '../session/type.ts.js';
     const isInfoLoading = ref(false);
     const isChatting = ref(false);
     let usedProtocol = protocol || new AGUIProtocol();
-    let abortController = null;
+    let chatAbortController = null;
+    let resumeAbortController = null;
+    let longPollTimer = null;
+    /** 用户主动中止（abortChat），禁止自动重连 */ let manualAbort = false;
+    /** 当前轮次已静默重连次数 */ let reconnectAttempt = 0;
+    /** 本轮流是否已收到 RUN_FINISHED */ let runFinished = false;
+    let reconnectTimer = null;
+    let reconnectWaitResolve = null;
+    let activeStreamContext = null;
     const getAgentInfo = ()=>{
         var _mediator_http;
         isInfoLoading.value = true;
@@ -121,27 +158,221 @@ import { SessionStatus } from '../session/type.ts.js';
             });
         }
     };
+    const userOperationStreamRequest = (sessionCode, operation, payload, config)=>{
+        var _mediator_http;
+        return (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.message.userOperation(sessionCode, operation, payload, config).then(()=>{
+            if (operation !== UserOperation.ApprovalCancel) {
+                streamRequest({
+                    sessionCode,
+                    config
+                });
+            } else {
+                clearLongPollTimer();
+                pollResumeSession(sessionCode);
+            }
+        });
+    };
+    const clearReconnectTimer = (continued = false)=>{
+        if (reconnectTimer) {
+            clearTimeout(reconnectTimer);
+            reconnectTimer = null;
+        }
+        if (reconnectWaitResolve) {
+            const resolve = reconnectWaitResolve;
+            reconnectWaitResolve = null;
+            resolve(continued);
+        }
+    };
+    const waitForReconnectDelay = (ms)=>new Promise((resolve)=>{
+            reconnectWaitResolve = resolve;
+            reconnectTimer = setTimeout(()=>{
+                reconnectTimer = null;
+                reconnectWaitResolve = null;
+                resolve(true);
+            }, ms);
+        });
+    const resetStreamReconnectState = ()=>{
+        manualAbort = false;
+        reconnectAttempt = 0;
+        runFinished = false;
+        clearReconnectTimer(false);
+    };
+    /** 裁掉尾部尚未完成的非用户消息，避免 DLQ 重放时重复气泡 */ const pruneTrailingIncompleteMessages = ()=>{
+        var _mediator_message;
+        const list = (_mediator_message = mediator.message) === null || _mediator_message === void 0 ? void 0 : _mediator_message.list.value;
+        if (!(list === null || list === void 0 ? void 0 : list.length)) {
+            return;
+        }
+        while(list.length > 0){
+            const last = list[list.length - 1];
+            if (last.role === MessageRole.User) {
+                break;
+            }
+            if (last.status === MessageStatus.Streaming || last.status === MessageStatus.Pending) {
+                list.pop();
+                continue;
+            }
+            break;
+        }
+    };
+    const isSameActiveSession = (sessionCode)=>{
+        var _mediator_session_current_value, _mediator_session_current, _mediator_session;
+        return sessionCode === ((_mediator_session = mediator.session) === null || _mediator_session === void 0 ? void 0 : (_mediator_session_current = _mediator_session.current) === null || _mediator_session_current === void 0 ? void 0 : (_mediator_session_current_value = _mediator_session_current.value) === null || _mediator_session_current_value === void 0 ? void 0 : _mediator_session_current_value.sessionCode);
+    };
+    /**
+   * 尝试静默重连。
+   * @returns reconnected | finished（服务端已结束）| failed（需对外报错或中止）
+   */ const attemptSilentReconnect = function() {
+        var _ref = _async_to_generator(function*(sessionCode) {
+            var _mediator_session_current_value, _mediator_session, _mediator_session_current_value1, _mediator_session1, _mediator_message_list_value_at, _mediator_message;
+            if (manualAbort || !isSameActiveSession(sessionCode)) {
+                return 'failed';
+            }
+            try {
+                var _mediator_session2;
+                yield (_mediator_session2 = mediator.session) === null || _mediator_session2 === void 0 ? void 0 : _mediator_session2.getSession(sessionCode);
+            } catch (e) {
+            // 状态刷新失败时沿用本地 session.status
+            }
+            if (manualAbort || !isSameActiveSession(sessionCode)) {
+                return 'failed';
+            }
+            if (((_mediator_session = mediator.session) === null || _mediator_session === void 0 ? void 0 : (_mediator_session_current_value = _mediator_session.current.value) === null || _mediator_session_current_value === void 0 ? void 0 : _mediator_session_current_value.status) !== SessionStatus.Running) {
+                return 'finished';
+            }
+            if (reconnectAttempt >= MAX_RECONNECT_ATTEMPTS) {
+                return 'failed';
+            }
+            reconnectAttempt += 1;
+            // 静默重连期间保持「生成中」态
+            isChatting.value = true;
+            const delayMs = getReconnectDelayMs(reconnectAttempt);
+            const waited = yield waitForReconnectDelay(delayMs);
+            if (!waited || manualAbort || !isSameActiveSession(sessionCode)) {
+                return 'failed';
+            }
+            if (((_mediator_session1 = mediator.session) === null || _mediator_session1 === void 0 ? void 0 : (_mediator_session_current_value1 = _mediator_session1.current.value) === null || _mediator_session_current_value1 === void 0 ? void 0 : _mediator_session_current_value1.status) !== SessionStatus.Running) {
+                return 'finished';
+            }
+            pruneTrailingIncompleteMessages();
+            const lastMessageId = (_mediator_message = mediator.message) === null || _mediator_message === void 0 ? void 0 : (_mediator_message_list_value_at = _mediator_message.list.value.at(-1)) === null || _mediator_message_list_value_at === void 0 ? void 0 : _mediator_message_list_value_at.id;
+            const ctx = activeStreamContext;
+            streamRequest({
+                sessionCode,
+                url: ctx === null || ctx === void 0 ? void 0 : ctx.url,
+                config: ctx === null || ctx === void 0 ? void 0 : ctx.config,
+                lastMessageId: lastMessageId !== undefined ? String(lastMessageId) : undefined,
+                isReconnect: true
+            });
+            return 'reconnected';
+        });
+        return function attemptSilentReconnect(sessionCode) {
+            return _ref.apply(this, arguments);
+        };
+    }();
     const streamRequest = function() {
-        var _ref = _async_to_generator(function*(sessionCode, url, config) {
-            var // 发起聊天
-            _mediator_http;
+        var _ref = _async_to_generator(function*({ sessionCode, url, config, resume, input, lastMessageId, isReconnect = false }) {
+            var _mediator_http;
+            if (!isReconnect) {
+                resetStreamReconnectState();
+            }
+            activeStreamContext = {
+                sessionCode,
+                url,
+                config
+            };
             // ag-ui 协议需要注入消息模块
             if (usedProtocol instanceof AGUIProtocol) {
                 usedProtocol.injectMessageModule(mediator.message);
             }
-            // 事件代理
-            const onDone = ()=>{
+            if (input) {
+                var // 列表新增一个用户消息
+                _mediator_message;
+                (_mediator_message = mediator.message) === null || _mediator_message === void 0 ? void 0 : _mediator_message.list.value.push({
+                    role: MessageRole.User,
+                    content: input,
+                    status: MessageStatus.Complete,
+                    sessionCode
+                });
+            }
+            const finishSuccessfully = ()=>{
                 var _usedProtocol_onDone;
                 isChatting.value = false;
                 (_usedProtocol_onDone = usedProtocol.onDone) === null || _usedProtocol_onDone === void 0 ? void 0 : _usedProtocol_onDone.call(usedProtocol);
+                if (input) {
+                    var // 刷新列表，获取前端 mock message 的 id，并更新到列表中
+                    _mediator_http_message, _mediator_http;
+                    (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : (_mediator_http_message = _mediator_http.message) === null || _mediator_http_message === void 0 ? void 0 : _mediator_http_message.getMessages(sessionCode).then((res)=>{
+                        var _mediator_message;
+                        const lastUserMessage = (_mediator_message = mediator.message) === null || _mediator_message === void 0 ? void 0 : _mediator_message.list.value.findLast((item)=>item.role === MessageRole.User);
+                        const lastApiUserMessage = res.findLast((item)=>item.role === MessageRole.User);
+                        if (lastUserMessage && lastApiUserMessage) {
+                            lastUserMessage.id = lastApiUserMessage.id;
+                        }
+                    });
+                }
+                // 轮询接口，判断是否可以继续聊天
+                pollResumeSession(sessionCode);
             };
-            const onError = (error)=>{
+            const failWithError = (error)=>{
                 var _usedProtocol_onError;
                 isChatting.value = false;
                 (_usedProtocol_onError = usedProtocol.onError) === null || _usedProtocol_onError === void 0 ? void 0 : _usedProtocol_onError.call(usedProtocol, error);
             };
+            // 事件代理
+            const onDone = ()=>{
+                // 用户 abort：FetchClient 将 AbortError 转为 onDone，不重连
+                if (manualAbort || runFinished) {
+                    finishSuccessfully();
+                    return;
+                }
+                // 连接被干净掐断且未收到 RUN_FINISHED：尝试静默重连
+                void attemptSilentReconnect(sessionCode).then((result)=>{
+                    if (result === 'reconnected') {
+                        // 保持 isChatting=true，不展示错误
+                        return;
+                    }
+                    if (result === 'finished') {
+                        finishSuccessfully();
+                        return;
+                    }
+                    // abort / 切会话导致的失败：静默结束，不展示错误气泡
+                    if (manualAbort || !isSameActiveSession(sessionCode)) {
+                        isChatting.value = false;
+                        return;
+                    }
+                    failWithError(new Error('Connection lost, please try again'));
+                });
+            };
+            const onError = (error)=>{
+                if (manualAbort) {
+                    return;
+                }
+                if (!isRecoverableStreamError(error)) {
+                    failWithError(error);
+                    return;
+                }
+                void attemptSilentReconnect(sessionCode).then((result)=>{
+                    if (result === 'reconnected') {
+                        return;
+                    }
+                    if (result === 'finished') {
+                        finishSuccessfully();
+                        return;
+                    }
+                    if (manualAbort || !isSameActiveSession(sessionCode)) {
+                        isChatting.value = false;
+                        return;
+                    }
+                    failWithError(error);
+                });
+            };
             const onMessage = (event)=>{
                 var _usedProtocol_onMessage;
+                const typedEvent = event;
+                if ((typedEvent === null || typedEvent === void 0 ? void 0 : typedEvent.type) === EventType.RunFinished) {
+                    runFinished = true;
+                }
                 (_usedProtocol_onMessage = usedProtocol.onMessage) === null || _usedProtocol_onMessage === void 0 ? void 0 : _usedProtocol_onMessage.call(usedProtocol, event);
             };
             const onStart = ()=>{
@@ -150,24 +381,31 @@ import { SessionStatus } from '../session/type.ts.js';
                 (_usedProtocol_onStart = usedProtocol.onStart) === null || _usedProtocol_onStart === void 0 ? void 0 : _usedProtocol_onStart.call(usedProtocol);
             };
             // 创建 AbortController
-            abortController = new AbortController();
-            (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.fetchClient.streamRequest(_object_spread({
+            chatAbortController = new AbortController();
+            // 发起聊天
+            void ((_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.fetchClient.streamRequest(_object_spread({
                 url: url || 'chat_completion/',
                 method: 'POST',
                 data: {
                     session_code: sessionCode,
+                    input,
                     execute_kwargs: {
-                        stream: true
+                        stream: true,
+                        persist_input: !!input,
+                        last_message_id: lastMessageId,
+                        resume
                     }
                 },
-                controller: abortController,
+                controller: chatAbortController,
                 onDone,
                 onError,
                 onMessage,
                 onStart
-            }, config));
+            }, config)).catch(()=>{
+            // 非 abort 错误已通过 onError 回调处理；abort 为正常结束
+            }));
         });
-        return function streamRequest(sessionCode, url, config) {
+        return function streamRequest(_) {
             return _ref.apply(this, arguments);
         };
     }();
@@ -191,7 +429,11 @@ import { SessionStatus } from '../session/type.ts.js';
                 property
             }));
             // 发起聊天
-            streamRequest(sessionCode, url, config);
+            streamRequest({
+                sessionCode,
+                url,
+                config
+            });
         });
         return function chat(userInput, sessionCode, url, config, property) {
             return _ref.apply(this, arguments);
@@ -206,15 +448,78 @@ import { SessionStatus } from '../session/type.ts.js';
    */ const resumeStreamingChat = (sessionCode, url, config)=>{
         var _mediator_session_current_value, _mediator_session;
         if (((_mediator_session = mediator.session) === null || _mediator_session === void 0 ? void 0 : (_mediator_session_current_value = _mediator_session.current.value) === null || _mediator_session_current_value === void 0 ? void 0 : _mediator_session_current_value.status) === SessionStatus.Running) {
-            streamRequest(sessionCode, url, config);
+            var _mediator_message_list_value_at, _mediator_message;
+            const lastMessageId = (_mediator_message = mediator.message) === null || _mediator_message === void 0 ? void 0 : (_mediator_message_list_value_at = _mediator_message.list.value.at(-1)) === null || _mediator_message_list_value_at === void 0 ? void 0 : _mediator_message_list_value_at.id;
+            streamRequest({
+                sessionCode,
+                url,
+                config,
+                lastMessageId
+            });
         }
     };
     /**
+   * 轮询接口，判断是否可以继续聊天
+   * @param sessionCode - 会话编码
+   * @returns 是否可以继续聊天
+   */ const pollResumeSession = (sessionCode)=>{
+        var _mediator_message, _lastMessage_content_outcome, _lastMessage_content;
+        const lastMessage = (_mediator_message = mediator.message) === null || _mediator_message === void 0 ? void 0 : _mediator_message.list.value.at(-1);
+        const pendingApprovalInterrupt = (lastMessage === null || lastMessage === void 0 ? void 0 : (_lastMessage_content = lastMessage.content) === null || _lastMessage_content === void 0 ? void 0 : (_lastMessage_content_outcome = _lastMessage_content.outcome) === null || _lastMessage_content_outcome === void 0 ? void 0 : _lastMessage_content_outcome.type) === RunFinishedOutcomeType.Interrupt ? lastMessage.content.outcome.interrupts.find((interrupt)=>{
+            var _interrupt_metadata_ticket, _interrupt_metadata;
+            return [
+                ApprovalInterruptTicketStatus.Pending,
+                ApprovalInterruptTicketStatus.Draft
+            ].includes((_interrupt_metadata = interrupt.metadata) === null || _interrupt_metadata === void 0 ? void 0 : (_interrupt_metadata_ticket = _interrupt_metadata.ticket) === null || _interrupt_metadata_ticket === void 0 ? void 0 : _interrupt_metadata_ticket.status);
+        }) : undefined;
+        const getIsTicketLoading = ()=>{
+            const isInterruptMessage = (lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.role) === MessageRole.Interrupt;
+            return isInterruptMessage && !!pendingApprovalInterrupt;
+        };
+        if (getIsTicketLoading()) {
+            var // 轮询接口，判断是否可以继续聊天
+            _mediator_http;
+            // 清除轮询定时器和中断轮询控制器
+            clearLongPollTimer();
+            resumeAbortController = new AbortController();
+            (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.session.isResumeSession(sessionCode, {
+                controller: resumeAbortController
+            }).then((res)=>{
+                if (res) {
+                    // 可以继续聊天，重新发起聊天（携带 execute_kwargs.resume 通知后端恢复中断）
+                    streamRequest({
+                        sessionCode,
+                        resume: {
+                            interruptId: pendingApprovalInterrupt.id,
+                            status: ResumeStatus.Resolved
+                        }
+                    });
+                } else {
+                    longPollTimer = setTimeout(()=>{
+                        var _mediator_session_current_value, _mediator_session_current, _mediator_session;
+                        // 如果会话不匹配，则不继续轮询
+                        if (sessionCode !== ((_mediator_session = mediator.session) === null || _mediator_session === void 0 ? void 0 : (_mediator_session_current = _mediator_session.current) === null || _mediator_session_current === void 0 ? void 0 : (_mediator_session_current_value = _mediator_session_current.value) === null || _mediator_session_current_value === void 0 ? void 0 : _mediator_session_current_value.sessionCode)) return;
+                        pollResumeSession(sessionCode);
+                    }, 10000);
+                }
+            });
+        }
+    };
+    const clearLongPollTimer = ()=>{
+        var _resumeAbortController_abort;
+        clearTimeout(longPollTimer);
+        longPollTimer = null;
+        resumeAbortController === null || resumeAbortController === void 0 ? void 0 : (_resumeAbortController_abort = resumeAbortController.abort) === null || _resumeAbortController_abort === void 0 ? void 0 : _resumeAbortController_abort.call(resumeAbortController);
+        resumeAbortController = null;
+    };
+    /**
    * 中止聊天（纯前端中止，后端继续处理）
    */ const abortChat = ()=>{
-        var _abortController_abort;
-        abortController === null || abortController === void 0 ? void 0 : (_abortController_abort = abortController.abort) === null || _abortController_abort === void 0 ? void 0 : _abortController_abort.call(abortController);
-        abortController = null;
+        var _chatAbortController_abort;
+        manualAbort = true;
+        clearReconnectTimer(false);
+        chatAbortController === null || chatAbortController === void 0 ? void 0 : (_chatAbortController_abort = chatAbortController.abort) === null || _chatAbortController_abort === void 0 ? void 0 : _chatAbortController_abort.call(chatAbortController);
+        chatAbortController = null;
     };
     /**
    * 停止会话，后端中止
@@ -269,7 +574,11 @@ import { SessionStatus } from '../session/type.ts.js';
                 property: originalProperty
             }));
             // 6. 立即发起流式请求（不等待删除和创建的 API 完成）
-            streamRequest(sessionCode, url, config);
+            streamRequest({
+                sessionCode,
+                url,
+                config
+            });
             // 7. 在后台等待 API 完成，处理可能的错误
             Promise.all([
                 deletePromise,
@@ -283,10 +592,15 @@ import { SessionStatus } from '../session/type.ts.js';
         };
     }();
     const reset = (protocol)=>{
-        // 重置状态
+        abortChat();
+        // 重置状态（保留 manualAbort=true，避免 abort 异步 onDone 误触发重连；下次 streamRequest 会复位）
         usedProtocol = protocol || new AGUIProtocol();
         info.value = null;
         isInfoLoading.value = false;
+        isChatting.value = false;
+        activeStreamContext = null;
+        reconnectAttempt = 0;
+        runFinished = false;
     };
     return {
         info,
@@ -299,6 +613,10 @@ import { SessionStatus } from '../session/type.ts.js';
         abortChat,
         stopChat,
         getAgentInfo,
-        reset
+        reset,
+        pollResumeSession,
+        clearLongPollTimer,
+        userOperationStreamRequest,
+        streamRequest
     };
 };

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

@@ -26,7 +26,7 @@ export declare class AGUIProtocol implements ISSEProtocol {
      * 处理活动快照事件
      * 记录活动的完整状态
      */
-    handleActivitySnapshotEvent(_event: IActivitySnapshotEvent): void;
+    handleActivitySnapshotEvent(event: IActivitySnapshotEvent): void;
     /**
      * 处理自定义事件
      */
@@ -35,6 +35,10 @@ export declare class AGUIProtocol implements ISSEProtocol {
      * 自定义事件 处理流程编排任务结束
      */
     handleFlowAgentEndCustomEvent(event: ICustomEvent): void;
+    /**
+     * 自定义事件 处理流程编排任务重启
+     */
+    handleFlowAgentRestartCustomEvent(event: ICustomEvent): void;
     /**
      * 自定义事件 处理流程编排任务结果（增量更新节点状态）
      */
@@ -43,6 +47,10 @@ export declare class AGUIProtocol implements ISSEProtocol {
      * 自定义事件 处理流程编排任务开始
      */
     handleFlowAgentStartCustomEvent(event: ICustomEvent): void;
+    /**
+     * 自定义事件 处理流程编排任务更新
+     */
+    handleFlowAgentUpdateCustomEvent(event: ICustomEvent): void;
     /**
      * 自定义事件 处理意图识别结束
      */
@@ -80,7 +88,7 @@ export declare class AGUIProtocol implements ISSEProtocol {
     /**
      * 处理运行完成事件
      */
-    handleRunFinishedEvent(_event: IRunFinishedEvent): void;
+    handleRunFinishedEvent(event: IRunFinishedEvent): void;
     /**
      * 处理运行开始事件
      */
@@ -105,6 +113,10 @@ export declare class AGUIProtocol implements ISSEProtocol {
      * 自定义事件 处理临时消息，转换为 AI 文本消息
      */
     handleTempMessageCustomEvent(event: ICustomEvent): void;
+    /**
+     * 自定义事件 处理审批结果
+     */
+    handleApprovalResultCustomEvent(event: ICustomEvent): void;
     /**
      * 处理文本消息块事件
      */
@@ -119,6 +131,7 @@ export declare class AGUIProtocol implements ISSEProtocol {
     handleTextMessageEndEvent(event: ITextMessageEndEvent): void;
     /**
      * 处理文本消息开始事件
+     * 断线续传时 DLQ 会重放事件：同 messageId 已存在则重置内容，避免重复气泡/文本翻倍
      */
     handleTextMessageStartEvent(event: ITextMessageStartEvent): void;
     /**