@lobehub/ui 1.11.0 → 1.11.1

package/lib/Chat/store/messageReducer.js

@@ -1,70 +1,66 @@
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+"use strict";
 
-// src/Chat/store/messageReducer.ts
-var messageReducer_exports = {};
-__export(messageReducer_exports, {
-  messagesReducer: () => messagesReducer
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
+Object.defineProperty(exports, "__esModule", {
+  value: true
 });
-module.exports = __toCommonJS(messageReducer_exports);
-var import_immer = require("immer");
-var messagesReducer = (state, payload) => {
+exports.messagesReducer = void 0;
+var _toConsumableArray2 = _interopRequireDefault(require("@babel/runtime/helpers/toConsumableArray"));
+var _immer = require("immer");
+var messagesReducer = function messagesReducer(state, payload) {
   switch (payload.type) {
-    case "addMessage":
-      return [...state, payload.message];
-    case "insertMessage":
-      return (0, import_immer.produce)(state, (draftState) => {
+    case 'addMessage':
+      return [].concat((0, _toConsumableArray2.default)(state), [payload.message]);
+    case 'insertMessage':
+      return (0, _immer.produce)(state, function (draftState) {
         draftState.splice(payload.index, 0, payload.message);
       });
-    case "deleteMessage":
-      return state.filter((_, i) => i !== payload.index);
-    case "resetMessages":
+    case 'deleteMessage':
+      return state.filter(function (_, i) {
+        return i !== payload.index;
+      });
+    case 'resetMessages':
       return [];
-    case "updateMessage":
-      return (0, import_immer.produce)(state, (draftState) => {
-        const { index, message } = payload;
+    case 'updateMessage':
+      return (0, _immer.produce)(state, function (draftState) {
+        var index = payload.index,
+          message = payload.message;
         draftState[index].content = message;
       });
-    case "updateMessageRole":
-      return (0, import_immer.produce)(state, (draftState) => {
-        const { index, role } = payload;
+    case 'updateMessageRole':
+      return (0, _immer.produce)(state, function (draftState) {
+        var index = payload.index,
+          role = payload.role;
         draftState[index].role = role;
       });
-    case "addUserMessage":
-      return (0, import_immer.produce)(state, (draftState) => {
-        draftState.push({ role: "user", content: payload.message });
+    case 'addUserMessage':
+      return (0, _immer.produce)(state, function (draftState) {
+        draftState.push({
+          role: 'user',
+          content: payload.message
+        });
       });
-    case "updateLatestBotMessage":
-      return (0, import_immer.produce)(state, () => {
-        const { responseStream } = payload;
-        const newMessage = { role: "assistant", content: responseStream.join("") };
-        return [...state.slice(0, -1), newMessage];
+    case 'updateLatestBotMessage':
+      return (0, _immer.produce)(state, function () {
+        var responseStream = payload.responseStream;
+        var newMessage = {
+          role: 'assistant',
+          content: responseStream.join('')
+        };
+        return [].concat((0, _toConsumableArray2.default)(state.slice(0, -1)), [newMessage]);
       });
-    case "setErrorMessage":
-      return (0, import_immer.produce)(state, (draftState) => {
-        const { index, error } = payload;
+    case 'setErrorMessage':
+      return (0, _immer.produce)(state, function (draftState) {
+        var index = payload.index,
+          error = payload.error;
         draftState[index].error = error;
       });
-    case "updateMessageChoice":
-      return (0, import_immer.produce)(state, (draftState) => {
-        const { index, message } = payload;
-        const botMessage = draftState[index];
-        const prevMsg = botMessage.content;
+    case 'updateMessageChoice':
+      return (0, _immer.produce)(state, function (draftState) {
+        var index = payload.index,
+          message = payload.message;
+        var botMessage = draftState[index];
+        var prevMsg = botMessage.content;
         botMessage.content = message;
         if (botMessage.choices) {
           botMessage.choices.push(prevMsg);
@@ -73,10 +69,7 @@ var messagesReducer = (state, payload) => {
         }
       });
     default:
-      throw Error("暂未实现的 type，请检查 reducer");
+      throw Error('暂未实现的 type，请检查 reducer');
   }
 };
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  messagesReducer
-});
+exports.messagesReducer = messagesReducer;

package/lib/Chat/store/selectors.js

@@ -1,53 +1,51 @@
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+"use strict";
 
-// src/Chat/store/selectors.ts
-var selectors_exports = {};
-__export(selectors_exports, {
-  chatSelectors: () => chatSelectors
+Object.defineProperty(exports, "__esModule", {
+  value: true
 });
-module.exports = __toCommonJS(selectors_exports);
-var import_tiktoken = require("@dqbd/tiktoken");
-var tiktoken = (0, import_tiktoken.get_encoding)("cl100k_base");
-var encode = (text) => {
-  if (!text)
-    return [];
-  const tokens = tiktoken.encode(text);
+exports.chatSelectors = void 0;
+var _tiktoken = require("@dqbd/tiktoken");
+var tiktoken = (0, _tiktoken.get_encoding)('cl100k_base');
+var encode = function encode(text) {
+  if (!text) return [];
+  var tokens = tiktoken.encode(text);
   return tokens;
 };
-var disableInputSel = (s) => s.changingSystemRole;
-var messagesTokens = (s) => encode(s.messages.map((m) => m.content).join(""));
-var agentContentTokens = (s) => encode(
-  s.messages.filter((s2) => s2.role === "system").map((c) => c.content).join("") || ""
-);
-var totalTokens = (s) => encode(s.messages.map((m) => m.content).join(""));
-var totalTokenCount = (s) => totalTokens(s).length;
-var agentTokenCount = (s) => agentContentTokens(s).length;
-var messagesTokenCount = (s) => messagesTokens(s).length;
+var disableInputSel = function disableInputSel(s) {
+  return s.changingSystemRole;
+};
+var messagesTokens = function messagesTokens(s) {
+  return encode(s.messages.map(function (m) {
+    return m.content;
+  }).join(''));
+};
+var agentContentTokens = function agentContentTokens(s) {
+  return encode(s.messages.filter(function (s) {
+    return s.role === 'system';
+  }).map(function (c) {
+    return c.content;
+  }).join('') || '');
+};
+var totalTokens = function totalTokens(s) {
+  return encode(s.messages.map(function (m) {
+    return m.content;
+  }).join(''));
+};
+var totalTokenCount = function totalTokenCount(s) {
+  return totalTokens(s).length;
+};
+var agentTokenCount = function agentTokenCount(s) {
+  return agentContentTokens(s).length;
+};
+var messagesTokenCount = function messagesTokenCount(s) {
+  return messagesTokens(s).length;
+};
 var chatSelectors = {
-  totalTokenCount,
-  agentTokenCount,
-  messagesTokenCount,
-  totalTokens,
-  messagesTokens,
+  totalTokenCount: totalTokenCount,
+  agentTokenCount: agentTokenCount,
+  messagesTokenCount: messagesTokenCount,
+  totalTokens: totalTokens,
+  messagesTokens: messagesTokens,
   disableInput: disableInputSel
 };
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  chatSelectors
-});
+exports.chatSelectors = chatSelectors;

package/lib/Chat/store/store.js

@@ -1,119 +1,246 @@
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+"use strict";
 
-// src/Chat/store/store.ts
-var store_exports = {};
-__export(store_exports, {
-  createStore: () => createStore
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
+Object.defineProperty(exports, "__esModule", {
+  value: true
 });
-module.exports = __toCommonJS(store_exports);
-var import_fetch = require("../utils/fetch");
-var import_const = require("../const");
-var import_initialState = require("./initialState");
-var import_messageReducer = require("./messageReducer");
-var createStore = (set, get) => ({
-  ...import_initialState.initialState,
-  dispatchMessage: (payload) => {
-    var _a, _b;
-    const { type, ...res } = payload;
-    const messages = (0, import_messageReducer.messagesReducer)(get().messages, payload);
-    set({ messages }, false, {
-      type: `dispatchMessage/${type}`,
-      payload: res
-    });
-    (_b = (_a = get()).onMessagesChange) == null ? void 0 : _b.call(_a, messages);
-  },
-  handleMessageEditing: (index) => {
-    set({ editingMessageId: index });
-  },
-  generateMessage: async (message, messages, options) => {
-    const { onResponseStart, request, onResponseFinished } = get();
-    if (!request)
-      return;
-    await (onResponseStart == null ? void 0 : onResponseStart(get().messages));
-    set({ loading: true });
-    const fetcher = () => request({ messages });
-    await (0, import_fetch.fetchSSE)(fetcher, options);
-    set({ loading: false });
-    onResponseFinished == null ? void 0 : onResponseFinished({ messages: get().messages });
-  },
-  sendMessage: async () => {
-    const { message, dispatchMessage, generateMessage, messages } = get();
-    if (!message)
-      return;
-    set({ message: "" });
-    dispatchMessage({ type: "addUserMessage", message });
-    dispatchMessage({ type: "addMessage", message: { role: "assistant", content: import_const.LOADING_FLAT } });
-    let currentResponse = [];
-    await generateMessage(message, messages, {
-      onMessageHandle: (text) => {
-        currentResponse = [...currentResponse, text];
-        dispatchMessage({ type: "updateLatestBotMessage", responseStream: currentResponse });
-        const item = document.getElementById("for-loading");
-        if (!item)
-          return;
-        item.scrollIntoView({ behavior: "smooth" });
-      },
-      onErrorHandle: (error) => {
-        dispatchMessage({ type: "setErrorMessage", error, index: get().messages.length - 1 });
-      }
-    });
-  },
-  resendMessage: async (index) => {
-    var _a;
-    const { dispatchMessage, sendMessage, generateMessage, messages } = get();
-    const lastMessage = messages.at(-1);
-    if (messages.length === index && (lastMessage == null ? void 0 : lastMessage.role) === "user") {
-      dispatchMessage({ type: "deleteMessage", index: index - 1 });
-      set({ message: lastMessage.content });
-      await sendMessage();
-      return;
-    }
-    const contextMessages = get().messages.slice(0, index);
-    const userMessage = (_a = contextMessages.at(-1)) == null ? void 0 : _a.content;
-    if (!userMessage)
-      return;
-    const targetMsg = messages[index];
-    if (targetMsg.role !== "assistant") {
-      dispatchMessage({
-        type: "insertMessage",
-        index,
-        message: { role: "assistant", content: import_const.LOADING_FLAT }
+exports.createStore = void 0;
+var _toConsumableArray2 = _interopRequireDefault(require("@babel/runtime/helpers/toConsumableArray"));
+var _regeneratorRuntime2 = _interopRequireDefault(require("@babel/runtime/helpers/regeneratorRuntime"));
+var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/asyncToGenerator"));
+var _objectWithoutProperties2 = _interopRequireDefault(require("@babel/runtime/helpers/objectWithoutProperties"));
+var _objectSpread2 = _interopRequireDefault(require("@babel/runtime/helpers/objectSpread2"));
+var _fetch = require("../utils/fetch");
+var _const = require("../const");
+var _initialState = require("./initialState");
+var _messageReducer = require("./messageReducer");
+var _excluded = ["type"];
+var createStore = function createStore(set, get) {
+  return (0, _objectSpread2.default)((0, _objectSpread2.default)({}, _initialState.initialState), {}, {
+    dispatchMessage: function dispatchMessage(payload) {
+      var _get$onMessagesChange, _get;
+      var type = payload.type,
+        res = (0, _objectWithoutProperties2.default)(payload, _excluded);
+      var messages = (0, _messageReducer.messagesReducer)(get().messages, payload);
+      set({
+        messages: messages
+      }, false, {
+        type: "dispatchMessage/".concat(type),
+        payload: res
+      });
+      (_get$onMessagesChange = (_get = get()).onMessagesChange) === null || _get$onMessagesChange === void 0 ? void 0 : _get$onMessagesChange.call(_get, messages);
+    },
+    handleMessageEditing: function handleMessageEditing(index) {
+      set({
+        editingMessageId: index
       });
-    } else {
-      const botPrevMsg = targetMsg.content;
-      dispatchMessage({ type: "updateMessageChoice", message: botPrevMsg, index });
-      dispatchMessage({ type: "updateMessage", message: import_const.LOADING_FLAT, index });
-    }
-    dispatchMessage({ type: "setErrorMessage", error: void 0, index });
-    let currentResponse = [];
-    await generateMessage(userMessage, contextMessages, {
-      onMessageHandle: (text) => {
-        currentResponse = [...currentResponse, text];
-        dispatchMessage({ type: "updateMessage", message: currentResponse.join(""), index });
-      },
-      onErrorHandle: (error) => {
-        dispatchMessage({ type: "setErrorMessage", error, index });
+    },
+    generateMessage: function () {
+      var _generateMessage = (0, _asyncToGenerator2.default)( /*#__PURE__*/(0, _regeneratorRuntime2.default)().mark(function _callee(message, messages, options) {
+        var _get2, onResponseStart, request, onResponseFinished, fetcher;
+        return (0, _regeneratorRuntime2.default)().wrap(function _callee$(_context) {
+          while (1) switch (_context.prev = _context.next) {
+            case 0:
+              _get2 = get(), onResponseStart = _get2.onResponseStart, request = _get2.request, onResponseFinished = _get2.onResponseFinished;
+              if (request) {
+                _context.next = 3;
+                break;
+              }
+              return _context.abrupt("return");
+            case 3:
+              _context.next = 5;
+              return onResponseStart === null || onResponseStart === void 0 ? void 0 : onResponseStart(get().messages);
+            case 5:
+              set({
+                loading: true
+              });
+              fetcher = function fetcher() {
+                return request({
+                  messages: messages
+                });
+              };
+              _context.next = 9;
+              return (0, _fetch.fetchSSE)(fetcher, options);
+            case 9:
+              set({
+                loading: false
+              });
+              onResponseFinished === null || onResponseFinished === void 0 ? void 0 : onResponseFinished({
+                messages: get().messages
+              });
+            case 11:
+            case "end":
+              return _context.stop();
+          }
+        }, _callee);
+      }));
+      function generateMessage(_x, _x2, _x3) {
+        return _generateMessage.apply(this, arguments);
       }
-    });
-  }
-});
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  createStore
-});
+      return generateMessage;
+    }(),
+    sendMessage: function () {
+      var _sendMessage = (0, _asyncToGenerator2.default)( /*#__PURE__*/(0, _regeneratorRuntime2.default)().mark(function _callee2() {
+        var _get3, message, dispatchMessage, generateMessage, messages, currentResponse;
+        return (0, _regeneratorRuntime2.default)().wrap(function _callee2$(_context2) {
+          while (1) switch (_context2.prev = _context2.next) {
+            case 0:
+              _get3 = get(), message = _get3.message, dispatchMessage = _get3.dispatchMessage, generateMessage = _get3.generateMessage, messages = _get3.messages;
+              if (message) {
+                _context2.next = 3;
+                break;
+              }
+              return _context2.abrupt("return");
+            case 3:
+              set({
+                message: ''
+              });
+              dispatchMessage({
+                type: 'addUserMessage',
+                message: message
+              });
+
+              // 添加一个空的信息用于放置 ai 响应
+              dispatchMessage({
+                type: 'addMessage',
+                message: {
+                  role: 'assistant',
+                  content: _const.LOADING_FLAT
+                }
+              });
+              currentResponse = []; // 生成 messages
+              _context2.next = 9;
+              return generateMessage(message, messages, {
+                onMessageHandle: function onMessageHandle(text) {
+                  currentResponse = [].concat((0, _toConsumableArray2.default)(currentResponse), [text]);
+                  dispatchMessage({
+                    type: 'updateLatestBotMessage',
+                    responseStream: currentResponse
+                  });
+
+                  // 滚动到最后一条消息
+                  var item = document.getElementById('for-loading');
+                  if (!item) return;
+                  item.scrollIntoView({
+                    behavior: 'smooth'
+                  });
+                },
+                onErrorHandle: function onErrorHandle(error) {
+                  dispatchMessage({
+                    type: 'setErrorMessage',
+                    error: error,
+                    index: get().messages.length - 1
+                  });
+                }
+              });
+            case 9:
+            case "end":
+              return _context2.stop();
+          }
+        }, _callee2);
+      }));
+      function sendMessage() {
+        return _sendMessage.apply(this, arguments);
+      }
+      return sendMessage;
+    }(),
+    resendMessage: function () {
+      var _resendMessage = (0, _asyncToGenerator2.default)( /*#__PURE__*/(0, _regeneratorRuntime2.default)().mark(function _callee3(index) {
+        var _contextMessages$at;
+        var _get4, dispatchMessage, sendMessage, generateMessage, messages, lastMessage, contextMessages, userMessage, targetMsg, botPrevMsg, currentResponse;
+        return (0, _regeneratorRuntime2.default)().wrap(function _callee3$(_context3) {
+          while (1) switch (_context3.prev = _context3.next) {
+            case 0:
+              _get4 = get(), dispatchMessage = _get4.dispatchMessage, sendMessage = _get4.sendMessage, generateMessage = _get4.generateMessage, messages = _get4.messages;
+              lastMessage = messages.at(-1); // 用户通过手动删除，造成了他的问题是最后一条消息
+              // 这种情况下，相当于用户重新发送消息
+              if (!(messages.length === index && (lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.role) === 'user')) {
+                _context3.next = 8;
+                break;
+              }
+              dispatchMessage({
+                type: 'deleteMessage',
+                index: index - 1
+              });
+              set({
+                message: lastMessage.content
+              });
+              _context3.next = 7;
+              return sendMessage();
+            case 7:
+              return _context3.abrupt("return");
+            case 8:
+              // 上下文消息就是当前消息之前的消息
+              contextMessages = get().messages.slice(0, index); // 上下文消息中最后一条消息
+              userMessage = (_contextMessages$at = contextMessages.at(-1)) === null || _contextMessages$at === void 0 ? void 0 : _contextMessages$at.content;
+              if (userMessage) {
+                _context3.next = 12;
+                break;
+              }
+              return _context3.abrupt("return");
+            case 12:
+              targetMsg = messages[index]; // 如果不是 assistant 的消息，那么需要额外插入一条消息
+              if (targetMsg.role !== 'assistant') {
+                dispatchMessage({
+                  type: 'insertMessage',
+                  index: index,
+                  message: {
+                    role: 'assistant',
+                    content: _const.LOADING_FLAT
+                  }
+                });
+              } else {
+                botPrevMsg = targetMsg.content; // 保存之前的消息为历史消息
+                dispatchMessage({
+                  type: 'updateMessageChoice',
+                  message: botPrevMsg,
+                  index: index
+                });
+                dispatchMessage({
+                  type: 'updateMessage',
+                  message: _const.LOADING_FLAT,
+                  index: index
+                });
+              }
+
+              // 重置错误信息
+              dispatchMessage({
+                type: 'setErrorMessage',
+                error: undefined,
+                index: index
+              });
+
+              // 开始更新消息
+              currentResponse = [];
+              _context3.next = 18;
+              return generateMessage(userMessage, contextMessages, {
+                onMessageHandle: function onMessageHandle(text) {
+                  currentResponse = [].concat((0, _toConsumableArray2.default)(currentResponse), [text]);
+                  dispatchMessage({
+                    type: 'updateMessage',
+                    message: currentResponse.join(''),
+                    index: index
+                  });
+                },
+                onErrorHandle: function onErrorHandle(error) {
+                  dispatchMessage({
+                    type: 'setErrorMessage',
+                    error: error,
+                    index: index
+                  });
+                }
+              });
+            case 18:
+            case "end":
+              return _context3.stop();
+          }
+        }, _callee3);
+      }));
+      function resendMessage(_x4) {
+        return _resendMessage.apply(this, arguments);
+      }
+      return resendMessage;
+    }()
+  });
+};
+exports.createStore = createStore;

package/lib/Chat/types.js

@@ -1,17 +1,5 @@
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+"use strict";
 
-// src/Chat/types.ts
-var types_exports = {};
-module.exports = __toCommonJS(types_exports);
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});