@zero-library/chat-agent 2.2.6 → 2.2.8

package/dist/index.cjs.js

@@ -281,7 +281,7 @@ var classifyTime = (timestamp) => {
     return "\u66F4\u65E9";
   }
 };
-var replaceThinkTags = (str) => {
+var replaceMarkdownTags = (str) => {
   return str.replace(/<think>/g, ":::alert type=think data={content:'").replace(/<\/think>/g, "'} :::");
 };
 var getChatSocketUrl = (baseURL, params) => {
@@ -468,8 +468,13 @@ function createChatStore() {
       config.services.websocketUrls = [url, ""];
     }
   };
-  const setPreview = (file = {}, isEdit = false) => {
-    if (common.shouldRender(config.layout.preview)) {
+  const setPreview = async (file = {}, isEdit = false) => {
+    const preview = common.shouldRender(config.layout.preview);
+    const canProceed = await config.hooks?.onBeforeFilePreview?.(file, preview);
+    if (canProceed === false) {
+      throw new Error("\u64CD\u4F5C\u88AB\u963B\u6B62");
+    }
+    if (preview) {
       config.preview = {
         file,
         isEdit
@@ -477,10 +482,11 @@ function createChatStore() {
     } else {
       if (file?.fileUrl) {
         window.open(file.fileUrl, "_blank");
-      } else {
+      } else if (file?.content) {
         console.log("\u5F53\u524D\u73AF\u5883\u4E0D\u652F\u6301\u9884\u89C8");
       }
     }
+    config.hooks?.onAfterFilePreview?.(file, preview);
   };
   const setLayout = (layout, receiverType = 3) => {
     const defaultLayout = receiverType === 3 ? defaultAgentLayout : defaultExpertLayout;
@@ -799,7 +805,7 @@ function createChatStore() {
       throw new Error("\u64CD\u4F5C\u88AB\u963B\u6B62");
     }
     await initConversation(conversationId);
-    setPreview();
+    await setPreview();
     config.hooks?.onAfterSwitchConversation?.(conversationId);
     try {
       if (conversation.active.member.agent) {
@@ -1081,28 +1087,50 @@ var styles_module_default2 = {
   loadingMessage: "styles_module_loadingMessage",
   chatAttachments: "styles_module_chatAttachments",
   chatSender: "styles_module_chatSender",
-  chatQuoteMsg: "styles_module_chatQuoteMsg"
+  chatQuoteMsg: "styles_module_chatQuoteMsg",
+  quoteTooltip: "styles_module_quoteTooltip",
+  quoteTooltipContainer: "styles_module_quoteTooltipContainer"
 };
-var MessageRender_default = ({ message: message2, placement }) => {
-  const content = react.useMemo(
+var MessageRender_default = ({ message: message2, placement, onFilePreview }) => {
+  const msgContent = react.useMemo(
     () => `${message2.msgContent || ""}${message2.msgContent && message2.tempContent ? "\n\n" : ""}${message2.tempContent || ""}`,
     [message2.msgContent, message2.tempContent]
   );
-  const paramsCitation = react.useMemo(() => {
-    let citation = "";
-    try {
-      citation = JSON.parse(message2.params || "{}").citation || "";
-    } catch (e) {
+  const quoteMsg = react.useMemo(() => {
+    const quoteMsg2 = {};
+    if (message2.quoteMsg?.id) {
+      if (message2.quoteMsg.msgContent) {
+        quoteMsg2.msgContent = message2.quoteMsg.msgContent;
+      }
+      if (message2.quoteMsg.msgFiles?.length) {
+        quoteMsg2.msgFiles = message2.quoteMsg.msgFiles;
+      }
+    } else if (message2.params) {
+      try {
+        const citation = JSON.parse(message2.params).citation;
+        if (citation) {
+          quoteMsg2.msgContent = citation;
+        }
+      } catch (e) {
+      }
+    }
+    if (quoteMsg2.msgContent) {
+      quoteMsg2.msgContentText = common.markdownToText(quoteMsg2.msgContent);
     }
-    return citation;
-  }, [message2.params]);
-  return /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { vertical: true, children: !(message2.msgContent || message2.tempContent || message2.msgFiles.length) ? /* @__PURE__ */ jsxRuntime.jsx(antd.Typography, { children: /* @__PURE__ */ jsxRuntime.jsx(antd.Spin, { size: "small" }) }) : /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
-    content && /* @__PURE__ */ jsxRuntime.jsx(
+    return common.isEmptyObj(quoteMsg2) ? null : quoteMsg2;
+  }, [message2.params, message2.quoteMsg]);
+  const filePreview = (file) => {
+    if (file.content && !["image"].includes(file.type)) {
+      onFilePreview?.(file);
+    }
+  };
+  return /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { vertical: true, children: !(msgContent || message2.msgFiles.length) ? /* @__PURE__ */ jsxRuntime.jsx(antd.Typography, { children: /* @__PURE__ */ jsxRuntime.jsx(antd.Spin, { size: "small" }) }) : /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
+    msgContent && /* @__PURE__ */ jsxRuntime.jsx(
       x.Bubble,
       {
         placement,
         className: classNames9__default.default({ [styles_module_default2.loadingMessage]: message2.type }),
-        content: /* @__PURE__ */ jsxRuntime.jsx(common.RenderMarkdown, { content: replaceThinkTags(content), customComponents, message: message2 })
+        content: /* @__PURE__ */ jsxRuntime.jsx(common.RenderMarkdown, { content: replaceMarkdownTags(msgContent), customComponents, message: message2 })
       }
     ),
     message2.msgFiles?.map((file) => /* @__PURE__ */ jsxRuntime.jsx(
@@ -1111,7 +1139,7 @@ var MessageRender_default = ({ message: message2, placement }) => {
         className: "m-t-8",
         variant: "borderless",
         placement,
-        content: /* @__PURE__ */ jsxRuntime.jsx(
+        content: /* @__PURE__ */ jsxRuntime.jsx("div", { className: "cursor-pointer", onClick: () => filePreview(file), children: /* @__PURE__ */ jsxRuntime.jsx(
           x.Attachments.FileCard,
           {
             item: {
@@ -1119,25 +1147,36 @@ var MessageRender_default = ({ message: message2, placement }) => {
               name: file.name,
               url: file.content
             }
-          },
-          file.content
-        )
+          }
+        ) })
       },
       file.content
     )),
-    message2.quoteMsg?.id ? /* @__PURE__ */ jsxRuntime.jsx(
+    quoteMsg && /* @__PURE__ */ jsxRuntime.jsx(
       x.Bubble,
       {
-        className: classNames9__default.default(styles_module_default2.chatQuoteMsg),
+        className: styles_module_default2.chatQuoteMsg,
         placement,
         content: /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { vertical: true, gap: 8, children: [
-          message2.quoteMsg.msgContent && /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { children: [
-            /* @__PURE__ */ jsxRuntime.jsx("span", { children: "\u3010\u5F15\u7528\u6D88\u606F\u3011\uFF1A" }),
-            /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { flex: 1, className: "text-ellipsis", title: message2.quoteMsg.msgContent, children: /* @__PURE__ */ jsxRuntime.jsx(common.RenderMarkdown, { content: message2.quoteMsg.msgContent }) })
-          ] }),
-          message2.quoteMsg?.msgFiles && message2.quoteMsg.msgFiles.length > 0 && /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { children: [
+          quoteMsg.msgContent && /* @__PURE__ */ jsxRuntime.jsx(
+            antd.Tooltip,
+            {
+              color: "#fff",
+              classNames: {
+                root: styles_module_default2.quoteTooltip,
+                body: `${styles_module_default2.quoteTooltipContainer} scroll-fade-in`
+              },
+              title: /* @__PURE__ */ jsxRuntime.jsx(common.RenderMarkdown, { content: quoteMsg.msgContent }),
+              getPopupContainer: (node) => node,
+              children: /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { children: [
+                /* @__PURE__ */ jsxRuntime.jsx("span", { children: "\u3010\u5F15\u7528\u6D88\u606F\u3011\uFF1A" }),
+                /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { flex: 1, className: "text-ellipsis", children: quoteMsg.msgContentText })
+              ] })
+            }
+          ),
+          quoteMsg?.msgFiles && /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { children: [
             /* @__PURE__ */ jsxRuntime.jsx("span", { children: "\u3010\u5F15\u7528\u6587\u4EF6\u3011\uFF1A" }),
-            /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { gap: 8, wrap: true, flex: 1, children: message2.quoteMsg.msgFiles.map((file) => /* @__PURE__ */ jsxRuntime.jsx(
+            /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { gap: 8, wrap: true, flex: 1, children: quoteMsg.msgFiles.map((file) => /* @__PURE__ */ jsxRuntime.jsx("div", { className: "cursor-pointer", onClick: () => filePreview(file), children: /* @__PURE__ */ jsxRuntime.jsx(
               x.Attachments.FileCard,
               {
                 item: {
@@ -1145,22 +1184,11 @@ var MessageRender_default = ({ message: message2, placement }) => {
                   name: file.name,
                   url: file.content
                 }
-              },
-              file.content
-            )) })
+              }
+            ) }, file.content)) })
           ] })
         ] })
       }
-    ) : paramsCitation && /* @__PURE__ */ jsxRuntime.jsx(
-      x.Bubble,
-      {
-        className: classNames9__default.default(styles_module_default2.chatQuoteMsg),
-        placement,
-        content: /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { vertical: true, gap: 8, children: /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { children: [
-          /* @__PURE__ */ jsxRuntime.jsx("span", { children: "\u3010\u5F15\u7528\u6D88\u606F\u3011\uFF1A" }),
-          /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { flex: 1, className: "text-ellipsis", title: paramsCitation, children: /* @__PURE__ */ jsxRuntime.jsx(common.RenderMarkdown, { content: paramsCitation }) })
-        ] }) })
-      }
     )
   ] }) });
 };
@@ -1332,8 +1360,19 @@ var BubbleListItems_default = ({ firstMessage = false, welcomeMessage = true, av
         placement: role.placement,
         variant: "borderless",
         avatar: role.avatar,
-        content: /* @__PURE__ */ jsxRuntime.jsx(MessageRender_default, { message: message2, placement: role.placement }),
-        footer: role.user === "agent" && /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { children: [
+        content: /* @__PURE__ */ jsxRuntime.jsx(
+          MessageRender_default,
+          {
+            message: message2,
+            placement: role.placement,
+            onFilePreview: (file) => chatStore.setPreview({
+              fileUrl: file.content,
+              suffix: file.extension,
+              fileName: file.name
+            })
+          }
+        ),
+        footer: /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { children: role.user === "agent" && /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
           /* @__PURE__ */ jsxRuntime.jsx(antd.Button, { onClick: () => common.copyText(message2.msgContent), color: "default", variant: "text", size: "small", icon: /* @__PURE__ */ jsxRuntime.jsx(icons.CopyOutlined, {}) }),
           /* @__PURE__ */ jsxRuntime.jsx(
             antd.Button,
@@ -1357,7 +1396,7 @@ var BubbleListItems_default = ({ firstMessage = false, welcomeMessage = true, av
               onClick: () => chatStore.feedback(conversationState.active.id, message2.id, 2, index)
             }
           )
-        ] })
+        ] }) })
         // ...(isHasTime
         //   ? { rootClassName: styles.hasSendTime, classNames: { header: styles.sendTime }, header: <span>{formatDate(message.sendDate)}</span> }
         //   : {})
@@ -1727,6 +1766,7 @@ var Attachments_default = react.forwardRef(({ fileUpload, fileUploadConfig = [],
       beforeUpload: (file, files) => onBeforeUpload(files),
       items: attachedFiles,
       onRemove,
+      overflow: "wrap",
       placeholder: (type) => type === "drop" ? { title: "\u5C06\u6587\u4EF6\u653E\u5230\u8FD9\u91CC" } : {
         icon: /* @__PURE__ */ jsxRuntime.jsx(icons.CloudUploadOutlined, {}),
         title: "\u63D0\u4EA4\u6587\u4EF6",
@@ -1799,7 +1839,7 @@ var ChatSender_default = react.forwardRef(
         autoSize: { minRows: 2, maxRows: 6 },
         onCancel,
         footer: ({ components }) => {
-          const { SendButton, LoadingButton, SpeechButton } = components;
+          const { SendButton, LoadingButton } = components;
           return /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { vertical: true, children: [
             /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { justify: "space-between", align: "center", gap: 12, children: [
               /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { className: "flex-1", gap: 6, align: "center", children: [
@@ -1807,10 +1847,7 @@ var ChatSender_default = react.forwardRef(
                 extraFooter,
                 " "
               ] }),
-              /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { align: "center", gap: 6, children: [
-                /* @__PURE__ */ jsxRuntime.jsx(SpeechButton, {}),
-                loading ? /* @__PURE__ */ jsxRuntime.jsx(LoadingButton, { disabled: false }) : /* @__PURE__ */ jsxRuntime.jsx(SendButton, { ...sendBtnProps })
-              ] })
+              /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { align: "center", gap: 6, children: loading ? /* @__PURE__ */ jsxRuntime.jsx(LoadingButton, { disabled: false }) : /* @__PURE__ */ jsxRuntime.jsx(SendButton, { ...sendBtnProps }) })
             ] }),
             extraFooterBelow
           ] });
@@ -1873,12 +1910,16 @@ var ChatSender_default2 = react.forwardRef(
       () => conversationState.messages[conversationState.active.id] || {},
       [conversationState.messages[conversationState.active.id]]
     );
+    const referenceContent = react.useMemo(() => {
+      const content = chatMessage.references?.content?.name;
+      if (content) {
+        return common.markdownToText(content);
+      }
+    }, [chatMessage.references]);
     const referenceHandle = (con) => {
       if (chatMessage?.loading) return;
       chatStore.setContent(con);
-      setTimeout(() => {
-        chatStore.sendMessage();
-      }, 10);
+      chatStore.sendMessage();
     };
     return /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { vertical: true, gap: 8, className: "zero-chat-sender", children: [
       /* @__PURE__ */ jsxRuntime.jsx(common.RenderWrapper, { control: prompts, DefaultComponent: SenderPromptsItems_default }),
@@ -1936,9 +1977,9 @@ var ChatSender_default2 = react.forwardRef(
             {
               title: /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { gap: 4, children: [
                 /* @__PURE__ */ jsxRuntime.jsx(icons.EnterOutlined, {}),
-                /* @__PURE__ */ jsxRuntime.jsx(antd.Typography.Text, { type: "secondary", ellipsis: true, children: chatMessage?.references?.content?.name })
+                /* @__PURE__ */ jsxRuntime.jsx(antd.Typography.Text, { type: "secondary", ellipsis: true, children: referenceContent })
               ] }),
-              open: !!chatMessage?.references?.content?.name,
+              open: !!referenceContent,
               onOpenChange: () => chatStore.setReferences(),
               classNames: {
                 header: styles_module_default3.nsSenderReferenceHeaderTitle,