@zero-library/chat-copilot 1.0.8 → 1.0.10

package/dist/index.cjs.js

@@ -445,6 +445,9 @@ var createChatService = (request, config) => {
   const getPreviewUrl = (agentId, conversationId, path) => {
     return `${config?.baseURL}/agents/${agentId}/${conversationId}/files/download?path=${encodeURIComponent(path)}&${TOKEN_KEY}=${tokenManager.get()}`;
   };
+  const docQuery = (params) => {
+    return request.get(`/library/cloud/document?${params}`);
+  };
   return {
     fetchModelList,
     fetchAgentInfo,
@@ -458,7 +461,8 @@ var createChatService = (request, config) => {
     getAgentUserInput,
     feedbackUpdate,
     chatUpload,
-    getPreviewUrl
+    getPreviewUrl,
+    docQuery
   };
 };
 
@@ -1160,7 +1164,7 @@ function createChatStore() {
           messages.push(finalMsg);
         } else {
           const prevContent = messages[idx]?.content || [];
-          messages[idx].content = [...prevContent, newMessage.content[0]];
+          messages[idx].content = [...prevContent, ...newMessage.content];
         }
         conversation.messages[newMessage.conversationId].loading = true;
         break;
@@ -2076,6 +2080,45 @@ var FileView_default = ({ data, loading }) => {
     /* @__PURE__ */ jsxRuntime.jsx("div", { className: "text-ellipsis", title: data.fileName, children: data.fileName })
   ] }) });
 };
+var DocDrawer_default = ({ title, open, onClose, paramsStr }) => {
+  const chatStore = useChatStore();
+  const configState = valtio.useSnapshot(chatStore.config);
+  const [content, setContent] = React6.useState("");
+  const [loading, setLoading] = React6.useState(false);
+  const getContent = async () => {
+    try {
+      setLoading(true);
+      const res = await configState.services.request.docQuery(paramsStr);
+      setContent(res.data?.content || "");
+    } finally {
+      setLoading(false);
+    }
+  };
+  React6.useEffect(() => {
+    getContent();
+  }, [paramsStr]);
+  return /* @__PURE__ */ jsxRuntime.jsx(antd.Drawer, { title, push: false, size: "100%", open, onClose, children: /* @__PURE__ */ jsxRuntime.jsx(antd.Spin, { spinning: loading, children: content ? /* @__PURE__ */ jsxRuntime.jsx(XMarkdown__default.default, { content }) : /* @__PURE__ */ jsxRuntime.jsx(antd.Empty, { className: "m-t-32" }) }) });
+};
+var IndexQuote_default = ({ data, loading, message: message3 }) => {
+  const [open, setOpen] = React6.useState(false);
+  const citation = React6.useMemo(() => {
+    const citations = message3.content?.find((item) => item.type === "citation")?.citations ?? [];
+    return citations?.find((item) => String(item.citationId) === String(data.citationId));
+  }, [data.citationId, message3.content]);
+  const onClick = () => {
+    if (!citation?.citationUrl) return;
+    if (common.isExternal(citation.citationUrl)) {
+      window.open(citation.citationUrl, "_blank");
+    } else {
+      setOpen(true);
+    }
+  };
+  const Content = /* @__PURE__ */ jsxRuntime.jsx("div", { children: /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { align: "center", gap: 8, children: /* @__PURE__ */ jsxRuntime.jsx("a", { onClick, children: citation?.citationName }) }) });
+  return /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "inline-block m-l-8", children: [
+    /* @__PURE__ */ jsxRuntime.jsx(antd.Spin, { spinning: loading, size: "small", children: /* @__PURE__ */ jsxRuntime.jsx(antd.Popover, { content: Content, children: /* @__PURE__ */ jsxRuntime.jsx(antd.Tag, { variant: "filled", color: "processing", className: "cursor-pointer", children: data?.citationName }) }) }),
+    open && /* @__PURE__ */ jsxRuntime.jsx(DocDrawer_default, { paramsStr: citation?.citationUrl, title: citation?.citationName, open, onClose: () => setOpen(false) })
+  ] });
+};
 var MdEdit_default = ({ data, loading }) => {
   const chatStore = useChatStore();
   const [value, setValue] = React6.useState("");
@@ -2133,11 +2176,15 @@ var PreviewLink_default = ({ data, loading, domNode, ...rest }) => {
   };
   return /* @__PURE__ */ jsxRuntime.jsx("a", { ...rest, onClick: handleClick, children: data });
 };
+var Think_default = ({ data }) => {
+  return /* @__PURE__ */ jsxRuntime.jsx(xV2.Think, { title: "\u6DF1\u5EA6\u601D\u8003", className: "m-b-16", children: data });
+};
 
 // src/components/XMarkdown/styles.module.less
 var styles_module_default4 = {
   xMarkdownStyle: "styles_module_xMarkdownStyle"
 };
+var MARKDOWN_PAYLOAD_PLACEHOLDER_PREFIX = "XMD_PAYLOAD_";
 var singleTildeExtension = {
   name: "singleTilde",
   level: "inline",
@@ -2164,44 +2211,86 @@ var getMarkdownConfig = (options) => {
     breaks: options?.breaks ?? false
   };
 };
-var adaptMarkdownComponent = (Component, message3) => {
-  const WrappedComponent = ({ children, streamStatus, ...rest }) => /* @__PURE__ */ jsxRuntime.jsx(Component, { message: message3, ...rest, data: common.safeParseJson(children, { repair: true }), loading: streamStatus === "loading" });
-  WrappedComponent.displayName = `AdaptedMarkdownComponent(${Component.displayName || Component.name || "Anonymous"})`;
-  return WrappedComponent;
-};
-var adaptMarkdownComponents = (components, message3) => Object.fromEntries(Object.entries(components ?? {}).map(([key, Component]) => [key, adaptMarkdownComponent(Component, message3)]));
+function makeMarkdownPayloadPlaceholder(index) {
+  return `${MARKDOWN_PAYLOAD_PLACEHOLDER_PREFIX}${index}`;
+}
+function resolveMappedMarkdownPayload(value, payloadMap) {
+  if (common.isString(value)) {
+    if (new RegExp(`^${MARKDOWN_PAYLOAD_PLACEHOLDER_PREFIX}\\d+$`).test(value)) {
+      return common.safeParseJson(payloadMap.get(value), { repair: true });
+    }
+  }
+  return value;
+}
 function formatMixedContent(content) {
   return content.replace(/<!DOCTYPE html>[\s\S]*?<\/html>/gi, (match) => `\`\`\`html
 ${match}
 \`\`\``);
 }
-var XMarkdown_default = ({ message: message3, components, content, ...rest }) => {
+function preprocessCustomTagPayloads(content = "", payloadTagPattern) {
+  let id = 0;
+  const payloadMap = /* @__PURE__ */ new Map();
+  const processedContent = content.replace(
+    new RegExp(`(?<opening><(?<tag>${payloadTagPattern})\\b[^>]*>)(?<inner>[\\s\\S]*?)(?<closing><\\/\\k<tag>\\s*>|$)`, "gi"),
+    (match, _opening, _tag, _inner, _closing, _offset, _source, groups) => {
+      const openingText = groups?.opening ?? "";
+      if (!openingText) {
+        return match;
+      }
+      const innerText = groups?.inner ?? "";
+      const closingText = groups?.closing ?? "";
+      const placeholder = makeMarkdownPayloadPlaceholder(id++);
+      payloadMap.set(placeholder, innerText.trim());
+      return `${openingText}${placeholder}${closingText}`;
+    }
+  );
+  return {
+    content: processedContent,
+    payloadMap
+  };
+}
+var adaptMarkdownComponent = (Component, payloadMap = /* @__PURE__ */ new Map(), message3) => {
+  const WrappedComponent = ({ children, streamStatus, ...rest }) => {
+    const rawData = resolveMappedMarkdownPayload(children, payloadMap);
+    return /* @__PURE__ */ jsxRuntime.jsx(Component, { message: message3, ...rest, data: rawData, loading: streamStatus === "loading" });
+  };
+  WrappedComponent.displayName = `AdaptedMarkdownComponent(${Component.displayName || Component.name || "Anonymous"})`;
+  return WrappedComponent;
+};
+var adaptMarkdownComponents = (components, payloadMap = /* @__PURE__ */ new Map(), message3) => Object.fromEntries(Object.entries(components ?? {}).map(([key, Component]) => [key, adaptMarkdownComponent(Component, payloadMap, message3)]));
+var XMarkdown_default = ({ message: message3, components = {}, content = "", ...rest }) => {
   const config = React6.useMemo(() => getMarkdownConfig(), []);
-  const markdownComponents = React6.useMemo(
-    () => adaptMarkdownComponents(
-      {
-        code: CodeHighlight_default,
-        agentnavigate: AgentNavigate_default,
-        a: PreviewLink_default,
-        appcard: AppCard_default,
-        fileview: FileView_default,
-        charts: Charts_default,
-        // indexquote: IndexQuote,
-        mdedit: MdEdit_default,
-        fileedit: FileEdit_default,
-        ...components
-      },
-      message3
-    ),
-    [components, message3]
+  const formattedContent = React6.useMemo(() => formatMixedContent(content), [content]);
+  const newComponents = React6.useMemo(
+    () => ({
+      think: Think_default,
+      code: CodeHighlight_default,
+      agentnavigate: AgentNavigate_default,
+      a: PreviewLink_default,
+      appcard: AppCard_default,
+      fileview: FileView_default,
+      charts: Charts_default,
+      indexquote: IndexQuote_default,
+      mdedit: MdEdit_default,
+      fileedit: FileEdit_default,
+      ...components
+    }),
+    [components]
+  );
+  const componentKeys = React6.useMemo(() => Object.keys(newComponents), [newComponents]);
+  const payloadTagPattern = React6.useMemo(() => componentKeys.join("|"), [componentKeys]);
+  const { content: processedContent, payloadMap } = React6.useMemo(
+    () => preprocessCustomTagPayloads(formattedContent, payloadTagPattern),
+    [formattedContent, payloadTagPattern]
   );
+  const markdownComponents = React6.useMemo(() => adaptMarkdownComponents(newComponents, payloadMap, message3), [componentKeys, message3, payloadMap]);
   return /* @__PURE__ */ jsxRuntime.jsx(
     XMarkdown__default.default,
     {
       className: classNames2__default.default("x-markdown-light", styles_module_default4.xMarkdownStyle),
       config,
       ...rest,
-      content: formatMixedContent(content),
+      content: processedContent,
       components: markdownComponents
     }
   );
@@ -2557,7 +2646,7 @@ var Button7 = ({ label, variant = "default", onClick, action, onAction }) => {
   return /* @__PURE__ */ jsxRuntime.jsx(antd.Button, { type: variant === "primary" ? "primary" : "default", danger: variant === "danger", onClick: handleClick, style: { borderRadius: 8 }, children: label });
 };
 var Divider = () => /* @__PURE__ */ jsxRuntime.jsx(antd.Divider, { style: { margin: "8px 0" } });
-var Tag = ({ text, color }) => /* @__PURE__ */ jsxRuntime.jsx(antd.Tag, { color, children: text });
+var Tag2 = ({ text, color }) => /* @__PURE__ */ jsxRuntime.jsx(antd.Tag, { color, children: text });
 var Progress = ({
   percent,
   status = "normal",
@@ -2575,7 +2664,7 @@ var A2uiComponents = {
   Switch,
   Button: Button7,
   Divider,
-  Tag,
+  Tag: Tag2,
   Progress,
   ArtifactCard,
   TableViewer,