@tdesign-react/chat 1.1.0-alpha.16 → 1.1.0-alpha.17

package/es/chat-engine/core/index.js

@@ -16,7 +16,8 @@ import { MessageStore } from './store/message.js';
 import './event-bus/index.js';
 import MessageProcessor from './processor/index.js';
 import './server/index.js';
-import { isAIMessage } from './utils/index.js';
+import { createStreamHandler } from './stream-handlers/index.js';
+export { createStreamHandler } from './stream-handlers/index.js';
 export { applyJsonPatch, findTargetElement, getMessageContentForCopy, isAIMessage, isActivityContent, isAttachmentContent, isImageContent, isMarkdownContent, isReasoningContent, isSearchContent, isSuggestionContent, isTextContent, isThinkingContent, isToolCallContent, isUserMessage, safeParseJSON } from './utils/index.js';
 import './adapters/index.js';
 import { ChatEventBus } from './event-bus/ChatEventBus.js';
@@ -24,6 +25,10 @@ export { ChatEventBus, createEventBus } from './event-bus/ChatEventBus.js';
 import { ChatEngineEventType } from './event-bus/types.js';
 export { ChatEngineEventType } from './event-bus/types.js';
 import { LLMService } from './server/llm-service.js';
+import { OpenClawStreamHandler } from './stream-handlers/openclaw-stream-handler.js';
+export { OpenClawStreamHandler } from './stream-handlers/openclaw-stream-handler.js';
+import { AGUIStreamHandler } from './stream-handlers/agui-stream-handler.js';
+export { AGUIStreamHandler } from './stream-handlers/agui-stream-handler.js';
 export { createAIMessageContent, createActivityContent, createMarkdownContent, createReasoningContent, createSuggestionContent, createTextContent, createThinkingContent, createToolCallContent, handleSuggestionToolCall, mergeStringContent, parseSSEData, updateToolCall } from './adapters/agui/utils.js';
 export { ActivityManagerImpl, activityManager } from './adapters/agui/ActivityManager.js';
 export { StateManagerImpl, stateManager } from './adapters/agui/StateManager.js';
@@ -32,12 +37,14 @@ export { SurfaceStateManager, surfaceStateManager } from './adapters/json-render
 export { applyA2UIDataUpdate, applyA2UIUpdates, convertA2UIMessagesToJsonRender } from './adapters/json-render/a2ui-to-jsonrender.js';
 export { OpenClawEventMapper } from './adapters/openclaw/event-mapper.js';
 export { OpenClawRPCHandler, RPCError } from './adapters/openclaw/rpc-handler.js';
+export { convertOpenClawHistory, convertOpenClawHistoryResponse } from './adapters/openclaw/history-converter.js';
 export { DEFAULT_OPENCLAW_CONFIG, mergeOpenClawConfig } from './adapters/openclaw/types/config.js';
 export { AgentDataType, AgentStreamType, ChatEventState, OpenClawConnectionState, OpenClawEventType, OpenClawMethod } from './adapters/openclaw/types/events.js';
 export { isEventFrame, isRequestFrame, isResponseFrame } from './adapters/openclaw/types/messages.js';
 export { OpenClawAdapter } from './adapters/openclaw/index.js';
-import '@babel/runtime/helpers/toConsumableArray';
+export { DefaultStreamHandler } from './stream-handlers/default-stream-handler.js';
 import './adapters/agui/event-mapper.js';
+import '@babel/runtime/helpers/toConsumableArray';
 import 'zod';
 import '@babel/runtime/helpers/possibleConstructorReturn';
 import '@babel/runtime/helpers/getPrototypeOf';
@@ -62,18 +69,15 @@ import './adapters/shared/index.js';
 import './adapters/shared/content-factory.js';
 import './adapters/shared/merge-utils.js';
 import './adapters/shared/activity-manager.js';
+import './adapters/openclaw/utils.js';
+import './adapters/openclaw/types/index.js';
 import './adapters/json-render/index.js';
 import './adapters/json-render/types/a2ui.js';
 import './adapters/json-render/types/index.js';
 import './adapters/json-render/types/core.js';
 import './server/batch-client.js';
-import './adapters/openclaw/utils.js';
-import './adapters/openclaw/types/index.js';
 
 var _excluded = ["prompt", "attachments"];
-function _createForOfIteratorHelper(r, e) { var t = "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (!t) { if (Array.isArray(r) || (t = _unsupportedIterableToArray(r)) || e && r && "number" == typeof r.length) { t && (r = t); var _n = 0, F = function F() {}; return { s: F, n: function n() { return _n >= r.length ? { done: !0 } : { done: !1, value: r[_n++] }; }, e: function e(r) { throw r; }, f: F }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } var o, a = !0, u = !1; return { s: function s() { t = t.call(r); }, n: function n() { var r = t.next(); return a = r.done, r; }, e: function e(r) { u = !0, o = r; }, f: function f() { try { a || null == t["return"] || t["return"](); } finally { if (u) throw o; } } }; }
-function _unsupportedIterableToArray(r, a) { if (r) { if ("string" == typeof r) return _arrayLikeToArray(r, a); var t = {}.toString.call(r).slice(8, -1); return "Object" === t && r.constructor && (t = r.constructor.name), "Map" === t || "Set" === t ? Array.from(r) : "Arguments" === t || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(t) ? _arrayLikeToArray(r, a) : void 0; } }
-function _arrayLikeToArray(r, a) { (null == a || a > r.length) && (a = r.length); for (var e = 0, n = Array(a); e < a; e++) n[e] = r[e]; return n; }
 function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
 function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
 var ChatEngine = /*#__PURE__*/function () {
@@ -99,6 +103,7 @@ var ChatEngine = /*#__PURE__*/function () {
   }, {
     key: "destroy",
     value: function destroy() {
+      var _this$streamHandler, _this$streamHandler$d;
       this.eventBus.emit(ChatEngineEventType.ENGINE_DESTROY, {
         timestamp: Date.now()
       });
@@ -106,17 +111,33 @@ var ChatEngine = /*#__PURE__*/function () {
       this.messageStore.clearHistory();
       this.messageStore.destroy();
       this.aguiAdapter = null;
+      (_this$streamHandler = this.streamHandler) === null || _this$streamHandler === void 0 || (_this$streamHandler$d = _this$streamHandler.destroy) === null || _this$streamHandler$d === void 0 || _this$streamHandler$d.call(_this$streamHandler);
       this.eventBus.destroy();
     }
   }, {
     key: "init",
     value: function init(configSetter, initialMessages) {
+      var _this = this;
       this.messageStore.initialize(this.convertMessages(initialMessages));
       this.config = typeof configSetter === "function" ? configSetter() : configSetter || {};
       this.llmService = new LLMService();
       if (this.config.protocol === "agui") {
         this.aguiAdapter = new AGUIAdapter();
       }
+      this.streamHandler = createStreamHandler({
+        protocol: this.config.protocol,
+        llmService: this.llmService,
+        aguiAdapter: this.aguiAdapter
+      });
+      if (this.config.protocol === "openclaw" && this.config.endpoint && this.streamHandler instanceof OpenClawStreamHandler) {
+        this.streamHandler.initConnection(this.config, function (historyMessages) {
+          if (historyMessages && historyMessages.length > 0) {
+            var _this$config$onHistor, _this$config;
+            _this.messageStore.setMessages(historyMessages, "replace");
+            (_this$config$onHistor = (_this$config = _this.config).onHistoryLoaded) === null || _this$config$onHistor === void 0 || _this$config$onHistor.call(_this$config, historyMessages);
+          }
+        });
+      }
       this.eventBus.emit(ChatEngineEventType.ENGINE_INIT, {
         timestamp: Date.now()
       });
@@ -249,24 +270,63 @@ var ChatEngine = /*#__PURE__*/function () {
       return sendAIMessage;
     }()
   }, {
-    key: "abortChat",
+    key: "resumeRun",
     value: function () {
-      var _abortChat = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee4() {
-        var _this$config;
-        var _this$messageStore$la, messageId;
+      var _resumeRun = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee4() {
+        var params,
+          newAIMessage,
+          _args4 = arguments;
         return _regeneratorRuntime.wrap(function (_context4) {
           while (1) switch (_context4.prev = _context4.next) {
+            case 0:
+              params = _args4.length > 0 && _args4[0] !== undefined ? _args4[0] : {};
+              newAIMessage = this.messageProcessor.createAssistantMessage({
+                content: [],
+                status: "pending"
+              });
+              this.messageStore.createMessage(newAIMessage);
+              this.eventBus.emit(ChatEngineEventType.MESSAGE_CREATE, {
+                message: newAIMessage,
+                messages: this.messages
+              });
+              params.messageID = newAIMessage.id;
+              _context4.next = 1;
+              return this.sendRequest(params);
+            case 1:
+              return _context4.abrupt("return", newAIMessage.id);
+            case 2:
+            case "end":
+              return _context4.stop();
+          }
+        }, _callee4, this);
+      }));
+      function resumeRun() {
+        return _resumeRun.apply(this, arguments);
+      }
+      return resumeRun;
+    }()
+  }, {
+    key: "abortChat",
+    value: function () {
+      var _abortChat = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee5() {
+        var _this$config2;
+        var _this$messageStore$la, messageId;
+        return _regeneratorRuntime.wrap(function (_context5) {
+          while (1) switch (_context5.prev = _context5.next) {
             case 0:
               this.stopReceive = true;
-              if (!((_this$config = this.config) !== null && _this$config !== void 0 && _this$config.onAbort)) {
-                _context4.next = 1;
+              if (!((_this$config2 = this.config) !== null && _this$config2 !== void 0 && _this$config2.onAbort)) {
+                _context5.next = 1;
                 break;
               }
-              _context4.next = 1;
+              _context5.next = 1;
               return this.config.onAbort();
             case 1:
               try {
                 this.llmService.closeConnect();
+                if (this.streamHandler instanceof OpenClawStreamHandler) {
+                  this.streamHandler.abort();
+                }
                 if (!this.config.stream) {
                   if ((_this$messageStore$la = this.messageStore.lastAIMessage) !== null && _this$messageStore$la !== void 0 && _this$messageStore$la.id) {
                     messageId = this.messageStore.lastAIMessage.id;
@@ -282,9 +342,9 @@ var ChatEngine = /*#__PURE__*/function () {
               }
             case 2:
             case "end":
-              return _context4.stop();
+              return _context5.stop();
           }
-        }, _callee4, this);
+        }, _callee5, this);
       }));
       function abortChat() {
         return _abortChat.apply(this, arguments);
@@ -313,23 +373,23 @@ var ChatEngine = /*#__PURE__*/function () {
   }, {
     key: "regenerateAIMessage",
     value: function () {
-      var _regenerateAIMessage = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee5() {
+      var _regenerateAIMessage = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee6() {
         var _this$lastRequestPara, _this$lastRequestPara2;
         var keepVersion,
           lastAIMessage,
           newAIMessage,
           params,
-          _args5 = arguments;
-        return _regeneratorRuntime.wrap(function (_context5) {
-          while (1) switch (_context5.prev = _context5.next) {
+          _args6 = arguments;
+        return _regeneratorRuntime.wrap(function (_context6) {
+          while (1) switch (_context6.prev = _context6.next) {
             case 0:
-              keepVersion = _args5.length > 0 && _args5[0] !== undefined ? _args5[0] : false;
+              keepVersion = _args6.length > 0 && _args6[0] !== undefined ? _args6[0] : false;
               lastAIMessage = this.messageStore.lastAIMessage;
               if (lastAIMessage) {
-                _context5.next = 1;
+                _context6.next = 1;
                 break;
               }
-              return _context5.abrupt("return");
+              return _context6.abrupt("return");
             case 1:
               if (!keepVersion) {
                 this.messageStore.removeMessage(lastAIMessage.id);
@@ -342,13 +402,13 @@ var ChatEngine = /*#__PURE__*/function () {
                 messageID: newAIMessage.id,
                 prompt: (_this$lastRequestPara = (_this$lastRequestPara2 = this.lastRequestParams) === null || _this$lastRequestPara2 === void 0 ? void 0 : _this$lastRequestPara2.prompt) !== null && _this$lastRequestPara !== void 0 ? _this$lastRequestPara : ""
               });
-              _context5.next = 2;
+              _context6.next = 2;
               return this.sendRequest(params);
             case 2:
             case "end":
-              return _context5.stop();
+              return _context6.stop();
           }
-        }, _callee5, this);
+        }, _callee6, this);
       }));
       function regenerateAIMessage() {
         return _regenerateAIMessage.apply(this, arguments);
@@ -358,37 +418,37 @@ var ChatEngine = /*#__PURE__*/function () {
   }, {
     key: "sendRequest",
     value: function () {
-      var _sendRequest = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee6(params) {
+      var _sendRequest = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee7(params) {
         var id, _t;
-        return _regeneratorRuntime.wrap(function (_context6) {
-          while (1) switch (_context6.prev = _context6.next) {
+        return _regeneratorRuntime.wrap(function (_context7) {
+          while (1) switch (_context7.prev = _context7.next) {
             case 0:
               id = params.messageID;
               this.eventBus.emit(ChatEngineEventType.REQUEST_START, {
                 params: params,
                 messageId: id
               });
-              _context6.prev = 1;
+              _context7.prev = 1;
               if (!this.config.stream) {
-                _context6.next = 3;
+                _context7.next = 3;
                 break;
               }
               this.stopReceive = false;
-              _context6.next = 2;
+              _context7.next = 2;
               return this.handleStreamRequest(params);
             case 2:
-              _context6.next = 4;
+              _context7.next = 4;
               break;
             case 3:
-              _context6.next = 4;
+              _context7.next = 4;
               return this.handleBatchRequest(params);
             case 4:
               this.lastRequestParams = params;
-              _context6.next = 6;
+              _context7.next = 6;
               break;
             case 5:
-              _context6.prev = 5;
-              _t = _context6["catch"](1);
+              _context7.prev = 5;
+              _t = _context7["catch"](1);
               this.setMessageStatus(id, "error");
               this.eventBus.emit(ChatEngineEventType.REQUEST_ERROR, {
                 messageId: id,
@@ -398,9 +458,9 @@ var ChatEngine = /*#__PURE__*/function () {
               throw _t;
             case 6:
             case "end":
-              return _context6.stop();
+              return _context7.stop();
           }
-        }, _callee6, this, [[1, 5]]);
+        }, _callee7, this, [[1, 5]]);
       }));
       function sendRequest(_x3) {
         return _sendRequest.apply(this, arguments);
@@ -416,23 +476,23 @@ var ChatEngine = /*#__PURE__*/function () {
   }, {
     key: "handleBatchRequest",
     value: function () {
-      var _handleBatchRequest = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee7(params) {
+      var _handleBatchRequest = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee8(params) {
         var id, result, message;
-        return _regeneratorRuntime.wrap(function (_context7) {
-          while (1) switch (_context7.prev = _context7.next) {
+        return _regeneratorRuntime.wrap(function (_context8) {
+          while (1) switch (_context8.prev = _context8.next) {
             case 0:
               id = params.messageID;
               if (id) {
-                _context7.next = 1;
+                _context8.next = 1;
                 break;
               }
-              return _context7.abrupt("return");
+              return _context8.abrupt("return");
             case 1:
               this.setMessageStatus(id, "pending");
-              _context7.next = 2;
+              _context8.next = 2;
               return this.llmService.handleBatchRequest(params, this.config);
             case 2:
-              result = _context7.sent;
+              result = _context8.sent;
               if (result) {
                 this.processMessageResult(id, result);
                 this.setMessageStatus(id, "complete");
@@ -454,9 +514,9 @@ var ChatEngine = /*#__PURE__*/function () {
               }
             case 3:
             case "end":
-              return _context7.stop();
+              return _context8.stop();
           }
-        }, _callee7, this);
+        }, _callee8, this);
       }));
       function handleBatchRequest(_x4) {
         return _handleBatchRequest.apply(this, arguments);
@@ -466,9 +526,9 @@ var ChatEngine = /*#__PURE__*/function () {
   }, {
     key: "handleError",
     value: function handleError(id, error) {
-      var _this$config$onError, _this$config2;
+      var _this$config$onError, _this$config3;
       this.setMessageStatus(id, "error");
-      (_this$config$onError = (_this$config2 = this.config).onError) === null || _this$config$onError === void 0 || _this$config$onError.call(_this$config2, error);
+      (_this$config$onError = (_this$config3 = this.config).onError) === null || _this$config$onError === void 0 || _this$config$onError.call(_this$config3, error);
       this.eventBus.emit(ChatEngineEventType.REQUEST_ERROR, {
         messageId: id,
         error: error
@@ -477,8 +537,8 @@ var ChatEngine = /*#__PURE__*/function () {
   }, {
     key: "handleComplete",
     value: function handleComplete(id, isAborted, params, chunk) {
-      var _this$config$onComple, _this$config3;
-      var customResult = (_this$config$onComple = (_this$config3 = this.config).onComplete) === null || _this$config$onComple === void 0 ? void 0 : _this$config$onComple.call(_this$config3, isAborted, params, chunk);
+      var _this$config$onComple, _this$config4;
+      var customResult = (_this$config$onComple = (_this$config4 = this.config).onComplete) === null || _this$config$onComple === void 0 ? void 0 : _this$config$onComple.call(_this$config4, isAborted, params, chunk);
       if (Array.isArray(customResult) || customResult && "status" in customResult) {
         this.processMessageResult(id, customResult);
       } else {
@@ -506,231 +566,61 @@ var ChatEngine = /*#__PURE__*/function () {
   }, {
     key: "handleStreamRequest",
     value: function () {
-      var _handleStreamRequest = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee8(params) {
-        var id, isAGUI, isOpenClaw;
-        return _regeneratorRuntime.wrap(function (_context8) {
-          while (1) switch (_context8.prev = _context8.next) {
+      var _handleStreamRequest = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee9(params) {
+        var id, context;
+        return _regeneratorRuntime.wrap(function (_context9) {
+          while (1) switch (_context9.prev = _context9.next) {
             case 0:
               id = params.messageID;
-              isAGUI = this.config.protocol === "agui";
-              isOpenClaw = this.config.protocol === "openclaw";
               if (id) {
                 this.setMessageStatus(id, "streaming");
               }
-              if (!(isAGUI && this.aguiAdapter)) {
-                _context8.next = 2;
-                break;
-              }
-              _context8.next = 1;
-              return this.handleAGUIStreamRequest(params, id);
-            case 1:
-              _context8.next = 5;
-              break;
-            case 2:
-              if (!isOpenClaw) {
-                _context8.next = 4;
-                break;
-              }
-              _context8.next = 3;
-              return this.handleOpenClawStreamRequest(params, id);
-            case 3:
-              _context8.next = 5;
-              break;
-            case 4:
-              _context8.next = 5;
-              return this.handleDefaultStreamRequest(params, id);
-            case 5:
-            case "end":
-              return _context8.stop();
-          }
-        }, _callee8, this);
-      }));
-      function handleStreamRequest(_x5) {
-        return _handleStreamRequest.apply(this, arguments);
-      }
-      return handleStreamRequest;
-    }()
-  }, {
-    key: "handleAGUIStreamRequest",
-    value: function () {
-      var _handleAGUIStreamRequest = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee9(params, messageId) {
-        var _this = this;
-        return _regeneratorRuntime.wrap(function (_context9) {
-          while (1) switch (_context9.prev = _context9.next) {
-            case 0:
+              context = this.buildStreamContext(id);
               _context9.next = 1;
-              return this.llmService.handleStreamRequest(params, _objectSpread(_objectSpread({}, this.config), {}, {
-                onMessage: function onMessage(chunk) {
-                  if (_this.stopReceive || !messageId) return null;
-                  var result = null;
-                  if (_this.aguiAdapter) {
-                    result = _this.aguiAdapter.handleAGUIEvent(chunk, {
-                      onRunStart: function onRunStart(event) {
-                        var _this$aguiAdapter2, _this$config$onStart, _this$config4;
-                        (_this$aguiAdapter2 = _this.aguiAdapter) === null || _this$aguiAdapter2 === void 0 || _this$aguiAdapter2.reset();
-                        (_this$config$onStart = (_this$config4 = _this.config).onStart) === null || _this$config$onStart === void 0 || _this$config$onStart.call(_this$config4, JSON.stringify(event));
-                        _this.eventBus.emit(ChatEngineEventType.AGUI_RUN_START, {
-                          runId: event.runId || "",
-                          threadId: event.threadId,
-                          timestamp: Date.now()
-                        });
-                      },
-                      onRunComplete: function onRunComplete(isAborted, requestParams, event) {
-                        _this.handleComplete(messageId, isAborted, requestParams, event);
-                        if (!isAborted) {
-                          _this.eventBus.emit(ChatEngineEventType.AGUI_RUN_COMPLETE, {
-                            runId: (event === null || event === void 0 ? void 0 : event.runId) || "",
-                            threadId: event === null || event === void 0 ? void 0 : event.threadId,
-                            timestamp: Date.now()
-                          });
-                        }
-                      },
-                      onRunError: function onRunError(error) {
-                        _this.handleError(messageId, error);
-                        _this.eventBus.emit(ChatEngineEventType.AGUI_RUN_ERROR, {
-                          error: error
-                        });
-                      }
-                    });
-                  }
-                  if (_this.config.onMessage) {
-                    var userResult = _this.config.onMessage(chunk, _this.messageStore.getMessageByID(messageId), result);
-                    if (userResult) {
-                      result = userResult;
-                    }
-                  }
-                  _this.eventBus.emit(ChatEngineEventType.REQUEST_STREAM, {
-                    messageId: messageId,
-                    chunk: chunk,
-                    content: result
-                  });
-                  _this.processMessageResult(messageId, result);
-                  return result;
-                },
-                onError: function onError(error) {
-                  if (messageId) _this.handleError(messageId, error);
-                },
-                onComplete: function onComplete(isAborted) {
-                  if (isAborted && messageId) {
-                    _this.handleComplete(messageId, isAborted, params);
-                  }
-                }
-              }));
+              return this.streamHandler.handleStream(params, context);
             case 1:
             case "end":
               return _context9.stop();
           }
         }, _callee9, this);
       }));
-      function handleAGUIStreamRequest(_x6, _x7) {
-        return _handleAGUIStreamRequest.apply(this, arguments);
-      }
-      return handleAGUIStreamRequest;
-    }()
-  }, {
-    key: "handleDefaultStreamRequest",
-    value: function () {
-      var _handleDefaultStreamRequest = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee0(params, messageId) {
-        var _this2 = this;
-        return _regeneratorRuntime.wrap(function (_context0) {
-          while (1) switch (_context0.prev = _context0.next) {
-            case 0:
-              _context0.next = 1;
-              return this.llmService.handleStreamRequest(params, _objectSpread(_objectSpread({}, this.config), {}, {
-                onStart: function onStart(chunk) {
-                  var _this2$config$onStart, _this2$config;
-                  (_this2$config$onStart = (_this2$config = _this2.config).onStart) === null || _this2$config$onStart === void 0 || _this2$config$onStart.call(_this2$config, chunk);
-                },
-                onMessage: function onMessage(chunk) {
-                  if (_this2.stopReceive || !messageId) return null;
-                  var result = null;
-                  if (_this2.config.onMessage) {
-                    result = _this2.config.onMessage(chunk, _this2.messageStore.getMessageByID(messageId));
-                  }
-                  _this2.eventBus.emit(ChatEngineEventType.REQUEST_STREAM, {
-                    messageId: messageId,
-                    chunk: chunk,
-                    content: result
-                  });
-                  _this2.processMessageResult(messageId, result);
-                  return result;
-                },
-                onError: function onError(error) {
-                  if (messageId) _this2.handleError(messageId, error);
-                },
-                onComplete: function onComplete(isAborted) {
-                  if (messageId) {
-                    _this2.handleComplete(messageId, isAborted, params);
-                  }
-                }
-              }));
-            case 1:
-            case "end":
-              return _context0.stop();
-          }
-        }, _callee0, this);
-      }));
-      function handleDefaultStreamRequest(_x8, _x9) {
-        return _handleDefaultStreamRequest.apply(this, arguments);
+      function handleStreamRequest(_x5) {
+        return _handleStreamRequest.apply(this, arguments);
       }
-      return handleDefaultStreamRequest;
+      return handleStreamRequest;
     }()
   }, {
-    key: "handleOpenClawStreamRequest",
-    value: function () {
-      var _handleOpenClawStreamRequest = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee1(params, messageId) {
-        var _this3 = this;
-        return _regeneratorRuntime.wrap(function (_context1) {
-          while (1) switch (_context1.prev = _context1.next) {
-            case 0:
-              _context1.next = 1;
-              return this.llmService.handleStreamRequest(params, _objectSpread(_objectSpread({}, this.config), {}, {
-                onStart: function onStart(chunk) {
-                  var _this3$config$onStart, _this3$config;
-                  (_this3$config$onStart = (_this3$config = _this3.config).onStart) === null || _this3$config$onStart === void 0 || _this3$config$onStart.call(_this3$config, chunk);
-                },
-                onMessage: function onMessage(chunk) {
-                  if (_this3.stopReceive || !messageId) return null;
-                  var result = chunk.data;
-                  if (_this3.config.onMessage) {
-                    var userResult = _this3.config.onMessage(chunk, _this3.messageStore.getMessageByID(messageId), result);
-                    if (userResult) {
-                      result = userResult;
-                    }
-                  }
-                  _this3.eventBus.emit(ChatEngineEventType.REQUEST_STREAM, {
-                    messageId: messageId,
-                    chunk: chunk,
-                    content: result
-                  });
-                  _this3.processMessageResult(messageId, result);
-                  return result;
-                },
-                onError: function onError(error) {
-                  if (messageId) _this3.handleError(messageId, error);
-                },
-                onComplete: function onComplete(isAborted) {
-                  if (messageId) {
-                    _this3.handleComplete(messageId, isAborted, params);
-                  }
-                }
-              }));
-            case 1:
-            case "end":
-              return _context1.stop();
-          }
-        }, _callee1, this);
-      }));
-      function handleOpenClawStreamRequest(_x0, _x1) {
-        return _handleOpenClawStreamRequest.apply(this, arguments);
-      }
-      return handleOpenClawStreamRequest;
-    }()
+    key: "buildStreamContext",
+    value: function buildStreamContext(messageId) {
+      var _this2 = this;
+      return {
+        messageId: messageId,
+        config: this.config,
+        getStopReceive: function getStopReceive() {
+          return _this2.stopReceive;
+        },
+        processMessageResult: function processMessageResult(id, result) {
+          return _this2.processMessageResult(id, result);
+        },
+        handleError: function handleError(id, error) {
+          return _this2.handleError(id, error);
+        },
+        handleComplete: function handleComplete(id, isAborted, params, chunk) {
+          return _this2.handleComplete(id, isAborted, params, chunk);
+        },
+        messageStore: this.messageStore,
+        eventBus: this.eventBus
+      };
+    }
   }, {
     key: "processMessageResult",
     value: function processMessageResult(messageId, result) {
       if (!result) return;
-      Array.isArray(result) ? this.messageStore.updateMultipleContents(messageId, result) : this.processContentUpdate(messageId, result);
+      if (Array.isArray(result) && result._isSnapshot) {
+        this.messageStore.replaceContent(messageId, result);
+      } else {
+        this.messageProcessor.applyContentUpdate(this.messageStore, messageId, result);
+      }
       var message = this.messageStore.getMessageByID(messageId);
       if (message) {
         this.eventBus.emit(ChatEngineEventType.MESSAGE_UPDATE, {
@@ -738,40 +628,9 @@ var ChatEngine = /*#__PURE__*/function () {
           content: result,
           message: message
         });
-        if (this.aguiAdapter) {
-          this.emitAGUIDetailEvents(messageId, result);
-        }
-      }
-    }
-  }, {
-    key: "emitAGUIDetailEvents",
-    value: function emitAGUIDetailEvents(messageId, result) {
-      var contents = Array.isArray(result) ? result : [result];
-      var _iterator = _createForOfIteratorHelper(contents),
-        _step;
-      try {
-        for (_iterator.s(); !(_step = _iterator.n()).done;) {
-          var _content$data2;
-          var content = _step.value;
-          if (content.data.activityType) {
-            var _content$data;
-            this.eventBus.emit(ChatEngineEventType.AGUI_ACTIVITY, {
-              activityType: content.data.activityType,
-              messageId: messageId,
-              content: content === null || content === void 0 || (_content$data = content.data) === null || _content$data === void 0 ? void 0 : _content$data.content
-            });
-          }
-          if (content !== null && content !== void 0 && (_content$data2 = content.data) !== null && _content$data2 !== void 0 && (_content$data2 = _content$data2.eventType) !== null && _content$data2 !== void 0 && _content$data2.startsWith("TOOL_CALL")) {
-            this.eventBus.emit(ChatEngineEventType.AGUI_TOOLCALL, {
-              toolCall: content.data,
-              eventType: content.data.eventType
-            });
-          }
+        if (this.streamHandler instanceof AGUIStreamHandler) {
+          this.streamHandler.emitDetailEvents(messageId, result, this.eventBus);
         }
-      } catch (err) {
-        _iterator.e(err);
-      } finally {
-        _iterator.f();
       }
     }
   }, {
@@ -802,32 +661,6 @@ var ChatEngine = /*#__PURE__*/function () {
         });
       }
     }
-  }, {
-    key: "processContentUpdate",
-    value: function processContentUpdate(messageId, rawChunk) {
-      var message = this.messageStore.messages.find(function (m) {
-        return m.id === messageId;
-      });
-      if (!message || !isAIMessage(message) || !message.content) return;
-      var targetIndex;
-      if ((rawChunk === null || rawChunk === void 0 ? void 0 : rawChunk.strategy) === "append") {
-        targetIndex = -1;
-      } else {
-        targetIndex = message.content.findIndex(function (content) {
-          return content.type === rawChunk.type;
-        });
-        if (targetIndex !== -1) {
-          for (var i = message.content.length - 1; i >= 0; i--) {
-            if (message.content[i].type === rawChunk.type) {
-              targetIndex = i;
-              break;
-            }
-          }
-        }
-      }
-      var processed = this.messageProcessor.processContentUpdate(targetIndex !== -1 ? message.content[targetIndex] : void 0, rawChunk);
-      this.messageStore.appendContent(messageId, processed, targetIndex);
-    }
   }]);
 }();