@zero-library/chat-agent 2.2.1 → 2.2.3

package/dist/index.cjs.js

@@ -62,8 +62,14 @@ var init_AppCard = __esm({
     init_styles_module();
     AppCard_default = ({ data, loading }) => {
       const chatStore = useChatStore();
-      const onClick = () => {
-        chatStore.switchAgentConversation(data.appKey);
+      const onClick = async () => {
+        if (data.appKey) {
+          const canProceed = await chatStore.config.hooks?.onAppRightVerify?.(data.appKey);
+          if (canProceed === false) {
+            return;
+          }
+        }
+        chatStore.switchAgentConversation(data.agentId);
       };
       return /* @__PURE__ */ jsxRuntime.jsx(antd.Spin, { spinning: loading, size: "small", children: /* @__PURE__ */ jsxRuntime.jsxs(antd.Flex, { vertical: true, gap: 4, align: "center", className: styles_module_default.appCard, onClick, children: [
         /* @__PURE__ */ jsxRuntime.jsx("img", { src: data.iconUrl }),
@@ -99,7 +105,7 @@ var DocDrawer_default;
 var init_DocDrawer = __esm({
   "src/ui/common/markdownAlert/components/DocDrawer.tsx"() {
     init_Context();
-    DocDrawer_default = ({ title, open, onClose, fileUrl }) => {
+    DocDrawer_default = ({ title, open, onClose, paramsStr }) => {
       const chatStore = useChatStore();
       const configState = valtio.useSnapshot(chatStore.config);
       const [content, setContent] = react.useState("");
@@ -107,7 +113,7 @@ var init_DocDrawer = __esm({
       const getContent = async () => {
         try {
           setLoading(true);
-          const res = await configState.services.request.docQuery(fileUrl);
+          const res = await configState.services.request.docQuery(paramsStr);
           setContent(res.data?.content);
         } finally {
           setLoading(false);
@@ -115,7 +121,7 @@ var init_DocDrawer = __esm({
       };
       react.useEffect(() => {
         getContent();
-      }, [fileUrl]);
+      }, [paramsStr]);
       return /* @__PURE__ */ jsxRuntime.jsx(antd.Drawer, { title, push: false, width: "100%", open, onClose, children: /* @__PURE__ */ jsxRuntime.jsx(antd.Spin, { spinning: loading, children: content ? /* @__PURE__ */ jsxRuntime.jsx(common.RenderMarkdown, { content }) : /* @__PURE__ */ jsxRuntime.jsx(antd.Empty, { className: "m-t-32" }) }) });
     };
   }
@@ -130,20 +136,28 @@ var IndexQuote_default;
 var init_IndexQuote = __esm({
   "src/ui/common/markdownAlert/IndexQuote.tsx"() {
     init_DocDrawer();
-    IndexQuote_default = ({ data, loading }) => {
+    IndexQuote_default = ({ data, loading, message: message2 }) => {
       const [open, setOpen] = react.useState(false);
+      const citation = react.useMemo(() => {
+        let citations = [];
+        try {
+          citations = JSON.parse(message2.citations);
+        } catch (e) {
+        }
+        return citations?.find((item) => String(item.citationId) === String(data.citationId));
+      }, [data.citationId, message2.citations]);
       const onClick = () => {
-        if (!data.fileUrl) return;
-        if (common.isExternal(data.fileUrl)) {
-          window.open(data.fileUrl, "_blank");
+        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: data.fileName }) }) });
+      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, { bordered: false, color: "processing", className: "cursor-pointer", children: data.index }) }) }),
-        open && /* @__PURE__ */ jsxRuntime.jsx(DocDrawer_default, { fileUrl: data.fileUrl, title: data.fileName, open, onClose: () => setOpen(false) })
+        /* @__PURE__ */ jsxRuntime.jsx(antd.Spin, { spinning: loading, size: "small", children: /* @__PURE__ */ jsxRuntime.jsx(antd.Popover, { content: Content, children: /* @__PURE__ */ jsxRuntime.jsx(antd.Tag, { bordered: false, color: "processing", className: "cursor-pointer", children: data?.citationName }) }) }),
+        open && /* @__PURE__ */ jsxRuntime.jsx(DocDrawer_default, { paramsStr: citation?.citationUrl, title: citation?.citationName, open, onClose: () => setOpen(false) })
       ] });
     };
   }
@@ -163,7 +177,7 @@ var init_MdEdit = __esm({
       const chatStore = useChatStore();
       const [value, setValue] = react.useState("");
       const onOk = () => {
-        chatStore.sendMessage("\u5F00\u59CB\u5199\u4F5C", [], { outline: value });
+        chatStore.sendMessage("\u5F00\u59CB\u5199\u4F5C", [], { citation: value });
       };
       react.useEffect(() => {
         setValue(data.content);
@@ -208,51 +222,6 @@ var init_FileEdit = __esm({
   }
 });
 
-// src/ui/common/markdownAlert/QuoteList.tsx
-var QuoteList_exports = {};
-__export(QuoteList_exports, {
-  default: () => QuoteList_default
-});
-var QuoteList_default;
-var init_QuoteList = __esm({
-  "src/ui/common/markdownAlert/QuoteList.tsx"() {
-    init_DocDrawer();
-    init_styles_module();
-    QuoteList_default = ({ data, loading }) => {
-      const [open, setOpen] = react.useState(false);
-      const [selectItem, setSelectItem] = react.useState({});
-      const onClick = (item) => {
-        if (!item.fileUrl) return;
-        if (common.isExternal(item.fileUrl)) {
-          window.open(item.fileUrl, "_blank");
-        } else {
-          setSelectItem(item);
-          setOpen(true);
-        }
-      };
-      return /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
-        /* @__PURE__ */ jsxRuntime.jsx(
-          antd.Collapse,
-          {
-            className: styles_module_default.quoteList,
-            bordered: false,
-            ghost: true,
-            expandIcon: ({ isActive }) => /* @__PURE__ */ jsxRuntime.jsx(icons.CaretRightOutlined, { rotate: isActive ? 90 : 0 }),
-            items: [
-              {
-                key: "1",
-                label: "\u53C2\u8003\u8D44\u6599",
-                children: /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { vertical: true, gap: 4, children: data.list.map((item) => /* @__PURE__ */ jsxRuntime.jsx("p", { onClick: () => !loading && onClick(item), className: "cursor-pointer", children: item.title }, item.fileUrl)) })
-              }
-            ]
-          }
-        ),
-        open && /* @__PURE__ */ jsxRuntime.jsx(DocDrawer_default, { fileUrl: selectItem.fileUrl, title: selectItem.title, open, onClose: () => setOpen(false) })
-      ] });
-    };
-  }
-});
-
 // src/ui/common/markdownAlert/Think.tsx
 var Think_exports = {};
 __export(Think_exports, {
@@ -808,8 +777,7 @@ function createChatStore() {
     }
   };
   const switchAgentConversation = async (id, strategy) => {
-    const agentId = String(id);
-    await switchAgent(agentId);
+    await switchAgent(String(id));
     const conversationId = await resolveConversationId(receiver.active.id, strategy);
     if (conversationId) {
       await switchConversation(conversationId);
@@ -920,60 +888,84 @@ function createChatStore() {
   const cancelReceive = async () => {
     console.log("\u53D6\u6D88\u63A5\u6536");
     await config.services.request.conversationStop(conversation.active.id);
+    conversation.messages[conversation.active.id].loading = false;
   };
   const updateConversationTitle = (conversationId, title) => {
-    const idx = conversations.list.items.findIndex((item) => item.id === conversationId);
-    if (idx === -1) return;
-    conversations.list.items[idx].title = title;
+    if (title && conversations.list.items.length) {
+      const idx = conversations.list.items.findIndex((item) => item.id === conversationId);
+      if (idx === -1) return;
+      conversations.list.items[idx].label = title;
+    }
   };
-  const findMsgIndex = (conversationId, msgId) => {
-    const messages = conversation.messages[conversationId]?.message || [];
-    return messages.findLastIndex((item) => item.id === msgId);
+  const findMsgIndex = (msgId, messages) => {
+    return (messages || []).findLastIndex((item) => item.id === msgId);
   };
   const startCallback = (msg) => {
-    if (msg.conversationTitle && conversations.list.items.length) {
-      updateConversationTitle(msg.conversationId, msg.conversationTitle);
-    }
     removeQuestionList(msg.conversationId);
-    const messages = conversation.messages[msg.conversationId].message;
-    messages.push(msg.sender.type === 3 ? { ...msg, msgContent: "\u6B63\u5728\u52AA\u529B\u601D\u8003\u4E2D" } : msg);
+    const messages = conversation.messages[msg.conversationId]?.message;
+    if (!messages) return;
+    messages.push(msg.sender.type === 3 ? { ...msg, tempContent: "\u6B63\u5728\u52AA\u529B\u601D\u8003\u4E2D" } : msg);
     conversation.messages[msg.conversationId].loading = true;
   };
   const stepCallback = (msg) => {
-    const messages = conversation.messages[msg.conversationId].message;
-    const idx = findMsgIndex(msg.conversationId, msg.id);
+    const messages = conversation.messages[msg.conversationId]?.message;
+    const idx = findMsgIndex(msg.id, messages);
     if (idx === -1) return;
-    messages[idx] = msg;
     conversation.messages[msg.conversationId].loading = true;
+    messages[idx] = {
+      ...messages[idx],
+      params: messages[idx]?.params || msg.params,
+      type: msg.type,
+      tempContent: msg.msgContent
+    };
   };
   const contentCallback = (msg) => {
-    const messages = conversation.messages[msg.conversationId].message;
-    const idx = findMsgIndex(msg.conversationId, msg.id);
+    const messages = conversation.messages[msg.conversationId]?.message;
+    const idx = findMsgIndex(msg.id, messages);
     if (idx === -1) return;
-    const message2 = messages[idx];
     conversation.messages[msg.conversationId].loading = true;
-    if (message2?.type === "STEP_STARTED" || message2?.type === "TEXT_MESSAGE_START") {
-      messages[idx] = msg;
-      return;
-    }
     messages[idx] = {
-      ...message2,
-      msgContent: message2.msgContent + (msg.msgContent || "")
+      ...messages[idx],
+      params: messages[idx]?.params || msg.params,
+      type: msg.type,
+      tempContent: "",
+      msgContent: messages[idx].msgContent + (msg.msgContent || "")
     };
   };
   const endCallback = (msg) => {
-    const messages = conversation.messages[msg.conversationId].message;
-    const idx = findMsgIndex(msg.conversationId, msg.id);
+    const messages = conversation.messages[msg.conversationId]?.message;
+    const idx = findMsgIndex(msg.id, messages);
     if (idx === -1) return;
-    messages[idx] = { ...messages[idx], type: void 0 };
+    messages[idx] = {
+      ...messages[idx],
+      tempContent: "",
+      type: void 0
+    };
     conversation.messages[msg.conversationId].loading = false;
     getQuestionList(msg.conversationId, msg.id);
   };
+  const citationCallback = (msg) => {
+    const messages = conversation.messages[msg.conversationId]?.message;
+    const idx = findMsgIndex(msg.id, messages);
+    if (idx === -1) return;
+    messages[idx] = {
+      ...messages[idx],
+      citations: msg.msgContent
+    };
+  };
   const errCallback = (msg) => {
     const messages = conversation.messages[msg.conversationId].message;
-    const idx = findMsgIndex(msg.conversationId, msg.id);
+    const idx = findMsgIndex(msg.id, messages);
     if (idx === -1) return;
-    messages[idx] = { ...msg, type: void 0 };
+    messages[idx] = {
+      ...messages[idx],
+      params: messages[idx]?.params || msg.params,
+      tempContent: msg.msgContent,
+      type: void 0,
+      // 报错后清空消息内容
+      msgContent: "",
+      citations: ""
+    };
     conversation.messages[msg.conversationId].loading = false;
   };
   const acceptMessage = async (newMessage) => {
@@ -983,6 +975,7 @@ function createChatStore() {
     if (canProceed === false) {
       throw new Error("\u64CD\u4F5C\u88AB\u963B\u6B62");
     }
+    updateConversationTitle(conversationId, newMessage.data.conversationTitle);
     switch (newMessage.data.type) {
       case "TEXT_MESSAGE_START":
         startCallback(newMessage.data);
@@ -996,6 +989,9 @@ function createChatStore() {
       case "TEXT_MESSAGE_END":
         endCallback(newMessage.data);
         break;
+      case "TEXT_MESSAGE_CITATION":
+        citationCallback(newMessage.data);
+        break;
       case "RUN_ERROR":
         errCallback(newMessage.data);
         break;
@@ -1076,7 +1072,7 @@ var customComponents = {
   indexQuote: () => Promise.resolve().then(() => (init_IndexQuote(), IndexQuote_exports)),
   mdEdit: () => Promise.resolve().then(() => (init_MdEdit(), MdEdit_exports)),
   fileEdit: () => Promise.resolve().then(() => (init_FileEdit(), FileEdit_exports)),
-  quoteList: () => Promise.resolve().then(() => (init_QuoteList(), QuoteList_exports)),
+  // quoteList: () => import('./QuoteList'),
   think: () => Promise.resolve().then(() => (init_Think(), Think_exports))
 };
 
@@ -1088,13 +1084,25 @@ var styles_module_default2 = {
   chatQuoteMsg: "styles_module_chatQuoteMsg"
 };
 var MessageRender_default = ({ message: message2, placement }) => {
-  return /* @__PURE__ */ jsxRuntime.jsx(antd.Flex, { vertical: true, children: !(message2.msgContent || message2.msgFiles.length) ? /* @__PURE__ */ jsxRuntime.jsx(antd.Typography, { children: /* @__PURE__ */ jsxRuntime.jsx(antd.Spin, { size: "small" }) }) : /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
-    message2.msgContent && /* @__PURE__ */ jsxRuntime.jsx(
+  const content = 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) {
+    }
+    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(
       x.Bubble,
       {
         placement,
         className: classNames9__default.default({ [styles_module_default2.loadingMessage]: message2.type }),
-        content: /* @__PURE__ */ jsxRuntime.jsx(common.RenderMarkdown, { content: replaceThinkTags(message2.msgContent), customComponents })
+        content: /* @__PURE__ */ jsxRuntime.jsx(common.RenderMarkdown, { content: replaceThinkTags(content), customComponents, message: message2 })
       }
     ),
     message2.msgFiles?.map((file) => /* @__PURE__ */ jsxRuntime.jsx(
@@ -1117,7 +1125,7 @@ var MessageRender_default = ({ message: message2, placement }) => {
       },
       file.content
     )),
-    message2.quoteMsg?.id && /* @__PURE__ */ jsxRuntime.jsx(
+    message2.quoteMsg?.id ? /* @__PURE__ */ jsxRuntime.jsx(
       x.Bubble,
       {
         className: classNames9__default.default(styles_module_default2.chatQuoteMsg),
@@ -1125,7 +1133,7 @@ var MessageRender_default = ({ message: message2, 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", children: message2.quoteMsg.msgContent })
+            /* @__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: [
             /* @__PURE__ */ jsxRuntime.jsx("span", { children: "\u3010\u5F15\u7528\u6587\u4EF6\u3011\uFF1A" }),
@@ -1143,6 +1151,16 @@ var MessageRender_default = ({ message: message2, placement }) => {
           ] })
         ] })
       }
+    ) : 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 }) })
+        ] }) })
+      }
     )
   ] }) });
 };
@@ -2097,13 +2115,13 @@ var layouts_default = react.forwardRef(({ theme, params, userInfo, hooks, layout
               value: configState.preview.file.content,
               onQuote: (text) => {
                 chatStore.setReferences({
-                  type: 1,
+                  type: 2,
                   content: {
                     name: text,
                     markdown: text
                   },
                   params: {
-                    outline: text
+                    citation: text
                   }
                 });
               },