@zero-library/chat-agent 2.2.15 → 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

@@ -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;