@theia/ai-chat 1.66.0-next.44 → 1.66.0-next.73

package/lib/common/chat-model.js

@@ -20,7 +20,7 @@
  *--------------------------------------------------------------------------------------------*/
 // Partially copied from https://github.com/microsoft/vscode/blob/a2cab7255c0df424027be05d58e1b7b941f4ea60/src/vs/workbench/contrib/chat/common/chatModel.ts
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ProgressChatResponseContentImpl = exports.ErrorChatResponseModel = exports.MutableChatResponseModel = exports.QuestionResponseContentImpl = exports.HorizontalLayoutChatResponseContentImpl = exports.CommandChatResponseContentImpl = exports.COMMAND_CHAT_RESPONSE_COMMAND = exports.ToolCallChatResponseContentImpl = exports.CodeChatResponseContentImpl = exports.InformationalChatResponseContentImpl = exports.MarkdownChatResponseContentImpl = exports.ThinkingChatResponseContentImpl = exports.TextChatResponseContentImpl = exports.ErrorChatResponseContentImpl = exports.MutableChatRequestModel = exports.ChatContextManagerImpl = exports.ChatRequestHierarchyBranchImpl = exports.ChatRequestHierarchyImpl = exports.ChatTreeChangeSet = exports.MutableChatModel = exports.QuestionResponseContent = exports.ProgressChatResponseContent = exports.ThinkingChatResponseContent = exports.ErrorChatResponseContent = exports.ToolCallChatResponseContent = exports.HorizontalLayoutChatResponseContent = exports.CodeChatResponseContent = exports.CommandChatResponseContent = exports.InformationalChatResponseContent = exports.MarkdownChatResponseContent = exports.TextChatResponseContent = exports.Location = exports.ChatResponseContent = exports.EditableChatRequestModel = exports.ChatRequestModel = exports.ChatSuggestionCallback = exports.ChatChangeEvent = exports.ChangeSetImpl = void 0;
+exports.UnknownChatResponseContentImpl = exports.ProgressChatResponseContentImpl = exports.ErrorChatResponseModel = exports.MutableChatResponseModel = exports.QuestionResponseContentImpl = exports.HorizontalLayoutChatResponseContentImpl = exports.CommandChatResponseContentImpl = exports.COMMAND_CHAT_RESPONSE_COMMAND = exports.ToolCallChatResponseContentImpl = exports.CodeChatResponseContentImpl = exports.InformationalChatResponseContentImpl = exports.MarkdownChatResponseContentImpl = exports.ThinkingChatResponseContentImpl = exports.TextChatResponseContentImpl = exports.ErrorChatResponseContentImpl = exports.MutableChatRequestModel = exports.ChatContextManagerImpl = exports.ChatRequestHierarchyBranchImpl = exports.ChatRequestHierarchyImpl = exports.ChatTreeChangeSet = exports.MutableChatModel = exports.QuestionResponseContent = exports.ProgressChatResponseContent = exports.ThinkingChatResponseContent = exports.ErrorChatResponseContent = exports.ToolCallChatResponseContent = exports.HorizontalLayoutChatResponseContent = exports.CodeChatResponseContent = exports.CommandChatResponseContent = exports.InformationalChatResponseContent = exports.MarkdownChatResponseContent = exports.TextChatResponseContent = exports.Location = exports.ChatResponseContent = exports.EditableChatRequestModel = exports.ChatRequestModel = exports.ChatSuggestionCallback = exports.ChatChangeEvent = exports.ChangeSetImpl = void 0;
 const ai_core_1 = require("@theia/ai-core");
 const core_1 = require("@theia/core");
 const markdown_rendering_1 = require("@theia/core/lib/common/markdown-rendering");
@@ -235,10 +235,9 @@ var QuestionResponseContent;
                 option && 'text' in option &&
                 typeof option.text === 'string' &&
                 ('value' in option ? typeof option.value === 'string' || typeof option.value === 'undefined' : true)) &&
-            'handler' in obj &&
-            typeof obj.handler === 'function' &&
-            'request' in obj &&
-            obj.request instanceof MutableChatRequestModel);
+            // handler and request are optional (undefined for restored/read-only questions)
+            ('handler' in obj ? obj.handler === undefined || typeof obj.handler === 'function' : true) &&
+            ('request' in obj ? obj.request === undefined || obj.request instanceof MutableChatRequestModel : true));
     }
     QuestionResponseContent.is = is;
 })(QuestionResponseContent || (exports.QuestionResponseContent = QuestionResponseContent = {}));
@@ -246,19 +245,28 @@ var QuestionResponseContent;
  * Implementations
  **********************/
 class MutableChatModel {
-    constructor(location = chat_agents_1.ChatAgentLocation.Panel) {
-        this.location = location;
+    get location() {
+        return this._location;
+    }
+    constructor(locationOrSerializedData = chat_agents_1.ChatAgentLocation.Panel) {
         this._onDidChangeEmitter = new core_1.Emitter();
         this.onDidChange = this._onDidChangeEmitter.event;
         this.toDispose = new core_1.DisposableCollection();
         this._suggestions = [];
         this._contextManager = new ChatContextManagerImpl();
-        // TODO accept serialized data as a parameter to restore a previously saved ChatModel
-        this._hierarchy = new ChatRequestHierarchyImpl();
-        this._changeSet = new ChatTreeChangeSet(this._hierarchy);
-        this.toDispose.push(this._changeSet);
-        this._changeSet.onDidChange(this._onDidChangeEmitter.fire, this._onDidChangeEmitter, this.toDispose);
-        this._id = (0, core_1.generateUuid)();
+        // Check if we're restoring from serialized data
+        if (this.isSerializedChatModel(locationOrSerializedData)) {
+            this.restoreFromSerializedData(locationOrSerializedData);
+        }
+        else {
+            // Normal creation path
+            this._location = locationOrSerializedData;
+            this._id = (0, core_1.generateUuid)();
+            this._hierarchy = new ChatRequestHierarchyImpl();
+            this._changeSet = new ChatTreeChangeSet(this._hierarchy);
+            this.toDispose.push(this._changeSet);
+            this._changeSet.onDidChange(this._onDidChangeEmitter.fire, this._onDidChangeEmitter, this.toDispose);
+        }
         this.toDispose.pushAll([
             this._onDidChangeEmitter,
             this._contextManager.onDidChange(this._onDidChangeEmitter.fire, this._onDidChangeEmitter),
@@ -270,6 +278,38 @@ class MutableChatModel {
             }),
         ]);
     }
+    /**
+     * Type guard to determine if we're receiving serialized data
+     */
+    isSerializedChatModel(data) {
+        return typeof data === 'object' && 'sessionId' in data && 'requests' in data && 'responses' in data;
+    }
+    /**
+     * Restore this chat model from serialized data
+     *
+     * Does not restore response content or changesets.
+     * This handled by the chat service using the deserializer registries.
+     */
+    restoreFromSerializedData(data) {
+        this._id = data.sessionId;
+        this._location = data.location;
+        // First, create all request models and build a map
+        const requestMap = new Map();
+        for (const reqData of data.requests) {
+            const respData = data.responses.find(r => r.requestId === reqData.id);
+            const requestModel = new MutableChatRequestModel(this, reqData, respData);
+            requestMap.set(requestModel.id, requestModel);
+        }
+        // Restore the hierarchy structure with all alternatives
+        this._hierarchy = new ChatRequestHierarchyImpl(data.hierarchy, requestMap);
+        // Register all requests with changeset
+        this._changeSet = new ChatTreeChangeSet(this._hierarchy);
+        this.toDispose.push(this._changeSet);
+        this._changeSet.onDidChange(this._onDidChangeEmitter.fire, this._onDidChangeEmitter, this.toDispose);
+        for (const requestModel of requestMap.values()) {
+            this._changeSet.registerRequest(requestModel);
+        }
+    }
     get id() {
         return this._id;
     }
@@ -334,6 +374,44 @@ class MutableChatModel {
     isEmpty() {
         return this.getRequests().length === 0;
     }
+    toSerializable() {
+        const hierarchy = this._hierarchy.toSerializable();
+        const allRequests = this.getAllRequests();
+        const serializedRequests = allRequests.map(req => req.toSerializable());
+        const serializedResponses = allRequests
+            .filter(req => req.response)
+            .map(req => req.response.toSerializable());
+        return {
+            sessionId: this._id,
+            location: this.location,
+            hierarchy,
+            requests: serializedRequests,
+            responses: serializedResponses
+        };
+    }
+    /**
+     * Get all requests from the hierarchy.
+     * This is used for operations that need to process all requests, such as serialization.
+     */
+    getAllRequests() {
+        const allRequests = [];
+        const visited = new Set();
+        const collectFromBranch = (branch) => {
+            for (const item of branch.items) {
+                // Avoid duplicates
+                if (!visited.has(item.element.id)) {
+                    visited.add(item.element.id);
+                    allRequests.push(item.element);
+                }
+                // Recursively collect from next branches
+                if (item.next) {
+                    collectFromBranch(item.next);
+                }
+            }
+        };
+        collectFromBranch(this._hierarchy.branch);
+        return allRequests;
+    }
     dispose() {
         this.toDispose.dispose();
     }
@@ -456,10 +534,70 @@ class ChatTreeChangeSet {
 }
 exports.ChatTreeChangeSet = ChatTreeChangeSet;
 class ChatRequestHierarchyImpl {
-    constructor() {
+    constructor(serializedHierarchy, requestMap) {
         this.onDidChangeActiveBranchEmitter = new core_1.Emitter();
         this.onDidChange = this.onDidChangeActiveBranchEmitter.event;
         this.branch = new ChatRequestHierarchyBranchImpl(this);
+        if (serializedHierarchy && requestMap) {
+            this.restoreFromSerialized(serializedHierarchy, requestMap);
+        }
+    }
+    /**
+     * Restore the hierarchy from serialized data.
+     */
+    restoreFromSerialized(serializedHierarchy, requestMap) {
+        // Build a map of branch IDs to restored branch objects
+        const branchMap = new Map();
+        // Function to restore a branch and its descendants
+        const restoreBranch = (branchId) => {
+            // Check if already restored
+            if (branchMap.has(branchId)) {
+                return branchMap.get(branchId);
+            }
+            const serializedBranch = serializedHierarchy.branches[branchId];
+            if (!serializedBranch) {
+                throw new Error(`Cannot find serialized branch with id: ${branchId}`);
+            }
+            // Restore items in this branch
+            const items = serializedBranch.items.map(serializedItem => {
+                const request = requestMap.get(serializedItem.requestId);
+                if (!request) {
+                    throw new Error(`Cannot find request with id: ${serializedItem.requestId}`);
+                }
+                // Restore next branch if present
+                const next = serializedItem.nextBranchId
+                    ? restoreBranch(serializedItem.nextBranchId)
+                    : undefined;
+                return {
+                    element: request,
+                    next
+                };
+            });
+            // Determine if this is the root branch
+            const isRoot = branchId === serializedHierarchy.rootBranchId;
+            if (isRoot) {
+                // For root branch, we need to replace the existing branch's internals
+                // Cast is safe here as we know this.branch is a ChatRequestHierarchyBranchImpl
+                const rootBranch = this.branch;
+                // Use Object.assign to update the readonly properties
+                Object.assign(rootBranch, {
+                    id: branchId,
+                    items,
+                    _activeIndex: serializedBranch.activeBranchIndex
+                });
+                branchMap.set(branchId, rootBranch);
+                return rootBranch;
+            }
+            else {
+                // For non-root branches, use constructor-based deserialization
+                const restoredBranch = new ChatRequestHierarchyBranchImpl(this, undefined, // previous will be set by parent
+                items, serializedBranch.activeBranchIndex, branchId);
+                branchMap.set(branchId, restoredBranch);
+                return restoredBranch;
+            }
+        };
+        // Start restoration from the root branch
+        restoreBranch(serializedHierarchy.rootBranchId);
     }
     append(request) {
         const branches = this.activeBranches();
@@ -514,6 +652,32 @@ class ChatRequestHierarchyImpl {
     notifyChange(event) {
         this.onDidChangeActiveBranchEmitter.fire(event);
     }
+    toSerializable() {
+        const branches = {};
+        // Recursively serialize all branches starting from the root
+        this.serializeBranch(this.branch, branches);
+        return {
+            rootBranchId: this.branch.id,
+            branches
+        };
+    }
+    serializeBranch(branch, branches) {
+        const items = branch.items.map(item => {
+            var _a;
+            if (item.next) {
+                this.serializeBranch(item.next, branches);
+            }
+            return {
+                requestId: item.element.id,
+                nextBranchId: (_a = item.next) === null || _a === void 0 ? void 0 : _a.id
+            };
+        });
+        branches[branch.id] = {
+            id: branch.id,
+            items,
+            activeBranchIndex: branch.activeBranchIndex
+        };
+    }
     dispose() {
         this.onDidChangeActiveBranchEmitter.dispose();
         this.branch.dispose();
@@ -521,12 +685,12 @@ class ChatRequestHierarchyImpl {
 }
 exports.ChatRequestHierarchyImpl = ChatRequestHierarchyImpl;
 class ChatRequestHierarchyBranchImpl {
-    constructor(hierarchy, previous, items = [], _activeIndex = -1) {
+    constructor(hierarchy, previous, items = [], _activeIndex = -1, id) {
         this.hierarchy = hierarchy;
         this.previous = previous;
         this.items = items;
         this._activeIndex = _activeIndex;
-        this.id = (0, core_1.generateUuid)();
+        this.id = id !== null && id !== void 0 ? id : (0, core_1.generateUuid)();
     }
     get activeBranchIndex() {
         return this._activeIndex;
@@ -670,24 +834,74 @@ class ChatContextManagerImpl {
 }
 exports.ChatContextManagerImpl = ChatContextManagerImpl;
 class MutableChatRequestModel {
-    constructor(session, message, agentId, context = { variables: [] }, data = {}) {
-        this.message = message;
+    constructor(session, messageOrData, agentIdOrResponseData, context = { variables: [] }, data = {}) {
         this._onDidChangeEmitter = new core_1.Emitter();
         this.onDidChange = this._onDidChangeEmitter.event;
         this._isEditing = false;
         this.toDispose = new core_1.DisposableCollection();
-        // TODO accept serialized data as a parameter to restore a previously saved ChatRequestModel
-        this._request = message.request;
-        this._id = (0, core_1.generateUuid)();
         this._session = session;
-        this._response = new MutableChatResponseModel(this._id, agentId);
-        this._context = context;
-        this._agentId = agentId;
-        this._data = data;
-        this.editContextManager = new ChatContextManagerImpl(context);
+        // Check if we're restoring from serialized data
+        if (this.isSerializedRequestData(messageOrData)) {
+            this.restoreFromSerializedData(messageOrData, agentIdOrResponseData);
+        }
+        else {
+            // Normal creation path
+            this._request = messageOrData.request;
+            this._id = (0, core_1.generateUuid)();
+            this._response = new MutableChatResponseModel(this._id, agentIdOrResponseData);
+            this._context = context;
+            this._agentId = agentIdOrResponseData;
+            this._data = data;
+            // Store the parsed message
+            this.message = messageOrData;
+        }
+        this.editContextManager = new ChatContextManagerImpl(this._context);
         this.editContextManager.onDidChange(this._onDidChangeEmitter.fire, this._onDidChangeEmitter, this.toDispose);
+        // Wire response changes to propagate through request to session
+        this._response.onDidChange(() => {
+            // Fire a generic addVariable event to propagate response changes
+            this._onDidChangeEmitter.fire({ kind: 'responseChanged' });
+        }, this, this.toDispose);
         this.toDispose.push(this._onDidChangeEmitter);
     }
+    /**
+     * Type guard to determine if we're receiving serialized data
+     */
+    isSerializedRequestData(data) {
+        return 'id' in data && 'text' in data && !('request' in data);
+    }
+    /**
+     * Restore this request model from serialized data
+     */
+    restoreFromSerializedData(reqData, respData) {
+        this._id = reqData.id;
+        this._request = { text: reqData.text };
+        this._agentId = reqData.agentId;
+        this._data = {};
+        // Create minimal context
+        this._context = { variables: [] };
+        // TODO: More sophisticated restoration?
+        // Casting required because 'message' is readonly
+        this.message = {
+            request: this._request,
+            parts: [{
+                    kind: 'text',
+                    text: reqData.text,
+                    promptText: reqData.text,
+                    range: { start: 0, endExclusive: reqData.text.length }
+                }],
+            toolRequests: new Map(),
+            variables: []
+        };
+        // Restore response if present
+        if (respData) {
+            this._response = new MutableChatResponseModel(this._id, this._agentId, respData);
+        }
+        else {
+            this._response = new MutableChatResponseModel(this._id, this._agentId);
+        }
+        // Note: ChangeSet restoration will be handled by ChatService using deserializer registry
+    }
     get changeSet() {
         return this._changeSet;
     }
@@ -759,6 +973,17 @@ class MutableChatRequestModel {
     cancel() {
         this.response.cancel();
     }
+    toSerializable() {
+        return {
+            id: this.id,
+            text: this.request.text,
+            agentId: this.agentId,
+            changeSet: this._changeSet ? {
+                title: this._changeSet.title,
+                elements: this._changeSet.getElements().map(elem => { var _a; return (_a = elem.toSerializable) === null || _a === void 0 ? void 0 : _a.call(elem); }).filter((elem) => elem !== undefined)
+            } : undefined
+        };
+    }
     dispose() {
         this.toDispose.dispose();
     }
@@ -812,6 +1037,15 @@ class ErrorChatResponseContentImpl {
     asString() {
         return undefined;
     }
+    toSerializable() {
+        return {
+            kind: 'error',
+            data: {
+                message: this._error.message,
+                stack: this._error.stack
+            }
+        };
+    }
 }
 exports.ErrorChatResponseContentImpl = ErrorChatResponseContentImpl;
 class TextChatResponseContentImpl {
@@ -839,6 +1073,12 @@ class TextChatResponseContentImpl {
             text: this.content
         };
     }
+    toSerializable() {
+        return {
+            kind: 'text',
+            data: { content: this._content }
+        };
+    }
 }
 exports.TextChatResponseContentImpl = TextChatResponseContentImpl;
 class ThinkingChatResponseContentImpl {
@@ -876,6 +1116,15 @@ class ThinkingChatResponseContentImpl {
             signature: this.signature
         };
     }
+    toSerializable() {
+        return {
+            kind: 'thinking',
+            data: {
+                content: this._content,
+                signature: this._signature
+            }
+        };
+    }
 }
 exports.ThinkingChatResponseContentImpl = ThinkingChatResponseContentImpl;
 class MarkdownChatResponseContentImpl {
@@ -904,6 +1153,12 @@ class MarkdownChatResponseContentImpl {
             text: this.content.value
         };
     }
+    toSerializable() {
+        return {
+            kind: 'markdownContent',
+            data: { content: this._content.value }
+        };
+    }
 }
 exports.MarkdownChatResponseContentImpl = MarkdownChatResponseContentImpl;
 class InformationalChatResponseContentImpl {
@@ -921,6 +1176,12 @@ class InformationalChatResponseContentImpl {
         this._content.appendMarkdown(nextChatResponseContent.content.value);
         return true;
     }
+    toSerializable() {
+        return {
+            kind: 'informational',
+            data: { content: this._content.value }
+        };
+    }
 }
 exports.InformationalChatResponseContentImpl = InformationalChatResponseContentImpl;
 class CodeChatResponseContentImpl {
@@ -947,6 +1208,16 @@ class CodeChatResponseContentImpl {
         this._code += `${nextChatResponseContent.code}`;
         return true;
     }
+    toSerializable() {
+        return {
+            kind: 'code',
+            data: {
+                code: this._code,
+                language: this._language,
+                location: this._location
+            }
+        };
+    }
 }
 exports.CodeChatResponseContentImpl = CodeChatResponseContentImpl;
 class ToolCallChatResponseContentImpl {
@@ -1058,6 +1329,18 @@ class ToolCallChatResponseContentImpl {
                 name: (_d = this.name) !== null && _d !== void 0 ? _d : ''
             }];
     }
+    toSerializable() {
+        return {
+            kind: 'toolCall',
+            data: {
+                id: this._id,
+                name: this._name,
+                arguments: this._arguments,
+                finished: this._finished,
+                result: this._result
+            }
+        };
+    }
 }
 exports.ToolCallChatResponseContentImpl = ToolCallChatResponseContentImpl;
 exports.COMMAND_CHAT_RESPONSE_COMMAND = {
@@ -1078,6 +1361,17 @@ class CommandChatResponseContentImpl {
         var _a, _b;
         return ((_a = this.command) === null || _a === void 0 ? void 0 : _a.id) || ((_b = this.customCallback) === null || _b === void 0 ? void 0 : _b.label) || 'command';
     }
+    toSerializable() {
+        var _a, _b;
+        return {
+            kind: 'command',
+            data: {
+                commandId: (_a = this.command) === null || _a === void 0 ? void 0 : _a.id,
+                commandLabel: (_b = this.customCallback) === null || _b === void 0 ? void 0 : _b.label,
+                arguments: this.args
+            }
+        };
+    }
 }
 exports.CommandChatResponseContentImpl = CommandChatResponseContentImpl;
 class HorizontalLayoutChatResponseContentImpl {
@@ -1103,22 +1397,52 @@ class HorizontalLayoutChatResponseContentImpl {
         }
         return true;
     }
+    toSerializable() {
+        return {
+            kind: 'horizontal',
+            data: {
+                content: this._content.map(child => {
+                    var _a, _b, _c;
+                    const serialized = (_a = child.toSerializable) === null || _a === void 0 ? void 0 : _a.call(child);
+                    if (!serialized) {
+                        return {
+                            kind: child.kind,
+                            fallbackMessage: (_b = child.asString) === null || _b === void 0 ? void 0 : _b.call(child),
+                            data: undefined
+                        };
+                    }
+                    return {
+                        ...serialized,
+                        fallbackMessage: (_c = child.asString) === null || _c === void 0 ? void 0 : _c.call(child)
+                    };
+                })
+            }
+        };
+    }
 }
 exports.HorizontalLayoutChatResponseContentImpl = HorizontalLayoutChatResponseContentImpl;
 /**
  * Default implementation for the QuestionResponseContent.
+ * Can be created with or without handler/request for read-only (restored) mode.
  */
 class QuestionResponseContentImpl {
-    constructor(question, options, request, handler) {
+    constructor(question, options, request, handler, selectedOption) {
         this.question = question;
         this.options = options;
         this.request = request;
         this.handler = handler;
         this.kind = 'question';
+        this._selectedOption = selectedOption;
+    }
+    get isReadOnly() {
+        return !this.handler || !this.request;
     }
     set selectedOption(option) {
         this._selectedOption = option;
-        this.request.response.response.responseContentChanged();
+        // Only trigger change notification if request is available (not in read-only mode)
+        if (this.request) {
+            this.request.response.response.responseContentChanged();
+        }
     }
     get selectedOption() {
         return this._selectedOption;
@@ -1131,13 +1455,22 @@ ${this.selectedOption ? `Answer: ${(_a = this.selectedOption) === null || _a ===
     merge() {
         return false;
     }
+    toSerializable() {
+        return {
+            kind: 'question',
+            data: {
+                question: this.question,
+                options: this.options,
+                selectedOption: this._selectedOption
+            }
+        };
+    }
 }
 exports.QuestionResponseContentImpl = QuestionResponseContentImpl;
 class ChatResponseImpl {
     constructor() {
         this._onDidChangeEmitter = new core_1.Emitter();
         this.onDidChange = this._onDidChangeEmitter.event;
-        // TODO accept serialized data as a parameter to restore a previously saved ChatResponse
         this._content = [];
     }
     get content() {
@@ -1222,21 +1555,46 @@ class ChatResponseImpl {
     }
 }
 class MutableChatResponseModel {
-    constructor(requestId, agentId) {
+    constructor(requestId, agentId, serializedData) {
         this._onDidChangeEmitter = new core_1.Emitter();
         this.onDidChange = this._onDidChangeEmitter.event;
         this.data = {};
-        // TODO accept serialized data as a parameter to restore a previously saved ChatResponseModel
         this._requestId = requestId;
-        this._id = (0, core_1.generateUuid)();
-        this._progressMessages = [];
+        this._agentId = agentId;
+        this._cancellationToken = new core_1.CancellationTokenSource();
+        // Check if we're restoring from serialized data
+        if (serializedData) {
+            this.restoreFromSerializedData(serializedData);
+        }
+        else {
+            // Normal creation path
+            this._id = (0, core_1.generateUuid)();
+            this._progressMessages = [];
+            this._isComplete = false;
+            this._isWaitingForInput = false;
+            this._isError = false;
+        }
         const response = new ChatResponseImpl();
         response.onDidChange(() => this._onDidChangeEmitter.fire());
         this._response = response;
-        this._isComplete = false;
+    }
+    /**
+     * Restore this response model from serialized data
+     */
+    restoreFromSerializedData(data) {
+        this._id = data.id;
+        // Always mark restored responses as complete since there's no active agent
+        this._isComplete = true;
+        this._isError = data.isError;
+        // Do not restore waitingForInput state - when a session is restored,
+        // the agent that was waiting for input is no longer running
         this._isWaitingForInput = false;
-        this._agentId = agentId;
-        this._cancellationToken = new core_1.CancellationTokenSource();
+        // TODO: Restore progressMessages?
+        this._progressMessages = [];
+        if (data.errorMessage) {
+            this._errorObject = new Error(data.errorMessage);
+        }
+        // Note: Content restoration will be handled by ChatService using deserializer registry
     }
     get id() {
         return this._id;
@@ -1341,6 +1699,32 @@ class MutableChatResponseModel {
     get isError() {
         return this._isError;
     }
+    toSerializable() {
+        var _a;
+        return {
+            id: this.id,
+            requestId: this.requestId,
+            isComplete: this.isComplete,
+            isError: this.isError,
+            errorMessage: (_a = this.errorObject) === null || _a === void 0 ? void 0 : _a.message,
+            content: this.response.content.map(c => {
+                var _a, _b, _c;
+                const serialized = (_a = c.toSerializable) === null || _a === void 0 ? void 0 : _a.call(c);
+                if (!serialized) {
+                    // Fallback if toSerializable not implemented
+                    return {
+                        kind: c.kind,
+                        fallbackMessage: (_b = c.asString) === null || _b === void 0 ? void 0 : _b.call(c),
+                        data: undefined
+                    };
+                }
+                return {
+                    ...serialized,
+                    fallbackMessage: (_c = c.asString) === null || _c === void 0 ? void 0 : _c.call(c)
+                };
+            })
+        };
+    }
 }
 exports.MutableChatResponseModel = MutableChatResponseModel;
 class ErrorChatResponseModel extends MutableChatResponseModel {
@@ -1378,6 +1762,30 @@ class ProgressChatResponseContentImpl {
             text: this.message
         };
     }
+    toSerializable() {
+        return {
+            kind: 'progress',
+            data: { message: this._message }
+        };
+    }
 }
 exports.ProgressChatResponseContentImpl = ProgressChatResponseContentImpl;
+class UnknownChatResponseContentImpl {
+    constructor(originalKind, fallbackMessage, data) {
+        this.originalKind = originalKind;
+        this.fallbackMessage = fallbackMessage;
+        this.data = data;
+        this.kind = 'unknown';
+    }
+    asString() {
+        return this.fallbackMessage;
+    }
+    toSerializable() {
+        return {
+            kind: 'unknown',
+            data: this.data
+        };
+    }
+}
+exports.UnknownChatResponseContentImpl = UnknownChatResponseContentImpl;
 //# sourceMappingURL=chat-model.js.map