@zero-library/chat-agent 2.2.14 → 2.2.16

package/dist/index.d.mts

@@ -930,6 +930,37 @@ interface ChatHooks {
      * @param isBuiltIn - 是否为内嵌布局
      */
     onAfterFilePreview?: (file?: FileUrlType | FileTextType, isBuiltIn?: boolean) => void;
+    /**
+     * 设置引用消息前回调
+     * 在设置引用消息前调用，返回false可阻止设置
+     *
+     * @param references - 要设置的引用消息, 空则表示正在清除引用消息
+     * @returns boolean | Promise<boolean> | void - 返回false阻止设置
+     *
+     * @example
+     * onBeforeSetReferences: (references) => {
+     *   // 验证引用消息内容
+     *   if (!references || !references.content) {
+     *     message.error('引用消息内容不能为空')
+     *     return false
+     *   }
+     *   return true
+     * }
+     */
+    onBeforeSetReferences?: (references?: ReferencesType) => boolean | Promise<boolean> | void;
+    /**
+     * 设置引用消息后回调
+     * 在设置引用消息完成后调用
+     *
+     * @param references - 已设置的引用消息, 空则表示已清除引用消息
+     *
+     * @example
+     * onAfterSetReferences: (references) => {
+     *   console.log('引用消息已设置:', references)
+     *   // 可以在这里更新界面状态或其他相关操作
+     * }
+     */
+    onAfterSetReferences?: (references?: ReferencesType) => void;
 }
 /**
  * 聊天参数接口
@@ -1387,7 +1418,7 @@ interface ChatHandle {
      * 设置引用消息
      * 在输入框中设置引用内容，将在下次发送消息时包含引用信息
      *
-     * @param references - 引用内容对象
+     * @param references - 引用内容对象, 空则表示清除引用消息
      *
      * @example
      * // 设置Markdown引用

package/dist/index.d.ts

@@ -930,6 +930,37 @@ interface ChatHooks {
      * @param isBuiltIn - 是否为内嵌布局
      */
     onAfterFilePreview?: (file?: FileUrlType | FileTextType, isBuiltIn?: boolean) => void;
+    /**
+     * 设置引用消息前回调
+     * 在设置引用消息前调用，返回false可阻止设置
+     *
+     * @param references - 要设置的引用消息, 空则表示正在清除引用消息
+     * @returns boolean | Promise<boolean> | void - 返回false阻止设置
+     *
+     * @example
+     * onBeforeSetReferences: (references) => {
+     *   // 验证引用消息内容
+     *   if (!references || !references.content) {
+     *     message.error('引用消息内容不能为空')
+     *     return false
+     *   }
+     *   return true
+     * }
+     */
+    onBeforeSetReferences?: (references?: ReferencesType) => boolean | Promise<boolean> | void;
+    /**
+     * 设置引用消息后回调
+     * 在设置引用消息完成后调用
+     *
+     * @param references - 已设置的引用消息, 空则表示已清除引用消息
+     *
+     * @example
+     * onAfterSetReferences: (references) => {
+     *   console.log('引用消息已设置:', references)
+     *   // 可以在这里更新界面状态或其他相关操作
+     * }
+     */
+    onAfterSetReferences?: (references?: ReferencesType) => void;
 }
 /**
  * 聊天参数接口
@@ -1387,7 +1418,7 @@ interface ChatHandle {
      * 设置引用消息
      * 在输入框中设置引用内容，将在下次发送消息时包含引用信息
      *
-     * @param references - 引用内容对象
+     * @param references - 引用内容对象, 空则表示清除引用消息
      *
      * @example
      * // 设置Markdown引用

package/dist/index.esm.js

@@ -274,7 +274,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 buildUrlParams(params, getWebSocketUrl(`${baseURL}/lolr/conversation/ws/subscribe`), "comma");
@@ -681,8 +681,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;
@@ -1185,13 +1190,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();
 
@@ -1276,6 +1274,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: (isBoolean(avatar) ? avatar : avatar?.user) ? /* @__PURE__ */ jsx(
@@ -1290,6 +1289,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: (isBoolean(avatar) ? avatar : avatar?.agent) ? { src: conversationState.active.member.agent?.memberAvatar, icon: /* @__PURE__ */ jsx(OpenAIOutlined, {}) } : null
@@ -1297,6 +1297,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: (isBoolean(avatar) ? avatar : avatar?.other) ? { src: conversationState.active.member.other?.memberAvatar, icon: /* @__PURE__ */ jsx(OpenAIOutlined, {}) } : null
@@ -1344,9 +1345,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 = 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,