@easyv/biz-components 1.0.30 → 2.1.1

package/dist/lib/easyv-biz-components/src/utils/fun-ASR/fun-ASR-manager.d.ts

@@ -29,7 +29,7 @@ export interface WebSocketConnectMethodConfig {
      * online 模式：采用流式处理方式，在语音数据输入的过程中，系统会逐块对语音进行实时识别。只要有新的语音数据块到达，就立即对其进行处理并输出部分识别结果，而无需等待整个语音输入结束。
      * offline 模式：需要等待完整的语音数据全部输入后，才开始进行识别处理。在处理过程中，系统会对整个语音数据进行全面的分析和处理，以得出最终的识别结果。
      */
-    mode?: '2pass' | 'online' | 'offline';
+    mode?: "2pass" | "online" | "offline";
     /** 是否是文件上传模式，当前模式暂不支持 */
     isFileMode?: boolean;
     /** 热词设置，key 是热词，value 是权重 */
@@ -40,7 +40,7 @@ export interface RecorderConfig {
      * 输出类型：mp3,wav，wav输出文件尺寸超大不推荐使用，但mp3编码支持会导致js文件超大，如果不需支持mp3可以使js文件大幅减小
      * pcm 文件无法直接播放，转成wav文件来播放，已提供转换函数 Recorder.pcm2wav
      */
-    type?: 'mp3' | 'wav' | 'pcm';
+    type?: "mp3" | "wav" | "pcm";
     /**
      * 比特率 wav:16或8位，MP3：8kbps 1k/s，8kbps 2k/s 录音文件很小
      */
@@ -67,7 +67,7 @@ export interface RecorderConfig {
     onProcess?: (...arg: any[]) => void;
 }
 export interface ErrorInfo {
-    type: 'webSocketError' | 'startRecord';
+    type: "webSocketError" | "startRecord";
     msg: string;
     isUserNotAllowed?: boolean;
 }
@@ -96,7 +96,7 @@ export declare class FunASRManager {
     sampleBuf: Int16Array<ArrayBuffer>;
     isRec: boolean;
     Recorder: RecorderFn;
-    status: 'live' | 'stop';
+    status: "live" | "stop";
     timer: number;
     propsOnError: undefined | ((errorInfo: ErrorInfo) => void);
     socketConfig: WebSocketConnectMethodConfig;

package/dist/lib/easyv-biz-components/src/utils/index.d.ts

@@ -4,4 +4,7 @@ export * from './vosk-browser-manager';
 export * from './wait-something';
 export * from './ai-agent-message-manager';
 export * from './common';
-export * from './fetchKnowledge';
+export * from './fetch-knowledge';
+export * from './show-modal';
+export * from './easyv-gui-helper';
+export * from './voice-record-manager';

package/dist/lib/easyv-biz-components/src/utils/show-modal/index.d.ts

@@ -0,0 +1 @@
+export * from './show-modal';

package/dist/lib/easyv-biz-components/src/utils/show-modal/show-modal.d.ts

@@ -0,0 +1,15 @@
+import { ReactElement } from 'react';
+
+export interface ShowModalParams {
+    children: (params: {
+        onOk: () => void;
+        onCancel: () => void;
+        visible: boolean;
+    }) => ReactElement;
+}
+export interface ShowModalReturn {
+    destroy: () => void;
+}
+export declare const showModalContainerClass = "easyv-biz-show-modal-container";
+/** children 是一个 render function，接收 onOk, onCancel 参数 */
+export declare const showModal: (params: ShowModalParams) => ShowModalReturn;

package/dist/lib/easyv-biz-components/src/utils/show-modal/test/show-modal.cy.d.ts

@@ -0,0 +1 @@
+

package/dist/lib/easyv-biz-components/src/utils/voice-record-manager/index.d.ts

@@ -0,0 +1 @@
+export * from './voice-record-manager';

package/dist/lib/easyv-biz-components/src/utils/voice-record-manager/voice-record-manager.d.ts

@@ -0,0 +1,114 @@
+export type VoiceStartOptions = {
+    /** 是否使用帧分析，默认 true, 启用后会触发 onFrame 回调 和 音量分析 */
+    useFrameAnalysis?: boolean;
+    onFrame?: (frame: Float32Array, info: {
+        sampleRate: number;
+        timestamp: number;
+        frameSize: number;
+    }) => void;
+    frameSize?: 256 | 512 | 1024 | 2048 | 4096 | 8192 | 16384;
+    channel?: number;
+    hardwareNoiseSuppression?: boolean;
+    noiseReduction?: {
+        enabled?: boolean;
+        mode?: "gate" | "expander";
+        thresholdOffsetDb?: number;
+        minGain?: number;
+        attackMs?: number;
+        releaseMs?: number;
+    };
+};
+export declare class VoiceRecordManager {
+    private mediaRecorder;
+    private audioChunks;
+    private stream;
+    private isRecording;
+    private startTime;
+    private stopTime;
+    private audioContext;
+    private sourceNode;
+    private workletNode;
+    private onFrameCb;
+    private lastWaveform;
+    private lastVolume;
+    private smoothRms;
+    private noiseFloorRms;
+    private lastLevelValue;
+    private readonly meterMinDb;
+    private readonly meterRangeDb;
+    private readonly attackMs;
+    private readonly releaseMs;
+    private readonly levelSmoothFactor;
+    private readonly hysteresisDelta;
+    private readonly epsilon;
+    private nrEnabled;
+    private nrMode;
+    private nrMinGain;
+    private nrAttackMs;
+    private nrReleaseMs;
+    private nrThresholdOffsetDb;
+    private nrGain;
+    /**
+     * 开始录制声音
+     * @throws 没有权限会抛出错误
+     */
+    start(options?: VoiceStartOptions): Promise<void>;
+    /**
+     * 停止录音，并返回音频数据（可指定采样率）
+     * @param sampleRate 采样率（可选），默认为 16k 采样率
+     * @returns Promise<Blob> 返回音频 Blob 数据
+     */
+    stop(sampleRate?: number): Promise<Blob | void>;
+    /**
+     * 释放所有资源
+     */
+    destroy(): void;
+    /** 获取录音状态 */
+    getRecordingStatus(): boolean;
+    /** 获取录音时长（毫秒） */
+    getDuration(): number;
+    /** 获取当前采样率（若启用帧管线） */
+    getSampleRate(): number | null;
+    /**
+     * 音频重采样（webm -> wav，采样率转换）
+     * @param blob 原始音频 Blob
+     * @param targetSampleRate 目标采样率
+     * @returns Promise<Blob> 返回重采样后的音频 Blob
+     */
+    private resampleAudio;
+    /** PCM数据编码为WAV格式 */
+    private encodeWAV;
+    /** 建立帧管线（AudioWorklet） */
+    private setupFramePipeline;
+    /** 关闭帧管线 */
+    private tearDownFramePipeline;
+    /**
+     * 获取当前录制的音量与波形
+     * - volume.rms: 均方根幅度 [0~1]，反映音频能量强度，范围 0–1（越大越响）
+     * - volume.peak: 峰值幅度，当前帧的最大绝对值，范围 0–1（越大越可能出现“爆音”）。
+     * - volume.db: dBFS（相对满刻度），范围 -∞~0（越大越响）
+     * - volume.level: 标准化音量值，范围 0–100（以 -60dB 为底噪阈值）
+     * - waveform: 最近一帧的 Float32 波形
+     */
+    getCurrentVolumeAndWaveform(): {
+        volume: {
+            rms: number;
+            peak: number;
+            db: number;
+            level: number;
+        };
+        waveform: Float32Array | null;
+    };
+    /** 辅助：构建硬件抑噪音频约束 */
+    private buildAudioConstraints;
+    /** 辅助：配置软件噪声抑制参数 */
+    private configureNoiseReduction;
+    /** 辅助：初始化 AudioWorklet 图与连接 */
+    private initAudioWorkletGraph;
+    /** 辅助：处理工作线程消息（拆分分析与抑噪） */
+    private handleWorkletMessage;
+    /** 辅助：分析帧并更新音量/噪声地板 */
+    private analyzeFrame;
+    /** 辅助：应用软件噪声抑制 */
+    private processFrameWithNoiseReduction;
+}

package/dist/lib/easyv-biz-components/src/utils/vosk-browser-manager/audio-manager/audio-manager.d.ts

@@ -1,7 +1,7 @@
 import { SampleRate } from './types';
 
 export declare const defaultSampleRate = 48000;
-export interface AudiManagerConfig {
+export interface AudioManagerConfig {
     /** 一个转换成 string  js 文件。其内容要符合 AudioWorkletProcessor 的要求。
      * - 需使用 registerProcessor 注册，第一个参数为 “audio-worklet-processor”，如： registerProcessor('audio-worklet-processor', audioToFloat32Processor);
      * - 参考：https://developer.mozilla.org/en-US/docs/Web/API/AudioWorkletProcessor
@@ -21,8 +21,8 @@ export declare class AudioManager {
     audioContext?: AudioContext;
     mediaSource?: MediaStreamAudioSourceNode;
     mediaStream?: MediaStream;
-    config: AudiManagerConfig;
-    constructor(config: AudiManagerConfig);
+    config: AudioManagerConfig;
+    constructor(config: AudioManagerConfig);
     init(): Promise<void>;
     start(): Promise<void>;
     stop(): Promise<void>;

package/dist/lib/easyv-biz-components/src/utils/vosk-browser-manager/vosk-browser-manager.d.ts

@@ -1,7 +1,7 @@
-import { AudiManagerConfig, AudioManager } from './audio-manager/audio-manager';
+import { AudioManager, AudioManagerConfig } from './audio-manager/audio-manager';
 import { SampleRate } from './audio-manager/types';
-import type * as Vosk from 'vosk-browser';
-interface VoskBrowserManagerConfig extends AudiManagerConfig {
+import type * as Vosk from "vosk-browser";
+interface VoskBrowserManagerConfig extends AudioManagerConfig {
     vosk: typeof Vosk;
     modelUrl: string;
     /** 加载模型等初始化配置完成后，自动开始监听 */
@@ -22,10 +22,10 @@ export declare class VoskBrowserManager {
     model?: Vosk.Model;
     autoStart: boolean;
     startResolve?: (value?: unknown) => void;
-    status?: 'initializing' | 'initialized';
+    status?: "initializing" | "initialized";
     recognizer?: Vosk.KaldiRecognizer;
     constructor(config: VoskBrowserManagerConfig);
-    setConfig(config: Partial<Pick<VoskBrowserManagerConfig, 'onResult' | 'onPartialResult'>>): void;
+    setConfig(config: Partial<Pick<VoskBrowserManagerConfig, "onResult" | "onPartialResult">>): void;
     init(vosk: typeof Vosk): Promise<void>;
     /** 设置热词会重启识别器
      * - 唤醒词识别阈值非常大，容易误唤醒，慎用！！！

package/dist/lib/easyv-biz-components/src/utils/wait-something/index.d.ts

@@ -6,4 +6,4 @@ export declare function wait(time: number): Promise<unknown>;
  * @param maxWaitTime 最大等待时间，单位是毫秒, 默认是 3 秒
  * @returns 'success' | 'timeout'
  */
-export declare function waitSomeThingBeTrue(waitFn: () => boolean, everyWaitTime: number, maxWaitTime?: number): Promise<'success' | 'timeout'>;
+export declare function waitSomeThingBeTrue(waitFn: () => boolean, everyWaitTime: number, maxWaitTime?: number): Promise<"success" | "timeout">;

package/dist/lib/easyv-biz-components/src/utils/xun-fei-voice/XF-stream-voice-manager.d.ts

@@ -1,7 +1,7 @@
 import { RecorderManager } from './recorder-manager';
 import { OnMessageInfo, RecordFrameInfo } from './types';
 
-type errorType = 'socketConnect' | 'message' | 'record' | 'recordPermission';
+type errorType = "socketConnect" | "message" | "record" | "recordPermission";
 export interface XFStreamVoiceManagerConfig {
     onError: (errInfo: {
         type: errorType;
@@ -40,7 +40,7 @@ export interface XFStreamVoiceManagerConfig {
      * - denied: 已拒绝
      * - granted: 已授权
      */
-    onGetMicrophonePermission?: (status: 'denied' | 'granted') => void;
+    onGetMicrophonePermission?: (status: "denied" | "granted") => void;
     /**
      * 讯飞语音的 appid
      */
@@ -65,12 +65,13 @@ export declare class XFStreamVoiceManager {
     resultTextTemp: string;
     continueRecord: boolean;
     startTimeStamp: number;
-    recordStatus: 'start' | 'stop';
+    recordStatus: "start" | "stop";
     readyToSend: boolean;
-    microphonePermissionStatus: 'denied' | 'granted' | 'prompt';
+    microphonePermissionStatus: "denied" | "granted" | "prompt" | "unknown";
     private inStopping;
     private stopResolve;
     constructor(config: XFStreamVoiceManagerConfig);
+    private checkBrowserCompatibility;
     renderResult(resultData: string): void;
     setConfig(newConfig: Partial<XFStreamVoiceManagerConfig>): void;
     sendMessage(message: string): void;

package/dist/lib/easyv-biz-components/src/utils/xun-fei-voice/recorder-manager/RecorderManager-draft.d.ts

@@ -20,14 +20,14 @@ export interface startConfig {
     toSampleRate?: number;
     fromSampleRate?: number;
     frameSize?: number;
-    arrayBufferType?: 'short16' | 'float32';
+    arrayBufferType?: "short16" | "float32";
 }
 export declare class RecorderManager {
     config: RecorderManagerConfig;
     audioContext: AudioContext | null;
     audioSource: MediaStreamAudioSourceNode | undefined;
     workletNode: AudioWorkletNode | undefined;
-    status: 'beforeInit' | 'initialized' | 'start' | 'stop';
+    status: "beforeInit" | "initialized" | "start" | "stop";
     audioChunks: any[];
     constructor(config: RecorderManagerConfig);
     setConfig: (newConfig: Partial<RecorderManagerConfig>) => void;

package/dist/lib/easyv-biz-components/src/utils/xun-fei-voice/types.d.ts

@@ -23,7 +23,7 @@ declare class RecorderManagerClass {
     start({ sampleRate, frameSize, arrayBufferType, }: {
         sampleRate?: number;
         frameSize?: number;
-        arrayBufferType?: 'short16' | 'float32';
+        arrayBufferType?: "short16" | "float32";
     }): Promise<void>;
     stop(): void;
 }
@@ -32,6 +32,7 @@ export interface OnMessageInfo {
     /** 优化后的结果。最终结果 */
     message: string;
     isLatest?: boolean;
+    temIsLatest?: boolean;
     /** 临时结果
      * 区别： 如用户再说 “一二三四五”， 临时结果为 “一、二、三、四、五”，最终结果会优化成阿拉伯数字 “12345
      */

package/dist/lib/hooks/use-ai-agent-message/constants.es.js

@@ -4,7 +4,20 @@ const aiAnswerContentTypeList = [
   AiAgentMessageType.DASHBOARD,
   AiAgentMessageType.EXECUTION_STEP
 ];
+var AiAgentKnowledgeEnum = /* @__PURE__ */ ((AiAgentKnowledgeEnum2) => {
+  AiAgentKnowledgeEnum2["easyMind"] = "easy_mind";
+  AiAgentKnowledgeEnum2["requestStandard"] = "request_standard";
+  AiAgentKnowledgeEnum2["thirdAgent"] = "third_party_agent";
+  return AiAgentKnowledgeEnum2;
+})(AiAgentKnowledgeEnum || {});
+const frontEndDirectShowKnowledgeResList = [
+  "third_party_agent",
+  "request_standard"
+  /* requestStandard */
+];
 export {
-  aiAnswerContentTypeList
+  AiAgentKnowledgeEnum,
+  aiAnswerContentTypeList,
+  frontEndDirectShowKnowledgeResList
 };
 //# sourceMappingURL=constants.es.js.map

package/dist/lib/hooks/use-ai-agent-message/constants.es.js.map

@@ -1 +1 @@
-{"version":3,"file":"constants.es.js","sources":["../../../../src/hooks/use-ai-agent-message/constants.ts"],"sourcesContent":["import { AiAgentMessageType } from '@/utils/ai-agent-message-manager';\n\n/** ai 回答的消息类型的合集 */\nexport const aiAnswerContentTypeList = [\n  AiAgentMessageType.TEXT,\n  AiAgentMessageType.DASHBOARD,\n  AiAgentMessageType.EXECUTION_STEP,\n];\n"],"names":[],"mappings":";AAGO,MAAM,0BAA0B;AAAA,EACrC,mBAAmB;AAAA,EACnB,mBAAmB;AAAA,EACnB,mBAAmB;AACrB;"}
+{"version":3,"file":"constants.es.js","sources":["../../../../src/hooks/use-ai-agent-message/constants.ts"],"sourcesContent":["import { AiAgentMessageType } from \"@/utils/ai-agent-message-manager\";\n\n/** ai 回答的消息类型的合集 */\nexport const aiAnswerContentTypeList = [\n  AiAgentMessageType.TEXT,\n  AiAgentMessageType.DASHBOARD,\n  AiAgentMessageType.EXECUTION_STEP,\n];\n\nexport enum AiAgentKnowledgeEnum {\n  /** easyMind */\n  easyMind = \"easy_mind\",\n  /** 自定义知识库时，请求配置为字段配置 */\n  requestStandard = \"request_standard\",\n  /** 第三发智能体的知识库 */\n  thirdAgent = \"third_party_agent\",\n}\n\n/** 前端直接展示知识库的响应类型 */\nexport const frontEndDirectShowKnowledgeResList = [\n  AiAgentKnowledgeEnum.thirdAgent,\n  AiAgentKnowledgeEnum.requestStandard,\n];\n"],"names":["AiAgentKnowledgeEnum"],"mappings":";AAGO,MAAM,0BAA0B;AAAA,EACrC,mBAAmB;AAAA,EACnB,mBAAmB;AAAA,EACnB,mBAAmB;AACrB;AAEY,IAAA,yCAAAA,0BAAL;AAELA,wBAAA,UAAW,IAAA;AAEXA,wBAAA,iBAAkB,IAAA;AAElBA,wBAAA,YAAa,IAAA;AANHA,SAAAA;AAAA,GAAA,wBAAA,CAAA,CAAA;AAUL,MAAM,qCAAqC;AAAA,EAChD;AAAA,EACA;AAAA;AACF;"}

package/dist/lib/hooks/use-ai-agent-message/use-ai-agent-message.es.js

@@ -2,17 +2,16 @@ import { useState, useRef } from "react";
 import useMemoizedFn from "../../node_modules/.pnpm/ahooks@3.8.5_react@18.2.0/node_modules/ahooks/es/useMemoizedFn/index.es.js";
 import useThrottleFn from "../../node_modules/.pnpm/ahooks@3.8.5_react@18.2.0/node_modules/ahooks/es/useThrottleFn/index.es.js";
 import useUnmount from "../../node_modules/.pnpm/ahooks@3.8.5_react@18.2.0/node_modules/ahooks/es/useUnmount/index.es.js";
-import { message } from "antd";
 import { AiAgentMessageManager } from "../../utils/ai-agent-message-manager/ai-agent-message-manager.es.js";
 import { useInitialized } from "../use-initialized.es.js";
 import { useWatchValue } from "../use-watch-value/use-watch-value.es.js";
-import { aiAnswerContentTypeList } from "./constants.es.js";
-import { generateUserMsgItem } from "./utils.es.js";
+import { frontEndDirectShowKnowledgeResList, aiAnswerContentTypeList } from "./constants.es.js";
+import { generateAiMsgPayload, generateUserMsgItem } from "./utils.es.js";
 import { RoomType } from "../use-socket-events/constant.es.js";
-import { fetchKnowledge } from "../../utils/fetchKnowledge/fetchKnowledge.es.js";
+import { fetchKnowledge } from "../../utils/fetch-knowledge/fetch-knowledge.es.js";
 import v4 from "../../node_modules/.pnpm/uuid@11.1.0/node_modules/uuid/dist/esm-browser/v4.es.js";
 import { wait } from "../../utils/wait-something/index.es.js";
-import { generateAiChatMessageFromSocketMessage } from "../../components/ai-message-render/utils.es.js";
+import { generateAiChatMessageFromSocketMessage } from "../../components/ai-components/ai-message-render/utils.es.js";
 import isNil from "../../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/isNil.es.js";
 import { AiAgentMessageType } from "../../utils/ai-agent-message-manager/constants.es.js";
 const useAiAgentMessage = (props) => {
@@ -34,7 +33,7 @@ const useAiAgentMessage = (props) => {
   const setMessageStatus = (status) => {
     messageStatusRef.current = status;
   };
-  const lastQuestionRef = useRef("");
+  const knowledgeMsgRenderFlagRef = useRef(true);
   const onStopChat = useMemoizedFn((id) => {
     var _a;
     (_a = aiAgentManagerConfig.onStopChat) == null ? void 0 : _a.call(aiAgentManagerConfig, id);
@@ -50,10 +49,6 @@ const useAiAgentMessage = (props) => {
     }
     if (payload.type === AiAgentMessageType.CONVERSION_START) {
       setMessageStatus("start");
-      socketMsgListRef.current = [];
-      setIsAnswer(true);
-      setChatLoading(true);
-      setCurrOneChatId(v4());
     }
     if (aiAnswerContentTypeList.includes(payload.type)) {
       if (!currOneChatIdRef.current) {
@@ -74,8 +69,10 @@ const useAiAgentMessage = (props) => {
       setChatLoading(false);
       setStepHintAnimation(false);
       setMessageStatus("error");
-      if (messageStatusRef.current !== "start") {
-        message.error(payload.content || "智能体消息出现错误！");
+      if (messageStatusRef.current === "end") {
+        console.error(
+          payload.content ? `智能体消息错误：${payload.content}` : "智能体消息出现错误！"
+        );
       } else {
         if (!currOneChatIdRef.current) {
           setCurrOneChatId(v4());
@@ -94,7 +91,7 @@ const useAiAgentMessage = (props) => {
       setStepHintAnimation(false);
       await wait(150);
       renderMessageList();
-      socketMsgListRef.current = [];
+      socketMsgListRef.current.splice(0);
       await wait(0);
       setCurrOneChatId("");
     }
@@ -156,23 +153,43 @@ const useAiAgentMessage = (props) => {
     trailing: false
   });
   const [socketManager, setSocketManager] = useState();
-  const onSendMsgToAi = useMemoizedFn((params) => {
+  const stopKnowledgeMsg = useMemoizedFn(() => {
+    knowledgeMsgRenderFlagRef.current = false;
+  });
+  const onSendMsgToAi = useMemoizedFn(() => {
     setMessageStatus("init");
-    lastQuestionRef.current = params.question;
+    setCurrOneChatId(v4());
+    socketMsgListRef.current = [];
+    setIsAnswer(true);
+    setChatLoading(true);
+    renderMessageList();
   });
   const onAiAgentToolCall = useMemoizedFn(async (data) => {
     var _a;
-    let result = "";
+    let result;
     try {
       const toolInfo = data.toolInfo;
       if (!toolInfo) {
         return;
       }
+      knowledgeMsgRenderFlagRef.current = true;
       result = await fetchKnowledge(toolInfo.query, {
         baseURL: toolInfo.baseURL,
         requestHandler: toolInfo.requestHandler,
         requestBody: toolInfo.requestBody,
-        requestHeader: toolInfo.requestHeader
+        requestHeader: toolInfo.requestHeader,
+        onOneChunkRes: (chunkRes) => {
+          if (frontEndDirectShowKnowledgeResList.includes(toolInfo.toolType)) {
+            if (!knowledgeMsgRenderFlagRef.current) {
+              return;
+            }
+            if (chatLoading) {
+              setChatLoading(false);
+            }
+            socketMsgListRef.current.push(generateAiMsgPayload(chunkRes));
+            renderMessageListThrottled();
+          }
+        }
       });
     } catch (error) {
       result = error.message;
@@ -234,7 +251,9 @@ const useAiAgentMessage = (props) => {
     setMessageList,
     currOneChatId: currOneChatIdRef.current,
     stepHintContent: stepHintContentRef.current,
-    stepHintAnimation
+    stepHintAnimation,
+    renderMessageList,
+    stopKnowledgeMsg
   };
 };
 export {

package/dist/lib/hooks/use-ai-agent-message/use-ai-agent-message.es.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-ai-agent-message.es.js","sources":["../../../../src/hooks/use-ai-agent-message/use-ai-agent-message.ts"],"sourcesContent":["import { useRef, useState } from 'react';\nimport { useMemoizedFn, useThrottleFn, useUnmount } from 'ahooks';\nimport { message } from 'antd';\nimport { isNil } from 'lodash-es';\nimport { v4 as uuid } from 'uuid';\nimport {\n  AiChatMessageItem,\n  generateAiChatMessageFromSocketMessage,\n} from '@/components/ai-message-render';\nimport { AiAgentMessageType, fetchKnowledge, ToolCallInfo, wait } from '@/utils';\nimport { AiAgentMessageManager } from '@/utils/ai-agent-message-manager/ai-agent-message-manager';\nimport {\n  AiAgentMessageManagerConfig,\n  AiAgentSocketMessagePayload,\n} from '@/utils/ai-agent-message-manager/types';\nimport { useInitialized } from '../use-initialized';\nimport { RoomType } from '../use-socket-events';\nimport { useWatchValue } from '../use-watch-value/use-watch-value';\nimport { aiAnswerContentTypeList } from './constants';\nimport { KnowledgeBaseDetailForRequestStandard } from './types';\nimport { generateUserMsgItem } from './utils';\n\nexport interface UseAiAgentMessageProps {\n  /** 这个 config 里的函数因为只在初始化调用，所以函数内部拿到的 state 是初始化时的 state，需要用 ref 透传才能拿到最新的 */\n  aiAgentManagerConfig: AiAgentMessageManagerConfig;\n  roomType: RoomType;\n  /** 智能体组件的话，传智能体组件 id */\n  agentId?: number;\n  /** 智能体整个会话的 id，如果是独立对话，那么需要，共享会话不需要，在智能体页面都是独立对话模式\n   * - 用 uuid 生成\n   */\n  agentSessionId?: string;\n  /** 后端返回对话标题 */\n  onReceiveConversationTitle?: (title: string) => void;\n}\n\nexport const useAiAgentMessage = (props: UseAiAgentMessageProps) => {\n  const { aiAgentManagerConfig, roomType, agentId, agentSessionId, onReceiveConversationTitle } =\n    props;\n  const [isAnswer, setIsAnswer] = useState(false);\n  const [chatLoading, setChatLoading] = useState(false);\n  const socketMsgListRef = useRef<AiAgentSocketMessagePayload[]>([]);\n  const [messageList, setMessageList] = useState<AiChatMessageItem[]>([]);\n  const [stepHintAnimation, setStepHintAnimation] = useState<boolean>(false);\n  const stepHintContentRef = useRef<string>('');\n  const setStepHintContent = (content: string) => {\n    stepHintContentRef.current = content;\n  };\n  /** 当前这个对话的 id，为空时不会在渲染消息。 */\n  const currOneChatIdRef = useRef<string>('');\n  const setCurrOneChatId = (id: string) => {\n    currOneChatIdRef.current = id;\n  };\n  const messageStatusRef = useRef<'start' | 'end' | 'error' | 'init'>('init');\n  const setMessageStatus = (status: 'start' | 'end' | 'error' | 'init') => {\n    messageStatusRef.current = status;\n  };\n  const lastQuestionRef = useRef<string>('');\n\n  const onStopChat = useMemoizedFn((id?: string) => {\n    aiAgentManagerConfig.onStopChat?.(id);\n    setIsAnswer(false);\n    setChatLoading(false);\n  });\n\n  const onAiAgentMessage = useMemoizedFn(async (payload: AiAgentSocketMessagePayload) => {\n    aiAgentManagerConfig.onAiAgentMessage?.(payload);\n    socketMsgListRef.current.push(payload);\n    if (payload.type === AiAgentMessageType.TITLE) {\n      onReceiveConversationTitle?.(payload.content || '');\n    }\n    if (payload.type === AiAgentMessageType.CONVERSION_START) {\n      setMessageStatus('start');\n      socketMsgListRef.current = [];\n      setIsAnswer(true);\n      setChatLoading(true);\n      setCurrOneChatId(uuid());\n    }\n    // 问答消息开始 或者 图表消息开始时 结束 loading\n    if (aiAnswerContentTypeList.includes(payload.type)) {\n      if (!currOneChatIdRef.current) {\n        setCurrOneChatId(uuid());\n      }\n      if (!isAnswer) {\n        setIsAnswer(true);\n      }\n      if (chatLoading) {\n        setChatLoading(false);\n      }\n      if (stepHintAnimation) {\n        setStepHintAnimation(false);\n      }\n    }\n    if (payload.type === AiAgentMessageType.ERROR) {\n      setIsAnswer(false);\n      setChatLoading(false);\n      setStepHintAnimation(false);\n      setMessageStatus('error');\n      // 在 start 和 end 之外的 error 直接 message 提示，不需要渲染。\n      if (messageStatusRef.current !== 'start') {\n        message.error(payload.content || '智能体消息出现错误！');\n      } else {\n        // error 的时候没有 start 的消息，所以手动设置一个\n        if (!currOneChatIdRef.current) {\n          setCurrOneChatId(uuid());\n        }\n        renderMessageList();\n      }\n      wait(0).then(() => {\n        setCurrOneChatId('');\n      });\n    }\n    if (payload.type === AiAgentMessageType.CONVERSION_END) {\n      setChatLoading(false);\n      setIsAnswer(false);\n      setMessageStatus('end');\n      setStepHintContent('结果已生成');\n      setStepHintAnimation(false);\n      await wait(150);\n      renderMessageList();\n      socketMsgListRef.current = [];\n      await wait(0);\n      setCurrOneChatId('');\n    }\n    if (payload.type === AiAgentMessageType.EXECUTION_STEP) {\n      setStepHintContent(payload.content || '');\n      setStepHintAnimation(true);\n    }\n    renderMessageListThrottled();\n  });\n\n  const renderMessageList = useMemoizedFn(() => {\n    const socketMsgList = [...socketMsgListRef.current];\n    const currOneChatId = currOneChatIdRef.current;\n    if (!currOneChatId) {\n      return;\n    }\n    if (socketMsgListRef.current.some((msg) => msg.type === AiAgentMessageType.CONVERSION_END)) {\n      socketMsgListRef.current = [];\n    }\n    const newMsgList = [...messageList];\n    // 如果消息列表中能 找不到 当前消息id，那么添加用户和ai的消息\n    const { aiMsg, userMsg } = generateAiChatMessageFromSocketMessage({\n      socketMsgList,\n      currMsgId: currOneChatId,\n    });\n    const isExistUserQuestion = newMsgList.some(\n      (msg) => msg.id === currOneChatId && msg.role === 'user',\n    );\n    if (!isExistUserQuestion && currOneChatId) {\n      if (userMsg) {\n        newMsgList.push(userMsg);\n      } else if (socketManager?.currChatInfo.question) {\n        newMsgList.push(generateUserMsgItem(currOneChatId, socketManager.currChatInfo.question));\n      }\n    }\n    const index = newMsgList.findIndex((msg) => msg.id === currOneChatId && msg.role === 'ai');\n    if (aiMsg) {\n      if (stepHintContentRef.current) {\n        const stepItem = aiMsg.contents.find((i) => Boolean(i.aiAgentStepMessage));\n        if (stepItem) {\n          stepItem.aiAgentStepMessage = stepHintContentRef.current;\n        } else {\n          aiMsg.contents.push({ aiAgentStepMessage: stepHintContentRef.current });\n        }\n      } else {\n        aiMsg.contents = aiMsg.contents.filter((i) => isNil(i.aiAgentStepMessage));\n      }\n      if (\n        messageStatusRef.current === 'end' &&\n        aiMsg.contents.every((i) => Boolean(i.aiAgentStepMessage))\n      ) {\n        aiMsg.contents = [];\n      }\n      if (index !== -1) {\n        newMsgList[index] = aiMsg;\n      } else {\n        newMsgList.push(aiMsg);\n      }\n    }\n    setMessageList(newMsgList);\n  });\n  const { run: renderMessageListThrottled } = useThrottleFn(renderMessageList, {\n    wait: 50,\n    trailing: false,\n  });\n\n  const [socketManager, setSocketManager] = useState<AiAgentMessageManager>();\n\n  const onSendMsgToAi = useMemoizedFn((params) => {\n    setMessageStatus('init');\n    lastQuestionRef.current = params.question;\n  });\n  const onAiAgentToolCall = useMemoizedFn(async (data: ToolCallInfo) => {\n    let result = '';\n    try {\n      const toolInfo = data.toolInfo as unknown as KnowledgeBaseDetailForRequestStandard & {\n        query: string;\n      };\n      if (!toolInfo) {\n        return;\n      }\n      result = await fetchKnowledge(toolInfo.query, {\n        baseURL: toolInfo.baseURL,\n        requestHandler: toolInfo.requestHandler,\n        requestBody: toolInfo.requestBody,\n        requestHeader: toolInfo.requestHeader,\n      });\n    } catch (error: any) {\n      result = error.message;\n    }\n    socketManager?.sendToolResultToAiAgent(result || '知识库请求错误');\n    aiAgentManagerConfig.onAiAgentToolCall?.(data);\n  });\n\n  useInitialized(() => {\n    // 初始化时加入房间\n    const _socketManager = new AiAgentMessageManager({\n      ...aiAgentManagerConfig,\n      onAiAgentMessage,\n      onStopChat,\n      onSendMsgToAi,\n      onAiAgentToolCall,\n    });\n    _socketManager.init();\n    setSocketManager(_socketManager);\n  });\n\n  useWatchValue({\n    value: { agentId, agentSessionId },\n    callback: (newInfo, oldInfo) => {\n      if (roomType === RoomType.AI_AGENT && !newInfo?.agentSessionId) {\n        return;\n      }\n      if (oldInfo?.agentId) {\n        socketManager?.leaveRoom({\n          roomType,\n          id: oldInfo.agentId,\n          subIds: oldInfo.agentSessionId ? [oldInfo.agentSessionId] : undefined,\n        });\n      }\n      if (newInfo.agentId) {\n        socketManager?.joinRoom({\n          roomType,\n          id: newInfo.agentId,\n          subIds: newInfo.agentSessionId ? [newInfo.agentSessionId] : undefined,\n        });\n      }\n    },\n  });\n\n  /** 卸载时，离开房间 */\n  useUnmount(() => {\n    if (!agentId) {\n      return;\n    }\n    socketManager?.leaveRoom({\n      roomType,\n      id: agentId,\n      subIds: agentSessionId ? [agentSessionId] : undefined,\n    });\n    socketManager?.closeWs();\n  });\n\n  return {\n    socketManager,\n    isAnswer,\n    setIsAnswer,\n    chatLoading,\n    setChatLoading,\n    messageList,\n    setMessageList,\n    currOneChatId: currOneChatIdRef.current,\n    stepHintContent: stepHintContentRef.current,\n    stepHintAnimation,\n  };\n};\n"],"names":["uuid"],"mappings":";;;;;;;;;;;;;;;;;AAoCa,MAAA,oBAAoB,CAAC,UAAkC;AAClE,QAAM,EAAE,sBAAsB,UAAU,SAAS,gBAAgB,+BAC/D;AACF,QAAM,CAAC,UAAU,WAAW,IAAI,SAAS,KAAK;AAC9C,QAAM,CAAC,aAAa,cAAc,IAAI,SAAS,KAAK;AAC9C,QAAA,mBAAmB,OAAsC,EAAE;AACjE,QAAM,CAAC,aAAa,cAAc,IAAI,SAA8B,CAAA,CAAE;AACtE,QAAM,CAAC,mBAAmB,oBAAoB,IAAI,SAAkB,KAAK;AACnE,QAAA,qBAAqB,OAAe,EAAE;AACtC,QAAA,qBAAqB,CAAC,YAAoB;AAC9C,uBAAmB,UAAU;AAAA,EAC/B;AAEM,QAAA,mBAAmB,OAAe,EAAE;AACpC,QAAA,mBAAmB,CAAC,OAAe;AACvC,qBAAiB,UAAU;AAAA,EAC7B;AACM,QAAA,mBAAmB,OAA2C,MAAM;AACpE,QAAA,mBAAmB,CAAC,WAA+C;AACvE,qBAAiB,UAAU;AAAA,EAC7B;AACM,QAAA,kBAAkB,OAAe,EAAE;AAEnC,QAAA,aAAa,cAAc,CAAC,OAAgB;;AAChD,+BAAqB,eAArB,8CAAkC;AAClC,gBAAY,KAAK;AACjB,mBAAe,KAAK;AAAA,EAAA,CACrB;AAEK,QAAA,mBAAmB,cAAc,OAAO,YAAyC;;AACrF,+BAAqB,qBAArB,8CAAwC;AACvB,qBAAA,QAAQ,KAAK,OAAO;AACjC,QAAA,QAAQ,SAAS,mBAAmB,OAAO;AAChB,+EAAA,QAAQ,WAAW;AAAA,IAAE;AAEhD,QAAA,QAAQ,SAAS,mBAAmB,kBAAkB;AACxD,uBAAiB,OAAO;AACxB,uBAAiB,UAAU,CAAC;AAC5B,kBAAY,IAAI;AAChB,qBAAe,IAAI;AACnB,uBAAiBA,IAAM;AAAA,IAAA;AAGzB,QAAI,wBAAwB,SAAS,QAAQ,IAAI,GAAG;AAC9C,UAAA,CAAC,iBAAiB,SAAS;AAC7B,yBAAiBA,IAAM;AAAA,MAAA;AAEzB,UAAI,CAAC,UAAU;AACb,oBAAY,IAAI;AAAA,MAAA;AAElB,UAAI,aAAa;AACf,uBAAe,KAAK;AAAA,MAAA;AAEtB,UAAI,mBAAmB;AACrB,6BAAqB,KAAK;AAAA,MAAA;AAAA,IAC5B;AAEE,QAAA,QAAQ,SAAS,mBAAmB,OAAO;AAC7C,kBAAY,KAAK;AACjB,qBAAe,KAAK;AACpB,2BAAqB,KAAK;AAC1B,uBAAiB,OAAO;AAEpB,UAAA,iBAAiB,YAAY,SAAS;AAChC,gBAAA,MAAM,QAAQ,WAAW,YAAY;AAAA,MAAA,OACxC;AAED,YAAA,CAAC,iBAAiB,SAAS;AAC7B,2BAAiBA,IAAM;AAAA,QAAA;AAEP,0BAAA;AAAA,MAAA;AAEf,WAAA,CAAC,EAAE,KAAK,MAAM;AACjB,yBAAiB,EAAE;AAAA,MAAA,CACpB;AAAA,IAAA;AAEC,QAAA,QAAQ,SAAS,mBAAmB,gBAAgB;AACtD,qBAAe,KAAK;AACpB,kBAAY,KAAK;AACjB,uBAAiB,KAAK;AACtB,yBAAmB,OAAO;AAC1B,2BAAqB,KAAK;AAC1B,YAAM,KAAK,GAAG;AACI,wBAAA;AAClB,uBAAiB,UAAU,CAAC;AAC5B,YAAM,KAAK,CAAC;AACZ,uBAAiB,EAAE;AAAA,IAAA;AAEjB,QAAA,QAAQ,SAAS,mBAAmB,gBAAgB;AACnC,yBAAA,QAAQ,WAAW,EAAE;AACxC,2BAAqB,IAAI;AAAA,IAAA;AAEA,+BAAA;AAAA,EAAA,CAC5B;AAEK,QAAA,oBAAoB,cAAc,MAAM;AAC5C,UAAM,gBAAgB,CAAC,GAAG,iBAAiB,OAAO;AAClD,UAAM,gBAAgB,iBAAiB;AACvC,QAAI,CAAC,eAAe;AAClB;AAAA,IAAA;AAEE,QAAA,iBAAiB,QAAQ,KAAK,CAAC,QAAQ,IAAI,SAAS,mBAAmB,cAAc,GAAG;AAC1F,uBAAiB,UAAU,CAAC;AAAA,IAAA;AAExB,UAAA,aAAa,CAAC,GAAG,WAAW;AAElC,UAAM,EAAE,OAAO,QAAQ,IAAI,uCAAuC;AAAA,MAChE;AAAA,MACA,WAAW;AAAA,IAAA,CACZ;AACD,UAAM,sBAAsB,WAAW;AAAA,MACrC,CAAC,QAAQ,IAAI,OAAO,iBAAiB,IAAI,SAAS;AAAA,IACpD;AACI,QAAA,CAAC,uBAAuB,eAAe;AACzC,UAAI,SAAS;AACX,mBAAW,KAAK,OAAO;AAAA,MAAA,WACd,+CAAe,aAAa,UAAU;AAC/C,mBAAW,KAAK,oBAAoB,eAAe,cAAc,aAAa,QAAQ,CAAC;AAAA,MAAA;AAAA,IACzF;AAEI,UAAA,QAAQ,WAAW,UAAU,CAAC,QAAQ,IAAI,OAAO,iBAAiB,IAAI,SAAS,IAAI;AACzF,QAAI,OAAO;AACT,UAAI,mBAAmB,SAAS;AACxB,cAAA,WAAW,MAAM,SAAS,KAAK,CAAC,MAAM,QAAQ,EAAE,kBAAkB,CAAC;AACzE,YAAI,UAAU;AACZ,mBAAS,qBAAqB,mBAAmB;AAAA,QAAA,OAC5C;AACL,gBAAM,SAAS,KAAK,EAAE,oBAAoB,mBAAmB,SAAS;AAAA,QAAA;AAAA,MACxE,OACK;AACC,cAAA,WAAW,MAAM,SAAS,OAAO,CAAC,MAAM,MAAM,EAAE,kBAAkB,CAAC;AAAA,MAAA;AAE3E,UACE,iBAAiB,YAAY,SAC7B,MAAM,SAAS,MAAM,CAAC,MAAM,QAAQ,EAAE,kBAAkB,CAAC,GACzD;AACA,cAAM,WAAW,CAAC;AAAA,MAAA;AAEpB,UAAI,UAAU,IAAI;AAChB,mBAAW,KAAK,IAAI;AAAA,MAAA,OACf;AACL,mBAAW,KAAK,KAAK;AAAA,MAAA;AAAA,IACvB;AAEF,mBAAe,UAAU;AAAA,EAAA,CAC1B;AACD,QAAM,EAAE,KAAK,+BAA+B,cAAc,mBAAmB;AAAA,IAC3E,MAAM;AAAA,IACN,UAAU;AAAA,EAAA,CACX;AAED,QAAM,CAAC,eAAe,gBAAgB,IAAI,SAAgC;AAEpE,QAAA,gBAAgB,cAAc,CAAC,WAAW;AAC9C,qBAAiB,MAAM;AACvB,oBAAgB,UAAU,OAAO;AAAA,EAAA,CAClC;AACK,QAAA,oBAAoB,cAAc,OAAO,SAAuB;;AACpE,QAAI,SAAS;AACT,QAAA;AACF,YAAM,WAAW,KAAK;AAGtB,UAAI,CAAC,UAAU;AACb;AAAA,MAAA;AAEO,eAAA,MAAM,eAAe,SAAS,OAAO;AAAA,QAC5C,SAAS,SAAS;AAAA,QAClB,gBAAgB,SAAS;AAAA,QACzB,aAAa,SAAS;AAAA,QACtB,eAAe,SAAS;AAAA,MAAA,CACzB;AAAA,aACM,OAAY;AACnB,eAAS,MAAM;AAAA,IAAA;AAEF,mDAAA,wBAAwB,UAAU;AACjD,+BAAqB,sBAArB,8CAAyC;AAAA,EAAI,CAC9C;AAED,iBAAe,MAAM;AAEb,UAAA,iBAAiB,IAAI,sBAAsB;AAAA,MAC/C,GAAG;AAAA,MACH;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IAAA,CACD;AACD,mBAAe,KAAK;AACpB,qBAAiB,cAAc;AAAA,EAAA,CAChC;AAEa,gBAAA;AAAA,IACZ,OAAO,EAAE,SAAS,eAAe;AAAA,IACjC,UAAU,CAAC,SAAS,YAAY;AAC9B,UAAI,aAAa,SAAS,YAAY,EAAC,mCAAS,iBAAgB;AAC9D;AAAA,MAAA;AAEF,UAAI,mCAAS,SAAS;AACpB,uDAAe,UAAU;AAAA,UACvB;AAAA,UACA,IAAI,QAAQ;AAAA,UACZ,QAAQ,QAAQ,iBAAiB,CAAC,QAAQ,cAAc,IAAI;AAAA,QAAA;AAAA,MAC7D;AAEH,UAAI,QAAQ,SAAS;AACnB,uDAAe,SAAS;AAAA,UACtB;AAAA,UACA,IAAI,QAAQ;AAAA,UACZ,QAAQ,QAAQ,iBAAiB,CAAC,QAAQ,cAAc,IAAI;AAAA,QAAA;AAAA,MAC7D;AAAA,IACH;AAAA,EACF,CACD;AAGD,aAAW,MAAM;AACf,QAAI,CAAC,SAAS;AACZ;AAAA,IAAA;AAEF,mDAAe,UAAU;AAAA,MACvB;AAAA,MACA,IAAI;AAAA,MACJ,QAAQ,iBAAiB,CAAC,cAAc,IAAI;AAAA,IAAA;AAE9C,mDAAe;AAAA,EAAQ,CACxB;AAEM,SAAA;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA,eAAe,iBAAiB;AAAA,IAChC,iBAAiB,mBAAmB;AAAA,IACpC;AAAA,EACF;AACF;"}
+{"version":3,"file":"use-ai-agent-message.es.js","sources":["../../../../src/hooks/use-ai-agent-message/use-ai-agent-message.ts"],"sourcesContent":["import { useRef, useState } from \"react\";\nimport { useMemoizedFn, useThrottleFn, useUnmount } from \"ahooks\";\nimport { isNil } from \"lodash-es\";\nimport { v4 as uuid } from \"uuid\";\nimport {\n  AiChatMessageItem,\n  generateAiChatMessageFromSocketMessage,\n} from \"@/components/ai-components/ai-message-render\";\nimport { AiAgentMessageType, fetchKnowledge, ToolCallInfo, wait } from \"@/utils\";\nimport { AiAgentMessageManager } from \"@/utils/ai-agent-message-manager/ai-agent-message-manager\";\nimport {\n  AiAgentMessageManagerConfig,\n  AiAgentSocketMessagePayload,\n} from \"@/utils/ai-agent-message-manager/types\";\nimport { useInitialized } from \"../use-initialized\";\nimport { RoomType } from \"../use-socket-events\";\nimport { useWatchValue } from \"../use-watch-value/use-watch-value\";\nimport {\n  AiAgentKnowledgeEnum,\n  aiAnswerContentTypeList,\n  frontEndDirectShowKnowledgeResList,\n} from \"./constants\";\nimport { KnowledgeBaseDetailForRequestStandard } from \"./types\";\nimport { generateAiMsgPayload, generateUserMsgItem } from \"./utils\";\n\nexport interface UseAiAgentMessageProps {\n  /** 这个 config 里的函数因为只在初始化调用，所以函数内部拿到的 state 是初始化时的 state，需要用 ref 透传才能拿到最新的 */\n  aiAgentManagerConfig: AiAgentMessageManagerConfig;\n  roomType: RoomType;\n  /** 智能体组件的话，传智能体组件 id */\n  agentId?: number;\n  /** 智能体整个会话的 id，如果是独立对话，那么需要，共享会话不需要，在智能体页面都是独立对话模式\n   * - 用 uuid 生成\n   */\n  agentSessionId?: string;\n  /** 后端返回对话标题 */\n  onReceiveConversationTitle?: (title: string) => void;\n}\n\nexport const useAiAgentMessage = (props: UseAiAgentMessageProps) => {\n  const { aiAgentManagerConfig, roomType, agentId, agentSessionId, onReceiveConversationTitle } =\n    props;\n  const [isAnswer, setIsAnswer] = useState(false);\n  const [chatLoading, setChatLoading] = useState(false);\n  const socketMsgListRef = useRef<AiAgentSocketMessagePayload[]>([]);\n  const [messageList, setMessageList] = useState<AiChatMessageItem[]>([]);\n  const [stepHintAnimation, setStepHintAnimation] = useState<boolean>(false);\n  const stepHintContentRef = useRef<string>(\"\");\n  const setStepHintContent = (content: string) => {\n    stepHintContentRef.current = content;\n  };\n  /** 当前这个对话的 id，为空时不会在渲染消息。 */\n  const currOneChatIdRef = useRef<string>(\"\");\n  const setCurrOneChatId = (id: string) => {\n    currOneChatIdRef.current = id;\n  };\n  const messageStatusRef = useRef<\"start\" | \"end\" | \"error\" | \"init\">(\"init\");\n  const setMessageStatus = (status: \"start\" | \"end\" | \"error\" | \"init\") => {\n    messageStatusRef.current = status;\n  };\n\n  const knowledgeMsgRenderFlagRef = useRef(true);\n\n  const onStopChat = useMemoizedFn((id?: string) => {\n    aiAgentManagerConfig.onStopChat?.(id);\n    setIsAnswer(false);\n    setChatLoading(false);\n  });\n\n  const onAiAgentMessage = useMemoizedFn(async (payload: AiAgentSocketMessagePayload) => {\n    aiAgentManagerConfig.onAiAgentMessage?.(payload);\n    socketMsgListRef.current.push(payload);\n    if (payload.type === AiAgentMessageType.TITLE) {\n      onReceiveConversationTitle?.((payload.content as string) || \"\");\n    }\n    if (payload.type === AiAgentMessageType.CONVERSION_START) {\n      setMessageStatus(\"start\");\n    }\n    // 问答消息开始 或者 图表消息开始时 结束 loading\n    if (aiAnswerContentTypeList.includes(payload.type)) {\n      if (!currOneChatIdRef.current) {\n        setCurrOneChatId(uuid());\n      }\n      if (!isAnswer) {\n        setIsAnswer(true);\n      }\n      if (chatLoading) {\n        setChatLoading(false);\n      }\n      if (stepHintAnimation) {\n        setStepHintAnimation(false);\n      }\n    }\n    if (payload.type === AiAgentMessageType.ERROR) {\n      setIsAnswer(false);\n      setChatLoading(false);\n      setStepHintAnimation(false);\n      setMessageStatus(\"error\");\n      // 在 start 和 end 之外的 error 打印在控制台，不需要渲染。\n      if (messageStatusRef.current === \"end\") {\n        console.error(\n          payload.content ? `智能体消息错误：${payload.content}` : \"智能体消息出现错误！\",\n        );\n      } else {\n        // error 的时候没有 start 的消息，所以手动设置一个\n        if (!currOneChatIdRef.current) {\n          setCurrOneChatId(uuid());\n        }\n        renderMessageList();\n      }\n      wait(0).then(() => {\n        setCurrOneChatId(\"\");\n      });\n    }\n    if (payload.type === AiAgentMessageType.CONVERSION_END) {\n      setChatLoading(false);\n      setIsAnswer(false);\n      setMessageStatus(\"end\");\n      setStepHintContent(\"结果已生成\");\n      setStepHintAnimation(false);\n      await wait(150);\n      renderMessageList();\n      socketMsgListRef.current.splice(0);\n      await wait(0);\n      setCurrOneChatId(\"\");\n    }\n    if (payload.type === AiAgentMessageType.EXECUTION_STEP) {\n      setStepHintContent((payload.content as string) || \"\");\n      setStepHintAnimation(true);\n    }\n    renderMessageListThrottled();\n  });\n\n  const renderMessageList = useMemoizedFn(() => {\n    const socketMsgList = [...socketMsgListRef.current];\n    const currOneChatId = currOneChatIdRef.current;\n    if (!currOneChatId) {\n      return;\n    }\n    if (socketMsgListRef.current.some((msg) => msg.type === AiAgentMessageType.CONVERSION_END)) {\n      socketMsgListRef.current = [];\n    }\n    const newMsgList = [...messageList];\n    // 如果消息列表中能 找不到 当前消息id，那么添加用户和ai的消息\n    const { aiMsg, userMsg } = generateAiChatMessageFromSocketMessage({\n      socketMsgList,\n      currMsgId: currOneChatId,\n    });\n    const isExistUserQuestion = newMsgList.some(\n      (msg) => msg.id === currOneChatId && msg.role === \"user\",\n    );\n    if (!isExistUserQuestion && currOneChatId) {\n      if (userMsg) {\n        newMsgList.push(userMsg);\n      } else if (socketManager?.currChatInfo.question) {\n        newMsgList.push(generateUserMsgItem(currOneChatId, socketManager.currChatInfo.question));\n      }\n    }\n    const index = newMsgList.findIndex((msg) => msg.id === currOneChatId && msg.role === \"ai\");\n    if (aiMsg) {\n      if (stepHintContentRef.current) {\n        const stepItem = aiMsg.contents.find((i) => Boolean(i.aiAgentStepMessage));\n        if (stepItem) {\n          stepItem.aiAgentStepMessage = stepHintContentRef.current;\n        } else {\n          aiMsg.contents.push({ aiAgentStepMessage: stepHintContentRef.current });\n        }\n      } else {\n        aiMsg.contents = aiMsg.contents.filter((i) => isNil(i.aiAgentStepMessage));\n      }\n      if (\n        messageStatusRef.current === \"end\" &&\n        aiMsg.contents.every((i) => Boolean(i.aiAgentStepMessage))\n      ) {\n        aiMsg.contents = [];\n      }\n      if (index !== -1) {\n        newMsgList[index] = aiMsg;\n      } else {\n        newMsgList.push(aiMsg);\n      }\n    }\n    setMessageList(newMsgList);\n  });\n  const { run: renderMessageListThrottled } = useThrottleFn(renderMessageList, {\n    wait: 50,\n    trailing: false,\n  });\n\n  const [socketManager, setSocketManager] = useState<AiAgentMessageManager>();\n\n  const stopKnowledgeMsg = useMemoizedFn(() => {\n    knowledgeMsgRenderFlagRef.current = false;\n  });\n\n  const onSendMsgToAi = useMemoizedFn(() => {\n    setMessageStatus(\"init\");\n    setCurrOneChatId(uuid());\n    socketMsgListRef.current = [];\n    setIsAnswer(true);\n    setChatLoading(true);\n    renderMessageList();\n  });\n  const onAiAgentToolCall = useMemoizedFn(async (data: ToolCallInfo) => {\n    let result;\n    try {\n      const toolInfo = data.toolInfo as unknown as KnowledgeBaseDetailForRequestStandard & {\n        query: string;\n      };\n      if (!toolInfo) {\n        return;\n      }\n      knowledgeMsgRenderFlagRef.current = true;\n      result = await fetchKnowledge(toolInfo.query, {\n        baseURL: toolInfo.baseURL,\n        requestHandler: toolInfo.requestHandler,\n        requestBody: toolInfo.requestBody,\n        requestHeader: toolInfo.requestHeader,\n        onOneChunkRes: (chunkRes) => {\n          if (\n            frontEndDirectShowKnowledgeResList.includes(toolInfo.toolType as AiAgentKnowledgeEnum)\n          ) {\n            if (!knowledgeMsgRenderFlagRef.current) {\n              return;\n            }\n            if (chatLoading) {\n              setChatLoading(false);\n            }\n            socketMsgListRef.current.push(generateAiMsgPayload(chunkRes));\n            renderMessageListThrottled();\n          }\n        },\n      });\n    } catch (error: any) {\n      result = error.message;\n    }\n    socketManager?.sendToolResultToAiAgent(result || \"知识库请求错误\");\n    aiAgentManagerConfig.onAiAgentToolCall?.(data);\n  });\n\n  useInitialized(() => {\n    // 初始化时加入房间\n    const _socketManager = new AiAgentMessageManager({\n      ...aiAgentManagerConfig,\n      onAiAgentMessage,\n      onStopChat,\n      onSendMsgToAi,\n      onAiAgentToolCall,\n    });\n    _socketManager.init();\n    setSocketManager(_socketManager);\n  });\n\n  useWatchValue({\n    value: { agentId, agentSessionId },\n    callback: (newInfo, oldInfo) => {\n      if (roomType === RoomType.AI_AGENT && !newInfo?.agentSessionId) {\n        return;\n      }\n      if (oldInfo?.agentId) {\n        socketManager?.leaveRoom({\n          roomType,\n          id: oldInfo.agentId,\n          subIds: oldInfo.agentSessionId ? [oldInfo.agentSessionId] : undefined,\n        });\n      }\n      if (newInfo.agentId) {\n        socketManager?.joinRoom({\n          roomType,\n          id: newInfo.agentId,\n          subIds: newInfo.agentSessionId ? [newInfo.agentSessionId] : undefined,\n        });\n      }\n    },\n  });\n\n  /** 卸载时，离开房间 */\n  useUnmount(() => {\n    if (!agentId) {\n      return;\n    }\n    socketManager?.leaveRoom({\n      roomType,\n      id: agentId,\n      subIds: agentSessionId ? [agentSessionId] : undefined,\n    });\n    socketManager?.closeWs();\n  });\n\n  return {\n    socketManager,\n    isAnswer,\n    setIsAnswer,\n    chatLoading,\n    setChatLoading,\n    messageList,\n    setMessageList,\n    currOneChatId: currOneChatIdRef.current,\n    stepHintContent: stepHintContentRef.current,\n    stepHintAnimation,\n    renderMessageList,\n    stopKnowledgeMsg,\n  };\n};\n"],"names":["uuid"],"mappings":";;;;;;;;;;;;;;;;AAuCa,MAAA,oBAAoB,CAAC,UAAkC;AAClE,QAAM,EAAE,sBAAsB,UAAU,SAAS,gBAAgB,+BAC/D;AACF,QAAM,CAAC,UAAU,WAAW,IAAI,SAAS,KAAK;AAC9C,QAAM,CAAC,aAAa,cAAc,IAAI,SAAS,KAAK;AAC9C,QAAA,mBAAmB,OAAsC,EAAE;AACjE,QAAM,CAAC,aAAa,cAAc,IAAI,SAA8B,CAAA,CAAE;AACtE,QAAM,CAAC,mBAAmB,oBAAoB,IAAI,SAAkB,KAAK;AACnE,QAAA,qBAAqB,OAAe,EAAE;AACtC,QAAA,qBAAqB,CAAC,YAAoB;AAC9C,uBAAmB,UAAU;AAAA,EAC/B;AAEM,QAAA,mBAAmB,OAAe,EAAE;AACpC,QAAA,mBAAmB,CAAC,OAAe;AACvC,qBAAiB,UAAU;AAAA,EAC7B;AACM,QAAA,mBAAmB,OAA2C,MAAM;AACpE,QAAA,mBAAmB,CAAC,WAA+C;AACvE,qBAAiB,UAAU;AAAA,EAC7B;AAEM,QAAA,4BAA4B,OAAO,IAAI;AAEvC,QAAA,aAAa,cAAc,CAAC,OAAgB;;AAChD,+BAAqB,eAArB,8CAAkC;AAClC,gBAAY,KAAK;AACjB,mBAAe,KAAK;AAAA,EAAA,CACrB;AAEK,QAAA,mBAAmB,cAAc,OAAO,YAAyC;;AACrF,+BAAqB,qBAArB,8CAAwC;AACvB,qBAAA,QAAQ,KAAK,OAAO;AACjC,QAAA,QAAQ,SAAS,mBAAmB,OAAO;AACf,+EAAA,QAAQ,WAAsB;AAAA,IAAE;AAE5D,QAAA,QAAQ,SAAS,mBAAmB,kBAAkB;AACxD,uBAAiB,OAAO;AAAA,IAAA;AAG1B,QAAI,wBAAwB,SAAS,QAAQ,IAAI,GAAG;AAC9C,UAAA,CAAC,iBAAiB,SAAS;AAC7B,yBAAiBA,IAAM;AAAA,MAAA;AAEzB,UAAI,CAAC,UAAU;AACb,oBAAY,IAAI;AAAA,MAAA;AAElB,UAAI,aAAa;AACf,uBAAe,KAAK;AAAA,MAAA;AAEtB,UAAI,mBAAmB;AACrB,6BAAqB,KAAK;AAAA,MAAA;AAAA,IAC5B;AAEE,QAAA,QAAQ,SAAS,mBAAmB,OAAO;AAC7C,kBAAY,KAAK;AACjB,qBAAe,KAAK;AACpB,2BAAqB,KAAK;AAC1B,uBAAiB,OAAO;AAEpB,UAAA,iBAAiB,YAAY,OAAO;AAC9B,gBAAA;AAAA,UACN,QAAQ,UAAU,WAAW,QAAQ,OAAO,KAAK;AAAA,QACnD;AAAA,MAAA,OACK;AAED,YAAA,CAAC,iBAAiB,SAAS;AAC7B,2BAAiBA,IAAM;AAAA,QAAA;AAEP,0BAAA;AAAA,MAAA;AAEf,WAAA,CAAC,EAAE,KAAK,MAAM;AACjB,yBAAiB,EAAE;AAAA,MAAA,CACpB;AAAA,IAAA;AAEC,QAAA,QAAQ,SAAS,mBAAmB,gBAAgB;AACtD,qBAAe,KAAK;AACpB,kBAAY,KAAK;AACjB,uBAAiB,KAAK;AACtB,yBAAmB,OAAO;AAC1B,2BAAqB,KAAK;AAC1B,YAAM,KAAK,GAAG;AACI,wBAAA;AACD,uBAAA,QAAQ,OAAO,CAAC;AACjC,YAAM,KAAK,CAAC;AACZ,uBAAiB,EAAE;AAAA,IAAA;AAEjB,QAAA,QAAQ,SAAS,mBAAmB,gBAAgB;AAClC,yBAAA,QAAQ,WAAsB,EAAE;AACpD,2BAAqB,IAAI;AAAA,IAAA;AAEA,+BAAA;AAAA,EAAA,CAC5B;AAEK,QAAA,oBAAoB,cAAc,MAAM;AAC5C,UAAM,gBAAgB,CAAC,GAAG,iBAAiB,OAAO;AAClD,UAAM,gBAAgB,iBAAiB;AACvC,QAAI,CAAC,eAAe;AAClB;AAAA,IAAA;AAEE,QAAA,iBAAiB,QAAQ,KAAK,CAAC,QAAQ,IAAI,SAAS,mBAAmB,cAAc,GAAG;AAC1F,uBAAiB,UAAU,CAAC;AAAA,IAAA;AAExB,UAAA,aAAa,CAAC,GAAG,WAAW;AAElC,UAAM,EAAE,OAAO,QAAQ,IAAI,uCAAuC;AAAA,MAChE;AAAA,MACA,WAAW;AAAA,IAAA,CACZ;AACD,UAAM,sBAAsB,WAAW;AAAA,MACrC,CAAC,QAAQ,IAAI,OAAO,iBAAiB,IAAI,SAAS;AAAA,IACpD;AACI,QAAA,CAAC,uBAAuB,eAAe;AACzC,UAAI,SAAS;AACX,mBAAW,KAAK,OAAO;AAAA,MAAA,WACd,+CAAe,aAAa,UAAU;AAC/C,mBAAW,KAAK,oBAAoB,eAAe,cAAc,aAAa,QAAQ,CAAC;AAAA,MAAA;AAAA,IACzF;AAEI,UAAA,QAAQ,WAAW,UAAU,CAAC,QAAQ,IAAI,OAAO,iBAAiB,IAAI,SAAS,IAAI;AACzF,QAAI,OAAO;AACT,UAAI,mBAAmB,SAAS;AACxB,cAAA,WAAW,MAAM,SAAS,KAAK,CAAC,MAAM,QAAQ,EAAE,kBAAkB,CAAC;AACzE,YAAI,UAAU;AACZ,mBAAS,qBAAqB,mBAAmB;AAAA,QAAA,OAC5C;AACL,gBAAM,SAAS,KAAK,EAAE,oBAAoB,mBAAmB,SAAS;AAAA,QAAA;AAAA,MACxE,OACK;AACC,cAAA,WAAW,MAAM,SAAS,OAAO,CAAC,MAAM,MAAM,EAAE,kBAAkB,CAAC;AAAA,MAAA;AAE3E,UACE,iBAAiB,YAAY,SAC7B,MAAM,SAAS,MAAM,CAAC,MAAM,QAAQ,EAAE,kBAAkB,CAAC,GACzD;AACA,cAAM,WAAW,CAAC;AAAA,MAAA;AAEpB,UAAI,UAAU,IAAI;AAChB,mBAAW,KAAK,IAAI;AAAA,MAAA,OACf;AACL,mBAAW,KAAK,KAAK;AAAA,MAAA;AAAA,IACvB;AAEF,mBAAe,UAAU;AAAA,EAAA,CAC1B;AACD,QAAM,EAAE,KAAK,+BAA+B,cAAc,mBAAmB;AAAA,IAC3E,MAAM;AAAA,IACN,UAAU;AAAA,EAAA,CACX;AAED,QAAM,CAAC,eAAe,gBAAgB,IAAI,SAAgC;AAEpE,QAAA,mBAAmB,cAAc,MAAM;AAC3C,8BAA0B,UAAU;AAAA,EAAA,CACrC;AAEK,QAAA,gBAAgB,cAAc,MAAM;AACxC,qBAAiB,MAAM;AACvB,qBAAiBA,IAAM;AACvB,qBAAiB,UAAU,CAAC;AAC5B,gBAAY,IAAI;AAChB,mBAAe,IAAI;AACD,sBAAA;AAAA,EAAA,CACnB;AACK,QAAA,oBAAoB,cAAc,OAAO,SAAuB;;AAChE,QAAA;AACA,QAAA;AACF,YAAM,WAAW,KAAK;AAGtB,UAAI,CAAC,UAAU;AACb;AAAA,MAAA;AAEF,gCAA0B,UAAU;AAC3B,eAAA,MAAM,eAAe,SAAS,OAAO;AAAA,QAC5C,SAAS,SAAS;AAAA,QAClB,gBAAgB,SAAS;AAAA,QACzB,aAAa,SAAS;AAAA,QACtB,eAAe,SAAS;AAAA,QACxB,eAAe,CAAC,aAAa;AAC3B,cACE,mCAAmC,SAAS,SAAS,QAAgC,GACrF;AACI,gBAAA,CAAC,0BAA0B,SAAS;AACtC;AAAA,YAAA;AAEF,gBAAI,aAAa;AACf,6BAAe,KAAK;AAAA,YAAA;AAEtB,6BAAiB,QAAQ,KAAK,qBAAqB,QAAQ,CAAC;AACjC,uCAAA;AAAA,UAAA;AAAA,QAC7B;AAAA,MACF,CACD;AAAA,aACM,OAAY;AACnB,eAAS,MAAM;AAAA,IAAA;AAEF,mDAAA,wBAAwB,UAAU;AACjD,+BAAqB,sBAArB,8CAAyC;AAAA,EAAI,CAC9C;AAED,iBAAe,MAAM;AAEb,UAAA,iBAAiB,IAAI,sBAAsB;AAAA,MAC/C,GAAG;AAAA,MACH;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IAAA,CACD;AACD,mBAAe,KAAK;AACpB,qBAAiB,cAAc;AAAA,EAAA,CAChC;AAEa,gBAAA;AAAA,IACZ,OAAO,EAAE,SAAS,eAAe;AAAA,IACjC,UAAU,CAAC,SAAS,YAAY;AAC9B,UAAI,aAAa,SAAS,YAAY,EAAC,mCAAS,iBAAgB;AAC9D;AAAA,MAAA;AAEF,UAAI,mCAAS,SAAS;AACpB,uDAAe,UAAU;AAAA,UACvB;AAAA,UACA,IAAI,QAAQ;AAAA,UACZ,QAAQ,QAAQ,iBAAiB,CAAC,QAAQ,cAAc,IAAI;AAAA,QAAA;AAAA,MAC7D;AAEH,UAAI,QAAQ,SAAS;AACnB,uDAAe,SAAS;AAAA,UACtB;AAAA,UACA,IAAI,QAAQ;AAAA,UACZ,QAAQ,QAAQ,iBAAiB,CAAC,QAAQ,cAAc,IAAI;AAAA,QAAA;AAAA,MAC7D;AAAA,IACH;AAAA,EACF,CACD;AAGD,aAAW,MAAM;AACf,QAAI,CAAC,SAAS;AACZ;AAAA,IAAA;AAEF,mDAAe,UAAU;AAAA,MACvB;AAAA,MACA,IAAI;AAAA,MACJ,QAAQ,iBAAiB,CAAC,cAAc,IAAI;AAAA,IAAA;AAE9C,mDAAe;AAAA,EAAQ,CACxB;AAEM,SAAA;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA,eAAe,iBAAiB;AAAA,IAChC,iBAAiB,mBAAmB;AAAA,IACpC;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;"}

package/dist/lib/hooks/use-ai-agent-message/utils.es.js

@@ -1,5 +1,6 @@
 import dayjs from "../../node_modules/.pnpm/dayjs@1.11.13/node_modules/dayjs/dayjs.min.es.js";
 import { generateRandomId } from "../../utils/common/common.es.js";
+import { AiAgentMessageType } from "../../utils/ai-agent-message-manager/constants.es.js";
 const generateUserMsgItem = (id, content) => {
   return {
     id,
@@ -18,8 +19,15 @@ const generateAiMsgItem = (content, showVoiceBtn = false) => {
     showVoiceBtn
   };
 };
+const generateAiMsgPayload = (content) => {
+  return {
+    type: AiAgentMessageType.TEXT,
+    content
+  };
+};
 export {
   generateAiMsgItem,
+  generateAiMsgPayload,
   generateUserMsgItem
 };
 //# sourceMappingURL=utils.es.js.map

package/dist/lib/hooks/use-ai-agent-message/utils.es.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.es.js","sources":["../../../../src/hooks/use-ai-agent-message/utils.ts"],"sourcesContent":["import dayjs from 'dayjs';\nimport { AiChatMessageItem } from '@/components';\nimport { generateRandomId } from '@/utils';\n\nexport const generateUserMsgItem = (id: string, content: string): AiChatMessageItem => {\n  return {\n    id,\n    contents: [{ text: content }],\n    role: 'user',\n    createTime: dayjs().toString(),\n    createTimeStamp: dayjs().unix(),\n  };\n};\nexport const generateAiMsgItem = (content: string, showVoiceBtn = false): AiChatMessageItem => {\n  return {\n    id: generateRandomId(),\n    role: 'ai',\n    contents: [{ text: content }],\n    createTime: new Date().toLocaleString(),\n    showVoiceBtn,\n  };\n};\n"],"names":[],"mappings":";;AAIa,MAAA,sBAAsB,CAAC,IAAY,YAAuC;AAC9E,SAAA;AAAA,IACL;AAAA,IACA,UAAU,CAAC,EAAE,MAAM,SAAS;AAAA,IAC5B,MAAM;AAAA,IACN,YAAY,MAAM,EAAE,SAAS;AAAA,IAC7B,iBAAiB,MAAM,EAAE,KAAK;AAAA,EAChC;AACF;AACO,MAAM,oBAAoB,CAAC,SAAiB,eAAe,UAA6B;AACtF,SAAA;AAAA,IACL,IAAI,iBAAiB;AAAA,IACrB,MAAM;AAAA,IACN,UAAU,CAAC,EAAE,MAAM,SAAS;AAAA,IAC5B,aAAY,oBAAI,KAAK,GAAE,eAAe;AAAA,IACtC;AAAA,EACF;AACF;"}
+{"version":3,"file":"utils.es.js","sources":["../../../../src/hooks/use-ai-agent-message/utils.ts"],"sourcesContent":["import dayjs from \"dayjs\";\nimport { AiChatMessageItem } from \"@/components\";\nimport { AiAgentMessageType, AiAgentSocketMessagePayload, generateRandomId } from \"@/utils\";\n\nexport const generateUserMsgItem = (id: string, content: string): AiChatMessageItem => {\n  return {\n    id,\n    contents: [{ text: content }],\n    role: \"user\",\n    createTime: dayjs().toString(),\n    createTimeStamp: dayjs().unix(),\n  };\n};\nexport const generateAiMsgItem = (content: string, showVoiceBtn = false): AiChatMessageItem => {\n  return {\n    id: generateRandomId(),\n    role: \"ai\",\n    contents: [{ text: content }],\n    createTime: new Date().toLocaleString(),\n    showVoiceBtn,\n  };\n};\n\nexport const generateAiMsgPayload = (content: string): AiAgentSocketMessagePayload => {\n  return {\n    type: AiAgentMessageType.TEXT,\n    content,\n  };\n};\n"],"names":[],"mappings":";;;AAIa,MAAA,sBAAsB,CAAC,IAAY,YAAuC;AAC9E,SAAA;AAAA,IACL;AAAA,IACA,UAAU,CAAC,EAAE,MAAM,SAAS;AAAA,IAC5B,MAAM;AAAA,IACN,YAAY,MAAM,EAAE,SAAS;AAAA,IAC7B,iBAAiB,MAAM,EAAE,KAAK;AAAA,EAChC;AACF;AACO,MAAM,oBAAoB,CAAC,SAAiB,eAAe,UAA6B;AACtF,SAAA;AAAA,IACL,IAAI,iBAAiB;AAAA,IACrB,MAAM;AAAA,IACN,UAAU,CAAC,EAAE,MAAM,SAAS;AAAA,IAC5B,aAAY,oBAAI,KAAK,GAAE,eAAe;AAAA,IACtC;AAAA,EACF;AACF;AAEa,MAAA,uBAAuB,CAAC,YAAiD;AAC7E,SAAA;AAAA,IACL,MAAM,mBAAmB;AAAA,IACzB;AAAA,EACF;AACF;"}

package/dist/lib/hooks/use-div-auto-scroll.es.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-div-auto-scroll.es.js","sources":["../../../src/hooks/use-div-auto-scroll.ts"],"sourcesContent":["import React, { useCallback, useEffect, useRef, useState } from 'react';\n\ninterface ScrollControllerProps {\n  scrollBuffer?: number;\n  scrollBehavior?: 'auto' | 'smooth';\n  containerRef: React.RefObject<HTMLDivElement | null>;\n}\n\n/**\n * 自动滚动 hook，用法是你传入一个容器的 ref，当你调用返回函数时，那么容器内的内容变化时，会自动滚动到最底部。\n * @param params 传入一个对象，包含以下属性：\n * scrollBuffer：滚动缓冲距离，默认为 50px。\n * scrollBehavior：滚动行为，默认为 'smooth'。\n * containerRef：容器的 React ref 引用对象。\n * @returns 一个触发自动滚动的函数\n */\nexport function useDivAutoScroll(params: ScrollControllerProps) {\n  const { scrollBuffer = 50, scrollBehavior = 'smooth', containerRef } = params;\n  const [isAutoScroll, setIsAutoScroll] = useState(true);\n  const isUserScrolling = useRef(false);\n  const scrollTimeout = useRef<number>();\n\n  // 检测是否接近底部\n  const isNearBottom = useCallback(() => {\n    if (!containerRef.current) return false;\n    const { scrollHeight, scrollTop, clientHeight } = containerRef.current;\n    return scrollHeight - scrollTop - clientHeight < scrollBuffer;\n  }, [containerRef, scrollBuffer]);\n\n  // 自动滚动逻辑\n  const autoScroll = useCallback(() => {\n    if (isAutoScroll && containerRef.current) {\n      containerRef.current.scrollTo({\n        top: containerRef.current.scrollHeight,\n        behavior: scrollBehavior,\n      });\n    }\n  }, [containerRef, isAutoScroll, scrollBehavior]);\n\n  // 处理滚动事件\n  const handleScroll = useCallback(() => {\n    if (!containerRef.current) return;\n\n    // 清除之前的定时器\n    if (scrollTimeout.current) clearTimeout(scrollTimeout.current);\n\n    // 标记用户滚动行为\n    if (!isUserScrolling.current) {\n      isUserScrolling.current = true;\n    }\n\n    // 设置滚动状态重置\n    scrollTimeout.current = setTimeout(() => {\n      isUserScrolling.current = false;\n    }, 100);\n\n    // 判断是否接近底部\n    if (isNearBottom()) {\n      setIsAutoScroll(true);\n    } else {\n      setIsAutoScroll(false);\n    }\n  }, [containerRef, isNearBottom]);\n\n  // 初始化事件监听\n  useEffect(() => {\n    const container = containerRef.current;\n    if (!container) return;\n\n    container.addEventListener('scroll', handleScroll);\n    return () => container.removeEventListener('scroll', handleScroll);\n  }, [containerRef, handleScroll]);\n\n  // 内容变化时自动滚动\n  const triggerAutoScroll = useCallback(() => {\n    if (!isUserScrolling.current) {\n      autoScroll();\n    }\n  }, [autoScroll, isUserScrolling]);\n\n  return triggerAutoScroll;\n}\n"],"names":[],"mappings":";AAgBO,SAAS,iBAAiB,QAA+B;AAC9D,QAAM,EAAE,eAAe,IAAI,iBAAiB,UAAU,iBAAiB;AACvE,QAAM,CAAC,cAAc,eAAe,IAAI,SAAS,IAAI;AAC/C,QAAA,kBAAkB,OAAO,KAAK;AACpC,QAAM,gBAAgB,OAAe;AAG/B,QAAA,eAAe,YAAY,MAAM;AACjC,QAAA,CAAC,aAAa,QAAgB,QAAA;AAClC,UAAM,EAAE,cAAc,WAAW,iBAAiB,aAAa;AACxD,WAAA,eAAe,YAAY,eAAe;AAAA,EAAA,GAChD,CAAC,cAAc,YAAY,CAAC;AAGzB,QAAA,aAAa,YAAY,MAAM;AAC/B,QAAA,gBAAgB,aAAa,SAAS;AACxC,mBAAa,QAAQ,SAAS;AAAA,QAC5B,KAAK,aAAa,QAAQ;AAAA,QAC1B,UAAU;AAAA,MAAA,CACX;AAAA,IAAA;AAAA,EAEF,GAAA,CAAC,cAAc,cAAc,cAAc,CAAC;AAGzC,QAAA,eAAe,YAAY,MAAM;AACjC,QAAA,CAAC,aAAa,QAAS;AAG3B,QAAI,cAAc,QAAsB,cAAA,cAAc,OAAO;AAGzD,QAAA,CAAC,gBAAgB,SAAS;AAC5B,sBAAgB,UAAU;AAAA,IAAA;AAId,kBAAA,UAAU,WAAW,MAAM;AACvC,sBAAgB,UAAU;AAAA,OACzB,GAAG;AAGN,QAAI,gBAAgB;AAClB,sBAAgB,IAAI;AAAA,IAAA,OACf;AACL,sBAAgB,KAAK;AAAA,IAAA;AAAA,EACvB,GACC,CAAC,cAAc,YAAY,CAAC;AAG/B,YAAU,MAAM;AACd,UAAM,YAAY,aAAa;AAC/B,QAAI,CAAC,UAAW;AAEN,cAAA,iBAAiB,UAAU,YAAY;AACjD,WAAO,MAAM,UAAU,oBAAoB,UAAU,YAAY;AAAA,EAAA,GAChE,CAAC,cAAc,YAAY,CAAC;AAGzB,QAAA,oBAAoB,YAAY,MAAM;AACtC,QAAA,CAAC,gBAAgB,SAAS;AACjB,iBAAA;AAAA,IAAA;AAAA,EACb,GACC,CAAC,YAAY,eAAe,CAAC;AAEzB,SAAA;AACT;"}
+{"version":3,"file":"use-div-auto-scroll.es.js","sources":["../../../src/hooks/use-div-auto-scroll.ts"],"sourcesContent":["import React, { useCallback, useEffect, useRef, useState } from \"react\";\n\ninterface ScrollControllerProps {\n  scrollBuffer?: number;\n  scrollBehavior?: \"auto\" | \"smooth\";\n  containerRef: React.RefObject<HTMLDivElement | null>;\n}\n\n/**\n * 自动滚动 hook，用法是你传入一个容器的 ref，当你调用返回函数时，那么容器内的内容变化时，会自动滚动到最底部。\n * @param params 传入一个对象，包含以下属性：\n * scrollBuffer：滚动缓冲距离，默认为 50px。\n * scrollBehavior：滚动行为，默认为 'smooth'。\n * containerRef：容器的 React ref 引用对象。\n * @returns 一个触发自动滚动的函数\n */\nexport function useDivAutoScroll(params: ScrollControllerProps) {\n  const { scrollBuffer = 50, scrollBehavior = \"smooth\", containerRef } = params;\n  const [isAutoScroll, setIsAutoScroll] = useState(true);\n  const isUserScrolling = useRef(false);\n  const scrollTimeout = useRef<number>();\n\n  // 检测是否接近底部\n  const isNearBottom = useCallback(() => {\n    if (!containerRef.current) return false;\n    const { scrollHeight, scrollTop, clientHeight } = containerRef.current;\n    return scrollHeight - scrollTop - clientHeight < scrollBuffer;\n  }, [containerRef, scrollBuffer]);\n\n  // 自动滚动逻辑\n  const autoScroll = useCallback(() => {\n    if (isAutoScroll && containerRef.current) {\n      containerRef.current.scrollTo({\n        top: containerRef.current.scrollHeight,\n        behavior: scrollBehavior,\n      });\n    }\n  }, [containerRef, isAutoScroll, scrollBehavior]);\n\n  // 处理滚动事件\n  const handleScroll = useCallback(() => {\n    if (!containerRef.current) return;\n\n    // 清除之前的定时器\n    if (scrollTimeout.current) clearTimeout(scrollTimeout.current);\n\n    // 标记用户滚动行为\n    if (!isUserScrolling.current) {\n      isUserScrolling.current = true;\n    }\n\n    // 设置滚动状态重置\n    scrollTimeout.current = setTimeout(() => {\n      isUserScrolling.current = false;\n    }, 100);\n\n    // 判断是否接近底部\n    if (isNearBottom()) {\n      setIsAutoScroll(true);\n    } else {\n      setIsAutoScroll(false);\n    }\n  }, [containerRef, isNearBottom]);\n\n  // 初始化事件监听\n  useEffect(() => {\n    const container = containerRef.current;\n    if (!container) return;\n\n    container.addEventListener(\"scroll\", handleScroll);\n    return () => container.removeEventListener(\"scroll\", handleScroll);\n  }, [containerRef, handleScroll]);\n\n  // 内容变化时自动滚动\n  const triggerAutoScroll = useCallback(() => {\n    if (!isUserScrolling.current) {\n      autoScroll();\n    }\n  }, [autoScroll, isUserScrolling]);\n\n  return triggerAutoScroll;\n}\n"],"names":[],"mappings":";AAgBO,SAAS,iBAAiB,QAA+B;AAC9D,QAAM,EAAE,eAAe,IAAI,iBAAiB,UAAU,iBAAiB;AACvE,QAAM,CAAC,cAAc,eAAe,IAAI,SAAS,IAAI;AAC/C,QAAA,kBAAkB,OAAO,KAAK;AACpC,QAAM,gBAAgB,OAAe;AAG/B,QAAA,eAAe,YAAY,MAAM;AACjC,QAAA,CAAC,aAAa,QAAgB,QAAA;AAClC,UAAM,EAAE,cAAc,WAAW,iBAAiB,aAAa;AACxD,WAAA,eAAe,YAAY,eAAe;AAAA,EAAA,GAChD,CAAC,cAAc,YAAY,CAAC;AAGzB,QAAA,aAAa,YAAY,MAAM;AAC/B,QAAA,gBAAgB,aAAa,SAAS;AACxC,mBAAa,QAAQ,SAAS;AAAA,QAC5B,KAAK,aAAa,QAAQ;AAAA,QAC1B,UAAU;AAAA,MAAA,CACX;AAAA,IAAA;AAAA,EAEF,GAAA,CAAC,cAAc,cAAc,cAAc,CAAC;AAGzC,QAAA,eAAe,YAAY,MAAM;AACjC,QAAA,CAAC,aAAa,QAAS;AAG3B,QAAI,cAAc,QAAsB,cAAA,cAAc,OAAO;AAGzD,QAAA,CAAC,gBAAgB,SAAS;AAC5B,sBAAgB,UAAU;AAAA,IAAA;AAId,kBAAA,UAAU,WAAW,MAAM;AACvC,sBAAgB,UAAU;AAAA,OACzB,GAAG;AAGN,QAAI,gBAAgB;AAClB,sBAAgB,IAAI;AAAA,IAAA,OACf;AACL,sBAAgB,KAAK;AAAA,IAAA;AAAA,EACvB,GACC,CAAC,cAAc,YAAY,CAAC;AAG/B,YAAU,MAAM;AACd,UAAM,YAAY,aAAa;AAC/B,QAAI,CAAC,UAAW;AAEN,cAAA,iBAAiB,UAAU,YAAY;AACjD,WAAO,MAAM,UAAU,oBAAoB,UAAU,YAAY;AAAA,EAAA,GAChE,CAAC,cAAc,YAAY,CAAC;AAGzB,QAAA,oBAAoB,YAAY,MAAM;AACtC,QAAA,CAAC,gBAAgB,SAAS;AACjB,iBAAA;AAAA,IAAA;AAAA,EACb,GACC,CAAC,YAAY,eAAe,CAAC;AAEzB,SAAA;AACT;"}

package/dist/lib/hooks/use-document-unmount/use-document-unmount.es.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-document-unmount.es.js","sources":["../../../../src/hooks/use-document-unmount/use-document-unmount.ts"],"sourcesContent":["import { useEffect } from 'react';\nimport { useMemoizedFn } from 'ahooks';\n\nexport function useDocumentUnmount(fn: () => void) {\n  const finalFn = useMemoizedFn(fn);\n  // 监听页面卸载事件\n  useEffect(() => {\n    const handleBeforeUnload = () => {\n      finalFn();\n    };\n\n    window.addEventListener('beforeunload', handleBeforeUnload);\n\n    return () => {\n      window.removeEventListener('beforeunload', handleBeforeUnload);\n    };\n  }, [finalFn]);\n}\n"],"names":[],"mappings":";;AAGO,SAAS,mBAAmB,IAAgB;AAC3C,QAAA,UAAU,cAAc,EAAE;AAEhC,YAAU,MAAM;AACd,UAAM,qBAAqB,MAAM;AACvB,cAAA;AAAA,IACV;AAEO,WAAA,iBAAiB,gBAAgB,kBAAkB;AAE1D,WAAO,MAAM;AACJ,aAAA,oBAAoB,gBAAgB,kBAAkB;AAAA,IAC/D;AAAA,EAAA,GACC,CAAC,OAAO,CAAC;AACd;"}
+{"version":3,"file":"use-document-unmount.es.js","sources":["../../../../src/hooks/use-document-unmount/use-document-unmount.ts"],"sourcesContent":["import { useEffect } from \"react\";\nimport { useMemoizedFn } from \"ahooks\";\n\nexport function useDocumentUnmount(fn: () => void) {\n  const finalFn = useMemoizedFn(fn);\n  // 监听页面卸载事件\n  useEffect(() => {\n    const handleBeforeUnload = () => {\n      finalFn();\n    };\n\n    window.addEventListener(\"beforeunload\", handleBeforeUnload);\n\n    return () => {\n      window.removeEventListener(\"beforeunload\", handleBeforeUnload);\n    };\n  }, [finalFn]);\n}\n"],"names":[],"mappings":";;AAGO,SAAS,mBAAmB,IAAgB;AAC3C,QAAA,UAAU,cAAc,EAAE;AAEhC,YAAU,MAAM;AACd,UAAM,qBAAqB,MAAM;AACvB,cAAA;AAAA,IACV;AAEO,WAAA,iBAAiB,gBAAgB,kBAAkB;AAE1D,WAAO,MAAM;AACJ,aAAA,oBAAoB,gBAAgB,kBAAkB;AAAA,IAC/D;AAAA,EAAA,GACC,CAAC,OAAO,CAAC;AACd;"}

package/dist/lib/hooks/use-initialized.es.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-initialized.es.js","sources":["../../../src/hooks/use-initialized.ts"],"sourcesContent":["import { useEffect, useRef, useState } from 'react';\n\ntype InitFunctionType = (() => Promise<void>) | (() => void);\n/** 传入的函数只会初始化运行一次\n * @param fn 初始化函数， fn 返回的如果是一个函数，会在卸载时运行。\n * @returns initialized 是否初始化，boolean 值是一个 react state，会触发重新渲染\n */\nexport function useInitialized(fn?: InitFunctionType) {\n  const [initialized, setInitialized] = useState(false);\n  const initializedRef = useRef(false);\n  const updateFnRef = useRef(fn);\n  updateFnRef.current = fn;\n\n  useEffect(() => {\n    const initFn = async () => {\n      if (!initializedRef.current) {\n        await updateFnRef.current?.();\n        setInitialized(true);\n        initializedRef.current = true;\n      }\n    };\n    initFn();\n  }, []);\n\n  return initialized;\n}\n"],"names":[],"mappings":";AAOO,SAAS,eAAe,IAAuB;AACpD,QAAM,CAAC,aAAa,cAAc,IAAI,SAAS,KAAK;AAC9C,QAAA,iBAAiB,OAAO,KAAK;AAC7B,QAAA,cAAc,OAAO,EAAE;AAC7B,cAAY,UAAU;AAEtB,YAAU,MAAM;AACd,UAAM,SAAS,YAAY;;AACrB,UAAA,CAAC,eAAe,SAAS;AAC3B,gBAAM,iBAAY,YAAZ;AACN,uBAAe,IAAI;AACnB,uBAAe,UAAU;AAAA,MAAA;AAAA,IAE7B;AACO,WAAA;AAAA,EACT,GAAG,EAAE;AAEE,SAAA;AACT;"}
+{"version":3,"file":"use-initialized.es.js","sources":["../../../src/hooks/use-initialized.ts"],"sourcesContent":["import { useEffect, useRef, useState } from \"react\";\n\ntype InitFunctionType = (() => Promise<void>) | (() => void);\n/** 传入的函数只会初始化运行一次\n * @param fn 初始化函数， fn 返回的如果是一个函数，会在卸载时运行。\n * @returns initialized 是否初始化，boolean 值是一个 react state，会触发重新渲染\n */\nexport function useInitialized(fn?: InitFunctionType) {\n  const [initialized, setInitialized] = useState(false);\n  const initializedRef = useRef(false);\n  const updateFnRef = useRef(fn);\n  updateFnRef.current = fn;\n\n  useEffect(() => {\n    const initFn = async () => {\n      if (!initializedRef.current) {\n        await updateFnRef.current?.();\n        setInitialized(true);\n        initializedRef.current = true;\n      }\n    };\n    initFn();\n  }, []);\n\n  return initialized;\n}\n"],"names":[],"mappings":";AAOO,SAAS,eAAe,IAAuB;AACpD,QAAM,CAAC,aAAa,cAAc,IAAI,SAAS,KAAK;AAC9C,QAAA,iBAAiB,OAAO,KAAK;AAC7B,QAAA,cAAc,OAAO,EAAE;AAC7B,cAAY,UAAU;AAEtB,YAAU,MAAM;AACd,UAAM,SAAS,YAAY;;AACrB,UAAA,CAAC,eAAe,SAAS;AAC3B,gBAAM,iBAAY,YAAZ;AACN,uBAAe,IAAI;AACnB,uBAAe,UAAU;AAAA,MAAA;AAAA,IAE7B;AACO,WAAA;AAAA,EACT,GAAG,EAAE;AAEE,SAAA;AACT;"}