@zeniai/client-epic-state 4.16.81 → 4.16.82

package/lib/entity/aiController/aiControllerPayload.d.ts

@@ -19,7 +19,6 @@ export interface AIControllerVisualizationPayload {
 }
 export interface AIControllerAnswerPayload {
     text: string;
-    follow_up_questions?: string[];
     state: string;
     type: string;
     visualization: AIControllerVisualizationPayload | null;
@@ -27,6 +26,7 @@ export interface AIControllerAnswerPayload {
     assumptions: string | null;
     insights: string | null;
     summarized_title: string | null;
+    follow_up_questions?: string[];
 }
 export interface ChatSessionPayload {
     chat_session_id: string;

package/lib/entity/aiController/aiControllerReducer.d.ts

@@ -12,6 +12,7 @@ export declare const setNewSession: import("@reduxjs/toolkit").ActionCreatorWith
 }, "aiController/setNewSession">, clearSession: import("@reduxjs/toolkit").ActionCreatorWithPayload<string, "aiController/clearSession">, setSessions: import("@reduxjs/toolkit").ActionCreatorWithPayload<ChatSessionPayload[], "aiController/setSessions">, setChatHistory: import("@reduxjs/toolkit").ActionCreatorWithPayload<{
     chatSessionId: ID;
     history: MessagePayload[];
+    isPaginationComplete?: boolean;
 }, "aiController/setChatHistory">, upsertAnswerPayload: import("@reduxjs/toolkit").ActionCreatorWithPayload<{
     chatSessionId: ID;
     answerPayload: AIControllerAnswerPayload;
@@ -40,6 +41,8 @@ export declare const setNewSession: import("@reduxjs/toolkit").ActionCreatorWith
 }, "aiController/addQuestionPayload">, clearAIController: import("@reduxjs/toolkit").ActionCreatorWithoutPayload<"aiController/clearAIController">, deleteQuestion: import("@reduxjs/toolkit").ActionCreatorWithPayload<{
     chatSessionId: ID;
     questionAnswerId: ID;
-}, "aiController/deleteQuestion">;
+}, "aiController/deleteQuestion">, updateNewSession: import("@reduxjs/toolkit").ActionCreatorWithPayload<{
+    chatSessionId: ID;
+}, "aiController/updateNewSession">;
 declare const _default: import("redux").Reducer<AIControllerState>;
 export default _default;

package/lib/entity/aiController/aiControllerReducer.js

@@ -1,105 +1,131 @@
 "use strict";
 var _a;
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.deleteQuestion = exports.clearAIController = exports.addQuestionPayload = exports.upsertOrAddQuestionAnswerPayload = exports.upsertAnswerPayload = exports.setChatHistory = exports.setSessions = exports.clearSession = exports.setNewSession = exports.toAIControllerVisualization = exports.initialAIControllerState = void 0;
+exports.updateNewSession = exports.deleteQuestion = exports.clearAIController = exports.addQuestionPayload = exports.upsertOrAddQuestionAnswerPayload = exports.upsertAnswerPayload = exports.setChatHistory = exports.setSessions = exports.clearSession = exports.setNewSession = exports.toAIControllerVisualization = exports.initialAIControllerState = void 0;
 const toolkit_1 = require("@reduxjs/toolkit");
 const zeniDayJS_1 = require("../../zeniDayJS");
 const aiControllerState_1 = require("./aiControllerState");
 exports.initialAIControllerState = {
     aiControllerByChatSessionId: {},
+    partialQuestionAnswers: {},
 };
+// Utility: Merge two arrays of Q&A pairs, combining responses for the same question
+function mergeQuestionAnswerPairs(existing, newQAs) {
+    const merged = new Map();
+    // Add existing Q&A pairs
+    for (const qa of existing) {
+        merged.set(qa.questionAnswerId, Object.assign({}, qa));
+    }
+    // Merge new Q&A pairs
+    for (const newQA of newQAs) {
+        const existingQA = merged.get(newQA.questionAnswerId);
+        if (existingQA) {
+            // Merge responses
+            const mergedResponses = Object.assign(Object.assign({}, existingQA.responses), newQA.responses);
+            merged.set(newQA.questionAnswerId, Object.assign(Object.assign({}, newQA), { responses: mergedResponses }));
+        }
+        else {
+            // Add new Q&A pair
+            merged.set(newQA.questionAnswerId, newQA);
+        }
+    }
+    // Convert back to array and sort by question creation time
+    return Array.from(merged.values()).sort((a, b) => a.question.createdAt.valueOf() - b.question.createdAt.valueOf());
+}
 // Utility: Convert chat history (array of MessagePayload) to AIControllerQuestionWithAnswer[]
 function chatHistoryToQA(history) {
+    var _a, _b, _c;
+    const questionGroups = new Map();
+    // First pass: collect all messages and group by message_id
+    for (const message of history) {
+        const messageId = message.message_id;
+        if (!questionGroups.has(messageId)) {
+            questionGroups.set(messageId, {
+                question: null,
+                responses: [],
+            });
+        }
+        const group = questionGroups.get(messageId);
+        if (!group) {
+            continue;
+        }
+        if (message.sender === 'user') {
+            // User message is the question
+            group.question = message;
+        }
+        else if (message.sender === 'agent') {
+            // Agent message is a response block
+            group.responses.push(message);
+        }
+    }
+    // Convert groups to AIControllerQuestionWithAnswer
     const questionAnswerPairs = [];
-    let userMessageIndex = 0;
-    while (userMessageIndex < history.length) {
-        // Skip all but the last consecutive user message
-        let lastUserMessageIndex = userMessageIndex;
-        while (lastUserMessageIndex + 1 < history.length &&
-            history[lastUserMessageIndex + 1].sender === 'user') {
-            lastUserMessageIndex++;
+    for (const [messageId, group] of questionGroups.entries()) {
+        // Skip groups that don't have a question
+        if (!group.question) {
+            continue;
         }
-        const userMessage = history[lastUserMessageIndex];
-        if (userMessage.sender === 'user') {
-            // Find the next agent message
-            let agentMessageIndex = lastUserMessageIndex + 1;
-            while (agentMessageIndex < history.length &&
-                history[agentMessageIndex].sender !== 'agent') {
-                agentMessageIndex++;
-            }
-            const agentMessage = agentMessageIndex < history.length &&
-                history[agentMessageIndex].sender === 'agent'
-                ? history[agentMessageIndex]
-                : null;
-            const questionAnswerId = `qa_${userMessage.message_id}`;
-            const sender = (0, aiControllerState_1.toMessageSender)(userMessage.sender);
-            const question = {
-                messageId: userMessage.message_id,
-                chatSessionId: userMessage.chat_session_id,
-                userId: userMessage.user_id,
-                sender,
-                state: 'complete',
-                questionText: userMessage.content,
-                createdAt: (0, zeniDayJS_1.date)(userMessage.created_at),
-            };
-            const responses = {};
-            if (agentMessage != null) {
-                try {
-                    const parsedAgentMessage = (0, aiControllerState_1.toMessageType)(agentMessage.message_type) === 'json'
-                        ? JSON.parse(agentMessage.content)
-                        : {
-                            text: agentMessage.content,
-                            state: 'complete',
-                            type: 'answer',
-                            visualization: undefined,
-                            follow_up_questions: [],
-                        };
-                    const responseType = (0, aiControllerState_1.toAIControllerAnswerResponseType)(parsedAgentMessage.type);
-                    const responseState = (0, aiControllerState_1.toAIControllerAnswerStateType)(parsedAgentMessage.state);
-                    if (typeof parsedAgentMessage.text === 'string' &&
-                        aiControllerState_1.ALL_AI_CONTROLLER_ANSWER_RESPONSE_TYPES.includes(responseType)) {
-                        responses[responseType] = {
-                            messageId: agentMessage.message_id,
-                            chatSessionId: agentMessage.chat_session_id,
-                            userId: agentMessage.user_id,
-                            sender: (0, aiControllerState_1.toMessageSender)(agentMessage.sender),
-                            type: responseType,
-                            state: responseState,
-                            content: {
-                                text: parsedAgentMessage.text,
-                                visualization: parsedAgentMessage.visualization
-                                    ? (0, exports.toAIControllerVisualization)(parsedAgentMessage.visualization)
-                                    : undefined,
-                                followUpQuestions: Array.isArray(parsedAgentMessage.follow_up_questions)
-                                    ? parsedAgentMessage.follow_up_questions
-                                    : [],
-                                assumptions: '',
-                                insights: '',
-                                summarizedTitle: undefined,
-                            },
-                            createdAt: (0, zeniDayJS_1.date)(agentMessage.created_at),
-                        };
-                    }
-                }
-                catch (error) {
-                    console.error('Failed to parse agent message: ', error);
+        const question = {
+            messageId: group.question.message_id,
+            chatSessionId: group.question.chat_session_id,
+            userId: group.question.user_id,
+            sender: (0, aiControllerState_1.toMessageSender)(group.question.sender),
+            state: 'complete',
+            questionText: group.question.content,
+            createdAt: (0, zeniDayJS_1.date)(group.question.created_at),
+        };
+        const responses = {};
+        // Process each response for this question
+        for (const agentMessage of group.responses) {
+            try {
+                const parsedAgentMessage = (0, aiControllerState_1.toMessageType)(agentMessage.message_type) === 'json'
+                    ? JSON.parse(agentMessage.content)
+                    : {
+                        text: agentMessage.content,
+                        state: 'complete',
+                        type: 'answer',
+                        visualization: null,
+                        follow_up_questions: [],
+                    };
+                const responseType = (0, aiControllerState_1.toAIControllerAnswerResponseType)(parsedAgentMessage.type);
+                const responseState = (0, aiControllerState_1.toAIControllerAnswerStateType)(parsedAgentMessage.state);
+                if (typeof parsedAgentMessage.text === 'string' &&
+                    aiControllerState_1.ALL_AI_CONTROLLER_ANSWER_RESPONSE_TYPES.includes(responseType)) {
+                    responses[responseType] = {
+                        messageId: agentMessage.message_id,
+                        chatSessionId: agentMessage.chat_session_id,
+                        userId: agentMessage.user_id,
+                        sender: (0, aiControllerState_1.toMessageSender)(agentMessage.sender),
+                        type: responseType,
+                        state: responseState,
+                        content: {
+                            text: parsedAgentMessage.text,
+                            visualization: parsedAgentMessage.visualization
+                                ? (0, exports.toAIControllerVisualization)(parsedAgentMessage.visualization)
+                                : undefined,
+                            followUpQuestions: Array.isArray(parsedAgentMessage.follow_up_questions)
+                                ? parsedAgentMessage.follow_up_questions
+                                : [],
+                            assumptions: (_a = parsedAgentMessage.assumptions) !== null && _a !== void 0 ? _a : '',
+                            insights: (_b = parsedAgentMessage.insights) !== null && _b !== void 0 ? _b : '',
+                            summarizedTitle: (_c = parsedAgentMessage.summarized_title) !== null && _c !== void 0 ? _c : undefined,
+                        },
+                        createdAt: (0, zeniDayJS_1.date)(agentMessage.created_at),
+                    };
                 }
             }
-            questionAnswerPairs.push({
-                questionAnswerId,
-                question,
-                responses,
-            });
-            userMessageIndex =
-                agentMessage !== null
-                    ? agentMessageIndex + 1
-                    : lastUserMessageIndex + 1;
-        }
-        else {
-            userMessageIndex++;
+            catch (error) {
+                console.error('Failed to parse agent message: ', error);
+            }
         }
+        questionAnswerPairs.push({
+            questionAnswerId: `qa_${messageId}`,
+            question,
+            responses,
+        });
     }
-    return questionAnswerPairs;
+    // Sort by question creation time (oldest first)
+    return questionAnswerPairs.sort((a, b) => a.question.createdAt.valueOf() - b.question.createdAt.valueOf());
 }
 const toAIControllerVisualization = (visualization) => {
     var _a;
@@ -178,10 +204,23 @@ const aiController = (0, toolkit_1.createSlice)({
                     chatSessionId,
                     userId,
                     chatSessionSummary: '',
+                    newSession: true,
                     createdAt,
                 },
                 questionAnswers: [],
             };
+            // Clear any existing partial Q&A when creating a new session
+            draft.partialQuestionAnswers[chatSessionId] = undefined;
+        },
+        updateNewSession(draft, action) {
+            const { chatSessionId } = action.payload;
+            if (draft.aiControllerByChatSessionId[chatSessionId] == null) {
+                console.warn(`session with id ${chatSessionId} not found`);
+                return;
+            }
+            if (draft.aiControllerByChatSessionId[chatSessionId].chatSession.newSession) {
+                draft.aiControllerByChatSessionId[chatSessionId].chatSession.newSession = false;
+            }
         },
         setSessions(draft, action) {
             action.payload.forEach((chatSessionPayload) => {
@@ -190,16 +229,145 @@ const aiController = (0, toolkit_1.createSlice)({
                         chatSession: toChatSession(chatSessionPayload),
                         questionAnswers: [],
                     };
+                // Clear any existing partial Q&A for this session
+                draft.partialQuestionAnswers[chatSessionPayload.chat_session_id] =
+                    undefined;
             });
         },
         setChatHistory(draft, action) {
-            const { chatSessionId, history } = action.payload;
+            const { chatSessionId, history, isPaginationComplete = false, } = action.payload;
             if (draft.aiControllerByChatSessionId[chatSessionId] == null) {
                 console.warn(`session with id ${chatSessionId} not found in setChatHistory`);
                 return;
             }
-            draft.aiControllerByChatSessionId[chatSessionId].questionAnswers =
-                chatHistoryToQA(history);
+            const session = draft.aiControllerByChatSessionId[chatSessionId];
+            // Check if we have an existing partial Q&A pair for this session
+            const existingPartialQA = draft.partialQuestionAnswers[chatSessionId];
+            // Process new messages into Q&A pairs
+            const newQAs = chatHistoryToQA(history);
+            // If we have an existing partial Q&A pair, merge with new Q&A pairs
+            let mergedQAs;
+            if (existingPartialQA && newQAs.length > 0) {
+                // Check if any new QAs have the same message_id as the existing partial QA
+                const hasMatchingMessageId = newQAs.some((newQA) => newQA.question.messageId === existingPartialQA.question.messageId);
+                if (hasMatchingMessageId) {
+                    // Merge the existing partial QA with new QAs that have matching message_id
+                    mergedQAs = mergeQuestionAnswerPairs([existingPartialQA], newQAs);
+                }
+                else {
+                    // No matching message_id - existing partial QA should be moved to complete
+                    // and new QAs should be processed separately
+                    mergedQAs = [existingPartialQA, ...newQAs];
+                }
+            }
+            else if (existingPartialQA && newQAs.length === 0) {
+                // Intermediate page scenario: existing partial QA but no new blocks to merge
+                // This means the partial QA is complete (no more blocks coming to complete it)
+                mergedQAs = [existingPartialQA];
+            }
+            else {
+                mergedQAs = newQAs;
+            }
+            // Separate complete and incomplete pairs
+            const completeQAs = [];
+            let incompleteQA;
+            for (const qa of mergedQAs) {
+                const questionText = qa.question.questionText;
+                const hasQuestion = questionText != null && questionText.trim().length > 0;
+                if (!hasQuestion) {
+                    // Skip if no question text
+                    continue;
+                }
+                // Check if this Q&A pair is complete
+                // A Q&A pair is considered complete if it has:
+                // 1. A question (already checked above)
+                // 2. At least one agent response (any type: thought, reasoning, analysis, answer, etc.)
+                // 3. AND EITHER:
+                //    - At least one response with state: 'complete' (not 'partial')
+                //    - OR multiple response blocks (thought, reasoning, analysis) even if they're all partial
+                const hasAnyResponse = Object.keys(qa.responses).length > 0;
+                const hasCompleteResponse = Object.values(qa.responses).some((response) => response.state === 'complete');
+                const hasMultipleResponseBlocks = Object.keys(qa.responses).length > 1;
+                // Check if this is an existing partial QA that should be moved to complete
+                // because no more blocks are coming to complete it
+                const isExistingPartialQA = existingPartialQA &&
+                    qa.question.messageId === existingPartialQA.question.messageId &&
+                    !hasCompleteResponse;
+                if (hasQuestion &&
+                    hasAnyResponse &&
+                    (hasCompleteResponse || hasMultipleResponseBlocks)) {
+                    // This Q&A pair is complete - has question and either:
+                    // - at least one complete response, OR
+                    // - multiple response blocks (thought, reasoning, analysis) even if all partial
+                    completeQAs.push(qa);
+                }
+                else if (hasQuestion &&
+                    hasAnyResponse &&
+                    isExistingPartialQA === true) {
+                    // This is an existing partial QA that should be moved to complete
+                    // because no more blocks are coming to complete it (intermediate page scenario)
+                    completeQAs.push(qa);
+                }
+                else if (hasQuestion && hasAnyResponse) {
+                    // This Q&A pair has a question and responses but they're all partial
+                    // Keep as incomplete for potential merging with future pages
+                    if (!incompleteQA) {
+                        incompleteQA = qa;
+                    }
+                }
+                else if (hasQuestion) {
+                    // This Q&A pair has only a question, no responses yet
+                    // Keep as incomplete for potential merging with future pages
+                    if (!incompleteQA) {
+                        incompleteQA = qa;
+                    }
+                }
+            }
+            // Update partialQuestionAnswers based on completion status
+            if (isPaginationComplete) {
+                // If pagination is complete, we should have processed all messages
+                // Any remaining incomplete Q&A pairs should be moved to complete
+                if (incompleteQA) {
+                    // When pagination is complete, move all incomplete Q&A pairs to complete
+                    // This handles cases where:
+                    // 1. Question with no responses (error in generation)
+                    // 2. Question with only partial responses (error in generation)
+                    // 3. Question with only answer blocks (no thought/reasoning blocks)
+                    completeQAs.push(incompleteQA);
+                    draft.partialQuestionAnswers[chatSessionId] = undefined;
+                }
+                else {
+                    // No incomplete Q&A pairs, clear partialQuestionAnswers
+                    draft.partialQuestionAnswers[chatSessionId] = undefined;
+                }
+            }
+            else if (incompleteQA) {
+                // Check if this is an intermediate page scenario
+                // If we had an existing partial QA but no new blocks were merged,
+                // then the partial QA should be considered complete
+                if (existingPartialQA && newQAs.length === 0) {
+                    // Intermediate page: existing partial QA but no new blocks to merge
+                    // Move the partial QA to complete since no more blocks are coming
+                    completeQAs.push(incompleteQA);
+                    draft.partialQuestionAnswers[chatSessionId] = undefined;
+                }
+                else {
+                    // Pagination is not complete, keep incomplete Q&A pairs for future merging
+                    draft.partialQuestionAnswers[chatSessionId] = incompleteQA;
+                }
+            }
+            else if (existingPartialQA && newQAs.length === 0) {
+                // Special case: existing partial QA but no new blocks and no incomplete QA in mergedQAs
+                // This means the existing partial QA should be moved to complete
+                completeQAs.push(existingPartialQA);
+                draft.partialQuestionAnswers[chatSessionId] = undefined;
+            }
+            else {
+                // No incomplete Q&A pairs, clear partialQuestionAnswers
+                draft.partialQuestionAnswers[chatSessionId] = undefined;
+            }
+            // Update state - append new complete Q&As to existing ones
+            session.questionAnswers = mergeQuestionAnswerPairs(session.questionAnswers, completeQAs);
         },
         /**
          * Upsert a Q/A pair: if the questionAnswerId exists, upsert the response block; if not, add a new Q/A entry.
@@ -281,10 +449,17 @@ const aiController = (0, toolkit_1.createSlice)({
             }
         },
         clearSession(draft, action) {
-            delete draft.aiControllerByChatSessionId[action.payload];
+            var _a;
+            const sessionId = action.payload;
+            delete draft.aiControllerByChatSessionId[sessionId];
+            // Clear partial Q&A if it belongs to the cleared session
+            if (((_a = draft.partialQuestionAnswers) === null || _a === void 0 ? void 0 : _a[sessionId]) != null) {
+                draft.partialQuestionAnswers[sessionId] = undefined;
+            }
         },
         clearAIController(draft) {
             draft.aiControllerByChatSessionId = {};
+            draft.partialQuestionAnswers = {};
         },
         deleteQuestion(draft, action) {
             const { chatSessionId, questionAnswerId } = action.payload;
@@ -305,9 +480,10 @@ const toChatSession = (chatSessionPayload) => {
     return {
         chatSessionId: chat_session_id,
         userId: user_id,
+        newSession: false,
         chatSessionSummary: chat_session_summary !== null && chat_session_summary !== void 0 ? chat_session_summary : undefined,
         createdAt: (0, zeniDayJS_1.date)(created_at),
     };
 };
-_a = aiController.actions, exports.setNewSession = _a.setNewSession, exports.clearSession = _a.clearSession, exports.setSessions = _a.setSessions, exports.setChatHistory = _a.setChatHistory, exports.upsertAnswerPayload = _a.upsertAnswerPayload, exports.upsertOrAddQuestionAnswerPayload = _a.upsertOrAddQuestionAnswerPayload, exports.addQuestionPayload = _a.addQuestionPayload, exports.clearAIController = _a.clearAIController, exports.deleteQuestion = _a.deleteQuestion;
+_a = aiController.actions, exports.setNewSession = _a.setNewSession, exports.clearSession = _a.clearSession, exports.setSessions = _a.setSessions, exports.setChatHistory = _a.setChatHistory, exports.upsertAnswerPayload = _a.upsertAnswerPayload, exports.upsertOrAddQuestionAnswerPayload = _a.upsertOrAddQuestionAnswerPayload, exports.addQuestionPayload = _a.addQuestionPayload, exports.clearAIController = _a.clearAIController, exports.deleteQuestion = _a.deleteQuestion, exports.updateNewSession = _a.updateNewSession;
 exports.default = aiController.reducer;

package/lib/entity/aiController/aiControllerState.d.ts

@@ -54,11 +54,11 @@ export interface ResponseBlockBase {
 }
 export interface AnswerContentBlockBase {
     text: string;
-    visualization?: AIControllerVisualization;
     followUpQuestions: string[];
     assumptions: string;
     insights: string;
     summarizedTitle?: string;
+    visualization?: AIControllerVisualization;
 }
 export interface QuestionContentBlockBase extends ResponseBlockBase {
     questionText: string;
@@ -76,6 +76,7 @@ export interface ChatSession {
     chatSessionId: ID;
     userId: ID;
     createdAt: ZeniDate;
+    newSession: boolean;
     chatSessionSummary?: string;
 }
 export interface ChatSessionWithMessages {
@@ -84,4 +85,5 @@ export interface ChatSessionWithMessages {
 }
 export interface AIControllerState {
     aiControllerByChatSessionId: Record<ID, ChatSessionWithMessages>;
+    partialQuestionAnswers: Record<ID, AIControllerQuestionWithAnswer | undefined>;
 }

package/lib/view/aiControllerView/aiControllerViewReducer.d.ts

@@ -1,5 +1,6 @@
 import { FetchStateAndError, ID } from '../../commonStateTypes/common';
 import { PageToken } from '../../commonStateTypes/viewAndReport/viewAndReport';
+import { ChatSessionPayload } from '../../entity/aiController/aiControllerPayload';
 import { AIControllerAnswerResponseType } from '../../entity/aiController/aiControllerState';
 import { ZeniAPIStatus } from '../../responsePayload';
 import { AIControllerViewState } from './aiControllerViewState';
@@ -25,8 +26,10 @@ export declare const submitQuestion: import("@reduxjs/toolkit").ActionCreatorWit
     sessionId: ID;
 }, "aiControllerView/createSessionSuccess">, createSessionFailure: import("@reduxjs/toolkit").ActionCreatorWithPayload<ZeniAPIStatus<Record<string, unknown>>, "aiControllerView/createSessionFailure">, fetchChatSessionsForUser: import("@reduxjs/toolkit").ActionCreatorWithoutPayload<"aiControllerView/fetchChatSessionsForUser">, fetchChatSessionsForUserSuccess: import("@reduxjs/toolkit").ActionCreatorWithPayload<{
     nextPageToken: PageToken;
-}, "aiControllerView/fetchChatSessionsForUserSuccess">, fetchChatSessionsForUserFailure: import("@reduxjs/toolkit").ActionCreatorWithPayload<ZeniAPIStatus<Record<string, unknown>>, "aiControllerView/fetchChatSessionsForUserFailure">, fetchChatHistory: import("@reduxjs/toolkit").ActionCreatorWithPreparedPayload<[chatSessionId: string], {
+    sessions: ChatSessionPayload[];
+}, "aiControllerView/fetchChatSessionsForUserSuccess">, fetchChatSessionsForUserFailure: import("@reduxjs/toolkit").ActionCreatorWithPayload<ZeniAPIStatus<Record<string, unknown>>, "aiControllerView/fetchChatSessionsForUserFailure">, fetchChatHistory: import("@reduxjs/toolkit").ActionCreatorWithPreparedPayload<[chatSessionId: string, userId: string], {
     chatSessionId: string;
+    userId: string;
 }, "aiControllerView/fetchChatHistory", never, never>, fetchChatHistorySuccess: import("@reduxjs/toolkit").ActionCreatorWithPayload<{
     chatSessionId: ID;
     nextPageToken: PageToken;

package/lib/view/aiControllerView/aiControllerViewReducer.js

@@ -11,8 +11,9 @@ exports.initialAIControllerViewState = {
         scrollPositionByChatSessionId: {},
         currentChatSessionId: undefined,
         agentId: undefined,
-        chatHistoryNextPageToken: { nextPageToken: null, hasMore: false },
+        chatHistoryNextPageToken: {},
         chatSessionsNextPageToken: { nextPageToken: null, hasMore: false },
+        chatSessionsFirstPageFetched: false,
     },
     allChatSessionIds: [],
     createSessionState: {
@@ -99,6 +100,13 @@ const aiControllerView = (0, toolkit_1.createSlice)({
                 action.payload.nextPageToken;
             draft.uiState.chatSessionsNextPageToken.hasMore =
                 action.payload.nextPageToken != null;
+            if (draft.uiState.chatSessionsFirstPageFetched == false) {
+                draft.uiState.chatSessionsFirstPageFetched = true;
+            }
+            draft.allChatSessionIds = Array.from(new Set([
+                ...draft.allChatSessionIds,
+                ...action.payload.sessions.map((session) => session.chat_session_id),
+            ]));
         },
         fetchChatSessionsForUserFailure(draft, action) {
             draft.fetchState = 'Error';
@@ -112,8 +120,8 @@ const aiControllerView = (0, toolkit_1.createSlice)({
                     error: undefined,
                 };
             },
-            prepare(chatSessionId) {
-                return { payload: { chatSessionId } };
+            prepare(chatSessionId, userId) {
+                return { payload: { chatSessionId, userId } };
             },
         },
         fetchChatHistorySuccess(draft, action) {
@@ -122,8 +130,18 @@ const aiControllerView = (0, toolkit_1.createSlice)({
                 fetchState: 'Completed',
                 error: undefined,
             };
-            draft.uiState.chatHistoryNextPageToken.nextPageToken = nextPageToken;
-            draft.uiState.chatHistoryNextPageToken.hasMore = nextPageToken != null;
+            draft.uiState.chatHistoryNextPageToken[chatSessionId] = {
+                nextPageToken,
+                hasMore: nextPageToken != null,
+            };
+            if (draft.responseStateByChatSessionId[chatSessionId] == null) {
+                draft.responseStateByChatSessionId[chatSessionId] = {
+                    thought: { fetchState: 'Completed', error: undefined },
+                    analysis: { fetchState: 'Completed', error: undefined },
+                    reasoning: { fetchState: 'Completed', error: undefined },
+                    answer: { fetchState: 'Completed', error: undefined },
+                };
+            }
         },
         fetchChatHistoryFailure(draft, action) {
             const { chatSessionId, error } = action.payload;
@@ -152,6 +170,10 @@ const aiControllerView = (0, toolkit_1.createSlice)({
             draft.createSessionState.fetchState = 'Completed';
             draft.createSessionState.error = undefined;
             draft.uiState.currentChatSessionId = sessionId;
+            draft.uiState.chatHistoryNextPageToken[sessionId] = {
+                nextPageToken: null,
+                hasMore: false,
+            };
         },
         createSessionFailure(draft, action) {
             draft.createSessionState.fetchState = 'Error';

package/lib/view/aiControllerView/aiControllerViewSelector.d.ts

@@ -7,6 +7,7 @@ import { AIControllerViewUIState } from './aiControllerViewState';
 export interface AIControllerViewSelector extends SelectorView, AIControllerSelectorView {
     createSessionState: FetchStateAndError;
     stopSubmitStateByChatSessionId: Record<ID, FetchStateAndError>;
+    fetchChatSessionHistoryByChatSessionId: Record<ID, FetchStateAndError>;
     responseStateByChatSessionId: Record<ID, {
         thought: FetchStateAndError;
         analysis: FetchStateAndError;

package/lib/view/aiControllerView/aiControllerViewSelector.js

@@ -3,12 +3,13 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.getAIControllerView = void 0;
 const aiControllerSelector_1 = require("../../entity/aiController/aiControllerSelector");
 const getAIControllerView = (state) => {
-    const { fetchState, error, uiState, createSessionState, stopSubmitStateByChatSessionId, responseStateByChatSessionId, } = state.aiControllerViewState;
+    const { fetchState, error, uiState, createSessionState, stopSubmitStateByChatSessionId, responseStateByChatSessionId, fetchChatSessionHistoryByChatSessionId, } = state.aiControllerViewState;
     return Object.assign(Object.assign({ fetchState,
         error,
         createSessionState,
         stopSubmitStateByChatSessionId,
         responseStateByChatSessionId,
-        uiState }, (0, aiControllerSelector_1.getAIControllerSelectorView)(state.aiControllerState)), { version: 1 });
+        uiState,
+        fetchChatSessionHistoryByChatSessionId }, (0, aiControllerSelector_1.getAIControllerSelectorView)(state.aiControllerState)), { version: 1 });
 };
 exports.getAIControllerView = getAIControllerView;

package/lib/view/aiControllerView/aiControllerViewState.d.ts

@@ -11,10 +11,13 @@ export interface AIControllerViewUIState {
             scrollTop: number;
         };
     };
+    chatHistoryNextPageToken: {
+        [chatSessionId: ID]: PageTokenState;
+    };
+    chatSessionsNextPageToken: PageTokenState;
+    chatSessionsFirstPageFetched: boolean;
     currentChatSessionId?: ID;
     agentId?: ID;
-    chatHistoryNextPageToken: PageTokenState;
-    chatSessionsNextPageToken: PageTokenState;
 }
 export interface AIControllerViewState extends FetchStateAndError {
     uiState: AIControllerViewUIState;

package/lib/view/aiControllerView/epics/fetchChatHistoryEpic.d.ts

@@ -5,4 +5,4 @@ import { RootState } from '../../../reducer';
 import { ZeniAPI } from '../../../zeniAPI';
 import { fetchChatHistory, fetchChatHistoryFailure, fetchChatHistorySuccess } from '../aiControllerViewReducer';
 export type ActionType = ReturnType<typeof fetchChatHistory> | ReturnType<typeof setChatHistory> | ReturnType<typeof fetchChatHistorySuccess> | ReturnType<typeof fetchChatHistoryFailure>;
-export declare const fetchChatHistoryEpic: (actions$: ActionsObservable<ActionType>, _: StateObservable<RootState>, zeniAPI: ZeniAPI) => Observable<ActionType>;
+export declare const fetchChatHistoryEpic: (actions$: ActionsObservable<ActionType>, state$: StateObservable<RootState>, zeniAPI: ZeniAPI) => Observable<ActionType>;

package/lib/view/aiControllerView/epics/fetchChatHistoryEpic.js

@@ -6,15 +6,27 @@ const operators_1 = require("rxjs/operators");
 const aiControllerReducer_1 = require("../../../entity/aiController/aiControllerReducer");
 const responsePayload_1 = require("../../../responsePayload");
 const aiControllerViewReducer_1 = require("../aiControllerViewReducer");
-const fetchChatHistoryEpic = (actions$, _, zeniAPI) => actions$.pipe((0, operators_1.filter)(aiControllerViewReducer_1.fetchChatHistory.match), (0, operators_1.switchMap)((action) => {
-    const { chatSessionId } = action.payload;
+const fetchChatHistoryEpic = (actions$, state$, zeniAPI) => actions$.pipe((0, operators_1.filter)(aiControllerViewReducer_1.fetchChatHistory.match), (0, operators_1.mergeMap)((action) => {
+    var _a, _b;
+    const { chatSessionId, userId } = action.payload;
+    const state = state$.value;
+    const pageToken = (_b = (_a = state.aiControllerViewState.uiState.chatHistoryNextPageToken[chatSessionId]) === null || _a === void 0 ? void 0 : _a.nextPageToken) !== null && _b !== void 0 ? _b : null;
+    const queryParams = new URLSearchParams();
+    queryParams.append('page_size', '20');
+    queryParams.append('chat_session_id', chatSessionId);
+    queryParams.append('user_id', userId);
+    if (pageToken != null) {
+        queryParams.append('page_token', pageToken);
+    }
     return zeniAPI
-        .getJSON(`${zeniAPI.apiEndPoints.aiControllerMicroServiceBaseUrl}/1.0/sessions/${chatSessionId}/messages`)
+        .getJSON(`${zeniAPI.apiEndPoints.aiControllerMicroServiceBaseUrl}/1.0/history?${queryParams.toString()}`)
         .pipe((0, operators_1.mergeMap)((response) => {
         if ((0, responsePayload_1.isSuccessResponse)(response) && response.data != null) {
+            const isPaginationComplete = response.data.next_page_token == null;
             return (0, rxjs_1.of)((0, aiControllerReducer_1.setChatHistory)({
                 chatSessionId,
                 history: response.data.messages,
+                isPaginationComplete,
             }), (0, aiControllerViewReducer_1.fetchChatHistorySuccess)({
                 chatSessionId,
                 nextPageToken: response.data.next_page_token,

package/lib/view/aiControllerView/epics/fetchChatSessionsForUserEpic.d.ts

@@ -5,4 +5,4 @@ import { RootState } from '../../../reducer';
 import { ZeniAPI } from '../../../zeniAPI';
 import { fetchChatSessionsForUser, fetchChatSessionsForUserFailure, fetchChatSessionsForUserSuccess } from '../aiControllerViewReducer';
 export type ActionType = ReturnType<typeof fetchChatSessionsForUser> | ReturnType<typeof setSessions> | ReturnType<typeof fetchChatSessionsForUserSuccess> | ReturnType<typeof fetchChatSessionsForUserFailure>;
-export declare const fetchChatSessionsForUserEpic: (actions$: ActionsObservable<ActionType>, _: StateObservable<RootState>, zeniAPI: ZeniAPI) => Observable<ActionType>;
+export declare const fetchChatSessionsForUserEpic: (actions$: ActionsObservable<ActionType>, state$: StateObservable<RootState>, zeniAPI: ZeniAPI) => Observable<ActionType>;

package/lib/view/aiControllerView/epics/fetchChatSessionsForUserEpic.js

@@ -6,12 +6,24 @@ const operators_1 = require("rxjs/operators");
 const aiControllerReducer_1 = require("../../../entity/aiController/aiControllerReducer");
 const responsePayload_1 = require("../../../responsePayload");
 const aiControllerViewReducer_1 = require("../aiControllerViewReducer");
-const fetchChatSessionsForUserEpic = (actions$, _, zeniAPI) => actions$.pipe((0, operators_1.filter)(aiControllerViewReducer_1.fetchChatSessionsForUser.match), (0, operators_1.switchMap)(() => {
+const fetchChatSessionsForUserEpic = (actions$, state$, zeniAPI) => actions$.pipe((0, operators_1.filter)(aiControllerViewReducer_1.fetchChatSessionsForUser.match), (0, operators_1.switchMap)(() => {
+    var _a;
+    const state = state$.value;
+    const pageToken = (_a = state.aiControllerViewState.uiState.chatSessionsNextPageToken
+        .nextPageToken) !== null && _a !== void 0 ? _a : null;
+    const queryParams = new URLSearchParams();
+    queryParams.append('page_size', '50');
+    if (pageToken != null) {
+        queryParams.append('page_token', pageToken);
+    }
     return zeniAPI
-        .getJSON(`${zeniAPI.apiEndPoints.aiControllerMicroServiceBaseUrl}/1.0/sessions`)
+        .getJSON(`${zeniAPI.apiEndPoints.aiControllerMicroServiceBaseUrl}/1.0/sessions?${queryParams.toString()}`)
         .pipe((0, operators_1.mergeMap)((response) => {
         if ((0, responsePayload_1.isSuccessResponse)(response) && response.data != null) {
-            return (0, rxjs_1.of)((0, aiControllerReducer_1.setSessions)(response.data.sessions), (0, aiControllerViewReducer_1.fetchChatSessionsForUserSuccess)({
+            // Filter out empty sessions that don't have a summary
+            const sessions = response.data.sessions.filter((session) => Boolean(session.chat_session_summary) === true);
+            return (0, rxjs_1.of)((0, aiControllerReducer_1.setSessions)(sessions), (0, aiControllerViewReducer_1.fetchChatSessionsForUserSuccess)({
+                sessions,
                 nextPageToken: response.data.next_page_token,
             }));
         }

package/lib/view/aiControllerView/epics/submitQuestionEpic.d.ts

@@ -1,8 +1,8 @@
 import { ActionsObservable, StateObservable } from 'redux-observable';
 import { Observable } from 'rxjs';
-import { upsertOrAddQuestionAnswerPayload } from '../../../entity/aiController/aiControllerReducer';
+import { updateNewSession, upsertOrAddQuestionAnswerPayload } from '../../../entity/aiController/aiControllerReducer';
 import { RootState } from '../../../reducer';
 import { ZeniAPI } from '../../../zeniAPI';
 import { submitQuestion, submitQuestionFailure, submitQuestionSuccess } from '../aiControllerViewReducer';
-export type ActionType = ReturnType<typeof submitQuestion> | ReturnType<typeof submitQuestionSuccess> | ReturnType<typeof submitQuestionFailure> | ReturnType<typeof upsertOrAddQuestionAnswerPayload>;
+export type ActionType = ReturnType<typeof submitQuestion> | ReturnType<typeof updateNewSession> | ReturnType<typeof submitQuestionSuccess> | ReturnType<typeof submitQuestionFailure> | ReturnType<typeof upsertOrAddQuestionAnswerPayload>;
 export declare const submitQuestionEpic: (actions$: ActionsObservable<ActionType>, state$: StateObservable<RootState>, zeniAPI: ZeniAPI) => Observable<ActionType>;

package/lib/view/aiControllerView/epics/submitQuestionEpic.js

@@ -6,7 +6,7 @@ const operators_1 = require("rxjs/operators");
 const aiControllerReducer_1 = require("../../../entity/aiController/aiControllerReducer");
 const responsePayload_1 = require("../../../responsePayload");
 const aiControllerViewReducer_1 = require("../aiControllerViewReducer");
-const submitQuestionEpic = (actions$, state$, zeniAPI) => actions$.pipe((0, operators_1.filter)(aiControllerViewReducer_1.submitQuestion.match), (0, operators_1.switchMap)((action) => {
+const submitQuestionEpic = (actions$, state$, zeniAPI) => actions$.pipe((0, operators_1.filter)(aiControllerViewReducer_1.submitQuestion.match), (0, operators_1.mergeMap)((action) => {
     const { questionAnswerId, userId, questionId, createdAt, question } = action.payload;
     const { aiControllerViewState } = state$.value;
     const { uiState } = aiControllerViewState;
@@ -16,7 +16,9 @@ const submitQuestionEpic = (actions$, state$, zeniAPI) => actions$.pipe((0, oper
     if (agentId == null || sessionId == null || Boolean(question) == false) {
         throw new Error('Invalid State');
     }
-    return zeniAPI
+    // First, update the session to mark it as no longer new
+    const updateSessionAction = (0, aiControllerReducer_1.updateNewSession)({ chatSessionId: sessionId });
+    return (0, rxjs_1.concat)((0, rxjs_1.of)(updateSessionAction), zeniAPI
         .postAndGetJSON(`${zeniAPI.apiEndPoints.aiControllerMicroServiceBaseUrl}/1.0/agents/${agentId}`, payload)
         .pipe((0, operators_1.mergeMap)((response) => {
         if ((0, responsePayload_1.isSuccessResponse)(response) && response.data != null) {
@@ -32,14 +34,13 @@ const submitQuestionEpic = (actions$, state$, zeniAPI) => actions$.pipe((0, oper
                 answerPayload: response.data,
                 userId,
                 insertQuestionAnswer: false, // Don't insert the Q/A here after stopSubmitSuccess
-            }));
-            actions.push((0, aiControllerViewReducer_1.submitQuestionSuccess)());
+            }), (0, aiControllerViewReducer_1.submitQuestionSuccess)());
             return (0, rxjs_1.of)(...actions);
         }
         else {
             return (0, rxjs_1.of)((0, aiControllerViewReducer_1.submitQuestionFailure)(response.status));
         }
     }), (0, operators_1.catchError)((error) => (0, rxjs_1.of)((0, aiControllerViewReducer_1.submitQuestionFailure)((0, responsePayload_1.createZeniAPIStatus)('Unexpected Error', 'Submit question REST API call errored out' +
-        JSON.stringify(error))))));
+        JSON.stringify(error)))))));
 }));
 exports.submitQuestionEpic = submitQuestionEpic;

package/lib/view/companyView/types/cockpitTypes.d.ts

@@ -4,9 +4,9 @@ import { SubscriptionViewSortKey } from '../../subscriptionView/types/subscripti
 import { FilterCategoryValueType } from '../helpers/cockpitHelpers';
 import { CockpitDownloadTabsID, CockpitTabsFileType } from './cockpitTabsID';
 export declare const toPortfolioSortKeyType: (v: string) => "expenses" | "runway" | "companyName" | "cashBalance" | "revenues" | "netIncome";
-export declare const toManagementSortKeyType: (v: string) => "status" | "bookCloseDate" | "taxStatus" | "controller" | "preparer" | "reviewer" | "annualRecurringRevenue" | "monthlyRecurringRevenue" | "companyName" | "group" | "task" | "notes" | "plans" | "customerSuccessManager" | "apAnalyst" | "reportSent" | "taxLead" | "satisfaction";
+export declare const toManagementSortKeyType: (v: string) => "status" | "bookCloseDate" | "group" | "taxStatus" | "controller" | "preparer" | "reviewer" | "annualRecurringRevenue" | "monthlyRecurringRevenue" | "companyName" | "task" | "notes" | "plans" | "customerSuccessManager" | "apAnalyst" | "reportSent" | "taxLead" | "satisfaction";
 export declare const toOnboardingSortKeyType: (v: string) => "companyName" | "identityVerified" | "businessVerified" | "bankAccountLinked" | "qboLinked" | "groupCreated" | "customerCreated" | "notes" | "enroller" | "invite" | "latestCreated";
-export declare const toHealthSortKeyType: (v: string) => "runway" | "controller" | "monthlyRecurringRevenue" | "companyName" | "group" | "companyEngagementMetricsUpdatedAt" | "netPromoterScore" | "customerSatisfactionScore" | "externalProductServiceExposure" | "customerSuccessManager" | "lastReportSent" | "issues" | "productVerticals" | "churnProbability" | "subscriptionDate" | "renewalDate" | "dashboardLastSeen" | "lastEngagement";
+export declare const toHealthSortKeyType: (v: string) => "group" | "runway" | "controller" | "monthlyRecurringRevenue" | "companyName" | "companyEngagementMetricsUpdatedAt" | "netPromoterScore" | "customerSatisfactionScore" | "externalProductServiceExposure" | "customerSuccessManager" | "lastReportSent" | "issues" | "productVerticals" | "churnProbability" | "subscriptionDate" | "renewalDate" | "dashboardLastSeen" | "lastEngagement";
 export declare const toTaskManagerSortKeyType: (v: string) => "status" | "name" | "dueDate" | "tags" | "createdBy" | "priority" | "taskName" | "assignee";
 export type PortfolioViewSortKey = ReturnType<typeof toPortfolioSortKeyType>;
 export type ManagementViewSortKey = ReturnType<typeof toManagementSortKeyType>;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zeniai/client-epic-state",
-  "version": "4.16.81",
+  "version": "4.16.82",
   "description": "Shared module between Web & Mobile containing required abstractions for state management, async network communication. ",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",