@blueking/chat-helper 0.0.1-beta.5 → 0.0.1-beta.6

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

@@ -131,7 +131,7 @@ export declare class AGUIProtocol implements ISSEProtocol {
     handleToolCallStartEvent(event: IToolCallStartEvent): void;
     injectMessageModule(message: IMessageModule): void;
     onDone(): void;
-    onError(error: unknown): void;
+    onError(error: Error): void;
     onMessage(message: unknown): void;
     onStart(): void;
 }

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

@@ -320,12 +320,31 @@ let tempTimestamp = 0;
         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);
+        // 处理完成
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message) {
+            message.status = MessageStatus.Complete;
+        }
     }
     onError(error) {
-        var _this_onErrorCallback, _this;
+        var // 回调给上层
+        _this_onErrorCallback, _this;
         (_this_onErrorCallback = (_this = this).onErrorCallback) === null || _this_onErrorCallback === void 0 ? void 0 : _this_onErrorCallback.call(_this, error);
+        // 处理错误
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message) {
+            message.status = MessageStatus.Error;
+            message.content = [
+                {
+                    type: MessageContentType.Text,
+                    data: error.message,
+                    status: MessageContentStatus.Error
+                }
+            ];
+        }
     }
     onMessage(message) {
         var // 回调给上层

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blueking/chat-helper",
-  "version": "0.0.1-beta.5",
+  "version": "0.0.1-beta.6",
   "description": "",
   "main": "./dist/index.ts.js",
   "scripts": {