@zero-library/chat-agent 2.2.14 → 2.2.16

package/dist/index.cjs.js

@@ -282,7 +282,7 @@ var classifyTime = (timestamp) => {
   }
 };
 var replaceMarkdownTags = (str) => {
-  return str.replace(/<think>/g, ":::alert type=think data={content:'").replace(/<\/think>/g, "'} :::");
+  return str.replace(/<think>/g, ":::alert type=think data={content:'").replace(/<\/think>/g, "'} :::\n");
 };
 var getChatSocketUrl = (baseURL, params) => {
   return common.buildUrlParams(params, common.getWebSocketUrl(`${baseURL}/lolr/conversation/ws/subscribe`), "comma");
@@ -689,8 +689,13 @@ function createChatStore() {
   const setSpeakHuman = (speakHuman) => {
     conversation.messages[conversation.active.id].speakHuman = speakHuman;
   };
-  const setReferences = (references) => {
+  const setReferences = async (references) => {
+    const canProceed = await config.hooks?.onBeforeSetReferences?.(references);
+    if (canProceed === false) {
+      throw new Error("\u64CD\u4F5C\u88AB\u963B\u6B62");
+    }
     conversation.messages[conversation.active.id].references = references;
+    config.hooks?.onAfterSetReferences?.(references);
   };
   const setContent = (content) => {
     conversation.messages[conversation.active.id].content = content;
@@ -1193,13 +1198,6 @@ var MessageRender_default = ({ message: message2, placement, onFilePreview }) =>
   ] }) });
 };
 
-// src/core/constants.ts
-var MEMBER_TYPE = {
-  1: "user",
-  2: "other",
-  3: "agent"
-};
-
 // src/ui/common/BubbleListItems.tsx
 init_Context();
 
@@ -1284,6 +1282,7 @@ var BubbleListItems_default = ({ firstMessage = false, welcomeMessage = true, av
     const roles = {};
     if (conversationState.active.member.user) {
       roles.user = {
+        id: conversationState.active.member.user.memberId,
         user: "user",
         placement: "end",
         avatar: (common.isBoolean(avatar) ? avatar : avatar?.user) ? /* @__PURE__ */ jsxRuntime.jsx(
@@ -1298,6 +1297,7 @@ var BubbleListItems_default = ({ firstMessage = false, welcomeMessage = true, av
     }
     if (conversationState.active.member.agent) {
       roles.agent = {
+        id: conversationState.active.member.agent.memberId,
         user: "agent",
         placement: "start",
         avatar: (common.isBoolean(avatar) ? avatar : avatar?.agent) ? { src: conversationState.active.member.agent?.memberAvatar, icon: /* @__PURE__ */ jsxRuntime.jsx(icons.OpenAIOutlined, {}) } : null
@@ -1305,6 +1305,7 @@ var BubbleListItems_default = ({ firstMessage = false, welcomeMessage = true, av
     }
     if (conversationState.active.member.other) {
       roles.other = {
+        id: conversationState.active.member.other.memberId,
         user: "other",
         placement: "start",
         avatar: (common.isBoolean(avatar) ? avatar : avatar?.other) ? { src: conversationState.active.member.other?.memberAvatar, icon: /* @__PURE__ */ jsxRuntime.jsx(icons.OpenAIOutlined, {}) } : null
@@ -1352,9 +1353,13 @@ var BubbleListItems_default = ({ firstMessage = false, welcomeMessage = true, av
     ] : [],
     [chatMessage?.questionList]
   );
+  const getRole = (sender) => {
+    const roleKey = Object.keys(conversationRoles).find((key) => conversationRoles[key].id === sender.id);
+    return conversationRoles[roleKey] || {};
+  };
   const chatRecords = react.useMemo(() => {
     return (chatMessage?.message || []).map((message2, index) => {
-      const role = conversationRoles[MEMBER_TYPE[message2.sender.type]] || {};
+      const role = getRole(message2.sender);
       return {
         key: message2.id,
         placement: role.placement,