@zero-library/chat-agent 2.1.15 → 2.1.17

package/dist/index.cjs.js

@@ -8,13 +8,13 @@ var react = require('react');
 var valtio = require('valtio');
 var icons = require('@ant-design/icons');
 var x = require('@ant-design/x');
-var classNames8 = require('classnames');
+var classNames9 = require('classnames');
 var InfiniteScroll = require('react-infinite-scroll-component');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 
 var dayjs__default = /*#__PURE__*/_interopDefault(dayjs);
-var classNames8__default = /*#__PURE__*/_interopDefault(classNames8);
+var classNames9__default = /*#__PURE__*/_interopDefault(classNames9);
 var InfiniteScroll__default = /*#__PURE__*/_interopDefault(InfiniteScroll);
 
 var __defProp = Object.defineProperty;
@@ -26,31 +26,9 @@ var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
 };
-function transform(source, fieldMap) {
-  const result = {};
-  for (const [targetKey, mapping] of Object.entries(fieldMap)) {
-    const key = targetKey;
-    if (typeof mapping === "function") {
-      result[key] = mapping(source);
-    } else if (typeof mapping === "string" && source && mapping in source) {
-      result[key] = source[mapping];
-    } else {
-      result[key] = mapping;
-    }
-  }
-  return result;
-}
-function transforms(sources, fieldMap) {
-  return sources?.map((source) => transform(source, fieldMap)) || [];
-}
-var copy, classifyTime, replaceThinkTags;
+var classifyTime, copy;
 var init_utils = __esm({
   "src/core/utils.ts"() {
-    copy = (value) => {
-      common.copyText(value).then(() => {
-        antd.message.success("\u590D\u5236\u6210\u529F");
-      });
-    };
     classifyTime = (timestamp) => {
       const now = dayjs__default.default();
       const target = dayjs__default.default(timestamp);
@@ -68,8 +46,10 @@ var init_utils = __esm({
         return "\u66F4\u65E9";
       }
     };
-    replaceThinkTags = (str) => {
-      return str.replace(/<think>/g, ":::alert type=think data={content:'").replace(/<\/think>/g, "'} :::");
+    copy = (value) => {
+      common.copyText(value).then(() => {
+        antd.message.success("\u590D\u5236\u6210\u529F");
+      });
     };
   }
 });
@@ -511,7 +491,7 @@ function createChatStore() {
     config.hooks = hooks;
   };
   const setUserInfo = (userInfo) => {
-    config.userInfo = transform(userInfo || common.getCurrentUser() || {}, {
+    config.userInfo = common.transform(userInfo || common.getCurrentUser() || {}, {
       id: (userInfo2) => {
         return String(userInfo2["id"]);
       },
@@ -542,7 +522,7 @@ function createChatStore() {
     try {
       receiver.loading = true;
       const { data } = await config.services.request.agentInfoQuery(agentId);
-      receiver.active = transform(data, {
+      receiver.active = common.transform(data, {
         name: "agentName",
         logo: "logo",
         description: "description",
@@ -574,7 +554,7 @@ function createChatStore() {
   };
   const switchExpert = (expert) => {
     if (receiver.active.id === expert.memberId) return;
-    receiver.active = transform(expert, {
+    receiver.active = common.transform(expert, {
       name: "memberName",
       logo: "memberAvatar",
       id: "memberId"
@@ -644,7 +624,7 @@ function createChatStore() {
         ...conversations.list.params
       });
       if (receiver.active.id !== targetId) return;
-      const items = transforms(data?.items || [], {
+      const items = common.transforms(data?.items || [], {
         key: "id",
         id: "id",
         label: "title",
@@ -832,9 +812,9 @@ function createChatStore() {
         conversationId
       });
       conversation.messages[conversationId].message = data.items;
-      config.hooks?.onAfterInitMessages?.(conversation.messages[conversationId].message);
-      config.hooks?.onAfterInit?.();
     }
+    config.hooks?.onAfterInitMessages?.(conversation.messages[conversationId].message);
+    config.hooks?.onAfterInit?.();
   };
   const pickReceiver = () => {
     const { agent, other } = conversation.active.member;
@@ -1009,9 +989,6 @@ function createChatStore() {
 // src/ui/layouts/index.tsx
 init_Context();
 
-// src/components/MessageRender.tsx
-init_utils();
-
 // src/ui/common/markdownAlert/index.ts
 var customComponents = {
   appCard: () => Promise.resolve().then(() => (init_AppCard(), AppCard_exports)),
@@ -1036,8 +1013,8 @@ var MessageRender_default = ({ message: message3, placement }) => {
       x.Bubble,
       {
         placement,
-        className: classNames8__default.default({ [styles_module_default2.loadingMessage]: message3.type }),
-        content: /* @__PURE__ */ jsxRuntime.jsx(common.RenderMarkdown, { content: replaceThinkTags(message3.msgContent), customComponents })
+        className: classNames9__default.default({ [styles_module_default2.loadingMessage]: message3.type }),
+        content: /* @__PURE__ */ jsxRuntime.jsx(common.RenderMarkdown, { content: message3.msgContent, customComponents })
       }
     ),
     message3.msgFiles?.map((file) => /* @__PURE__ */ jsxRuntime.jsx(
@@ -1063,7 +1040,7 @@ var MessageRender_default = ({ message: message3, placement }) => {
     message3.quoteMsg?.id && /* @__PURE__ */ jsxRuntime.jsx(
       x.Bubble,
       {
-        className: classNames8__default.default(styles_module_default2.chatQuoteMsg),
+        className: classNames9__default.default(styles_module_default2.chatQuoteMsg),
         placement,
         content: /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { vertical: true, gap: 8, children: [
           message3.quoteMsg.msgContent && /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { children: [
@@ -1124,7 +1101,73 @@ var styles_module_default3 = {
   nsAvatarListItem: "styles_module_nsAvatarListItem",
   nsAvatarListItemIcon: "styles_module_nsAvatarListItemIcon",
   nsAvatarListItemIconActive: "styles_module_nsAvatarListItemIconActive",
-  nsAvatarListItemName: "styles_module_nsAvatarListItemName"
+  nsAvatarListItemName: "styles_module_nsAvatarListItemName",
+  think: "styles_module_think2",
+  loadingMessage: "styles_module_loadingMessage2"
+};
+function extractThinkContent(content) {
+  let main = content;
+  const thinkContents = [];
+  let buffer = "";
+  const regex = /<think>(.*?)<\/think>/gs;
+  for (const match of content.matchAll(regex)) {
+    const thinkText = match[1].trim();
+    thinkContents.push(thinkText);
+    main = main.replace(match[0], "");
+  }
+  const lastStart = content.lastIndexOf("<think>");
+  const lastEnd = content.lastIndexOf("</think>");
+  if (lastStart > lastEnd) {
+    buffer = content.slice(lastStart).replace("<think>", "");
+    main = main.slice(0, lastStart);
+  }
+  return { mainContent: main, thinkContents, thinkBuffer: buffer };
+}
+function renderMarkdownPanel(content, label, key) {
+  return {
+    key,
+    label,
+    children: /* @__PURE__ */ jsxRuntime.jsx(common.RenderMarkdown, { content, customComponents })
+  };
+}
+var MessageAIRender_default = ({ message: message3, placement }) => {
+  const { mainContent, thinkContents, thinkBuffer } = react.useMemo(() => {
+    if (!message3?.msgContent) return { mainContent: "", thinkContents: [], thinkBuffer: "" };
+    return extractThinkContent(message3.msgContent);
+  }, [message3?.msgContent]);
+  if (!message3.msgContent) {
+    return /* @__PURE__ */ jsxRuntime.jsx(antd.Typography, { children: /* @__PURE__ */ jsxRuntime.jsx(antd.Spin, { size: "small" }) });
+  }
+  return /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { vertical: true, children: /* @__PURE__ */ jsxRuntime.jsx(
+    x.Bubble,
+    {
+      placement,
+      className: classNames9__default.default({ [styles_module_default3.loadingMessage]: message3.type && message3.type !== "TEXT_MESSAGE_END" }),
+      content: /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
+        thinkContents.length > 0 && /* @__PURE__ */ jsxRuntime.jsx(
+          antd.Collapse,
+          {
+            ghost: true,
+            size: "small",
+            className: styles_module_default3.think,
+            defaultActiveKey: thinkContents.map((_, i) => `think-${i}`),
+            items: thinkContents.map((content, index) => renderMarkdownPanel(content, `\u601D\u8003\u8FC7\u7A0B`, `think-${index}`))
+          }
+        ),
+        thinkBuffer && /* @__PURE__ */ jsxRuntime.jsx(
+          antd.Collapse,
+          {
+            ghost: true,
+            size: "small",
+            className: styles_module_default3.think,
+            defaultActiveKey: ["buffer-think"],
+            items: [renderMarkdownPanel(thinkBuffer, "\u4ED4\u7EC6\u601D\u8003\u4E2D...", "buffer-think")]
+          }
+        ),
+        /* @__PURE__ */ jsxRuntime.jsx(common.RenderMarkdown, { content: mainContent, customComponents })
+      ] })
+    }
+  ) });
 };
 var BubbleListItems_default = ({ firstMessage = false, avatar = { user: false, agent: true, other: true } }) => {
   const chatStore = useChatStore();
@@ -1174,7 +1217,7 @@ var BubbleListItems_default = ({ firstMessage = false, avatar = { user: false, a
           /* @__PURE__ */ jsxRuntime.jsx(
             x.Welcome,
             {
-              className: classNames8__default.default(styles_module_default3.chatWelcome, "p-t-32"),
+              className: classNames9__default.default(styles_module_default3.chatWelcome, "p-t-32"),
               variant: "borderless",
               icon: /* @__PURE__ */ jsxRuntime.jsx(antd.Avatar, { shape: "square", size: 58, src: receiverState.active.logo }),
               title: `\u4F60\u597D\uFF0C\u6211\u662F${receiverState.active.name || ""}`,
@@ -1192,7 +1235,7 @@ var BubbleListItems_default = ({ firstMessage = false, avatar = { user: false, a
                   label: "\u{1F914} \u63A8\u8350\u95EE\u9898:",
                   children: receiverState.active.config.recommendQuestions.map(({ question }) => ({
                     key: question,
-                    description: /* @__PURE__ */ jsxRuntime.jsx("span", { onClick: () => chatStore.sendMessage(question), className: classNames8__default.default(styles_module_default3.chatWelcomePrompts, "text-ellipsis"), children: question })
+                    description: /* @__PURE__ */ jsxRuntime.jsx("span", { onClick: () => chatStore.sendMessage(question), className: classNames9__default.default(styles_module_default3.chatWelcomePrompts, "text-ellipsis"), children: question })
                   }))
                 }
               ]
@@ -1238,7 +1281,8 @@ var BubbleListItems_default = ({ firstMessage = false, avatar = { user: false, a
         placement: role.placement,
         variant: "borderless",
         avatar: role.avatar,
-        content: /* @__PURE__ */ jsxRuntime.jsx(MessageRender_default, { message: message3, placement: role.placement }),
+        // content: <MessageRender message={message as ConversationMessage} placement={role.placement} />,
+        content: role.user === "agent" ? /* @__PURE__ */ jsxRuntime.jsx(MessageAIRender_default, { message: message3, placement: role.placement }) : /* @__PURE__ */ jsxRuntime.jsx(MessageRender_default, { message: message3, placement: role.placement }),
         footer: role.user === "agent" && /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { children: [
           /* @__PURE__ */ jsxRuntime.jsx(antd.Button, { onClick: () => copy(message3.msgContent), color: "default", variant: "text", size: "small", icon: /* @__PURE__ */ jsxRuntime.jsx(icons.CopyOutlined, {}) }),
           /* @__PURE__ */ jsxRuntime.jsx(
@@ -1324,7 +1368,7 @@ var BubbleListItems_default = ({ firstMessage = false, avatar = { user: false, a
       autoScroll: false,
       ref: listRef,
       items: bubbleListItems,
-      className: classNames8__default.default(styles_module_default3.nsBubbleList, "height-full", "scroll-fade-in", "zero-chat-bubbles"),
+      className: classNames9__default.default(styles_module_default3.nsBubbleList, "height-full", "scroll-fade-in", "zero-chat-bubbles"),
       onScroll: handleScroll
     },
     conversationState.active.id
@@ -1359,7 +1403,7 @@ var CharacterList_default = () => {
         /* @__PURE__ */ jsxRuntime.jsx(
           antd.Avatar,
           {
-            className: classNames8__default.default(styles_module_default3.nsAvatarListItemIcon, "cursor-pointer", {
+            className: classNames9__default.default(styles_module_default3.nsAvatarListItemIcon, "cursor-pointer", {
               [styles_module_default3.nsAvatarListItemIconActive]: activeCharacter.id === item.id
             }),
             size: 50,
@@ -1454,7 +1498,7 @@ var ChatHeader_default = ({
   const chatStore = useChatStore();
   const receiverState = valtio.useSnapshot(chatStore.receiver);
   const configState = valtio.useSnapshot(chatStore.config);
-  return /* @__PURE__ */ jsxRuntime.jsx(jsxRuntime.Fragment, { children: /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { justify: "space-between", align: "center", className: classNames8__default.default(styles_module_default3.nsChatHeader, "zero-chat-header"), children: [
+  return /* @__PURE__ */ jsxRuntime.jsx(jsxRuntime.Fragment, { children: /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { justify: "space-between", align: "center", className: classNames9__default.default(styles_module_default3.nsChatHeader, "zero-chat-header"), children: [
     /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { gap: 4, align: "center", children: [
       /* @__PURE__ */ jsxRuntime.jsx(
         common.RenderWrapper,
@@ -1764,7 +1808,7 @@ var SenderPromptsItems_default = () => {
           title: /* @__PURE__ */ jsxRuntime.jsx(
             "div",
             {
-              className: classNames8__default.default(styles_module_default3.nsSenderListTitle, "text-ellipsis"),
+              className: classNames9__default.default(styles_module_default3.nsSenderListTitle, "text-ellipsis"),
               children: `${receiverState.active.name}\u5F00\u59CB\u5173\u6CE8${question.name}\u5185\u5BB9\uFF01`
             }
           ),
@@ -1893,7 +1937,7 @@ var ConversationListHeader_default = () => {
         type: "primary",
         shape: "round",
         onClick: () => chatStore.createConversation(),
-        className: classNames8__default.default("m-t-16"),
+        className: classNames9__default.default("m-t-16"),
         icon: /* @__PURE__ */ jsxRuntime.jsx(icons.PlusOutlined, {}),
         children: "\u65B0\u5EFA\u4F1A\u8BDD"
       }
@@ -1901,7 +1945,7 @@ var ConversationListHeader_default = () => {
   ] });
 };
 var ConversationListPanel_default = ({ header }) => {
-  return /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { vertical: true, className: classNames8__default.default("height-full", "zero-chat-conversations", styles_module_default3.nsConversationListPanel), children: [
+  return /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { vertical: true, className: classNames9__default.default("height-full", "zero-chat-conversations", styles_module_default3.nsConversationListPanel), children: [
     /* @__PURE__ */ jsxRuntime.jsx(common.RenderWrapper, { control: header, DefaultComponent: ConversationListHeader_default }),
     /* @__PURE__ */ jsxRuntime.jsx("div", { className: "full-scroll", children: /* @__PURE__ */ jsxRuntime.jsx(ConversationList_default, {}) })
   ] });
@@ -1999,7 +2043,7 @@ var layouts_default = react.forwardRef(({ theme, params, userInfo, hooks, layout
   react.useEffect(() => {
     configState.hooks?.onBeforeInit?.();
   }, []);
-  return /* @__PURE__ */ jsxRuntime.jsx(x.XProvider, { theme: { cssVar: true, ...theme }, children: /* @__PURE__ */ jsxRuntime.jsx(ChatProvider, { store: chatStore, children: /* @__PURE__ */ jsxRuntime.jsx(antd.Spin, { spinning: receiverState.loading, wrapperClassName: "full-spin", children: /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { vertical: true, className: classNames8__default.default(styles_module_default4.nsChatLayout, "zero-chat-layout", "height-full"), children: [
+  return /* @__PURE__ */ jsxRuntime.jsx(x.XProvider, { theme: { cssVar: true, ...theme }, children: /* @__PURE__ */ jsxRuntime.jsx(ChatProvider, { store: chatStore, children: /* @__PURE__ */ jsxRuntime.jsx(antd.Spin, { spinning: receiverState.loading, wrapperClassName: "full-spin", children: /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { vertical: true, className: classNames9__default.default(styles_module_default4.nsChatLayout, "zero-chat-layout", "height-full"), children: [
     /* @__PURE__ */ jsxRuntime.jsx(common.RenderWrapper, { control: configState.layout.globalHeader, DefaultComponent: ChatHeader_default }),
     /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { className: "full-scroll", children: [
       /* @__PURE__ */ jsxRuntime.jsx(common.RenderWrapper, { control: configState.layout.leftPanel }),
@@ -2053,15 +2097,15 @@ var layouts_default = react.forwardRef(({ theme, params, userInfo, hooks, layout
             }
           )
         ] }) }),
-        /* @__PURE__ */ jsxRuntime.jsx(antd.Splitter.Panel, { collapsible: false, max: 800, min: 400, size: sizes[1], children: /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { vertical: true, className: classNames8__default.default("height-full"), children: [
+        /* @__PURE__ */ jsxRuntime.jsx(antd.Splitter.Panel, { collapsible: false, max: 800, min: 400, size: sizes[1], children: /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { vertical: true, className: classNames9__default.default("height-full"), children: [
           /* @__PURE__ */ jsxRuntime.jsx(common.RenderWrapper, { control: configState.layout.chatHeader, DefaultComponent: ChatHeader_default }),
-          /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { vertical: true, className: classNames8__default.default("full-scroll"), children: /* @__PURE__ */ jsxRuntime.jsxs(
+          /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { vertical: true, className: classNames9__default.default("full-scroll"), children: /* @__PURE__ */ jsxRuntime.jsxs(
             antd.Flex,
             {
               justify: "center",
               vertical: true,
               gap: 24,
-              className: classNames8__default.default("height-full", styles_module_default4.nsChatBody, "zero-chat-body", styles_module_default4.nsBodyWidth),
+              className: classNames9__default.default("height-full", styles_module_default4.nsChatBody, "zero-chat-body", styles_module_default4.nsBodyWidth),
               children: [
                 common.shouldRender(configState.layout.messageList) && /* @__PURE__ */ jsxRuntime.jsx("div", { className: "full-scroll", children: /* @__PURE__ */ jsxRuntime.jsx(common.RenderWrapper, { control: configState.layout.messageList, DefaultComponent: BubbleListItems_default }) }),
                 /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { vertical: true, gap: 8, children: [