@iblai/web-utils 1.7.2 → 1.8.0

package/dist/index.js

@@ -1181,17 +1181,19 @@ async function redirectToAuthSpa(options) {
         logoutTimestamp: "ibl_logout_timestamp",
         loginTimestamp: "ibl_login_timestamp",
         tenantSwitching: "ibl_tenant_switching",
-    }, hasNonExpiredAuthToken, isOffline, preserveTokenKey, authRedirectProxy, isNativeApp, } = options;
-    console.log("[redirectToAuthSpa] starting redirect to auth spa", redirectTo, platformKey, logout, saveRedirect);
+    }, hasNonExpiredAuthToken, isOffline, preserveTokenKey, authRedirectProxy, isNativeApp, forceRedirect = false, } = options;
+    console.log("[redirectToAuthSpa] starting redirect to auth spa", redirectTo, platformKey, logout, saveRedirect, forceRedirect);
     // Skip if a tenant switch is already in progress
-    if (cookieNames.tenantSwitching &&
+    if (!forceRedirect &&
+        cookieNames.tenantSwitching &&
         document.cookie.includes(cookieNames.tenantSwitching)) {
         console.log("[redirectToAuthSpa] Tenant switch in progress, skipping redirect");
         return;
     }
     // Skip if a login occurred after the last logout (login takes precedence)
     // but only if this app actually has a valid auth token
-    if (hasNonExpiredAuthToken &&
+    if (!forceRedirect &&
+        hasNonExpiredAuthToken &&
         cookieNames.loginTimestamp &&
         cookieNames.logoutTimestamp) {
         const loginTs = getCookieValue(cookieNames.loginTimestamp);
@@ -1203,7 +1205,8 @@ async function redirectToAuthSpa(options) {
             hasValidToken,
             loginAhead: loginTs && logoutTs ? Number(loginTs) > Number(logoutTs) : false,
         });
-        if (hasValidToken &&
+        if (!forceRedirect &&
+            hasValidToken &&
             loginTs &&
             logoutTs &&
             Number(loginTs) > Number(logoutTs)) {
@@ -11106,6 +11109,30 @@ const chatSlice = createSlice({
         resetCurrentStreamingMessage: (state) => {
             state.currentStreamingMessage = { id: "", content: "" };
         },
+        updateStreamingMessageId: (state, action) => {
+            const oldId = state.currentStreamingMessage.id;
+            const { newId } = action.payload;
+            if (!newId || oldId === newId) {
+                return;
+            }
+            if (oldId) {
+                const activeMessages = state.chats[state.activeTab];
+                if (activeMessages === null || activeMessages === void 0 ? void 0 : activeMessages.length) {
+                    const lastIndex = activeMessages.length - 1;
+                    const last = activeMessages[lastIndex];
+                    if (last.id === oldId) {
+                        state.chats = {
+                            ...state.chats,
+                            [state.activeTab]: [
+                                ...activeMessages.slice(0, lastIndex),
+                                { ...last, id: newId },
+                            ],
+                        };
+                    }
+                }
+            }
+            state.currentStreamingMessage.id = newId;
+        },
         setActiveTab: (state, action) => {
             state.activeTab = action.payload;
         },
@@ -12375,7 +12402,7 @@ const useChat = ({ wsUrl, wsToken, flowConfig, sessionId, stopGenerationWsUrl, e
                 onStreamingChange === null || onStreamingChange === void 0 ? void 0 : onStreamingChange(false);
             });
             socket.addEventListener("message", (event) => {
-                var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _0, _1;
+                var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _0, _1, _2, _3, _4;
                 if (isWebSocketPaused.current) {
                     console.log("[ws-message] Message received but WebSocket is paused");
                     return;
@@ -12948,8 +12975,23 @@ const useChat = ({ wsUrl, wsToken, flowConfig, sessionId, stopGenerationWsUrl, e
                         onStreamingChange(true);
                         return;
                     }
+                    //HANDLE RECEPTION OF CHAT MESSAGES INFO
+                    if ((messageData === null || messageData === void 0 ? void 0 : messageData.type) === "chat_messages" &&
+                        ((_t = messageData === null || messageData === void 0 ? void 0 : messageData.messages) === null || _t === void 0 ? void 0 : _t.length) > 0) {
+                        const chatMessageID = (_v = (_u = messageData === null || messageData === void 0 ? void 0 : messageData.messages) === null || _u === void 0 ? void 0 : _u.find((pre) => pre.is_ai)) === null || _v === void 0 ? void 0 : _v.id;
+                        if (chatMessageID) {
+                            dispatch(chatActions.updateStreamingMessageId({ newId: chatMessageID }));
+                            currentStreamingMessage.current = {
+                                ...currentStreamingMessage.current,
+                                id: chatMessageID,
+                            };
+                            onStreamingMessageUpdate === null || onStreamingMessageUpdate === void 0 ? void 0 : onStreamingMessageUpdate(currentStreamingMessage.current);
+                            return;
+                        }
+                    }
                     // Handle streaming message content (chat text)
-                    if (messageData.data !== undefined || messageData.eos !== undefined) {
+                    if ((messageData === null || messageData === void 0 ? void 0 : messageData.type) !== "chat_messages" &&
+                        (messageData.data !== undefined || messageData.eos !== undefined)) {
                         const messageText = messageData.data || "";
                         // If we have content to add, update the in-memory streaming message and Redux
                         if (messageText && currentStreamingMessage.current) {
@@ -12965,28 +13007,28 @@ const useChat = ({ wsUrl, wsToken, flowConfig, sessionId, stopGenerationWsUrl, e
                             };
                             onStreamingMessageUpdate === null || onStreamingMessageUpdate === void 0 ? void 0 : onStreamingMessageUpdate(currentStreamingMessage.current);
                             // Update or add message in Redux store
-                            if ((_t = currentStreamingMessage.current) === null || _t === void 0 ? void 0 : _t.id) {
+                            if ((_w = currentStreamingMessage.current) === null || _w === void 0 ? void 0 : _w.id) {
                                 // TODO: Investigate why `appendMessageToActiveTab` is consuming this action payload
                                 dispatch(chatActions.appendMessageToActiveTab(undefined));
                             }
                         }
-                        // Handle end of stream
                         if (messageData.eos) {
+                            // Handle end of stream
                             const hasPendingArtifact = !!pendingArtifactVersion.current;
-                            const hasMessageContent = !!((_u = currentStreamingMessage.current) === null || _u === void 0 ? void 0 : _u.content) &&
+                            const hasMessageContent = !!((_x = currentStreamingMessage.current) === null || _x === void 0 ? void 0 : _x.content) &&
                                 currentStreamingMessage.current.content.length > 0;
                             console.log("[ws-message] End of stream (EOS) received:", {
-                                generationId: (_v = currentStreamingMessage.current) === null || _v === void 0 ? void 0 : _v.id,
-                                finalContentLength: (_x = (_w = currentStreamingMessage.current) === null || _w === void 0 ? void 0 : _w.content) === null || _x === void 0 ? void 0 : _x.length,
-                                hasContent: !!((_y = currentStreamingMessage.current) === null || _y === void 0 ? void 0 : _y.content),
+                                generationId: (_y = currentStreamingMessage.current) === null || _y === void 0 ? void 0 : _y.id,
+                                finalContentLength: (_0 = (_z = currentStreamingMessage.current) === null || _z === void 0 ? void 0 : _z.content) === null || _0 === void 0 ? void 0 : _0.length,
+                                hasContent: !!((_1 = currentStreamingMessage.current) === null || _1 === void 0 ? void 0 : _1.content),
                                 hasPendingArtifact,
                             });
                             // Final update to Redux store with complete message
-                            if (((_z = currentStreamingMessage.current) === null || _z === void 0 ? void 0 : _z.id) &&
+                            if (((_2 = currentStreamingMessage.current) === null || _2 === void 0 ? void 0 : _2.id) &&
                                 (hasMessageContent || hasPendingArtifact)) {
                                 console.log("[ws-message] Is currentStreamingMessage.current", {
-                                    id: (_0 = currentStreamingMessage.current) === null || _0 === void 0 ? void 0 : _0.id,
-                                    content: (_1 = currentStreamingMessage.current) === null || _1 === void 0 ? void 0 : _1.content,
+                                    id: (_3 = currentStreamingMessage.current) === null || _3 === void 0 ? void 0 : _3.id,
+                                    content: (_4 = currentStreamingMessage.current) === null || _4 === void 0 ? void 0 : _4.content,
                                 });
                                 // TODO: Investigate why `appendMessageToActiveTab` is consuming this action payload
                                 dispatch(chatActions.appendMessageToActiveTab(undefined));
@@ -13041,7 +13083,7 @@ const useChat = ({ wsUrl, wsToken, flowConfig, sessionId, stopGenerationWsUrl, e
                     onStatusChange("error");
                 }
             });
-            socket.addEventListener("error", (error) => {
+            socket.addEventListener("error", () => {
                 // Don't show error if we're offline in Tauri - this is expected
                 // Check both the ref AND navigator.onLine (the ref may not be updated yet)
                 const isLikelyOffline = isOfflineRef.current || (isTauriApp() && !navigator.onLine);
@@ -13710,7 +13752,7 @@ function useAdvancedChat({ tenantKey, mentorId, username = ANONYMOUS_USERNAME, t
                             ...result,
                             role: result.type === "human" ? "user" : "assistant",
                             visible: true,
-                            id: new Date().getTime(),
+                            id: result.id || new Date().getTime(),
                             content: typeof result.content === "object" &&
                                 result.content.length > 0
                                 ? (_c = result.content.find((item) => item.text)) === null || _c === void 0 ? void 0 : _c.text