@gooddata/sdk-ui-gen-ai 11.57.0 → 11.58.0-alpha.1

package/esm/store/messages/messagesSlice.js

@@ -2,8 +2,7 @@
 import { createSlice } from "@reduxjs/toolkit";
 import { selectDefaultAgentId } from "../../components/utils/agentSelection.js";
 import { DEFAULT_EFFORT, deriveConversationEffort } from "../../components/utils/effortSelection.js";
-import { isAssistantMessage, isChatConversationLocalItem, isUserMessage, isVisualizationContents, makeAgentChangeItem, makeErrorContent, makeErrorContents, } from "../../model.js";
-import { convertMessageToChatConversation } from "../sideEffects/utils.js";
+import { makeAgentChangeItem, makeErrorContent, } from "../../model.js";
 import { createEmptyConversation, getConversationData, getConversationLocalId, isConversationWithLocalId, } from "../utils.js";
 export const LS_VERBOSE_KEY = "gd-gen-ai-verbose";
 export const messagesSliceName = "messages";
@@ -21,10 +20,6 @@ const initialState = {
     // Start with loading state to avoid re-render from empty state on startup
     loaded: false,
     verbose: getInitialVerboseState(),
-    //old messages
-    messageAsyncProcess: "loading",
-    messageOrder: [],
-    messages: {},
     //conversations
     conversations: undefined,
     currentConversation: undefined,
@@ -35,14 +30,6 @@ const initialState = {
     conversationsData: {},
     refocus: 0,
 };
-const setNormalizedMessages = (state, messages) => {
-    state.messages = messages.reduce((acc, message) => {
-        acc[message.localId] = message;
-        return acc;
-    }, {});
-    state.messageOrder = messages.map((message) => message.localId);
-    state.loaded = true;
-};
 const setNormalizedConversations = (state, conversations) => {
     const originalConversations = state.conversations ?? [];
     // Iterate new conversations and try to find existing conversation by id
@@ -102,131 +89,64 @@ const getResponseTrace = (state, conversationId, responseId) => {
     return (trace[responseId] ??= { steps: [], detailsByStepId: {}, responseDetails: [] });
 };
 const getAssistantMessageStrict = (state, assistantMessageId, conversationId) => {
-    if (state.currentConversation) {
-        const data = getConversationData(state.conversationsData, conversationId);
-        if (!data) {
-            throw new Error(`Unexpected error during message evaluation.`);
-        }
-        const message = data.items[assistantMessageId];
-        if (message.role !== "assistant") {
-            throw new Error(`Unexpected error during message evaluation.`);
-        }
-        return message;
+    const data = getConversationData(state.conversationsData, conversationId);
+    if (!data) {
+        throw new Error(`Unexpected error during message evaluation.`);
     }
-    else {
-        const message = state.messages[assistantMessageId];
-        if (!isAssistantMessage(message)) {
-            throw new Error(`Unexpected error during message evaluation.`);
-        }
-        return message;
+    const message = data.items[assistantMessageId];
+    if (message.role !== "assistant") {
+        throw new Error(`Unexpected error during message evaluation.`);
     }
+    return message;
 };
 const getUserMessageStrict = (state, assistantMessageId, conversationId) => {
-    if (state.currentConversation) {
-        const data = getConversationData(state.conversationsData, conversationId);
-        if (!data) {
-            throw new Error(`Unexpected error during message evaluation.`);
-        }
-        const message = data.items[assistantMessageId];
-        if (message.role !== "user") {
-            throw new Error(`Unexpected error during message evaluation.`);
-        }
-        return message;
+    const data = getConversationData(state.conversationsData, conversationId);
+    if (!data) {
+        throw new Error(`Unexpected error during message evaluation.`);
     }
-    else {
-        const message = state.messages[assistantMessageId];
-        if (!isUserMessage(message)) {
-            throw new Error(`Unexpected error during message evaluation.`);
-        }
-        return message;
+    const message = data.items[assistantMessageId];
+    if (message.role !== "user") {
+        throw new Error(`Unexpected error during message evaluation.`);
     }
+    return message;
 };
 const getMessageExists = (state, assistantMessageId, conversationId) => {
-    if (state.currentConversation) {
-        if (!conversationId) {
-            return false;
-        }
-        const data = getConversationData(state.conversationsData, conversationId);
-        return !!data?.items[assistantMessageId];
+    if (!conversationId) {
+        return false;
     }
-    return !!state.messages[assistantMessageId];
+    const data = getConversationData(state.conversationsData, conversationId);
+    return !!data?.items[assistantMessageId];
 };
 /**
  * Get the user message before the assistant message, or undefined if there isn't one.
  * This is useful for dynamically created assistant messages that don't have a preceding user message.
  */
 const getUserMessageBeforeSafe = (state, assistantMessageId, conversationId) => {
-    if (state.currentConversation) {
-        const data = getConversationData(state.conversationsData, conversationId);
-        if (!data) {
-            return undefined;
-        }
-        const messageIndex = data.order.indexOf(assistantMessageId);
-        if (messageIndex <= 0) {
-            return undefined;
-        }
-        const message = data.items[data.order[messageIndex - 1]];
-        return message.role === "user" ? message : undefined;
+    const data = getConversationData(state.conversationsData, conversationId);
+    if (!data) {
+        return undefined;
     }
-    const messageIndex = state.messageOrder.indexOf(assistantMessageId);
+    const messageIndex = data.order.indexOf(assistantMessageId);
     if (messageIndex <= 0) {
         return undefined;
     }
-    const message = state.messages[state.messageOrder[messageIndex - 1]];
-    return isUserMessage(message) ? message : undefined;
+    const message = data.items[data.order[messageIndex - 1]];
+    return message.role === "user" ? message : undefined;
 };
 const messagesSlice = createSlice({
     name: messagesSliceName,
     initialState,
     reducers: {
         loadThreadAction: (state) => {
-            if (state.currentConversation) {
-                const data = getConversationData(state.conversationsData, state.currentConversation.localId);
-                if (data) {
-                    data.asyncProcess = "loading";
-                }
-            }
-            else {
-                state.messageAsyncProcess = "loading";
+            const data = getConversationData(state.conversationsData, state.currentConversation?.localId);
+            if (data) {
+                data.asyncProcess = "loading";
             }
         },
         loadThreadErrorAction: (state, { payload: { error } }) => {
             state.globalError = errorToObject(error);
-            if (state.currentConversation) {
-                const data = getConversationData(state.conversationsData, state.currentConversation.localId);
-                delete data?.asyncProcess;
-            }
-            else {
-                delete state.messageAsyncProcess;
-            }
-        },
-        loadThreadSuccessAction: (state, { payload: { messages, threadId } }) => {
-            setNormalizedMessages(state, messages);
-            state.threadId = threadId;
-            if (state.currentConversation) {
-                const data = getConversationData(state.conversationsData, state.currentConversation.localId);
-                delete data?.asyncProcess;
-            }
-            else {
-                delete state.messageAsyncProcess;
-                state.conversationsLoaded = true;
-            }
-        },
-        /**
-         * Restore previously cached messages immediately while the backend is still loading.
-         * Unlike loadThreadSuccessAction, this does not mark the thread as fully loaded,
-         * so the backend fetch continues and will replace these messages when complete.
-         * Sets asyncProcess to "restoring" so that:
-         * - The skeleton spinner is hidden and cached messages are rendered (Messages checks for "loading"/"clearing" only).
-         * - The input remains disabled (Input checks !!asyncProcess, and "restoring" is truthy).
-         */
-        restoreCachedMessagesAction: (state, { payload: { messages } }) => {
-            state.messages = messages.reduce((acc, message) => {
-                acc[message.localId] = message;
-                return acc;
-            }, {});
-            state.messageOrder = messages.map((message) => message.localId);
-            state.messageAsyncProcess = "restoring";
+            const data = getConversationData(state.conversationsData, state.currentConversation?.localId);
+            delete data?.asyncProcess;
         },
         loadConversationsSuccessAction: (state, { payload: { conversations }, }) => {
             setNormalizedConversations(state, conversations);
@@ -236,50 +156,28 @@ const messagesSlice = createSlice({
             state.selectedAgentId = currentConversation.agentId ?? state.selectedAgentId;
             seedConversationEffort(state, currentConversation.localId);
             state.threadId = threadId;
-            if (state.currentConversation) {
-                const data = getConversationData(state.conversationsData, state.currentConversation.localId);
-                delete data?.asyncProcess;
-            }
-            else {
-                delete state.messageAsyncProcess;
-            }
+            const data = getConversationData(state.conversationsData, state.currentConversation?.localId);
+            delete data?.asyncProcess;
         },
         clearThreadAction: (state) => {
             // Don't clobber an in-flight reply's "evaluating" marker: in multi-conversation mode this
             // "clear" navigates to a fresh draft and leaves the outgoing conversation intact, so it must
             // stay busy until its reply resolves (LX-2644).
-            if (state.currentConversation) {
-                const data = getConversationData(state.conversationsData, state.currentConversation.localId);
-                if (data && data.asyncProcess !== "evaluating") {
-                    data.asyncProcess = "clearing";
-                }
-            }
-            else if (state.messageAsyncProcess !== "evaluating") {
-                state.messageAsyncProcess = "clearing";
+            const data = getConversationData(state.conversationsData, state.currentConversation?.localId);
+            if (data && data.asyncProcess !== "evaluating") {
+                data.asyncProcess = "clearing";
             }
         },
         clearThreadErrorAction: (state, { payload: { error } }) => {
             state.globalError = errorToObject(error);
-            if (state.currentConversation) {
-                const data = getConversationData(state.conversationsData, state.currentConversation.localId);
-                delete data?.asyncProcess;
-            }
-            else {
-                delete state.messageAsyncProcess;
-            }
+            const data = getConversationData(state.conversationsData, state.currentConversation?.localId);
+            delete data?.asyncProcess;
         },
         clearThreadSuccessAction: (state) => {
-            state.messages = {};
-            state.messageOrder = [];
             state.loaded = false;
             delete state.globalError;
-            if (state.currentConversation) {
-                const data = getConversationData(state.conversationsData, state.currentConversation.localId);
-                delete data?.asyncProcess;
-            }
-            else {
-                delete state.messageAsyncProcess;
-            }
+            const data = getConversationData(state.conversationsData, state.currentConversation?.localId);
+            delete data?.asyncProcess;
         },
         clearConversationSuccessAction: (state, { payload: { conversation, threadId }, }) => {
             state.conversations = [conversation, ...(state.conversations ?? [])];
@@ -290,108 +188,60 @@ const messagesSlice = createSlice({
             const data = getConversationData(state.conversationsData, state.currentConversation.localId);
             delete data?.asyncProcess;
         },
-        /**
-         * Add message to the stack
-         */
         newMessageAction: (state, action) => {
-            let message = action.payload;
-            if (state.currentConversation && !isChatConversationLocalItem(message)) {
-                message = convertMessageToChatConversation(message);
+            const message = action.payload;
+            // keep list of conversations, but clear current conversation data in the view
+            if (!state.currentConversation) {
+                state.currentConversation = createEmptyConversation();
             }
-            if (isChatConversationLocalItem(message)) {
-                // keep list of conversations, but clear current conversation data in the view
-                if (!state.currentConversation) {
-                    state.currentConversation = createEmptyConversation();
+            const currentConversation = state.currentConversation;
+            currentConversation.updatedAt = new Date().toISOString();
+            state.conversations = state.conversations?.map((conversation) => {
+                if (conversation.localId === currentConversation.localId) {
+                    return {
+                        ...conversation,
+                        updatedAt: currentConversation.updatedAt,
+                    };
                 }
-                const currentConversation = state.currentConversation;
-                currentConversation.updatedAt = new Date().toISOString();
-                state.conversations = state.conversations?.map((conversation) => {
-                    if (conversation.localId === currentConversation.localId) {
-                        return {
-                            ...conversation,
-                            updatedAt: currentConversation.updatedAt,
-                        };
-                    }
-                    return conversation;
+                return conversation;
+            });
+            const data = getConversationData(state.conversationsData, currentConversation.localId);
+            if (data) {
+                // We need to mark all messages as filled
+                Object.values(data.items).forEach((item) => {
+                    item.filled = true;
                 });
-                const data = getConversationData(state.conversationsData, currentConversation.localId);
-                if (data) {
-                    // We need to mark all messages as filled
-                    Object.values(data.items).forEach((item) => {
-                        item.filled = true;
-                    });
-                    data.items[message.localId] = message;
-                    data.order.push(message.localId);
-                }
-            }
-            else {
-                if (state.currentConversation) {
-                    throw new Error("Working with thread message but conversation mode is active.");
-                }
-                state.messages[message.localId] = message;
-                state.messageOrder.push(action.payload.localId);
+                data.items[message.localId] = message;
+                data.order.push(message.localId);
             }
             state.loaded = true;
         },
-        /**
-         * Start the message evaluation, adding new assistant message as an incomplete placeholder
-         */
         evaluateMessageAction: (state, { payload: { message, conversationId }, }) => {
-            if (isChatConversationLocalItem(message)) {
-                if (!conversationId) {
-                    throw new Error("Working with conversation message but thread mode is active.");
-                }
-                const data = getConversationData(state.conversationsData, conversationId);
-                if (data) {
-                    data.items[message.localId] = message;
-                    data.order.push(message.localId);
-                }
-                setConversationInProgress(state, true, conversationId);
-            }
-            else {
-                if (state.currentConversation) {
-                    throw new Error("Working with thread message but conversation mode is active.");
-                }
-                state.messages[message.localId] = message;
-                state.messageOrder.push(message.localId);
+            if (!conversationId) {
+                throw new Error("Working with conversation message but thread mode is active.");
             }
-            if (state.currentConversation) {
-                const data = getConversationData(state.conversationsData, conversationId);
-                if (data) {
-                    data.asyncProcess = "evaluating";
-                }
-            }
-            else {
-                state.messageAsyncProcess = "evaluating";
+            const data = getConversationData(state.conversationsData, conversationId);
+            if (data) {
+                data.items[message.localId] = message;
+                data.order.push(message.localId);
+                data.asyncProcess = "evaluating";
             }
+            setConversationInProgress(state, true, conversationId);
         },
-        /**
-         * The evaluation failed, need to update the assistant message.
-         */
         evaluateMessageErrorAction: (state, { payload, }) => {
             // Always clear the async process first to avoid stranding the UI
             const data = getConversationData(state.conversationsData, payload.conversationId);
             if (data) {
                 delete data.asyncProcess;
             }
-            delete state.messageAsyncProcess;
             if (!getMessageExists(state, payload.assistantMessageId, payload.conversationId)) {
                 return;
             }
             const assistantMessage = getAssistantMessageStrict(state, payload.assistantMessageId, payload.conversationId);
-            if (isChatConversationLocalItem(assistantMessage)) {
-                assistantMessage.complete = true;
-                assistantMessage.streaming = false;
-                assistantMessage.content = makeErrorContent(payload.error);
-            }
-            else {
-                assistantMessage.complete = true;
-                assistantMessage.content.push(makeErrorContents(payload.error));
-            }
+            assistantMessage.complete = true;
+            assistantMessage.streaming = false;
+            assistantMessage.content = makeErrorContent(payload.error);
         },
-        /**
-         * Received new chunk from server over SSE.
-         */
         evaluateMessageStreamingAction: (state, { payload, }) => {
             //NOTE: During streaming a message, user can choose to close or reset a chat
             // and without this check, we would get an unwanted error
@@ -402,20 +252,14 @@ const messagesSlice = createSlice({
             // Update assistant message
             const assistantMessage = getAssistantMessageStrict(state, payload.assistantMessageId, payload.conversationId);
             assistantMessage.id = payload.interactionId ?? assistantMessage.id;
-            if (isChatConversationLocalItem(assistantMessage)) {
-                assistantMessage.content = payload.content ?? {
-                    type: "text",
-                    text: "",
-                };
-                assistantMessage.streaming = true;
-                assistantMessage.cancelled = false;
-                assistantMessage.responseId = payload.item?.responseId ?? "";
-                assistantMessage.replyTo = payload.item?.replyTo;
-            }
-            else {
-                assistantMessage.content.push(...(payload.contents ?? []));
-                assistantMessage.cancelled = false;
-            }
+            assistantMessage.content = payload.content ?? {
+                type: "text",
+                text: "",
+            };
+            assistantMessage.streaming = true;
+            assistantMessage.cancelled = false;
+            assistantMessage.responseId = payload.item?.responseId ?? "";
+            assistantMessage.replyTo = payload.item?.replyTo;
             // Also update the interaction id in the relevant user message (if one exists)
             // Note: dynamically created assistant messages (for multi-interaction streams)
             // may not have a preceding user message
@@ -447,16 +291,10 @@ const messagesSlice = createSlice({
             // Update assistant message
             const userMessage = getUserMessageStrict(state, payload.userMessageId, payload.conversationId);
             userMessage.id = payload.interactionId ?? userMessage.id;
-            if (isChatConversationLocalItem(userMessage)) {
-                const message = payload.message;
-                userMessage.responseId = message.responseId;
-                userMessage.replyTo = message.replyTo;
-                userMessage.createdAt = message.createdAt;
-            }
-            else {
-                const message = payload.message;
-                userMessage.created = message.created;
-            }
+            const message = payload.message;
+            userMessage.responseId = message.responseId;
+            userMessage.replyTo = message.replyTo;
+            userMessage.createdAt = message.createdAt;
         },
         evaluateMessageCompleteAction: (state, { payload, }) => {
             // Always clear the async process first to avoid stranding the UI
@@ -465,25 +303,19 @@ const messagesSlice = createSlice({
             if (data) {
                 delete data.asyncProcess;
             }
-            delete state.messageAsyncProcess;
             if (!getMessageExists(state, payload.assistantMessageId, payload.conversationId)) {
                 return;
             }
             const assistantMessage = getAssistantMessageStrict(state, payload.assistantMessageId, payload.conversationId);
             assistantMessage.complete = true;
             assistantMessage.cancelled = payload.cancelled ?? assistantMessage.cancelled;
-            if (isChatConversationLocalItem(assistantMessage)) {
-                assistantMessage.streaming = false;
-            }
+            assistantMessage.streaming = false;
         },
-        setMessagesAction: (state, { payload: { messages, items }, }) => {
+        setMessagesAction: (state, { payload: { items } }) => {
             const conversation = state.currentConversation;
             if (conversation) {
                 setNormalizedConversation(state, conversation, items ?? []);
             }
-            else {
-                setNormalizedMessages(state, messages ?? []);
-            }
         },
         setVerboseAction: (state, { payload: { verbose } }) => {
             state.verbose = verbose;
@@ -668,9 +500,6 @@ const messagesSlice = createSlice({
             // orphan the "loading" flag on the conversation that was actually loading -
             // its skeleton then spins forever when it is reopened (LX-2577). Clear the
             // load flag wherever it sits; there is only ever one load in flight.
-            if (state.messageAsyncProcess === "loading" || state.messageAsyncProcess === "restoring") {
-                delete state.messageAsyncProcess;
-            }
             for (const data of Object.values(state.conversationsData)) {
                 if (data.asyncProcess === "loading" || data.asyncProcess === "restoring") {
                     delete data.asyncProcess;
@@ -678,13 +507,8 @@ const messagesSlice = createSlice({
             }
             // Preserve the original behaviour for the current conversation's non-load
             // async states (e.g. "evaluating"/"clearing").
-            if (state.currentConversation) {
-                const data = getConversationData(state.conversationsData, state.currentConversation.localId);
-                delete data?.asyncProcess;
-            }
-            else {
-                delete state.messageAsyncProcess;
-            }
+            const data = getConversationData(state.conversationsData, state.currentConversation?.localId);
+            delete data?.asyncProcess;
         },
         /**
          * Clear the "loading"/"restoring" flag for a specific conversation.
@@ -701,16 +525,11 @@ const messagesSlice = createSlice({
                     delete data.asyncProcess;
                 }
             }
-            else if (state.messageAsyncProcess === "loading" || state.messageAsyncProcess === "restoring") {
-                delete state.messageAsyncProcess;
-            }
         },
         filledFormAction: (state, { payload, }) => {
             const conversationId = getConversationLocalId(state.currentConversation);
             const assistantMessage = getAssistantMessageStrict(state, payload.assistantMessageId, conversationId);
-            if (isChatConversationLocalItem(assistantMessage)) {
-                assistantMessage.filled = true;
-            }
+            assistantMessage.filled = true;
         },
         refocusInput: (state) => {
             state.refocus++;
@@ -718,172 +537,104 @@ const messagesSlice = createSlice({
         setUserFeedback: (state, { payload, }) => {
             const conversationId = getConversationLocalId(state.currentConversation);
             const assistantMessage = getAssistantMessageStrict(state, payload.assistantMessageId, conversationId);
-            if (isChatConversationLocalItem(assistantMessage)) {
-                const original = assistantMessage.feedback ?? {
-                    type: "feedback",
-                    createdAt: new Date().getTime(),
-                    updatedAt: new Date().getTime(),
-                };
-                assistantMessage.feedback = {
-                    ...original,
-                    feedback: payload.feedback,
-                };
-            }
-            else {
-                assistantMessage.feedback = payload.feedback;
-            }
+            const original = assistantMessage.feedback ?? {
+                type: "feedback",
+                createdAt: new Date().getTime(),
+                updatedAt: new Date().getTime(),
+            };
+            assistantMessage.feedback = {
+                ...original,
+                feedback: payload.feedback,
+            };
         },
         setUserFeedbackError: (state, { payload, }) => {
             // Reset feedback to NONE and set error when submission fails
             const assistantMessage = getAssistantMessageStrict(state, payload.assistantMessageId, payload.conversationId);
-            if (isChatConversationLocalItem(assistantMessage)) {
-                const original = assistantMessage.feedback ?? {
-                    type: "feedback",
-                    createdAt: new Date().getTime(),
-                    updatedAt: new Date().getTime(),
-                };
-                assistantMessage.feedback = {
-                    ...original,
-                    feedback: "NONE",
-                    error: payload.error,
-                };
-            }
-            else {
-                assistantMessage.feedback = "NONE";
-                assistantMessage.feedbackError = payload.error;
-            }
-        },
-        clearUserFeedbackError: (state, { payload, }) => {
-            // Clear feedback error after showing toast
-            const assistantMessage = getAssistantMessageStrict(state, payload.assistantMessageId, payload.conversationId);
-            if (isChatConversationLocalItem(assistantMessage)) {
-                delete assistantMessage.feedback?.error;
-            }
-            else {
-                delete assistantMessage.feedbackError;
-            }
+            const original = assistantMessage.feedback ?? {
+                type: "feedback",
+                createdAt: new Date().getTime(),
+                updatedAt: new Date().getTime(),
+            };
+            assistantMessage.feedback = {
+                ...original,
+                feedback: "NONE",
+                error: payload.error,
+            };
         },
         saveVisualizationAction: (state, { payload, }) => {
             const conversationId = getConversationLocalId(state.currentConversation);
             const assistantMessage = getAssistantMessageStrict(state, payload.assistantMessageId, conversationId);
-            if (isChatConversationLocalItem(assistantMessage)) {
-                if (assistantMessage.content.type !== "multipart") {
-                    throw new Error("Unexpected message type");
-                }
-                const visualization = assistantMessage.content.parts
-                    .filter((filter) => filter.type === "visualization")
-                    .find((content) => content.visualization?.insight.identifier === payload.visualizationId);
-                if (visualization?.visualization) {
-                    visualization.visualization.insight.title = payload.visualizationTitle;
-                    visualization.saving = {
-                        started: true,
-                        completed: false,
-                    };
-                }
-            }
-            else {
-                const visualization = assistantMessage.content
-                    .filter(isVisualizationContents)
-                    .flatMap((content) => content.createdVisualizations)
-                    .find((content) => content.id === payload.visualizationId);
-                if (visualization) {
-                    visualization.saving = true;
-                }
+            if (assistantMessage.content.type !== "multipart") {
+                throw new Error("Unexpected message type");
+            }
+            const visualization = assistantMessage.content.parts
+                .filter((filter) => filter.type === "visualization")
+                .find((content) => content.visualization?.insight.identifier === payload.visualizationId);
+            if (visualization?.visualization) {
+                visualization.visualization.insight.title = payload.visualizationTitle;
+                visualization.saving = {
+                    started: true,
+                    completed: false,
+                };
             }
         },
         savedVisualizationAction: (state, { payload, }) => {
             const conversationId = getConversationLocalId(state.currentConversation);
             const assistantMessage = getAssistantMessageStrict(state, payload.assistantMessageId, conversationId);
-            if (isChatConversationLocalItem(assistantMessage)) {
-                if (assistantMessage.content.type !== "multipart") {
-                    throw new Error("Unexpected message type");
-                }
-                const visualization = assistantMessage.content.parts
-                    .filter((filter) => filter.type === "visualization")
-                    .find((content) => content.visualization?.insight.identifier === payload.visualizationId);
-                if (visualization) {
-                    delete visualization.saving;
-                }
+            if (assistantMessage.content.type !== "multipart") {
+                throw new Error("Unexpected message type");
+            }
+            const visualization = assistantMessage.content.parts
+                .filter((filter) => filter.type === "visualization")
+                .find((content) => content.visualization?.insight.identifier === payload.visualizationId);
+            if (visualization) {
+                delete visualization.saving;
             }
         },
         saveVisualizationErrorAction: (state, { payload, }) => {
             const assistantMessage = getAssistantMessageStrict(state, payload.assistantMessageId, payload.conversationId);
-            if (isChatConversationLocalItem(assistantMessage)) {
-                if (assistantMessage.content.type !== "multipart") {
-                    throw new Error("Unexpected message type");
-                }
-                const visualization = assistantMessage.content.parts
-                    .filter((filter) => filter.type === "visualization")
-                    .find((content) => content.visualization?.insight.identifier === payload.visualizationId);
-                if (visualization) {
-                    visualization.saving = {
-                        started: false,
-                        completed: false,
-                    };
-                    visualization.error = payload.error;
-                }
-            }
-            else {
-                const visualization = assistantMessage.content
-                    .filter(isVisualizationContents)
-                    .flatMap((content) => content.createdVisualizations)
-                    .find((content) => content.id === payload.visualizationId);
-                if (visualization) {
-                    visualization.saving = false;
-                }
+            if (assistantMessage.content.type !== "multipart") {
+                throw new Error("Unexpected message type");
+            }
+            const visualization = assistantMessage.content.parts
+                .filter((filter) => filter.type === "visualization")
+                .find((content) => content.visualization?.insight.identifier === payload.visualizationId);
+            if (visualization) {
+                visualization.saving = {
+                    started: false,
+                    completed: false,
+                };
+                visualization.error = payload.error;
             }
         },
         saveVisualizationSuccessAction: (state, { payload, }) => {
             const assistantMessage = getAssistantMessageStrict(state, payload.assistantMessageId, payload.conversationId);
-            if (isChatConversationLocalItem(assistantMessage)) {
-                if (assistantMessage.content.type !== "multipart") {
-                    throw new Error("Unexpected message type");
-                }
-                const visualization = assistantMessage.content.parts
-                    .filter((filter) => filter.type === "visualization")
-                    .find((content) => content.visualization?.insight.identifier === payload.visualizationId);
-                if (visualization?.visualization) {
-                    visualization.saving = {
-                        started: true,
-                        completed: true,
-                    };
-                    visualization.visualization.insight.identifier = payload.savedVisualizationId;
-                }
-            }
-            else {
-                const visualization = assistantMessage.content
-                    .filter(isVisualizationContents)
-                    .flatMap((content) => content.createdVisualizations)
-                    .find((content) => content.id === payload.visualizationId);
-                if (visualization) {
-                    visualization.saving = false;
-                    visualization.savedVisualizationId = payload.savedVisualizationId;
-                }
+            if (assistantMessage.content.type !== "multipart") {
+                throw new Error("Unexpected message type");
+            }
+            const visualization = assistantMessage.content.parts
+                .filter((filter) => filter.type === "visualization")
+                .find((content) => content.visualization?.insight.identifier === payload.visualizationId);
+            if (visualization?.visualization) {
+                visualization.saving = {
+                    started: true,
+                    completed: true,
+                };
+                visualization.visualization.insight.identifier = payload.savedVisualizationId;
             }
         },
         saveVisualisationRenderStatusAction: (state, _action) => state,
         saveVisualisationRenderStatusSuccessAction: (state, { payload, }) => {
             const conversationId = getConversationLocalId(state.currentConversation);
             const assistantMessage = getAssistantMessageStrict(state, payload.assistantMessageId, conversationId);
-            if (isChatConversationLocalItem(assistantMessage)) {
-                if (assistantMessage.content.type !== "multipart") {
-                    throw new Error("Unexpected message type");
-                }
-                const visualization = assistantMessage.content.parts
-                    .filter((filter) => filter.type === "visualization")
-                    .find((content) => content.visualization?.insight.identifier === payload.visualizationId);
-                if (visualization) {
-                    delete visualization.reporting;
-                }
+            if (assistantMessage.content.type !== "multipart") {
+                throw new Error("Unexpected message type");
             }
-            else {
-                const visualization = assistantMessage.content
-                    .filter(isVisualizationContents)
-                    .flatMap((content) => content.createdVisualizations)
-                    .find((content) => content.id === payload.visualizationId);
-                if (visualization) {
-                    delete visualization.statusReportPending;
-                }
+            const visualization = assistantMessage.content.parts
+                .filter((filter) => filter.type === "visualization")
+                .find((content) => content.visualization?.insight.identifier === payload.visualizationId);
+            if (visualization) {
+                delete visualization.reporting;
             }
         },
         visualizationErrorAction: (state, _action) => state,
@@ -962,7 +713,7 @@ const messagesSlice = createSlice({
 });
 const errorToObject = (error) => Object.fromEntries(Object.getOwnPropertyNames(error).map((key) => [key, error[key]]));
 export const messagesSliceReducer = messagesSlice.reducer;
-export const { loadThreadAction, loadThreadErrorAction, loadThreadSuccessAction, loadConversationsSuccessAction, loadConversationSuccessAction, restoreCachedMessagesAction, clearThreadErrorAction, clearThreadSuccessAction, clearConversationSuccessAction, evaluateConversationTitleAction, evaluateMessageAction, evaluateMessageErrorAction, evaluateMessageStreamingAction, evaluateMessageUpdateAction, evaluateMessageCompleteAction, setMessagesAction, setVerboseAction, setGlobalErrorAction, cancelAsyncAction, clearConversationLoadingAction, setUserFeedback, setUserFeedbackError, clearUserFeedbackError, saveVisualizationAction, savedVisualizationAction, saveVisualizationErrorAction, saveVisualizationSuccessAction, saveVisualisationRenderStatusAction, saveVisualisationRenderStatusSuccessAction, visualizationErrorAction, appendInteractionStepAction, appendTracedActionAction, appendResponseDetailAction, applyPendingAgentSwitchAction, revertAgentSwitchAction, setAgentsAction, pinConversationAction, pinConversationSuccessAction, pinConversationFailureAction, renameConversationAction, renameConversationSuccessAction, renameConversationFailureAction, deleteConversationAction, deleteConversationStartAction, deleteConversationSuccessAction, deleteConversationFailureAction, filledFormAction, refocusInput, 
+export const { loadThreadAction, loadThreadErrorAction, loadConversationsSuccessAction, loadConversationSuccessAction, clearThreadErrorAction, clearThreadSuccessAction, clearConversationSuccessAction, evaluateConversationTitleAction, evaluateMessageAction, evaluateMessageErrorAction, evaluateMessageStreamingAction, evaluateMessageUpdateAction, evaluateMessageCompleteAction, setMessagesAction, setVerboseAction, setGlobalErrorAction, cancelAsyncAction, clearConversationLoadingAction, setUserFeedback, setUserFeedbackError, saveVisualizationAction, savedVisualizationAction, saveVisualizationErrorAction, saveVisualizationSuccessAction, saveVisualisationRenderStatusAction, saveVisualisationRenderStatusSuccessAction, visualizationErrorAction, appendInteractionStepAction, appendTracedActionAction, appendResponseDetailAction, applyPendingAgentSwitchAction, revertAgentSwitchAction, setAgentsAction, pinConversationAction, pinConversationSuccessAction, pinConversationFailureAction, renameConversationAction, renameConversationSuccessAction, renameConversationFailureAction, deleteConversationAction, deleteConversationStartAction, deleteConversationSuccessAction, deleteConversationFailureAction, filledFormAction, refocusInput, 
 /**
  * @public
  */