@codingame/monaco-vscode-api 13.1.6 → 14.0.0

package/vscode/src/vs/workbench/contrib/chat/common/chatModel.js

@@ -0,0 +1,991 @@
+
+import { __decorate, __param } from '../../../../../../../external/tslib/tslib.es6.js';
+import { asArray } from '../../../../base/common/arrays.js';
+import { DeferredPromise } from '../../../../base/common/async.js';
+import { Emitter } from '../../../../base/common/event.js';
+import { isMarkdownString, MarkdownString } from '../../../../base/common/htmlContent.js';
+import { Disposable } from '../../../../base/common/lifecycle.js';
+import { revive } from '../../../../base/common/marshalling.js';
+import { Schemas } from '../../../../base/common/network.js';
+import { equals } from '../../../../base/common/objects.js';
+import { isEqual, basename } from '../../../../base/common/resources.js';
+import { URI, isUriComponents } from '../../../../base/common/uri.js';
+import { generateUuid } from '../../../../base/common/uuid.js';
+import { OffsetRange } from '../../../../editor/common/core/offsetRange.js';
+import { localize } from '../../../../nls.js';
+import { ILogService } from '../../../../platform/log/common/log.service.js';
+import { ChatAgentLocation, reviveSerializedAgent } from './chatAgents.js';
+import { IChatAgentService } from './chatAgents.service.js';
+import { reviveParsedChatRequest, ChatRequestTextPart } from './chatParserTypes.js';
+import { isIUsedContext } from './chatService.js';
+
+var ChatModel_1;
+function isPasteVariableEntry(obj) {
+    return obj.kind === 'paste';
+}
+function isLinkVariableEntry(obj) {
+    return obj.kind === 'link';
+}
+const nonHistoryKinds = ( new Set(['toolInvocation', 'toolInvocationSerialized']));
+function isChatProgressHistoryResponseContent(content) {
+    return !( nonHistoryKinds.has(content.kind));
+}
+function toChatHistoryContent(content) {
+    return content.filter(isChatProgressHistoryResponseContent);
+}
+class ChatRequestModel {
+    get session() {
+        return this._session;
+    }
+    get username() {
+        return this.session.requesterUsername;
+    }
+    get avatarIconUri() {
+        return this.session.requesterAvatarIconUri;
+    }
+    get attempt() {
+        return this._attempt;
+    }
+    get variableData() {
+        return this._variableData;
+    }
+    set variableData(v) {
+        this._variableData = v;
+    }
+    get confirmation() {
+        return this._confirmation;
+    }
+    get locationData() {
+        return this._locationData;
+    }
+    get attachedContext() {
+        return this._attachedContext;
+    }
+    get workingSet() {
+        return this._workingSet;
+    }
+    get paused() {
+        return this._paused;
+    }
+    set paused(paused) {
+        this._paused = paused;
+    }
+    constructor(_session, message, _variableData, timestamp, _attempt = 0, _confirmation, _locationData, _attachedContext, _workingSet, isCompleteAddedRequest = false, restoredId) {
+        this._session = _session;
+        this.message = message;
+        this._variableData = _variableData;
+        this.timestamp = timestamp;
+        this._attempt = _attempt;
+        this._confirmation = _confirmation;
+        this._locationData = _locationData;
+        this._attachedContext = _attachedContext;
+        this._workingSet = _workingSet;
+        this.isCompleteAddedRequest = isCompleteAddedRequest;
+        this.shouldBeRemovedOnSend = false;
+        this._paused = false;
+        this.id = restoredId ?? 'request_' + generateUuid();
+    }
+    adoptTo(session) {
+        this._session = session;
+    }
+}
+class Response extends Disposable {
+    get onDidChangeValue() {
+        return this._onDidChangeValue.event;
+    }
+    get value() {
+        return this._responseParts;
+    }
+    constructor(value) {
+        super();
+        this._onDidChangeValue = this._register(( new Emitter()));
+        this._responseRepr = '';
+        this._markdownContent = '';
+        this._citations = [];
+        this._responseParts = ( asArray(value).map((v) => (isMarkdownString(v) ?
+            { content: v, kind: 'markdownContent' } :
+            'kind' in v ? v : { kind: 'treeData', treeData: v })));
+        this._updateRepr(true);
+    }
+    toString() {
+        return this._responseRepr;
+    }
+    getMarkdown() {
+        return this._markdownContent;
+    }
+    clear() {
+        this._responseParts = [];
+        this._updateRepr(true);
+    }
+    updateContent(progress, quiet) {
+        if (progress.kind === 'markdownContent') {
+            const lastResponsePart = this._responseParts
+                .filter(p => p.kind !== 'textEditGroup')
+                .at(-1);
+            if (!lastResponsePart || lastResponsePart.kind !== 'markdownContent' || !canMergeMarkdownStrings(lastResponsePart.content, progress.content)) {
+                this._responseParts.push(progress);
+            }
+            else {
+                const idx = this._responseParts.indexOf(lastResponsePart);
+                this._responseParts[idx] = { ...lastResponsePart, content: appendMarkdownString(lastResponsePart.content, progress.content) };
+            }
+            this._updateRepr(quiet);
+        }
+        else if (progress.kind === 'textEdit') {
+            let found = false;
+            for (let i = 0; !found && i < this._responseParts.length; i++) {
+                const candidate = this._responseParts[i];
+                if (candidate.kind === 'textEditGroup' && isEqual(candidate.uri, progress.uri)) {
+                    candidate.edits.push(progress.edits);
+                    candidate.done = progress.done;
+                    found = true;
+                }
+            }
+            if (!found) {
+                this._responseParts.push({
+                    kind: 'textEditGroup',
+                    uri: progress.uri,
+                    edits: [progress.edits],
+                    done: progress.done
+                });
+            }
+            this._updateRepr(quiet);
+        }
+        else if (progress.kind === 'progressTask') {
+            const responsePosition = this._responseParts.push(progress) - 1;
+            this._updateRepr(quiet);
+            const disp = progress.onDidAddProgress(() => {
+                this._updateRepr(false);
+            });
+            progress.task?.().then((content) => {
+                disp.dispose();
+                if (typeof content === 'string') {
+                    this._responseParts[responsePosition].content = ( new MarkdownString(content));
+                }
+                this._updateRepr(false);
+            });
+        }
+        else if (progress.kind === 'toolInvocation') {
+            if (progress.confirmationMessages) {
+                progress.confirmed.p.then(() => {
+                    this._updateRepr(false);
+                });
+            }
+            this._responseParts.push(progress);
+            this._updateRepr(quiet);
+        }
+        else {
+            this._responseParts.push(progress);
+            this._updateRepr(quiet);
+        }
+    }
+    addCitation(citation) {
+        this._citations.push(citation);
+        this._updateRepr();
+    }
+    _updateRepr(quiet) {
+        this._responseRepr = this.partsToRepr(this._responseParts);
+        this._responseRepr += this._citations.length ? '\n\n' + getCodeCitationsMessage(this._citations) : '';
+        this._markdownContent = ( this._responseParts.map(part => {
+            if (part.kind === 'inlineReference') {
+                return this.inlineRefToRepr(part);
+            }
+            else if (part.kind === 'markdownContent' || part.kind === 'markdownVuln') {
+                return part.content.value;
+            }
+            else {
+                return '';
+            }
+        }))
+            .filter(s => s.length > 0)
+            .join('');
+        if (!quiet) {
+            this._onDidChangeValue.fire();
+        }
+    }
+    partsToRepr(parts) {
+        const blocks = [];
+        let currentBlockSegments = [];
+        for (const part of parts) {
+            let segment;
+            switch (part.kind) {
+                case 'treeData':
+                case 'progressMessage':
+                case 'codeblockUri':
+                case 'toolInvocation':
+                case 'toolInvocationSerialized':
+                    continue;
+                case 'inlineReference':
+                    segment = { text: this.inlineRefToRepr(part) };
+                    break;
+                case 'command':
+                    segment = { text: part.command.title, isBlock: true };
+                    break;
+                case 'textEditGroup':
+                    segment = { text: ( localize(4754, "Made changes.")), isBlock: true };
+                    break;
+                case 'confirmation':
+                    segment = { text: `${part.title}\n${part.message}`, isBlock: true };
+                    break;
+                default:
+                    segment = { text: part.content.value };
+                    break;
+            }
+            if (segment.isBlock) {
+                if (currentBlockSegments.length) {
+                    blocks.push(currentBlockSegments.join(''));
+                    currentBlockSegments = [];
+                }
+                blocks.push(segment.text);
+            }
+            else {
+                currentBlockSegments.push(segment.text);
+            }
+        }
+        if (currentBlockSegments.length) {
+            blocks.push(currentBlockSegments.join(''));
+        }
+        return blocks.join('\n\n');
+    }
+    inlineRefToRepr(part) {
+        if ('uri' in part.inlineReference) {
+            return this.uriToRepr(part.inlineReference.uri);
+        }
+        return 'name' in part.inlineReference
+            ? '`' + part.inlineReference.name + '`'
+            : this.uriToRepr(part.inlineReference);
+    }
+    uriToRepr(uri) {
+        if (uri.scheme === Schemas.http || uri.scheme === Schemas.https) {
+            return ( uri.toString(false));
+        }
+        return basename(uri);
+    }
+}
+class ChatResponseModel extends Disposable {
+    get session() {
+        return this._session;
+    }
+    get shouldBeRemovedOnSend() {
+        return this._shouldBeRemovedOnSend;
+    }
+    get isComplete() {
+        return this._isComplete;
+    }
+    set shouldBeRemovedOnSend(hidden) {
+        this._shouldBeRemovedOnSend = hidden;
+        this._onDidChange.fire();
+    }
+    get isCanceled() {
+        return this._isCanceled;
+    }
+    get vote() {
+        return this._vote;
+    }
+    get voteDownReason() {
+        return this._voteDownReason;
+    }
+    get followups() {
+        return this._followups;
+    }
+    get response() {
+        return this._response;
+    }
+    get result() {
+        return this._result;
+    }
+    get username() {
+        return this.session.responderUsername;
+    }
+    get avatarIcon() {
+        return this.session.responderAvatarIcon;
+    }
+    get agent() {
+        return this._agent;
+    }
+    get slashCommand() {
+        return this._slashCommand;
+    }
+    get agentOrSlashCommandDetected() {
+        return this._agentOrSlashCommandDetected ?? false;
+    }
+    get usedContext() {
+        return this._usedContext;
+    }
+    get contentReferences() {
+        return Array.from(this._contentReferences);
+    }
+    get codeCitations() {
+        return this._codeCitations;
+    }
+    get progressMessages() {
+        return this._progressMessages;
+    }
+    get isStale() {
+        return this._isStale;
+    }
+    get isPendingConfirmation() {
+        return ( this._response.value.some(part => part.kind === 'toolInvocation' && part.isConfirmed === undefined
+            || part.kind === 'confirmation' && part.isUsed === false));
+    }
+    constructor(_response, _session, _agent, _slashCommand, requestId, _isComplete = false, _isCanceled = false, _vote, _voteDownReason, _result, followups, isCompleteAddedRequest = false, _shouldBeRemovedOnSend = false, restoredId) {
+        super();
+        this._session = _session;
+        this._agent = _agent;
+        this._slashCommand = _slashCommand;
+        this.requestId = requestId;
+        this._isComplete = _isComplete;
+        this._isCanceled = _isCanceled;
+        this._vote = _vote;
+        this._voteDownReason = _voteDownReason;
+        this._result = _result;
+        this.isCompleteAddedRequest = isCompleteAddedRequest;
+        this._shouldBeRemovedOnSend = _shouldBeRemovedOnSend;
+        this._onDidChange = this._register(( new Emitter()));
+        this.onDidChange = this._onDidChange.event;
+        this._contentReferences = [];
+        this._codeCitations = [];
+        this._progressMessages = [];
+        this._isStale = false;
+        this._isStale = Array.isArray(_response) && (_response.length !== 0 || isMarkdownString(_response) && _response.value.length !== 0);
+        this._followups = followups ? [...followups] : undefined;
+        this._response = this._register(( new Response(_response)));
+        this._register(this._response.onDidChangeValue(() => this._onDidChange.fire()));
+        this.id = restoredId ?? 'response_' + generateUuid();
+    }
+    updateContent(responsePart, quiet) {
+        this._response.updateContent(responsePart, quiet);
+    }
+    applyReference(progress) {
+        if (progress.kind === 'usedContext') {
+            this._usedContext = progress;
+        }
+        else if (progress.kind === 'reference') {
+            this._contentReferences.push(progress);
+            this._onDidChange.fire();
+        }
+    }
+    applyCodeCitation(progress) {
+        this._codeCitations.push(progress);
+        this._response.addCitation(progress);
+        this._onDidChange.fire();
+    }
+    setAgent(agent, slashCommand) {
+        this._agent = agent;
+        this._slashCommand = slashCommand;
+        this._agentOrSlashCommandDetected = !agent.isDefault || !!slashCommand;
+        this._onDidChange.fire();
+    }
+    setResult(result) {
+        this._result = result;
+        this._onDidChange.fire();
+    }
+    complete() {
+        if (this._result?.errorDetails?.responseIsRedacted) {
+            this._response.clear();
+        }
+        this._isComplete = true;
+        this._onDidChange.fire();
+    }
+    cancel() {
+        this._isComplete = true;
+        this._isCanceled = true;
+        this._onDidChange.fire();
+    }
+    setFollowups(followups) {
+        this._followups = followups;
+        this._onDidChange.fire();
+    }
+    setVote(vote) {
+        this._vote = vote;
+        this._onDidChange.fire();
+    }
+    setVoteDownReason(reason) {
+        this._voteDownReason = reason;
+        this._onDidChange.fire();
+    }
+    setEditApplied(edit, editCount) {
+        if (!this.response.value.includes(edit)) {
+            return false;
+        }
+        if (!edit.state) {
+            return false;
+        }
+        edit.state.applied = editCount;
+        this._onDidChange.fire();
+        return true;
+    }
+    adoptTo(session) {
+        this._session = session;
+        this._onDidChange.fire();
+    }
+}
+var ChatPauseState;
+(function (ChatPauseState) {
+    ChatPauseState[ChatPauseState["NotPausable"] = 0] = "NotPausable";
+    ChatPauseState[ChatPauseState["Paused"] = 1] = "Paused";
+    ChatPauseState[ChatPauseState["Unpaused"] = 2] = "Unpaused";
+})(ChatPauseState || (ChatPauseState = {}));
+function normalizeSerializableChatData(raw) {
+    normalizeOldFields(raw);
+    if (!('version' in raw)) {
+        return {
+            version: 3,
+            ...raw,
+            lastMessageDate: raw.creationDate,
+            customTitle: undefined,
+        };
+    }
+    if (raw.version === 2) {
+        return {
+            ...raw,
+            version: 3,
+            customTitle: raw.computedTitle
+        };
+    }
+    return raw;
+}
+function normalizeOldFields(raw) {
+    if (!raw.sessionId) {
+        raw.sessionId = generateUuid();
+    }
+    if (!raw.creationDate) {
+        raw.creationDate = getLastYearDate();
+    }
+    if ('version' in raw && (raw.version === 2 || raw.version === 3)) {
+        if (!raw.lastMessageDate) {
+            raw.lastMessageDate = getLastYearDate();
+        }
+    }
+}
+function getLastYearDate() {
+    const lastYearDate = ( new Date());
+    lastYearDate.setFullYear(lastYearDate.getFullYear() - 1);
+    return lastYearDate.getTime();
+}
+function isExportableSessionData(obj) {
+    const data = obj;
+    return typeof data === 'object' &&
+        typeof data.requesterUsername === 'string';
+}
+function isSerializableSessionData(obj) {
+    const data = obj;
+    return isExportableSessionData(obj) &&
+        typeof data.creationDate === 'number' &&
+        typeof data.sessionId === 'string' &&
+        obj.requests.every((request) => !request.usedContext  || isIUsedContext(request.usedContext));
+}
+var ChatRequestRemovalReason;
+(function (ChatRequestRemovalReason) {
+    ChatRequestRemovalReason[ChatRequestRemovalReason["Removal"] = 0] = "Removal";
+    ChatRequestRemovalReason[ChatRequestRemovalReason["Resend"] = 1] = "Resend";
+    ChatRequestRemovalReason[ChatRequestRemovalReason["Adoption"] = 2] = "Adoption";
+})(ChatRequestRemovalReason || (ChatRequestRemovalReason = {}));
+var ChatModelInitState;
+(function (ChatModelInitState) {
+    ChatModelInitState[ChatModelInitState["Created"] = 0] = "Created";
+    ChatModelInitState[ChatModelInitState["Initializing"] = 1] = "Initializing";
+    ChatModelInitState[ChatModelInitState["Initialized"] = 2] = "Initialized";
+})(ChatModelInitState || (ChatModelInitState = {}));
+let ChatModel = ChatModel_1 = class ChatModel extends Disposable {
+    static getDefaultTitle(requests) {
+        const firstRequestMessage = requests.at(0)?.message ?? '';
+        const message = typeof firstRequestMessage === 'string' ?
+            firstRequestMessage :
+            firstRequestMessage.text;
+        return message.split('\n')[0].substring(0, 50);
+    }
+    get welcomeMessage() {
+        return this._welcomeMessage;
+    }
+    get sampleQuestions() {
+        return this._sampleQuestions;
+    }
+    get sessionId() {
+        return this._sessionId;
+    }
+    get requestInProgress() {
+        const lastRequest = this.lastRequest;
+        if (!lastRequest?.response) {
+            return false;
+        }
+        if (lastRequest.response.isPendingConfirmation) {
+            return false;
+        }
+        return !lastRequest.response.isComplete;
+    }
+    get requestPausibility() {
+        const lastRequest = this.lastRequest;
+        if (!lastRequest?.response?.agent || lastRequest.response.isComplete || lastRequest.response.isPendingConfirmation) {
+            return ChatPauseState.NotPausable;
+        }
+        return lastRequest.paused ? ChatPauseState.Paused : ChatPauseState.Unpaused;
+    }
+    get hasRequests() {
+        return this._requests.length > 0;
+    }
+    get lastRequest() {
+        return this._requests.at(-1);
+    }
+    get creationDate() {
+        return this._creationDate;
+    }
+    get lastMessageDate() {
+        return this._lastMessageDate;
+    }
+    get _defaultAgent() {
+        return this.chatAgentService.getDefaultAgent(ChatAgentLocation.Panel);
+    }
+    get requesterUsername() {
+        return this._defaultAgent?.metadata.requester?.name ??
+            this.initialData?.requesterUsername ?? '';
+    }
+    get responderUsername() {
+        return this._defaultAgent?.fullName ??
+            this.initialData?.responderUsername ?? '';
+    }
+    get requesterAvatarIconUri() {
+        return this._defaultAgent?.metadata.requester?.icon ??
+            this._initialRequesterAvatarIconUri;
+    }
+    get responderAvatarIcon() {
+        return this._defaultAgent?.metadata.themeIcon ??
+            this._initialResponderAvatarIconUri;
+    }
+    get initState() {
+        return this._initState;
+    }
+    get isImported() {
+        return this._isImported;
+    }
+    get customTitle() {
+        return this._customTitle;
+    }
+    get title() {
+        return this._customTitle || ChatModel_1.getDefaultTitle(this._requests);
+    }
+    get initialLocation() {
+        return this._initialLocation;
+    }
+    constructor(initialData, _initialLocation, logService, chatAgentService) {
+        super();
+        this.initialData = initialData;
+        this._initialLocation = _initialLocation;
+        this.logService = logService;
+        this.chatAgentService = chatAgentService;
+        this._onDidDispose = this._register(( new Emitter()));
+        this.onDidDispose = this._onDidDispose.event;
+        this._onDidChange = this._register(( new Emitter()));
+        this.onDidChange = this._onDidChange.event;
+        this._initState = ChatModelInitState.Created;
+        this._isInitializedDeferred = ( new DeferredPromise());
+        this._isImported = false;
+        this._checkpoint = undefined;
+        const isValid = isSerializableSessionData(initialData);
+        if (initialData && !isValid) {
+            this.logService.warn(`ChatModel#constructor: Loaded malformed session data: ${JSON.stringify(initialData)}`);
+        }
+        this._isImported = (!!initialData && !isValid) || (initialData?.isImported ?? false);
+        this._sessionId = (isValid && initialData.sessionId) || generateUuid();
+        this._requests = initialData ? this._deserialize(initialData) : [];
+        this._creationDate = (isValid && initialData.creationDate) || Date.now();
+        this._lastMessageDate = (isValid && initialData.lastMessageDate) || this._creationDate;
+        this._customTitle = isValid ? initialData.customTitle : undefined;
+        this._initialRequesterAvatarIconUri = initialData?.requesterAvatarIconUri && URI.revive(initialData.requesterAvatarIconUri);
+        this._initialResponderAvatarIconUri = isUriComponents(initialData?.responderAvatarIconUri) ? URI.revive(initialData.responderAvatarIconUri) : initialData?.responderAvatarIconUri;
+    }
+    _deserialize(obj) {
+        const requests = obj.requests;
+        if (!Array.isArray(requests)) {
+            this.logService.error(`Ignoring malformed session data: ${JSON.stringify(obj)}`);
+            return [];
+        }
+        try {
+            return ( requests.map((raw) => {
+                const parsedRequest = typeof raw.message === 'string'
+                    ? this.getParsedRequestFromString(raw.message)
+                    : reviveParsedChatRequest(raw.message);
+                const variableData = this.reviveVariableData(raw.variableData);
+                const request = ( new ChatRequestModel(
+                    this,
+                    parsedRequest,
+                    variableData,
+                    raw.timestamp ?? -1,
+                    undefined,
+                    undefined,
+                    undefined,
+                    undefined,
+                    raw.workingSet?.map((uri) => URI.revive(uri)),
+                    undefined,
+                    raw.requestId
+                ));
+                request.shouldBeRemovedOnSend = !!raw.isHidden;
+                if (raw.response || raw.result || raw.responseErrorDetails) {
+                    const agent = (raw.agent && 'metadata' in raw.agent) ?
+                        reviveSerializedAgent(raw.agent) : undefined;
+                    const result = 'responseErrorDetails' in raw ?
+                        { errorDetails: raw.responseErrorDetails } : raw.result;
+                    request.response = ( new ChatResponseModel(raw.response ?? [( new MarkdownString(raw.response))], this, agent, raw.slashCommand, request.id, true, raw.isCanceled, raw.vote, raw.voteDownReason, result, raw.followups, undefined, undefined, raw.responseId));
+                    request.response.shouldBeRemovedOnSend = !!raw.isHidden;
+                    if (raw.usedContext) {
+                        request.response.applyReference(revive(raw.usedContext));
+                    }
+                    raw.contentReferences?.forEach(r => request.response.applyReference(revive(r)));
+                    raw.codeCitations?.forEach(c => request.response.applyCodeCitation(revive(c)));
+                }
+                return request;
+            }));
+        }
+        catch (error) {
+            this.logService.error('Failed to parse chat data', error);
+            return [];
+        }
+    }
+    reviveVariableData(raw) {
+        const variableData = raw && Array.isArray(raw.variables)
+            ? raw :
+            { variables: [] };
+        variableData.variables = ( variableData.variables.map((v) => {
+            if (v && 'values' in v && Array.isArray(v.values)) {
+                return {
+                    id: v.id ?? '',
+                    name: v.name,
+                    value: v.values[0]?.value,
+                    range: v.range,
+                    modelDescription: v.modelDescription,
+                    references: v.references
+                };
+            }
+            else {
+                return v;
+            }
+        }));
+        return variableData;
+    }
+    getParsedRequestFromString(message) {
+        const parts = [( new ChatRequestTextPart(( new OffsetRange(0, message.length)), { startColumn: 1, startLineNumber: 1, endColumn: 1, endLineNumber: 1 }, message))];
+        return {
+            text: message,
+            parts
+        };
+    }
+    toggleLastRequestPaused(isPaused) {
+        if (this.requestPausibility !== ChatPauseState.NotPausable && this.lastRequest?.response?.agent) {
+            const pausedValue = isPaused ?? !this.lastRequest.paused;
+            this.lastRequest.paused = pausedValue;
+            this.chatAgentService.setRequestPaused(this.lastRequest.response.agent.id, this.lastRequest.id, pausedValue);
+            this._onDidChange.fire({ kind: 'changedRequest', request: this.lastRequest });
+        }
+    }
+    startInitialize() {
+        if (this.initState !== ChatModelInitState.Created) {
+            throw ( new Error(
+                `ChatModel is in the wrong state for startInitialize: ${ChatModelInitState[this.initState]}`
+            ));
+        }
+        this._initState = ChatModelInitState.Initializing;
+    }
+    deinitialize() {
+        this._initState = ChatModelInitState.Created;
+        this._isInitializedDeferred = ( new DeferredPromise());
+    }
+    initialize(welcomeMessage, sampleQuestions) {
+        if (this.initState !== ChatModelInitState.Initializing) {
+            throw ( new Error(
+                `ChatModel is in the wrong state for initialize: ${ChatModelInitState[this.initState]}`
+            ));
+        }
+        this._initState = ChatModelInitState.Initialized;
+        this._welcomeMessage = welcomeMessage;
+        this._sampleQuestions = sampleQuestions;
+        this._isInitializedDeferred.complete();
+        this._onDidChange.fire({ kind: 'initialize' });
+    }
+    setInitializationError(error) {
+        if (this.initState !== ChatModelInitState.Initializing) {
+            throw ( new Error(
+                `ChatModel is in the wrong state for setInitializationError: ${ChatModelInitState[this.initState]}`
+            ));
+        }
+        if (!this._isInitializedDeferred.isSettled) {
+            this._isInitializedDeferred.error(error);
+        }
+    }
+    waitForInitialization() {
+        return this._isInitializedDeferred.p;
+    }
+    getRequests() {
+        return this._requests;
+    }
+    get checkpoint() {
+        return this._checkpoint;
+    }
+    disableRequests(requestIds) {
+        const toHide = ( new Set(requestIds));
+        this._requests.forEach((request) => {
+            const shouldBeRemovedOnSend = ( toHide.has(request.id));
+            request.shouldBeRemovedOnSend = shouldBeRemovedOnSend;
+            if (request.response) {
+                request.response.shouldBeRemovedOnSend = shouldBeRemovedOnSend;
+            }
+        });
+        this._onDidChange.fire({
+            kind: 'setHidden',
+            hiddenRequestIds: ( new Set(requestIds)),
+        });
+    }
+    addRequest(message, variableData, attempt, chatAgent, slashCommand, confirmation, locationData, attachments, workingSet, isCompleteAddedRequest) {
+        const request = ( new ChatRequestModel(
+            this,
+            message,
+            variableData,
+            Date.now(),
+            attempt,
+            confirmation,
+            locationData,
+            attachments,
+            workingSet,
+            isCompleteAddedRequest
+        ));
+        request.response = ( new ChatResponseModel(
+            [],
+            this,
+            chatAgent,
+            slashCommand,
+            request.id,
+            undefined,
+            undefined,
+            undefined,
+            undefined,
+            undefined,
+            undefined,
+            isCompleteAddedRequest
+        ));
+        this._requests.push(request);
+        this._lastMessageDate = Date.now();
+        this._onDidChange.fire({ kind: 'addRequest', request });
+        return request;
+    }
+    setCustomTitle(title) {
+        this._customTitle = title;
+    }
+    setRequestPaused(request, isPaused) {
+        request.paused = isPaused;
+        this._onDidChange.fire({ kind: 'changedRequest', request });
+    }
+    updateRequest(request, variableData) {
+        request.variableData = variableData;
+        this._onDidChange.fire({ kind: 'changedRequest', request });
+    }
+    adoptRequest(request) {
+        const oldOwner = request.session;
+        const index = oldOwner._requests.findIndex(candidate => candidate.id === request.id);
+        if (index === -1) {
+            return;
+        }
+        oldOwner._requests.splice(index, 1);
+        request.adoptTo(this);
+        request.response?.adoptTo(this);
+        this._requests.push(request);
+        oldOwner._onDidChange.fire({ kind: 'removeRequest', requestId: request.id, responseId: request.response?.id, reason: ChatRequestRemovalReason.Adoption });
+        this._onDidChange.fire({ kind: 'addRequest', request });
+    }
+    acceptResponseProgress(request, progress, quiet) {
+        if (!request.response) {
+            request.response = ( new ChatResponseModel([], this, undefined, undefined, request.id));
+        }
+        if (request.response.isComplete) {
+            throw ( new Error('acceptResponseProgress: Adding progress to a completed response'));
+        }
+        if (progress.kind === 'markdownContent' ||
+            progress.kind === 'treeData' ||
+            progress.kind === 'inlineReference' ||
+            progress.kind === 'codeblockUri' ||
+            progress.kind === 'markdownVuln' ||
+            progress.kind === 'progressMessage' ||
+            progress.kind === 'command' ||
+            progress.kind === 'textEdit' ||
+            progress.kind === 'warning' ||
+            progress.kind === 'progressTask' ||
+            progress.kind === 'confirmation' ||
+            progress.kind === 'toolInvocation') {
+            request.response.updateContent(progress, quiet);
+        }
+        else if (progress.kind === 'usedContext' || progress.kind === 'reference') {
+            request.response.applyReference(progress);
+        }
+        else if (progress.kind === 'agentDetection') {
+            const agent = this.chatAgentService.getAgent(progress.agentId);
+            if (agent) {
+                request.response.setAgent(agent, progress.command);
+                this._onDidChange.fire({ kind: 'setAgent', agent, command: progress.command });
+            }
+        }
+        else if (progress.kind === 'codeCitation') {
+            request.response.applyCodeCitation(progress);
+        }
+        else if (progress.kind === 'move') {
+            this._onDidChange.fire({ kind: 'move', target: progress.uri, range: progress.range });
+        }
+        else {
+            this.logService.error(`Couldn't handle progress: ${JSON.stringify(progress)}`);
+        }
+    }
+    removeRequest(id, reason = ChatRequestRemovalReason.Removal) {
+        const index = this._requests.findIndex(request => request.id === id);
+        const request = this._requests[index];
+        if (index !== -1) {
+            this._onDidChange.fire({ kind: 'removeRequest', requestId: request.id, responseId: request.response?.id, reason });
+            this._requests.splice(index, 1);
+            request.response?.dispose();
+        }
+    }
+    cancelRequest(request) {
+        if (request.response) {
+            request.response.cancel();
+        }
+    }
+    setResponse(request, result) {
+        if (!request.response) {
+            request.response = ( new ChatResponseModel([], this, undefined, undefined, request.id));
+        }
+        request.response.setResult(result);
+    }
+    completeResponse(request) {
+        if (!request.response) {
+            throw ( new Error('Call setResponse before completeResponse'));
+        }
+        request.response.complete();
+    }
+    setFollowups(request, followups) {
+        if (!request.response) {
+            return;
+        }
+        request.response.setFollowups(followups);
+    }
+    setResponseModel(request, response) {
+        request.response = response;
+        this._onDidChange.fire({ kind: 'addResponse', response });
+    }
+    toExport() {
+        return {
+            requesterUsername: this.requesterUsername,
+            requesterAvatarIconUri: this.requesterAvatarIconUri,
+            responderUsername: this.responderUsername,
+            responderAvatarIconUri: this.responderAvatarIcon,
+            initialLocation: this.initialLocation,
+            requests: ( this._requests.map((r) => {
+                const message = {
+                    ...r.message,
+                    parts: ( r.message.parts.map(p => p && 'toJSON' in p ? p.toJSON() : p))
+                };
+                const agent = r.response?.agent;
+                const agentJson = agent && 'toJSON' in agent ? agent.toJSON() :
+                    agent ? { ...agent } : undefined;
+                return {
+                    requestId: r.id,
+                    message,
+                    variableData: r.variableData,
+                    response: r.response ?
+                        ( r.response.response.value.map(item => {
+                            if (item.kind === 'treeData') {
+                                return item.treeData;
+                            }
+                            else if (item.kind === 'markdownContent') {
+                                return item.content;
+                            }
+                            else {
+                                return item;
+                            }
+                        }))
+                        : undefined,
+                    responseId: r.response?.id,
+                    isHidden: r.shouldBeRemovedOnSend,
+                    result: r.response?.result,
+                    followups: r.response?.followups,
+                    isCanceled: r.response?.isCanceled,
+                    vote: r.response?.vote,
+                    voteDownReason: r.response?.voteDownReason,
+                    agent: agentJson,
+                    workingSet: r.workingSet,
+                    slashCommand: r.response?.slashCommand,
+                    usedContext: r.response?.usedContext,
+                    contentReferences: r.response?.contentReferences,
+                    codeCitations: r.response?.codeCitations,
+                    timestamp: r.timestamp
+                };
+            })),
+        };
+    }
+    toJSON() {
+        return {
+            version: 3,
+            ...this.toExport(),
+            sessionId: this.sessionId,
+            creationDate: this._creationDate,
+            isImported: this._isImported,
+            lastMessageDate: this._lastMessageDate,
+            customTitle: this._customTitle
+        };
+    }
+    dispose() {
+        this._requests.forEach(r => r.response?.dispose());
+        this._onDidDispose.fire();
+        super.dispose();
+    }
+};
+ChatModel = ChatModel_1 = ( __decorate([
+    ( __param(2, ILogService)),
+    ( __param(3, IChatAgentService))
+], ChatModel));
+function updateRanges(variableData, diff) {
+    return {
+        variables: ( variableData.variables.map(v => ({
+            ...v,
+            range: v.range && {
+                start: v.range.start - diff,
+                endExclusive: v.range.endExclusive - diff
+            }
+        })))
+    };
+}
+function canMergeMarkdownStrings(md1, md2) {
+    if (md1.baseUri && md2.baseUri) {
+        const baseUriEquals = md1.baseUri.scheme === md2.baseUri.scheme
+            && md1.baseUri.authority === md2.baseUri.authority
+            && md1.baseUri.path === md2.baseUri.path
+            && md1.baseUri.query === md2.baseUri.query
+            && md1.baseUri.fragment === md2.baseUri.fragment;
+        if (!baseUriEquals) {
+            return false;
+        }
+    }
+    else if (md1.baseUri || md2.baseUri) {
+        return false;
+    }
+    return equals(md1.isTrusted, md2.isTrusted) &&
+        md1.supportHtml === md2.supportHtml &&
+        md1.supportThemeIcons === md2.supportThemeIcons;
+}
+function appendMarkdownString(md1, md2) {
+    const appendedValue = typeof md2 === 'string' ? md2 : md2.value;
+    return {
+        value: md1.value + appendedValue,
+        isTrusted: md1.isTrusted,
+        supportThemeIcons: md1.supportThemeIcons,
+        supportHtml: md1.supportHtml,
+        baseUri: md1.baseUri
+    };
+}
+function getCodeCitationsMessage(citations) {
+    if (citations.length === 0) {
+        return '';
+    }
+    const licenseTypes = citations.reduce((set, c) => set.add(c.license), ( new Set()));
+    const label = licenseTypes.size === 1 ?
+        ( localize(4755, "Similar code found with 1 license type", licenseTypes.size)) :
+        ( localize(4756, "Similar code found with {0} license types", licenseTypes.size));
+    return label;
+}
+
+export { ChatModel, ChatModelInitState, ChatPauseState, ChatRequestModel, ChatRequestRemovalReason, ChatResponseModel, Response, appendMarkdownString, canMergeMarkdownStrings, getCodeCitationsMessage, isExportableSessionData, isLinkVariableEntry, isPasteVariableEntry, isSerializableSessionData, normalizeSerializableChatData, toChatHistoryContent, updateRanges };