@ai-group/chat-sdk 3.6.3 → 3.6.5

package/dist/cjs/hooks/useADKChat.js

@@ -49,6 +49,202 @@ var combineTextParts = (parts) => {
   }
   return result;
 };
+var DEFAULT_SESSION_EVENTS_POLLING_INTERVAL = 1e4;
+var getMessageEventId = (messageItem) => {
+  var _a;
+  return messageItem.eventId || ((_a = messageItem.raw) == null ? void 0 : _a.id);
+};
+var mapSessionMessages = (sessionEvents) => {
+  const mapped = [];
+  sessionEvents.forEach((item) => {
+    if (!item || !item.content || !Array.isArray(item.content.parts))
+      return;
+    if (item.author === "user" && item.content.parts.some((part) => part.task))
+      return;
+    const role = (item.content.role || "").toLowerCase() === "user" ? "user" : "bot";
+    const parts = item.content.parts.filter((part) => {
+      if (!part)
+        return false;
+      return Boolean(
+        part.text || part.inlineData || part.functionCall || part.functionResponse || part.fileData || part.executableCode || part.codeExecutionResult || part.errorMessage
+      );
+    });
+    if (parts.length === 0)
+      return;
+    const textParts = parts.filter(
+      (part) => part.text || part.errorMessage || part.inlineData
+    );
+    const fileDataParts = parts.filter((part) => part.fileData);
+    const otherParts = parts.filter(
+      (part) => !part.text && !part.errorMessage && !part.inlineData && !part.fileData
+    );
+    if (textParts.length > 0 || fileDataParts.length > 0) {
+      const messageItem = {
+        id: (0, import_uuid.v4)(),
+        author: item.author,
+        invocationId: item.invocationId,
+        eventId: item.id,
+        timestamp: item.timestamp,
+        isLike: item.isLike,
+        role,
+        modelCode: item.modelCode,
+        usageMetadata: item.usageMetadata,
+        finishReason: item.finishReason,
+        raw: item
+      };
+      textParts.forEach((part) => {
+        if (part.inlineData) {
+          messageItem.inlineData = {
+            displayName: part.inlineData.displayName,
+            data: part.inlineData.data,
+            mimeType: part.inlineData.mimeType
+          };
+        }
+        if (part.text) {
+          messageItem.text = (messageItem.text ? `${messageItem.text}
+` : "") + part.text;
+          if (typeof part.thought !== "undefined")
+            messageItem.thought = part.thought;
+        }
+        if (part.errorMessage) {
+          messageItem.text = (messageItem.text ? `${messageItem.text}
+` : "") + part.errorMessage;
+        }
+      });
+      if (fileDataParts.length > 0) {
+        messageItem.fileData = fileDataParts.map((part) => ({
+          displayName: part.fileData.displayName || "",
+          fileUri: part.fileData.fileUri,
+          mimeType: part.fileData.mimeType
+        }));
+      }
+      mapped.push(messageItem);
+    }
+    otherParts.forEach((part) => {
+      const messageItem = {
+        id: (0, import_uuid.v4)(),
+        author: item.author,
+        invocationId: item.invocationId,
+        eventId: item.id,
+        timestamp: item.timestamp,
+        isLike: item.isLike,
+        role,
+        modelCode: item.modelCode,
+        usageMetadata: item.usageMetadata,
+        finishReason: item.finishReason,
+        raw: item
+      };
+      if (part.functionCall)
+        messageItem.functionCall = part.functionCall;
+      if (part.executableCode)
+        messageItem.executableCode = part.executableCode;
+      if (part.codeExecutionResult)
+        messageItem.codeExecutionResult = part.codeExecutionResult;
+      if (part.functionResponse) {
+        const functionCallMessage = mapped.find(
+          (currentMessage) => {
+            var _a, _b;
+            return ((_a = currentMessage.functionCall) == null ? void 0 : _a.id) === ((_b = part.functionResponse) == null ? void 0 : _b.id);
+          }
+        );
+        if (functionCallMessage) {
+          functionCallMessage.functionResponse = part.functionResponse;
+        }
+        return;
+      }
+      mapped.push(messageItem);
+    });
+  });
+  return mapped;
+};
+var getComparableMessageContent = (messageItem) => {
+  var _a;
+  return JSON.stringify({
+    role: messageItem.role,
+    text: ((_a = messageItem.text) == null ? void 0 : _a.replace(/\r\n/g, "\n").trim()) || "",
+    thought: Boolean(messageItem.thought),
+    inlineData: messageItem.inlineData || null,
+    fileData: messageItem.fileData || null,
+    functionCall: messageItem.functionCall || null,
+    functionResponse: messageItem.functionResponse || null,
+    executableCode: messageItem.executableCode || null,
+    codeExecutionResult: messageItem.codeExecutionResult || null
+  });
+};
+var isStreamMessageWithoutServerEventId = (messageItem) => {
+  const eventId = getMessageEventId(messageItem);
+  return Boolean(messageItem.invocationId) && (!eventId || eventId === messageItem.invocationId);
+};
+var insertMissingSessionEvents = (currentMessages, sessionEvents) => {
+  const renderedEventIds = new Set(
+    currentMessages.map(getMessageEventId).filter(Boolean)
+  );
+  const unmatchedStreamMessageIndexes = new Set(
+    currentMessages.map(
+      (messageItem, index) => isStreamMessageWithoutServerEventId(messageItem) ? index : -1
+    ).filter((index) => index >= 0)
+  );
+  const missingEvents = sessionEvents.filter((event) => {
+    var _a, _b;
+    const parts = (_a = event.content) == null ? void 0 : _a.parts;
+    const canRender = Boolean(event.id) && ((_b = event.content) == null ? void 0 : _b.role) === "model" && Array.isArray(parts) && parts.length > 0;
+    if (!canRender || renderedEventIds.has(event.id))
+      return false;
+    const eventMessages = mapSessionMessages([event]);
+    const matchedIndexes = [];
+    const isRenderedByStream = eventMessages.length > 0 && eventMessages.every((eventMessage) => {
+      const comparableContent = getComparableMessageContent(eventMessage);
+      const matchedIndex = currentMessages.findIndex(
+        (currentMessage, index) => unmatchedStreamMessageIndexes.has(index) && !matchedIndexes.includes(index) && currentMessage.invocationId === eventMessage.invocationId && getComparableMessageContent(currentMessage) === comparableContent
+      );
+      if (matchedIndex < 0)
+        return false;
+      matchedIndexes.push(matchedIndex);
+      return true;
+    });
+    if (isRenderedByStream) {
+      matchedIndexes.forEach(
+        (index) => unmatchedStreamMessageIndexes.delete(index)
+      );
+      return false;
+    }
+    return true;
+  });
+  if (missingEvents.length === 0)
+    return currentMessages;
+  const missingMessages = mapSessionMessages(missingEvents);
+  if (missingMessages.length === 0)
+    return currentMessages;
+  const eventOrder = new Map(
+    sessionEvents.map((event, index) => [event.id, index])
+  );
+  const getMessageOrder = (messageItem) => {
+    var _a;
+    const eventId = getMessageEventId(messageItem);
+    if (eventId && eventOrder.has(eventId))
+      return eventOrder.get(eventId);
+    const timestamp = messageItem.timestamp ?? ((_a = messageItem.raw) == null ? void 0 : _a.timestamp);
+    if (timestamp !== void 0) {
+      const nextEventIndex = sessionEvents.findIndex(
+        (event) => (event.timestamp || 0) > timestamp
+      );
+      return nextEventIndex < 0 ? sessionEvents.length + 0.5 : nextEventIndex - 0.5;
+    }
+    return Number.POSITIVE_INFINITY;
+  };
+  const nextMessages = [...currentMessages];
+  missingMessages.forEach((messageItem) => {
+    const targetOrder = getMessageOrder(messageItem);
+    let insertIndex = nextMessages.findIndex((currentMessage) => {
+      const currentOrder = getMessageOrder(currentMessage);
+      return Number.isFinite(currentOrder) && currentOrder > targetOrder;
+    });
+    if (insertIndex < 0)
+      insertIndex = nextMessages.length;
+    nextMessages.splice(insertIndex, 0, messageItem);
+  });
+  return nextMessages;
+};
 function useADKChat({
   url = window.location.origin,
   token,
@@ -62,13 +258,24 @@ function useADKChat({
   onSuccess,
   onStream
 }) {
-  const [loading, setLoading] = (0, import_react.useState)(false);
+  var _a, _b;
+  const [loading, setLoadingState] = (0, import_react.useState)(false);
+  const loadingRef = (0, import_react.useRef)(false);
+  const setLoading = (0, import_react.useCallback)((nextLoading) => {
+    loadingRef.current = nextLoading;
+    setLoadingState(nextLoading);
+  }, []);
   const [sessionLoading, setSessionLoading] = (0, import_react.useState)(false);
   const sendingRef = (0, import_react.useRef)(false);
   const ctrl = (0, import_react.useRef)(null);
   const tokenRef = (0, import_react.useRef)(token);
   tokenRef.current = token;
   const { appNo, showFirstSession } = config || {};
+  const pollSessionEvents = ((_a = config == null ? void 0 : config.session) == null ? void 0 : _a.pollSessionEvents) ?? true;
+  const sessionEventsPollingInterval = Math.max(
+    ((_b = config == null ? void 0 : config.session) == null ? void 0 : _b.sessionEventsPollingInterval) ?? DEFAULT_SESSION_EVENTS_POLLING_INTERVAL,
+    1e3
+  );
   const [appInfo, setAppInfo] = (0, import_react.useState)(null);
   const [initialized, setInitialized] = (0, import_react.useState)(false);
   const [sessionList, setSessionList] = (0, import_react.useState)([]);
@@ -81,6 +288,10 @@ function useADKChat({
   const [prologue, setPrologue] = (0, import_react.useState)("");
   const [suggestedQuestions, setSuggestedQuestions] = (0, import_react.useState)([]);
   const [messages, setMessages] = (0, import_react.useState)([]);
+  const pendingSessionEventsRef = (0, import_react.useRef)(null);
+  const sessionEventsFlushTimerRef = (0, import_react.useRef)(null);
+  const sessionEventsSyncingRef = (0, import_react.useRef)(false);
+  const sessionDetailLoadingRef = (0, import_react.useRef)(false);
   const stateDeltaRef = (0, import_react.useRef)(void 0);
   const backgroundRef = (0, import_react.useRef)(/* @__PURE__ */ new Map());
   const messagesRef = (0, import_react.useRef)(messages);
@@ -90,17 +301,17 @@ function useADKChat({
   const mergedMessages = (0, import_react.useMemo)(() => {
     const fnResMap = {};
     messages.forEach((msg) => {
-      var _a;
+      var _a2;
       if (msg.functionResponse) {
-        fnResMap[((_a = msg == null ? void 0 : msg.functionResponse) == null ? void 0 : _a.id) || ""] = msg.functionResponse;
+        fnResMap[((_a2 = msg == null ? void 0 : msg.functionResponse) == null ? void 0 : _a2.id) || ""] = msg.functionResponse;
       }
     });
     return messages.map((msg) => {
-      var _a;
+      var _a2;
       if (msg.functionCall) {
         return {
           ...msg,
-          functionResponse: fnResMap[((_a = msg == null ? void 0 : msg.functionCall) == null ? void 0 : _a.id) || ""]
+          functionResponse: fnResMap[((_a2 = msg == null ? void 0 : msg.functionCall) == null ? void 0 : _a2.id) || ""]
         };
       }
       return msg;
@@ -108,6 +319,66 @@ function useADKChat({
   }, [messages]);
   const textMsgRef = (0, import_react.useRef)(null);
   const eventDataRef = (0, import_react.useRef)(/* @__PURE__ */ new Map());
+  const cancelSessionEventsFlush = (0, import_react.useCallback)(() => {
+    if (sessionEventsFlushTimerRef.current === null)
+      return;
+    clearTimeout(sessionEventsFlushTimerRef.current);
+    sessionEventsFlushTimerRef.current = null;
+  }, []);
+  const flushPendingSessionEvents = (0, import_react.useCallback)(() => {
+    const pending = pendingSessionEventsRef.current;
+    const replyInProgress = loadingRef.current || sendingRef.current || textMsgRef.current !== null;
+    if (!pending || replyInProgress || pending.sessionId !== currentSessionIdRef.current)
+      return;
+    const previewMessages = insertMissingSessionEvents(
+      messagesRef.current,
+      pending.events
+    );
+    pendingSessionEventsRef.current = null;
+    if (previewMessages === messagesRef.current)
+      return;
+    sessionEventsSyncingRef.current = true;
+    setLoading(true);
+    sessionEventsFlushTimerRef.current = setTimeout(() => {
+      sessionEventsFlushTimerRef.current = null;
+      if (pending.sessionId === currentSessionIdRef.current) {
+        setMessages((prev) => {
+          const nextMessages = insertMissingSessionEvents(
+            prev,
+            pending.events
+          );
+          messagesRef.current = nextMessages;
+          return nextMessages;
+        });
+      }
+      sessionEventsSyncingRef.current = false;
+      setLoading(false);
+    }, 0);
+  }, [setLoading]);
+  const syncSessionEvents = (0, import_react.useCallback)(
+    (events, targetSessionId) => {
+      if (targetSessionId !== currentSessionIdRef.current)
+        return;
+      pendingSessionEventsRef.current = {
+        sessionId: targetSessionId,
+        events
+      };
+      flushPendingSessionEvents();
+    },
+    [flushPendingSessionEvents]
+  );
+  (0, import_react.useEffect)(() => {
+    pendingSessionEventsRef.current = null;
+    cancelSessionEventsFlush();
+    if (sessionEventsSyncingRef.current) {
+      sessionEventsSyncingRef.current = false;
+      setLoading(false);
+    }
+  }, [cancelSessionEventsFlush, currentSessionId, setLoading]);
+  (0, import_react.useEffect)(() => {
+    if (!loading)
+      flushPendingSessionEvents();
+  }, [flushPendingSessionEvents, loading]);
   const insertMessage = (0, import_react.useCallback)((msg) => {
     setMessages((prev) => {
       const lastMessage = prev[prev.length - 1];
@@ -167,9 +438,9 @@ function useADKChat({
     [withEventMeta]
   );
   const storeEvents = (0, import_react.useCallback)((part, event) => {
-    var _a, _b;
+    var _a2, _b2;
     let title = "";
-    if (part == null && ((_a = event.actions) == null ? void 0 : _a.artifactDelta)) {
+    if (part == null && ((_a2 = event.actions) == null ? void 0 : _a2.artifactDelta)) {
       title += "eventAction: artifact";
     } else if (part) {
       if (part.text) {
@@ -179,7 +450,7 @@ function useADKChat({
       } else if (part.functionResponse) {
         title += `functionResponse:${part.functionResponse.name}`;
       } else if (part.executableCode) {
-        title += `executableCode:${(_b = part.executableCode.code) == null ? void 0 : _b.slice(0, 10)}`;
+        title += `executableCode:${(_b2 = part.executableCode.code) == null ? void 0 : _b2.slice(0, 10)}`;
       } else if (part.codeExecutionResult) {
         title += `codeExecutionResult:${part.codeExecutionResult.outcome}`;
       } else if (part.errorMessage) {
@@ -193,7 +464,7 @@ function useADKChat({
   }, []);
   const storeMessage = (0, import_react.useCallback)(
     (part, event, role) => {
-      var _a, _b;
+      var _a2, _b2;
       const msg = {
         id: (0, import_uuid.v4)(),
         author: event.author,
@@ -217,7 +488,7 @@ function useADKChat({
         } else if (part.text) {
           msg.text = part.text;
           msg.thought = part.thought;
-          if ((_b = (_a = event == null ? void 0 : event.groundingMetadata) == null ? void 0 : _a.searchEntryPoint) == null ? void 0 : _b.renderedContent) {
+          if ((_b2 = (_a2 = event == null ? void 0 : event.groundingMetadata) == null ? void 0 : _a2.searchEntryPoint) == null ? void 0 : _b2.renderedContent) {
             msg.renderedContent = event.groundingMetadata.searchEntryPoint.renderedContent;
           }
         } else if (part.fileData) {
@@ -240,8 +511,8 @@ function useADKChat({
   );
   const processPart = (0, import_react.useCallback)(
     (event, part, ignore = false) => {
-      var _a, _b;
-      const renderedContent = (_b = (_a = event.groundingMetadata) == null ? void 0 : _a.searchEntryPoint) == null ? void 0 : _b.renderedContent;
+      var _a2, _b2;
+      const renderedContent = (_b2 = (_a2 = event.groundingMetadata) == null ? void 0 : _a2.searchEntryPoint) == null ? void 0 : _b2.renderedContent;
       if (part.text) {
         if (ignore)
           return;
@@ -295,8 +566,8 @@ function useADKChat({
   );
   const processFollowupPart = (0, import_react.useCallback)(
     (event) => {
-      var _a, _b;
-      const part = ((_b = (_a = event == null ? void 0 : event.content) == null ? void 0 : _a.parts) == null ? void 0 : _b[0]) || null;
+      var _a2, _b2;
+      const part = ((_b2 = (_a2 = event == null ? void 0 : event.content) == null ? void 0 : _a2.parts) == null ? void 0 : _b2[0]) || null;
       const text = (part == null ? void 0 : part.text) || "";
       storeEvents(part, event);
       if (text) {
@@ -381,7 +652,7 @@ function useADKChat({
                 content: rawData.data,
                 invocationId: rawData.result.invocationId,
                 sessionId: rawData.result.sessionId,
-                id: rawData.result.invocationId
+                id: rawData.data.id || rawData.result.id || rawData.result.eventId || rawData.result.invocationId
               };
             }
             if (chunkJson.error) {
@@ -480,7 +751,7 @@ function useADKChat({
             }
           },
           onclose: () => {
-            var _a;
+            var _a2;
             const isBackground = currentSessionIdRef.current !== targetSessionId;
             if (isBackground) {
               const bg = backgroundRef.current.get(targetSessionId);
@@ -491,7 +762,7 @@ function useADKChat({
             } else {
               setLoading(false);
               if (textMsgRef.current) {
-                onMessage == null ? void 0 : onMessage(((_a = textMsgRef.current) == null ? void 0 : _a.text) || "", textMsgRef.current);
+                onMessage == null ? void 0 : onMessage(((_a2 = textMsgRef.current) == null ? void 0 : _a2.text) || "", textMsgRef.current);
               }
               textMsgRef.current = null;
             }
@@ -527,7 +798,7 @@ function useADKChat({
     files = [],
     functionResponse
   }) => {
-    if (loading || sendingRef.current)
+    if (loadingRef.current || sendingRef.current)
       return;
     if (!text.trim() && !(files == null ? void 0 : files.length) && !functionResponse)
       return;
@@ -547,11 +818,11 @@ function useADKChat({
           id: (0, import_uuid.v4)(),
           role: "user",
           fileData: files.map((file) => {
-            var _a, _b, _c;
+            var _a2, _b2, _c;
             return {
               displayName: file.fileName || file.name,
-              mimeType: file.mimeType || ((_a = file.response) == null ? void 0 : _a.mimeType),
-              fileUri: file.tempUrl || ((_b = file.response) == null ? void 0 : _b.fileUrl) || ((_c = file.response) == null ? void 0 : _c.tempUrl)
+              mimeType: file.mimeType || ((_a2 = file.response) == null ? void 0 : _a2.mimeType),
+              fileUri: file.tempUrl || ((_b2 = file.response) == null ? void 0 : _b2.fileUrl) || ((_c = file.response) == null ? void 0 : _c.tempUrl)
             };
           })
         });
@@ -574,10 +845,10 @@ function useADKChat({
           parts: functionResponse ? [{ functionResponse }] : [{ text }]
         },
         files: files.map((file) => {
-          var _a, _b, _c, _d, _e, _f;
+          var _a2, _b2, _c, _d, _e, _f;
           return {
-            fileName: file.fileName || ((_a = file.response) == null ? void 0 : _a.fileName),
-            fileId: file.fileId || ((_b = file.response) == null ? void 0 : _b.fileId),
+            fileName: file.fileName || ((_a2 = file.response) == null ? void 0 : _a2.fileName),
+            fileId: file.fileId || ((_b2 = file.response) == null ? void 0 : _b2.fileId),
             tempUrl: file.tempUrl || ((_c = file.response) == null ? void 0 : _c.fileUrl) || ((_d = file.response) == null ? void 0 : _d.tempUrl),
             type: file.type || ((_e = file.response) == null ? void 0 : _e.fileType),
             mimeType: file.mimeType || ((_f = file.response) == null ? void 0 : _f.mimeType)
@@ -593,7 +864,7 @@ function useADKChat({
     initAppConversations();
   };
   const reChat = async () => {
-    if (loading || sendingRef.current)
+    if (loadingRef.current || sendingRef.current)
       return;
     if (messages.length === 0)
       return;
@@ -632,8 +903,8 @@ function useADKChat({
     }
   };
   const confirmFnCall = (fnCall, confirmed) => {
-    var _a;
-    const functionResponse = ((_a = strategies == null ? void 0 : strategies.createFunctionResponse) == null ? void 0 : _a.call(
+    var _a2;
+    const functionResponse = ((_a2 = strategies == null ? void 0 : strategies.createFunctionResponse) == null ? void 0 : _a2.call(
       strategies,
       fnCall,
       confirmed
@@ -645,7 +916,7 @@ function useADKChat({
     startChat({ functionResponse });
   };
   const suggestChat = (text) => {
-    if (loading)
+    if (loadingRef.current)
       return;
     if (!text.trim())
       return;
@@ -653,126 +924,43 @@ function useADKChat({
     startChat({ text });
   };
   const stopChat = (0, import_react.useCallback)(() => {
-    var _a;
-    (_a = ctrl.current) == null ? void 0 : _a.abort();
+    var _a2;
+    pendingSessionEventsRef.current = null;
+    cancelSessionEventsFlush();
+    sessionEventsSyncingRef.current = false;
+    (_a2 = ctrl.current) == null ? void 0 : _a2.abort();
     ctrl.current = null;
     setLoading(false);
     textMsgRef.current = null;
-  }, []);
+  }, [cancelSessionEventsFlush, setLoading]);
   const clearChat = () => {
-    var _a, _b;
+    var _a2, _b2;
     const newSessionId = (0, import_uuid.v4)();
     setCurrentSessionId(newSessionId);
     stopChat();
-    setPrologue(((_a = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _a.prologue) || "");
+    setPrologue(((_a2 = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _a2.prologue) || "");
     setMessages([]);
-    setSuggestedQuestions(((_b = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _b.suggested_questions) || []);
-  };
-  const formatMessages = (messages2, isReplace = false) => {
-    const mapped = [];
-    messages2.forEach((item) => {
-      if (!item || !item.content || !Array.isArray(item.content.parts))
-        return;
-      const role = (item.content.role || "").toLowerCase() === "user" ? "user" : "bot";
-      const parts = Array.isArray(item.content.parts) ? item.content.parts.filter((p) => {
-        if (!p)
-          return false;
-        return Boolean(
-          p.text || p.inlineData || p.functionCall || p.functionResponse || p.fileData || p.executableCode || p.codeExecutionResult || p.errorMessage
-        );
-      }) : [];
-      if (parts.length === 0)
-        return;
-      const textParts = parts.filter(
-        (p) => p.text || p.errorMessage || p.inlineData
-      );
-      const fileDataParts = parts.filter((p) => p.fileData);
-      const otherParts = parts.filter(
-        (p) => !p.text && !p.errorMessage && !p.inlineData && !p.fileData
-      );
-      if (textParts.length > 0 || fileDataParts.length > 0) {
-        const msg = {
-          id: (0, import_uuid.v4)(),
-          author: item.author,
-          invocationId: item.invocationId,
-          eventId: item.id,
-          timestamp: item.timestamp,
-          isLike: item.isLike,
-          role,
-          modelCode: item.modelCode,
-          usageMetadata: item.usageMetadata,
-          finishReason: item.finishReason,
-          raw: item
-        };
-        for (const part of textParts) {
-          if (part.inlineData) {
-            msg.inlineData = {
-              displayName: part.inlineData.displayName,
-              data: part.inlineData.data,
-              mimeType: part.inlineData.mimeType
-            };
-          }
-          if (part.text) {
-            msg.text = (msg.text ? msg.text + "\n" : "") + part.text;
-            if (typeof part.thought !== "undefined")
-              msg.thought = part.thought;
-          }
-          if (part.errorMessage) {
-            msg.text = (msg.text ? msg.text + "\n" : "") + part.errorMessage;
-          }
-        }
-        if (fileDataParts.length > 0) {
-          msg.fileData = fileDataParts.map((part) => ({
-            displayName: part.fileData.displayName || "",
-            fileUri: part.fileData.fileUri,
-            mimeType: part.fileData.mimeType
-          }));
-        }
-        mapped.push(msg);
-      }
-      otherParts.forEach((part) => {
-        const msg = {
-          id: (0, import_uuid.v4)(),
-          author: item.author,
-          invocationId: item.invocationId,
-          eventId: item.id,
-          timestamp: item.timestamp,
-          isLike: item.isLike,
-          role,
-          modelCode: item.modelCode,
-          usageMetadata: item.usageMetadata,
-          finishReason: item.finishReason,
-          raw: item
-        };
-        if (part.functionCall)
-          msg.functionCall = part.functionCall;
-        if (part.executableCode)
-          msg.executableCode = part.executableCode;
-        if (part.codeExecutionResult)
-          msg.codeExecutionResult = part.codeExecutionResult;
-        if (part.functionResponse) {
-          const functionCallMsg = mapped.find(
-            (msg2) => {
-              var _a, _b;
-              return ((_a = msg2.functionCall) == null ? void 0 : _a.id) === ((_b = part.functionResponse) == null ? void 0 : _b.id);
-            }
-          );
-          if (functionCallMsg) {
-            functionCallMsg.functionResponse = part.functionResponse;
-          }
-          return;
-        }
-        mapped.push(msg);
-      });
-    });
-    setMessages((prev) => isReplace ? mapped : [...prev, ...mapped]);
+    setSuggestedQuestions(((_b2 = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _b2.suggested_questions) || []);
   };
+  const formatMessages = (0, import_react.useCallback)(
+    (sessionEvents, isReplace = false) => {
+      const mapped = mapSessionMessages(sessionEvents);
+      setMessages((prev) => isReplace ? mapped : [...prev, ...mapped]);
+    },
+    []
+  );
   const setCurrentSession = async (sessionId) => {
-    var _a, _b;
+    var _a2, _b2;
     if (sessionId) {
       if (sessionId === currentSessionId) {
         return;
       }
+      pendingSessionEventsRef.current = null;
+      cancelSessionEventsFlush();
+      if (sessionEventsSyncingRef.current) {
+        sessionEventsSyncingRef.current = false;
+        setLoading(false);
+      }
       const prevSessionId = currentSessionId;
       if (ctrl.current) {
         backgroundRef.current.set(prevSessionId, {
@@ -805,29 +993,34 @@ function useADKChat({
       }
       setInitialized(false);
       setCurrentSessionId(sessionId);
-      const { data, result } = await (0, import_api.fetchSessionDetail)({
-        url,
-        appNo,
-        sessionId,
-        token: tokenRef.current
-      });
-      if ((result == null ? void 0 : result.code) === import_constants.API_SUCCESS_CODE) {
-        setPrologue(((_a = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _a.prologue) || "");
-        if (Array.isArray(data) && data.length > 0) {
-          formatMessages(data, true);
-        } else {
-          setSuggestedQuestions(
-            ((_b = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _b.suggested_questions) || []
-          );
+      sessionDetailLoadingRef.current = true;
+      try {
+        const { data, result } = await (0, import_api.fetchSessionDetail)({
+          url,
+          appNo,
+          sessionId,
+          token: tokenRef.current
+        });
+        if ((result == null ? void 0 : result.code) === import_constants.API_SUCCESS_CODE) {
+          setPrologue(((_a2 = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _a2.prologue) || "");
+          if (Array.isArray(data) && data.length > 0) {
+            formatMessages(data, true);
+          } else {
+            setSuggestedQuestions(
+              ((_b2 = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _b2.suggested_questions) || []
+            );
+          }
         }
+      } finally {
+        sessionDetailLoadingRef.current = false;
+        setInitialized(true);
       }
-      setInitialized(true);
     } else {
       setCurrentSessionId((0, import_uuid.v4)());
     }
   };
   const initAppConversations = async (fetchDetail = false) => {
-    var _a, _b;
+    var _a2, _b2;
     try {
       setSessionLoading(true);
       const {
@@ -863,9 +1056,9 @@ function useADKChat({
           if (showFirstSession) {
             setCurrentSession(sessionId);
           } else {
-            fetchDetail && setPrologue(((_a = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _a.prologue) || "");
+            fetchDetail && setPrologue(((_a2 = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _a2.prologue) || "");
             fetchDetail && setSuggestedQuestions(
-              ((_b = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _b.suggested_questions) || []
+              ((_b2 = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _b2.suggested_questions) || []
             );
           }
           if (!initialized) {
@@ -992,17 +1185,59 @@ function useADKChat({
       initAppConversations(true);
     }
   }, [appInfo]);
+  (0, import_react.useEffect)(() => {
+    if (!enabled || !initialized || !pollSessionEvents || !appNo || !currentSessionId)
+      return;
+    let disposed = false;
+    let requesting = false;
+    const pollingSessionId = currentSessionId;
+    const poll = async () => {
+      if (requesting || sessionDetailLoadingRef.current)
+        return;
+      requesting = true;
+      try {
+        const { data, result } = await (0, import_api.fetchSessionDetail)({
+          url,
+          appNo,
+          sessionId: pollingSessionId,
+          token: tokenRef.current
+        });
+        if (!disposed && (result == null ? void 0 : result.code) === import_constants.API_SUCCESS_CODE && Array.isArray(data)) {
+          syncSessionEvents(data, pollingSessionId);
+        }
+      } catch {
+      } finally {
+        requesting = false;
+      }
+    };
+    void poll();
+    const timer = window.setInterval(poll, sessionEventsPollingInterval);
+    return () => {
+      disposed = true;
+      window.clearInterval(timer);
+    };
+  }, [
+    appNo,
+    currentSessionId,
+    enabled,
+    initialized,
+    pollSessionEvents,
+    sessionEventsPollingInterval,
+    syncSessionEvents,
+    url
+  ]);
   (0, import_react.useEffect)(() => {
     return () => {
-      var _a;
-      (_a = ctrl.current) == null ? void 0 : _a.abort();
+      var _a2;
+      cancelSessionEventsFlush();
+      (_a2 = ctrl.current) == null ? void 0 : _a2.abort();
       backgroundRef.current.forEach((bg) => {
-        var _a2;
-        return (_a2 = bg.ctrl) == null ? void 0 : _a2.abort();
+        var _a3;
+        return (_a3 = bg.ctrl) == null ? void 0 : _a3.abort();
       });
       backgroundRef.current.clear();
     };
-  }, []);
+  }, [cancelSessionEventsFlush]);
   return {
     appInfo,
     startChat,