@tdesign-react/chat 1.0.0-beta.4 → 1.0.1

package/es/chatbot/core/index.js

@@ -1,585 +0,0 @@
-/**
- * tdesign v1.0.0-beta.3
- * (c) 2025 tdesign
- * @license MIT
- */
-
-import _objectWithoutProperties from '@babel/runtime/helpers/objectWithoutProperties';
-import _asyncToGenerator from '@babel/runtime/helpers/asyncToGenerator';
-import _classCallCheck from '@babel/runtime/helpers/classCallCheck';
-import _createClass from '@babel/runtime/helpers/createClass';
-import _defineProperty from '@babel/runtime/helpers/defineProperty';
-import _regeneratorRuntime from '@babel/runtime/regenerator';
-import { MessageStore } from './store/message.js';
-import './server/index.js';
-import MessageProcessor from './processor/index.js';
-import { AGUIAdapter } from './adapters/agui/index.js';
-export { AGUIAdapter } from './adapters/agui/index.js';
-import { isAIMessage } from './utils/index.js';
-export { applyJsonPatch, findTargetElement, getMessageContentForCopy, isAIMessage, isAttachmentContent, isImageContent, isMarkdownContent, isSearchContent, isSuggestionContent, isTextContent, isThinkingContent, isToolCallContent, isUserMessage, safeParseJSON } from './utils/index.js';
-import { LLMService } from './server/llm-service.js';
-export { StateManagerImpl, stateManager } from './adapters/agui/state-manager.js';
-import '@babel/runtime/helpers/toConsumableArray';
-import '@babel/runtime/helpers/possibleConstructorReturn';
-import '@babel/runtime/helpers/getPrototypeOf';
-import '@babel/runtime/helpers/get';
-import '@babel/runtime/helpers/inherits';
-import './store/reactiveState.js';
-import '@babel/runtime/helpers/slicedToArray';
-import '@babel/runtime/helpers/wrapNativeSuper';
-import '@babel/runtime/helpers/typeof';
-import './server/types.js';
-import './server/errors.js';
-import './server/connection-manager.js';
-import './utils/logger.js';
-import './server/sse-client.js';
-import './utils/eventEmitter.js';
-import './server/sse-parser.js';
-import './adapters/agui/event-mapper.js';
-import './adapters/agui/events.js';
-import 'zod';
-import './adapters/agui/types.js';
-import './adapters/agui/utils.js';
-import './server/batch-client.js';
-
-var _excluded = ["prompt", "attachments"];
-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 () {
-  function ChatEngine() {
-    _classCallCheck(this, ChatEngine);
-    _defineProperty(this, "stopReceive", false);
-    _defineProperty(this, "aguiAdapter", null);
-    this.messageProcessor = new MessageProcessor();
-    this.messageStore = new MessageStore();
-  }
-  return _createClass(ChatEngine, [{
-    key: "messages",
-    get: function get() {
-      return this.messageStore.messages;
-    }
-  }, {
-    key: "status",
-    get: function get() {
-      var _this$messages$at;
-      return ((_this$messages$at = this.messages.at(-1)) === null || _this$messages$at === void 0 ? void 0 : _this$messages$at.status) || "idle";
-    }
-  }, {
-    key: "destroy",
-    value: function destroy() {
-      this.abortChat();
-      this.messageStore.clearHistory();
-      this.aguiAdapter = null;
-    }
-  }, {
-    key: "init",
-    value: function init(configSetter, initialMessages) {
-      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();
-      }
-    }
-  }, {
-    key: "sendUserMessage",
-    value: function () {
-      var _sendUserMessage = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee(requestParams) {
-        var prompt, attachments, customParams, hasValidPrompt, hasValidAttachments, userMessage, aiMessage, params;
-        return _regeneratorRuntime.wrap(function (_context) {
-          while (1) switch (_context.prev = _context.next) {
-            case 0:
-              prompt = requestParams.prompt, attachments = requestParams.attachments, customParams = _objectWithoutProperties(requestParams, _excluded);
-              hasValidPrompt = prompt && prompt.trim() !== "";
-              hasValidAttachments = Array.isArray(attachments) && attachments.length > 0;
-              if (!(!hasValidPrompt && !hasValidAttachments)) {
-                _context.next = 1;
-                break;
-              }
-              console.warn("[ChatEngine] sendUserMessage: \u5FC5\u987B\u63D0\u4F9B\u6709\u6548\u7684prompt\u6216attachments");
-              return _context.abrupt("return");
-            case 1:
-              userMessage = this.messageProcessor.createUserMessage(prompt, attachments);
-              aiMessage = this.messageProcessor.createAssistantMessage();
-              this.messageStore.createMultiMessages([userMessage, aiMessage]);
-              params = _objectSpread(_objectSpread({}, requestParams), {}, {
-                messageID: aiMessage.id
-              }, customParams);
-              this.sendRequest(params);
-            case 2:
-            case "end":
-              return _context.stop();
-          }
-        }, _callee, this);
-      }));
-      function sendUserMessage(_x) {
-        return _sendUserMessage.apply(this, arguments);
-      }
-      return sendUserMessage;
-    }()
-  }, {
-    key: "sendSystemMessage",
-    value: function () {
-      var _sendSystemMessage = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee2(msg) {
-        return _regeneratorRuntime.wrap(function (_context2) {
-          while (1) switch (_context2.prev = _context2.next) {
-            case 0:
-              this.messageStore.createMessage({
-                role: "system",
-                content: [{
-                  type: "text",
-                  data: msg
-                }]
-              });
-            case 1:
-            case "end":
-              return _context2.stop();
-          }
-        }, _callee2, this);
-      }));
-      function sendSystemMessage(_x2) {
-        return _sendSystemMessage.apply(this, arguments);
-      }
-      return sendSystemMessage;
-    }()
-  }, {
-    key: "sendAIMessage",
-    value: function () {
-      var _sendAIMessage = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee3() {
-        var options,
-          _options$params,
-          params,
-          content,
-          _options$sendRequest,
-          sendRequest,
-          newAIMessage,
-          _args3 = arguments;
-        return _regeneratorRuntime.wrap(function (_context3) {
-          while (1) switch (_context3.prev = _context3.next) {
-            case 0:
-              options = _args3.length > 0 && _args3[0] !== undefined ? _args3[0] : {};
-              _options$params = options.params, params = _options$params === void 0 ? {} : _options$params, content = options.content, _options$sendRequest = options.sendRequest, sendRequest = _options$sendRequest === void 0 ? true : _options$sendRequest;
-              newAIMessage = this.messageProcessor.createAssistantMessage({
-                content: content,
-                status: sendRequest ? "pending" : "complete"
-              });
-              this.messageStore.createMessage(newAIMessage);
-              if (!sendRequest) {
-                _context3.next = 1;
-                break;
-              }
-              params.messageID = newAIMessage.id;
-              _context3.next = 1;
-              return this.sendRequest(params);
-            case 1:
-            case "end":
-              return _context3.stop();
-          }
-        }, _callee3, this);
-      }));
-      function sendAIMessage() {
-        return _sendAIMessage.apply(this, arguments);
-      }
-      return sendAIMessage;
-    }()
-  }, {
-    key: "abortChat",
-    value: function () {
-      var _abortChat = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee4() {
-        var _this$config;
-        var _this$messageStore$la;
-        return _regeneratorRuntime.wrap(function (_context4) {
-          while (1) switch (_context4.prev = _context4.next) {
-            case 0:
-              this.stopReceive = true;
-              if (!((_this$config = this.config) !== null && _this$config !== void 0 && _this$config.onAbort)) {
-                _context4.next = 1;
-                break;
-              }
-              _context4.next = 1;
-              return this.config.onAbort();
-            case 1:
-              try {
-                this.llmService.closeConnect();
-                if (!this.config.stream) {
-                  if ((_this$messageStore$la = this.messageStore.lastAIMessage) !== null && _this$messageStore$la !== void 0 && _this$messageStore$la.id) {
-                    this.messageStore.removeMessage(this.messageStore.lastAIMessage.id);
-                  }
-                }
-              } catch (error) {
-                console.warn("Error during service cleanup:", error);
-              }
-            case 2:
-            case "end":
-              return _context4.stop();
-          }
-        }, _callee4, this);
-      }));
-      function abortChat() {
-        return _abortChat.apply(this, arguments);
-      }
-      return abortChat;
-    }()
-  }, {
-    key: "registerMergeStrategy",
-    value: function registerMergeStrategy(type, handler) {
-      this.messageProcessor.registerHandler(type, handler);
-    }
-  }, {
-    key: "setMessages",
-    value: function setMessages(messages) {
-      var mode = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : "replace";
-      this.messageStore.setMessages(messages, mode);
-    }
-  }, {
-    key: "clearMessages",
-    value: function clearMessages() {
-      this.messageStore.clearHistory();
-    }
-  }, {
-    key: "regenerateAIMessage",
-    value: function () {
-      var _regenerateAIMessage = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee5() {
-        var _this$lastRequestPara, _this$lastRequestPara2;
-        var keepVersion,
-          lastAIMessage,
-          newAIMessage,
-          params,
-          _args5 = arguments;
-        return _regeneratorRuntime.wrap(function (_context5) {
-          while (1) switch (_context5.prev = _context5.next) {
-            case 0:
-              keepVersion = _args5.length > 0 && _args5[0] !== undefined ? _args5[0] : false;
-              lastAIMessage = this.messageStore.lastAIMessage;
-              if (lastAIMessage) {
-                _context5.next = 1;
-                break;
-              }
-              return _context5.abrupt("return");
-            case 1:
-              if (!keepVersion) {
-                this.messageStore.removeMessage(lastAIMessage.id);
-              } else {
-                this.messageStore.createMessageBranch(lastAIMessage.id);
-              }
-              newAIMessage = this.messageProcessor.createAssistantMessage();
-              this.messageStore.createMessage(newAIMessage);
-              params = _objectSpread(_objectSpread({}, this.lastRequestParams || {}), {}, {
-                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;
-              return this.sendRequest(params);
-            case 2:
-            case "end":
-              return _context5.stop();
-          }
-        }, _callee5, this);
-      }));
-      function regenerateAIMessage() {
-        return _regenerateAIMessage.apply(this, arguments);
-      }
-      return regenerateAIMessage;
-    }()
-  }, {
-    key: "sendRequest",
-    value: function () {
-      var _sendRequest = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee6(params) {
-        var id, _t;
-        return _regeneratorRuntime.wrap(function (_context6) {
-          while (1) switch (_context6.prev = _context6.next) {
-            case 0:
-              id = params.messageID;
-              _context6.prev = 1;
-              if (!this.config.stream) {
-                _context6.next = 3;
-                break;
-              }
-              this.stopReceive = false;
-              _context6.next = 2;
-              return this.handleStreamRequest(params);
-            case 2:
-              _context6.next = 4;
-              break;
-            case 3:
-              _context6.next = 4;
-              return this.handleBatchRequest(params);
-            case 4:
-              this.lastRequestParams = params;
-              _context6.next = 6;
-              break;
-            case 5:
-              _context6.prev = 5;
-              _t = _context6["catch"](1);
-              this.setMessageStatus(id, "error");
-              throw _t;
-            case 6:
-            case "end":
-              return _context6.stop();
-          }
-        }, _callee6, this, [[1, 5]]);
-      }));
-      function sendRequest(_x3) {
-        return _sendRequest.apply(this, arguments);
-      }
-      return sendRequest;
-    }()
-  }, {
-    key: "getToolcallByName",
-    value: function getToolcallByName(name) {
-      var _this$aguiAdapter;
-      return (_this$aguiAdapter = this.aguiAdapter) === null || _this$aguiAdapter === void 0 ? void 0 : _this$aguiAdapter.getToolcallByName(name);
-    }
-  }, {
-    key: "handleBatchRequest",
-    value: function () {
-      var _handleBatchRequest = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee7(params) {
-        var id, result;
-        return _regeneratorRuntime.wrap(function (_context7) {
-          while (1) switch (_context7.prev = _context7.next) {
-            case 0:
-              id = params.messageID;
-              if (id) {
-                _context7.next = 1;
-                break;
-              }
-              return _context7.abrupt("return");
-            case 1:
-              this.setMessageStatus(id, "pending");
-              _context7.next = 2;
-              return this.llmService.handleBatchRequest(params, this.config);
-            case 2:
-              result = _context7.sent;
-              if (result) {
-                this.processMessageResult(id, result);
-                this.setMessageStatus(id, "complete");
-              } else {
-                this.setMessageStatus(id, "error");
-              }
-            case 3:
-            case "end":
-              return _context7.stop();
-          }
-        }, _callee7, this);
-      }));
-      function handleBatchRequest(_x4) {
-        return _handleBatchRequest.apply(this, arguments);
-      }
-      return handleBatchRequest;
-    }()
-  }, {
-    key: "handleError",
-    value: function handleError(id, error) {
-      var _this$config$onError, _this$config2;
-      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);
-    }
-  }, {
-    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);
-      if (Array.isArray(customResult) || customResult && "status" in customResult) {
-        this.processMessageResult(id, customResult);
-      } else {
-        var allContentFailed = this.messageStore.messages.every(function (content) {
-          return content.status === "error";
-        });
-        this.setMessageStatus(id, isAborted ? "stop" : allContentFailed ? "error" : "complete");
-      }
-    }
-  }, {
-    key: "handleStreamRequest",
-    value: function () {
-      var _handleStreamRequest = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee8(params) {
-        var id, isAGUI;
-        return _regeneratorRuntime.wrap(function (_context8) {
-          while (1) switch (_context8.prev = _context8.next) {
-            case 0:
-              id = params.messageID;
-              isAGUI = this.config.protocol === "agui";
-              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 = 3;
-              break;
-            case 2:
-              _context8.next = 3;
-              return this.handleDefaultStreamRequest(params, id);
-            case 3:
-            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:
-              _context9.next = 1;
-              return this.llmService.handleStreamRequest(params, _objectSpread(_objectSpread({}, this.config), {}, {
-                onMessage: function onMessage(chunk) {
-                  if (_this.stopReceive || !messageId) return null;
-                  var result;
-                  if (_this.aguiAdapter) {
-                    result = _this.aguiAdapter.handleAGUIEvent(chunk, {
-                      onRunStart: function onRunStart(event) {
-                        var _this$config$onStart, _this$config4;
-                        return (_this$config$onStart = (_this$config4 = _this.config).onStart) === null || _this$config$onStart === void 0 ? void 0 : _this$config$onStart.call(_this$config4, JSON.stringify(event));
-                      },
-                      onRunComplete: function onRunComplete(isAborted, params2, event) {
-                        return _this.handleComplete(messageId, isAborted, params2, event);
-                      },
-                      onRunError: function onRunError(error) {
-                        return _this.handleError(messageId, error);
-                      }
-                    });
-                  }
-                  if (_this.config.onMessage) {
-                    var userResult = _this.config.onMessage(chunk, _this.messageStore.getMessageByID(messageId), result);
-                    if (userResult) {
-                      result = userResult;
-                    }
-                  }
-                  _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);
-                  }
-                }
-              }));
-            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.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);
-      }
-      return handleDefaultStreamRequest;
-    }()
-  }, {
-    key: "processMessageResult",
-    value: function processMessageResult(messageId, result) {
-      if (!result) return;
-      if (Array.isArray(result)) {
-        this.messageStore.updateMultipleContents(messageId, result);
-      } else {
-        this.processContentUpdate(messageId, result);
-      }
-    }
-  }, {
-    key: "convertMessages",
-    value: function convertMessages(messages) {
-      if (!messages) return {
-        messageIds: [],
-        messages: []
-      };
-      return {
-        messageIds: messages.map(function (msg) {
-          return msg.id;
-        }),
-        messages: messages
-      };
-    }
-  }, {
-    key: "setMessageStatus",
-    value: function setMessageStatus(messageId, status) {
-      this.messageStore.setMessageStatus(messageId, status);
-    }
-  }, {
-    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);
-    }
-  }]);
-}();
-
-export { ChatEngine as default };
-//# sourceMappingURL=index.js.map