@iblai/web-utils 1.7.1 → 1.7.3

package/dist/index.js

@@ -9884,9 +9884,8 @@ function TenantProvider({ children, fallback, onAuthSuccess, onAuthFailure, curr
         }
     }
     async function handleLoadTenantMetadata(tenantKey) {
-        var _a, _b;
+        var _a;
         setIsLoading(true);
-        const MAIN_TENANT_KEY = "main";
         try {
             const data = await fetchTenantMetadata([
                 {
@@ -9911,18 +9910,6 @@ function TenantProvider({ children, fallback, onAuthSuccess, onAuthFailure, curr
             const { data: tenants } = await fetchUserTenants();
             const enhancedTenants = await enhanceTenants(tenants, data);
             saveUserTenants(enhancedTenants);
-            // If we're in an advertising tenant and user has other tenants,
-            // switch to a non-advertising, non-main tenant
-            const isCurrentTenantAdvertising = ((_b = data.metadata) === null || _b === void 0 ? void 0 : _b.is_advertising) || isAdvertisingTenant;
-            if (isCurrentTenantAdvertising &&
-                enhancedTenants &&
-                enhancedTenants.length > 0) {
-                const nonAdvertisingTenant = enhancedTenants.find((t) => !t.is_advertising && t.key !== MAIN_TENANT_KEY);
-                if (nonAdvertisingTenant) {
-                    handleTenantSwitch(nonAdvertisingTenant.key, false, false);
-                    return;
-                }
-            }
             if (currentTenant && currentTenant !== tenantKey) {
                 setTenantKey(data.platform_key);
                 const userAlreadyInTenant = enhancedTenants === null || enhancedTenants === void 0 ? void 0 : enhancedTenants.find((t) => t.key === tenantKey);
@@ -11119,6 +11106,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;
         },
@@ -12388,7 +12399,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;
@@ -12961,8 +12972,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) {
@@ -12978,28 +13004,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));
@@ -13054,7 +13080,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);
@@ -13723,7 +13749,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