@iblai/web-utils 1.7.1 → 1.7.3

package/dist/data-layer/src/index.d.ts

@@ -1,6 +1,7 @@
 export * from './features/api-keys/api-slice';
 export * from './features/chat-history/api-slice';
 export * from './features/chat/api-slice';
+export * from './features/chat/custom-api-slice';
 export * from './features/chat-files/api-slice';
 export * from './features/chat-files/types';
 export * from './features/llms/api-slice';

package/dist/index.d.ts

@@ -511,6 +511,12 @@ interface ArtifactVersion {
     created_by?: string;
     change_summary?: string;
 }
+interface RawMessageFromWS {
+    id: string;
+    is_ai: boolean;
+    is_human: boolean;
+    generation_id: string;
+}
 interface Message$1 {
     id: string;
     role: "user" | "assistant" | "system";
@@ -2342,4 +2348,4 @@ declare const WithPermissions: ({ rbacResource, rbacPermissions, children, }: Wi
 declare const checkRbacPermission: (rbacPermissions: object, rbacResource: string, enableRBAC?: boolean) => boolean;
 
 export { ALPHANUMERIC_32_REGEX, ANONYMOUS_USERNAME, AuthContext, AuthContextProvider, AuthProvider, CHAT_AREA_SIZE, CacheKeys, DEFAULT_DISCLAIMER_CONTENT, LOCAL_STORAGE_KEYS, MAX_INITIAL_WEBSOCKET_CONNECTION_ATTEMPTS, MENTOR_CHAT_DOCUMENTS_EXTENSIONS, MENTOR_VISIBILITY, MENTOR_VISIBILITY_VALUES, METADATAS, MentorProvider, REQUIRED_ACTIONS_FOR_GROUPS, RemoteEvents, STREAMING_CONTENT_BUFFER_THRESHOLD, STREAMING_CONTENT_FLUSH_INTERVAL, SUBSCRIPTION_MESSAGES, SUBSCRIPTION_PACKAGES, SUBSCRIPTION_PACKAGES_V2, SUBSCRIPTION_TRIGGERS, SUBSCRIPTION_V2_TRIGGERS, ServiceWorkerProvider, SubscriptionFlow, SubscriptionFlowV2, TOOLS, TenantContext, TenantContextProvider, TenantProvider, TimeTracker, WithFormPermissions, WithPermissions, addFiles, addMessage, addProtocolToUrl, advancedTabs, advancedTabsProperties, appleRestrictionReducer, appleRestrictionSlice, chatActions, chatInputSlice, chatInputSliceActions, chatInputSliceReducer, chatInputSliceSelectors, chatSliceReducerShared, checkModelAvailable, checkOllamaHealth, checkRbacPermission, clearAllCaches, clearApiCache, clearAuthCookies, clearCookies, clearCurrentTenantCookie, clearFiles, clearMessages, combineCSVData, convertToOllamaMessages, createFileReference, createMultipleFileReferences, csvDataToText, defaultSessionIds, deleteCookie, deleteCookieOnAllDomains, enableChatActionsPopup, eventBus, fetchWithCache, filesReducer, filesSlice, formatRelativeTime, getAuthSpaJoinUrl, getCacheStatus, getCachedApiResponse, getCookieValue, getDomainParts, getFileInfo, getInitials, getLocalLLMSystemPrompt, getNextNavigation, getParentDomain, getPlatform, getPlatformKey, getServiceWorkerStatus, getStoredUserName, getTimeAgo, getUserEmail, getUserName, handleLogout, hostChatReducer, hostChatSlice, initServiceWorker, isAlphaNumeric32, isExpo, isFileAccepted, isInIframe, isJSON, isLoggedIn, isNode, isReactNative, isSafariBrowser, isServiceWorkerSupported, isStripeActivated, isTauri, isTauriOfflineMode, isWeb, loadMetadataConfig, markdownToPlainText, monetizationSlice, onStatusChange, onUpdate, parseCSV, preCacheMentorData, rbacReducer, redirectToAuthSpa, redirectToAuthSpaJoinTenant, registerServiceWorker, removeFile, requestPresignedUrl, safeRequire, selectActiveChatMessages, selectActiveTab, selectArtifactsEnabled, selectAttachedFiles, selectChats, selectCurrentStreamingArtifact, selectCurrentStreamingMessage, selectDocumentFilter, selectEnableChatActionsPopup, selectIframeContext, selectIsError, selectIsPending, selectIsStopped, selectIsTyping, selectLastArtifactContentFlushTime, selectMetadata, selectNumberOfActiveChatMessages, selectRbacPermissions, selectSessionId, selectSessionIds, selectShouldStartNewChat, selectShowingSharedChat, selectStatus, selectStreaming, selectStreamingArtifactContentBuffer, selectStreamingArtifactFullContent, selectToken, selectTokenEnabled, selectTools, sendMessageToParentWebsite, setAccessCheckResponse, setAdvancedDisplayMonetizationCheckoutModal, setCachedApiResponse, setCookieForAuth, setDisplayMonetizationCheckoutModal, setError402Detected, setFreeTrialUsageOptions, setOfflineStatus, setOpenAppleRestrictionModal, setOpenPricingModal, setPricingModalData, setSubscriptionStatus, setTauriMode, setTopBannerOptions, setupNetworkListeners, showMonetizationCheckoutModal, skipWaiting, streamOllamaChat, subscriptionReducer, subscriptionSlice, syncAuthToCookies, tenantKeySchema, tenantSchema, topBannerReducer, topBannerSlice, translatePrompt, unregisterServiceWorker, updateFileMetadata, updateFileProgress, updateFileRetryCount, updateFileStatus, updateFileUrl, updateFileUrlFromWebSocket, updateRbacPermissions, uploadToS3, use402ErrorCheck, useAccessingPublicRoute, useAdvancedChat, useAuthContext, useAuthProvider, useAxdToken, useCachedSessionId, useChat, useChatFileUpload, useCurrentTenant, useDayJs, useDmToken, useEmbedMode, useEventCallback, useEventListener, useExternalPricingPlan, useFileDragDrop, useIsAdmin, useIsomorphicLayoutEffect, useLocalStorage, useMentorSettings, useMentorTools, useModelFileUploadCapabilities, useOS, useProfileImageUpload, useResponsive, useServiceWorker, useShowAttachment, useShowFreeTrialDialog, useShowVoiceCall, useShowVoiceRecorder, useStripeUpgrade, useSubscriptionHandler, useSubscriptionHandlerV2, useTenantContext, useTenantMetadata, useTimeTracker, useTimeTrackerNative, useTimer, useUserAgreement, useUserData, useUserProfileUpdate, useUserTenants, useUsername, useVisitingTenant, useVoiceChat, useWelcome as useWelcomeMessage, userDataSchema, validateFile };
-export type { AdvancedTab, AppleRestrictionState, ArtifactData, ArtifactPayload, ArtifactVersion, AttachedFile, AuthContextType, ChatInputState, ChatMode, ChatSliceState, ChatState, ChatStatus, CreateStripeCustomerPortalRequest, Error402MessageData, FileAttachment, FileInfo, FileProcessingEvent, FileReference, FileUploadCapabilities, FileUploadState, FilesState, HandleLogoutOptions, HostChatState, Message$1 as Message, MessageAction, MonetizationState, OAuthRequiredData, OAuthResolvedData, OllamaChatRequest, OllamaChatStreamResponse, OllamaMessage, OllamaStreamCallbacks, PricingModalData$1 as PricingModalData, PricingTablePayload, Prompt, RbacPermissions, RbacState, RedirectToAuthSpaOptions, SendMessageOptions, ServiceWorkerStatus, SessionIds, StreamingArtifact, StreamingMessage, StripeUpgradePlan, SubscriptionFlowConfig, SubscriptionFlowConfigV2, SubscriptionState, Tenant, TenantContextType, TenantKeyMentorIdParams, TenantMetadata, TimeTrackerConfig, TimeTrackerState, TopBannerOptions, TopBannerState, TopTrialBannerProps, UploadProgressCallback, UseChatProps, UseChatReturn, UseExternalPricingProps, UseProfileImageUploadOptions, UseProfileImageUploadReturn, UseTimeTrackerConfig, UseTimeTrackerNativeConfig, UseTimeTrackerNativeReturn, UseTimeTrackerReturn, UseUserProfileUpdateResult, UserProfileUpdateData };
+export type { AdvancedTab, AppleRestrictionState, ArtifactData, ArtifactPayload, ArtifactVersion, AttachedFile, AuthContextType, ChatInputState, ChatMode, ChatSliceState, ChatState, ChatStatus, CreateStripeCustomerPortalRequest, Error402MessageData, FileAttachment, FileInfo, FileProcessingEvent, FileReference, FileUploadCapabilities, FileUploadState, FilesState, HandleLogoutOptions, HostChatState, Message$1 as Message, MessageAction, MonetizationState, OAuthRequiredData, OAuthResolvedData, OllamaChatRequest, OllamaChatStreamResponse, OllamaMessage, OllamaStreamCallbacks, PricingModalData$1 as PricingModalData, PricingTablePayload, Prompt, RawMessageFromWS, RbacPermissions, RbacState, RedirectToAuthSpaOptions, SendMessageOptions, ServiceWorkerStatus, SessionIds, StreamingArtifact, StreamingMessage, StripeUpgradePlan, SubscriptionFlowConfig, SubscriptionFlowConfigV2, SubscriptionState, Tenant, TenantContextType, TenantKeyMentorIdParams, TenantMetadata, TimeTrackerConfig, TimeTrackerState, TopBannerOptions, TopBannerState, TopTrialBannerProps, UploadProgressCallback, UseChatProps, UseChatReturn, UseExternalPricingProps, UseProfileImageUploadOptions, UseProfileImageUploadReturn, UseTimeTrackerConfig, UseTimeTrackerNativeConfig, UseTimeTrackerNativeReturn, UseTimeTrackerReturn, UseUserProfileUpdateResult, UserProfileUpdateData };

package/dist/index.esm.js

@@ -9864,9 +9864,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([
                 {
@@ -9891,18 +9890,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);
@@ -11099,6 +11086,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;
         },
@@ -12368,7 +12379,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;
@@ -12941,8 +12952,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) {
@@ -12958,28 +12984,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));
@@ -13034,7 +13060,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);
@@ -13703,7 +13729,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