@codingame/monaco-vscode-chat-service-override 22.1.8 → 23.0.0

package/vscode/src/vs/workbench/contrib/chat/common/promptSyntax/service/promptsServiceImpl.js

@@ -1,34 +1,37 @@
 
 import { __decorate, __param } from '@codingame/monaco-vscode-api/external/tslib/tslib.es6';
-import { localize } from '@codingame/monaco-vscode-api/vscode/vs/nls';
-import { PromptsType, getPromptsTypeForLanguageId, PROMPT_LANGUAGE_ID, MODE_LANGUAGE_ID } from '@codingame/monaco-vscode-e28ac690-06d5-5ee9-92d1-02df70296354-common/vscode/vs/workbench/contrib/chat/common/promptSyntax/promptTypes';
-import { basename } from '@codingame/monaco-vscode-api/vscode/vs/base/common/path';
-import { PromptFilesLocator } from '../utils/promptFilesLocator.js';
-import { Disposable } from '@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle';
-import { Emitter } from '@codingame/monaco-vscode-api/vscode/vs/base/common/event';
-import { ILogService } from '@codingame/monaco-vscode-api/vscode/vs/platform/log/common/log.service';
-import { ILabelService } from '@codingame/monaco-vscode-api/vscode/vs/platform/label/common/label.service';
-import { IModelService } from '@codingame/monaco-vscode-api/vscode/vs/editor/common/services/model.service';
+import { Delayer } from '@codingame/monaco-vscode-api/vscode/vs/base/common/async';
 import { CancellationToken } from '@codingame/monaco-vscode-api/vscode/vs/base/common/cancellation';
-import { IInstantiationService } from '@codingame/monaco-vscode-api/vscode/vs/platform/instantiation/common/instantiation';
-import { IUserDataProfileService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/userDataProfile/common/userDataProfile.service';
-import { PromptsStorage } from './promptsService.js';
-import { getCleanPromptName, PROMPT_FILE_EXTENSION } from '@codingame/monaco-vscode-e28ac690-06d5-5ee9-92d1-02df70296354-common/vscode/vs/workbench/contrib/chat/common/promptSyntax/config/promptFileLocations';
+import { CancellationError } from '@codingame/monaco-vscode-api/vscode/vs/base/common/errors';
+import { Emitter } from '@codingame/monaco-vscode-api/vscode/vs/base/common/event';
+import { Disposable } from '@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle';
+import { ResourceMap, ResourceSet } from '@codingame/monaco-vscode-api/vscode/vs/base/common/map';
+import { basename } from '@codingame/monaco-vscode-api/vscode/vs/base/common/path';
+import { isEqual, dirname } from '@codingame/monaco-vscode-api/vscode/vs/base/common/resources';
+import { URI } from '@codingame/monaco-vscode-api/vscode/vs/base/common/uri';
+import { OffsetRange } from '@codingame/monaco-vscode-api/vscode/vs/editor/common/core/ranges/offsetRange';
 import { ILanguageService } from '@codingame/monaco-vscode-api/vscode/vs/editor/common/languages/language.service';
-import { PromptsConfig } from '@codingame/monaco-vscode-e28ac690-06d5-5ee9-92d1-02df70296354-common/vscode/vs/workbench/contrib/chat/common/promptSyntax/config/config';
+import { IModelService } from '@codingame/monaco-vscode-api/vscode/vs/editor/common/services/model.service';
+import { localize } from '@codingame/monaco-vscode-api/vscode/vs/nls';
 import { IConfigurationService } from '@codingame/monaco-vscode-api/vscode/vs/platform/configuration/common/configuration.service';
-import { NewPromptsParser } from './newPromptsParser.js';
 import { IFileService } from '@codingame/monaco-vscode-api/vscode/vs/platform/files/common/files.service';
-import { ResourceMap } from '@codingame/monaco-vscode-api/vscode/vs/base/common/map';
-import { CancellationError } from '@codingame/monaco-vscode-api/vscode/vs/base/common/errors';
-import { OffsetRange } from '@codingame/monaco-vscode-api/vscode/vs/editor/common/core/ranges/offsetRange';
-import { isEqual, dirname } from '@codingame/monaco-vscode-api/vscode/vs/base/common/resources';
-import { Delayer } from '@codingame/monaco-vscode-api/vscode/vs/base/common/async';
+import { IInstantiationService } from '@codingame/monaco-vscode-api/vscode/vs/platform/instantiation/common/instantiation';
+import { ILabelService } from '@codingame/monaco-vscode-api/vscode/vs/platform/label/common/label.service';
+import { ILogService } from '@codingame/monaco-vscode-api/vscode/vs/platform/log/common/log.service';
 import { IFilesConfigurationService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/filesConfiguration/common/filesConfigurationService.service';
+import { StorageScope, StorageTarget } from '@codingame/monaco-vscode-api/vscode/vs/platform/storage/common/storage';
+import { IStorageService } from '@codingame/monaco-vscode-api/vscode/vs/platform/storage/common/storage.service';
+import { IUserDataProfileService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/userDataProfile/common/userDataProfile.service';
+import { PromptsConfig } from '@codingame/monaco-vscode-8d4ef243-d8c1-540d-9f11-38df46f54c9f-common/vscode/vs/workbench/contrib/chat/common/promptSyntax/config/config';
+import { getCleanPromptName, PROMPT_FILE_EXTENSION } from '@codingame/monaco-vscode-8d4ef243-d8c1-540d-9f11-38df46f54c9f-common/vscode/vs/workbench/contrib/chat/common/promptSyntax/config/promptFileLocations';
+import { PromptsType, getPromptsTypeForLanguageId, PROMPT_LANGUAGE_ID, getLanguageIdForPromptsType } from '@codingame/monaco-vscode-8d4ef243-d8c1-540d-9f11-38df46f54c9f-common/vscode/vs/workbench/contrib/chat/common/promptSyntax/promptTypes';
+import { PromptFilesLocator } from '../utils/promptFilesLocator.js';
+import { PromptFileParser, PromptHeaderAttributes } from '@codingame/monaco-vscode-8d4ef243-d8c1-540d-9f11-38df46f54c9f-common/vscode/vs/workbench/contrib/chat/common/promptSyntax/promptFileParser';
+import { PromptsStorage } from '@codingame/monaco-vscode-2339d6ac-d6bb-53cd-95ee-81911735d1c7-common/vscode/vs/workbench/contrib/chat/common/promptSyntax/service/promptsService';
 
-var ChatModeUpdateTracker_1;
+var UpdateTracker_1;
 let PromptsService = class PromptsService extends Disposable {
-    constructor(logger, labelService, modelService, instantiationService, userDataService, languageService, configurationService, fileService, filesConfigService) {
+    constructor(logger, labelService, modelService, instantiationService, userDataService, languageService, configurationService, fileService, filesConfigService, storageService) {
         super();
         this.logger = logger;
         this.labelService = labelService;
@@ -39,27 +42,53 @@ let PromptsService = class PromptsService extends Disposable {
         this.configurationService = configurationService;
         this.fileService = fileService;
         this.filesConfigService = filesConfigService;
+        this.storageService = storageService;
         this.parsedPromptFileCache = ( new ResourceMap());
+        this.promptFileByCommandCache = ( new Map());
+        this.onDidChangeParsedPromptFilesCacheEmitter = ( new Emitter());
         this.contributedFiles = {
             [PromptsType.prompt]: ( new ResourceMap()),
             [PromptsType.instructions]: ( new ResourceMap()),
-            [PromptsType.mode]: ( new ResourceMap()),
+            [PromptsType.agent]: ( new ResourceMap()),
         };
+        this.disabledPromptsStorageKeyPrefix = 'chat.disabledPromptFiles.';
+        this.onDidChangeParsedPromptFilesCacheEmitter = this._register(( new Emitter()));
         this.fileLocator = this._register(this.instantiationService.createInstance(PromptFilesLocator));
+        const promptUpdateTracker = this._register(( new UpdateTracker(this.fileLocator, PromptsType.prompt, this.modelService)));
+        this._register(promptUpdateTracker.onDidPromptChange((event) => {
+            if (event.kind === 'fileSystem') {
+                this.promptFileByCommandCache.clear();
+            }
+            else {
+                const pendingDeletes = [];
+                for (const [key, value] of this.promptFileByCommandCache) {
+                    if (isEqual(value.value?.uri, event.uri)) {
+                        pendingDeletes.push(key);
+                    }
+                }
+                for (const key of pendingDeletes) {
+                    this.promptFileByCommandCache.delete(key);
+                }
+            }
+            this.onDidChangeParsedPromptFilesCacheEmitter.fire();
+        }));
         this._register(this.modelService.onModelRemoved((model) => {
             this.parsedPromptFileCache.delete(model.uri);
         }));
     }
-    get onDidChangeCustomChatModes() {
-        if (!this.onDidChangeCustomChatModesEmitter) {
-            const emitter = this.onDidChangeCustomChatModesEmitter = this._register(( new Emitter()));
-            const chatModelTracker = this._register(( new ChatModeUpdateTracker(this.fileLocator, this.modelService)));
-            this._register(chatModelTracker.onDidChangeContent(() => {
-                this.cachedCustomChatModes = undefined;
+    get onDidChangeCustomAgents() {
+        if (!this.onDidChangeCustomAgentsEmitter) {
+            const emitter = this.onDidChangeCustomAgentsEmitter = this._register(( new Emitter()));
+            const updateTracker = this._register(( new UpdateTracker(this.fileLocator, PromptsType.agent, this.modelService)));
+            this._register(updateTracker.onDidPromptChange((event) => {
+                this.cachedCustomAgents = undefined;
                 emitter.fire();
             }));
         }
-        return this.onDidChangeCustomChatModesEmitter.event;
+        return this.onDidChangeCustomAgentsEmitter.event;
+    }
+    get onDidChangeParsedPromptFilesCache() {
+        return this.onDidChangeParsedPromptFilesCacheEmitter.event;
     }
     getPromptFileType(uri) {
         const model = this.modelService.getModel(uri);
@@ -71,16 +100,13 @@ let PromptsService = class PromptsService extends Disposable {
         if (cached && cached[0] === textModel.getVersionId()) {
             return cached[1];
         }
-        const ast = ( new NewPromptsParser()).parse(textModel.uri, textModel.getValue());
+        const ast = ( new PromptFileParser()).parse(textModel.uri, textModel.getValue());
         if (!cached || cached[0] < textModel.getVersionId()) {
             this.parsedPromptFileCache.set(textModel.uri, [textModel.getVersionId(), ast]);
         }
         return ast;
     }
     async listPromptFiles(type, token) {
-        if (!PromptsConfig.enabled(this.configurationService)) {
-            return [];
-        }
         const prompts = await Promise.all([
             this.fileLocator.listFiles(type, PromptsStorage.user, token).then(uris => ( uris.map(uri => ({ uri, storage: PromptsStorage.user, type })))),
             this.fileLocator.listFiles(type, PromptsStorage.local, token).then(uris => ( uris.map(uri => ({ uri, storage: PromptsStorage.local, type })))),
@@ -89,9 +115,6 @@ let PromptsService = class PromptsService extends Disposable {
         return [...prompts.flat()];
     }
     async listPromptFilesForStorage(type, storage, token) {
-        if (!PromptsConfig.enabled(this.configurationService)) {
-            return [];
-        }
         switch (storage) {
             case PromptsStorage.extension:
                 return this.getExtensionContributions(type);
@@ -107,12 +130,17 @@ let PromptsService = class PromptsService extends Disposable {
         return Promise.all(( this.contributedFiles[type].values()));
     }
     getSourceFolders(type) {
-        if (!PromptsConfig.enabled(this.configurationService)) {
-            return [];
-        }
         const result = [];
-        for (const uri of this.fileLocator.getConfigBasedSourceFolders(type)) {
-            result.push({ uri, storage: PromptsStorage.local, type });
+        if (type === PromptsType.agent) {
+            const folders = this.fileLocator.getAgentSourceFolder();
+            for (const uri of folders) {
+                result.push({ uri, storage: PromptsStorage.local, type });
+            }
+        }
+        else {
+            for (const uri of this.fileLocator.getConfigBasedSourceFolders(type)) {
+                result.push({ uri, storage: PromptsStorage.local, type });
+            }
         }
         const userHome = this.userDataService.currentProfile.promptsHome;
         result.push({ uri: userHome, storage: PromptsStorage.user, type });
@@ -120,12 +148,12 @@ let PromptsService = class PromptsService extends Disposable {
     }
     asPromptSlashCommand(command) {
         if (command.match(/^[\p{L}\d_\-\.]+$/u)) {
-            return { command, detail: ( localize(5681, 'Prompt file: {0}', command)) };
+            return { command, detail: ( localize(5922, 'Prompt file: {0}', command)) };
         }
         return undefined;
     }
     async resolvePromptSlashCommand(data, token) {
-        const promptUri = await this.getPromptPath(data);
+        const promptUri = data.promptPath?.uri ?? (await this.getPromptPath(data.command));
         if (!promptUri) {
             return undefined;
         }
@@ -137,15 +165,46 @@ let PromptsService = class PromptsService extends Disposable {
             return undefined;
         }
     }
-    async getPromptPath(data) {
-        if (data.promptPath) {
-            return data.promptPath.uri;
+    async populatePromptCommandCache(command) {
+        let cache = this.promptFileByCommandCache.get(command);
+        if (cache && cache.pendingPromise) {
+            return cache.pendingPromise;
+        }
+        const newPromise = this.resolvePromptSlashCommand({ command, detail: '' }, CancellationToken.None);
+        if (cache) {
+            cache.pendingPromise = newPromise;
+        }
+        else {
+            cache = { value: undefined, pendingPromise: newPromise };
+            this.promptFileByCommandCache.set(command, cache);
+        }
+        const newValue = await newPromise.finally(() => cache.pendingPromise = undefined);
+        cache.value = newValue;
+        this.onDidChangeParsedPromptFilesCacheEmitter.fire();
+        return newValue;
+    }
+    resolvePromptSlashCommandFromCache(command) {
+        const cache = this.promptFileByCommandCache.get(command);
+        const value = cache?.value;
+        if (value === undefined) {
+            void this.populatePromptCommandCache(command).catch((error) => { });
         }
+        return value;
+    }
+    async getPromptDetails(promptPath) {
+        const parsedPromptFile = await this.parseNew(promptPath.uri, CancellationToken.None).catch(() => undefined);
+        return {
+            name: parsedPromptFile?.header?.name ?? promptPath.name ?? getCleanPromptName(promptPath.uri),
+            description: parsedPromptFile?.header?.description ?? promptPath.description
+        };
+    }
+    async getPromptPath(command) {
         const promptPaths = await this.listPromptFiles(PromptsType.prompt, CancellationToken.None);
-        const command = data.command;
-        const result = promptPaths.find(promptPath => getCommandNameFromPromptPath(promptPath) === command);
-        if (result) {
-            return result.uri;
+        for (const promptPath of promptPaths) {
+            const details = await this.getPromptDetails(promptPath);
+            if (details.name === command) {
+                return promptPath.uri;
+            }
         }
         const textModel = this.modelService.getModels().find(model => model.getLanguageId() === PROMPT_LANGUAGE_ID && getCommandNameFromURI(model.uri) === command);
         if (textModel) {
@@ -155,44 +214,47 @@ let PromptsService = class PromptsService extends Disposable {
     }
     async getPromptCommandName(uri) {
         const promptPaths = await this.listPromptFiles(PromptsType.prompt, CancellationToken.None);
-        const promptPath = promptPaths.find(promptPath => isEqual(promptPath.uri, uri));
+        let promptPath = promptPaths.find(promptPath => isEqual(promptPath.uri, uri));
         if (!promptPath) {
-            return getCommandNameFromURI(uri);
+            promptPath = { uri, storage: PromptsStorage.local, type: PromptsType.prompt };
         }
-        return getCommandNameFromPromptPath(promptPath);
+        const { name } = await this.getPromptDetails(promptPath);
+        return name;
     }
     async findPromptSlashCommands() {
         const promptFiles = await this.listPromptFiles(PromptsType.prompt, CancellationToken.None);
-        return (promptFiles.map(promptPath => {
-            const command = getCommandNameFromPromptPath(promptPath);
+        return Promise.all(( promptFiles.map(async (promptPath) => {
+            const { name } = await this.getPromptDetails(promptPath);
             return {
-                command,
+                command: name,
                 detail: ( localize(
-                    5681,
+                    5922,
                     'Prompt file: {0}',
                     this.labelService.getUriLabel(promptPath.uri, { relative: true })
                 )),
                 promptPath
             };
-        }));
+        })));
     }
-    async getCustomChatModes(token) {
-        if (!this.cachedCustomChatModes) {
-            const customChatModes = this.computeCustomChatModes(token);
-            if (!this.onDidChangeCustomChatModesEmitter) {
-                return customChatModes;
+    async getCustomAgents(token) {
+        let customAgents = this.cachedCustomAgents;
+        if (!customAgents) {
+            customAgents = this.computeCustomAgents(token);
+            if (this.onDidChangeCustomAgentsEmitter) {
+                this.cachedCustomAgents = customAgents;
             }
-            this.cachedCustomChatModes = customChatModes;
         }
-        return this.cachedCustomChatModes;
+        const disabledAgents = this.getDisabledPromptFiles(PromptsType.agent);
+        return (await customAgents).filter(agent => !( disabledAgents.has(agent.uri)));
     }
-    async computeCustomChatModes(token) {
-        const modeFiles = await this.listPromptFiles(PromptsType.mode, token);
-        const customChatModes = await Promise.all(( modeFiles.map(async ({ uri, name: modeName }) => {
+    async computeCustomAgents(token) {
+        const agentFiles = await this.listPromptFiles(PromptsType.agent, token);
+        const customAgents = await Promise.all(( agentFiles.map(async (promptPath) => {
+            const uri = promptPath.uri;
             const ast = await this.parseNew(uri, token);
             let metadata;
             if (ast.header) {
-                const advanced = ast.header.getAttribute('advancedOptions');
+                const advanced = ast.header.getAttribute(PromptHeaderAttributes.advancedOptions);
                 if (advanced && advanced.value.type === 'object') {
                     metadata = {};
                     for (const [key, value] of Object.entries(advanced.value)) {
@@ -212,19 +274,20 @@ let PromptsService = class PromptsService extends Disposable {
                     toolReferences.push({ name, range });
                 }
             }
-            const modeInstructions = {
+            const agentInstructions = {
                 content: ast.body?.getContent() ?? '',
                 toolReferences,
                 metadata,
             };
-            const name = modeName ?? getCleanPromptName(uri);
+            const name = ast.header?.name ?? promptPath.name ?? getCleanPromptName(uri);
+            const source = IAgentSource.fromPromptPath(promptPath);
             if (!ast.header) {
-                return { uri, name, modeInstructions };
+                return { uri, name, agentInstructions, source };
             }
-            const { description, model, tools } = ast.header;
-            return { uri, name, description, model, tools, modeInstructions };
+            const { description, model, tools, handOffs, argumentHint, target } = ast.header;
+            return { uri, name, description, model, tools, handOffs, argumentHint, target, agentInstructions, source };
         })));
-        return customChatModes;
+        return customAgents;
     }
     async parseNew(uri, token) {
         const model = this.modelService.getModel(uri);
@@ -235,7 +298,7 @@ let PromptsService = class PromptsService extends Disposable {
         if (token.isCancellationRequested) {
             throw ( new CancellationError());
         }
-        return ( new NewPromptsParser()).parse(uri, ( fileContent.value.toString()));
+        return ( new PromptFileParser()).parse(uri, ( fileContent.value.toString()));
     }
     registerContributedFile(type, name, description, uri, extension) {
         const bucket = this.contributedFiles[type];
@@ -253,27 +316,27 @@ let PromptsService = class PromptsService extends Disposable {
             return { uri, name, description, storage: PromptsStorage.extension, type, extension };
         })();
         bucket.set(uri, entryPromise);
-        const updateModesIfRequired = () => {
-            if (type === PromptsType.mode) {
-                this.cachedCustomChatModes = undefined;
-                this.onDidChangeCustomChatModesEmitter?.fire();
+        const updateAgentsIfRequired = () => {
+            if (type === PromptsType.agent) {
+                this.cachedCustomAgents = undefined;
+                this.onDidChangeCustomAgentsEmitter?.fire();
             }
         };
-        updateModesIfRequired();
+        updateAgentsIfRequired();
         return {
             dispose: () => {
                 bucket.delete(uri);
-                updateModesIfRequired();
+                updateAgentsIfRequired();
             }
         };
     }
     getPromptLocationLabel(promptPath) {
         switch (promptPath.storage) {
             case PromptsStorage.local: return this.labelService.getUriLabel(dirname(promptPath.uri), { relative: true });
-            case PromptsStorage.user: return localize(5682, 'User Data');
+            case PromptsStorage.user: return localize(5923, 'User Data');
             case PromptsStorage.extension: {
                 return localize(
-                    5683,
+                    5924,
                     'Extension: {0}',
                     promptPath.extension.displayName ?? promptPath.extension.id
                 );
@@ -284,6 +347,55 @@ let PromptsService = class PromptsService extends Disposable {
     findAgentMDsInWorkspace(token) {
         return this.fileLocator.findAgentMDsInWorkspace(token);
     }
+    async listAgentMDs(token, includeNested) {
+        const useAgentMD = this.configurationService.getValue(PromptsConfig.USE_AGENT_MD);
+        if (!useAgentMD) {
+            return [];
+        }
+        if (includeNested) {
+            return await this.fileLocator.findAgentMDsInWorkspace(token);
+        }
+        else {
+            return await this.fileLocator.findAgentMDsInWorkspaceRoots(token);
+        }
+    }
+    async listCopilotInstructionsMDs(token) {
+        const useCopilotInstructionsFiles = this.configurationService.getValue(PromptsConfig.USE_COPILOT_INSTRUCTION_FILES);
+        if (!useCopilotInstructionsFiles) {
+            return [];
+        }
+        return await this.fileLocator.findCopilotInstructionsMDsInWorkspace(token);
+    }
+    getAgentFileURIFromModeFile(oldURI) {
+        return this.fileLocator.getAgentFileURIFromModeFile(oldURI);
+    }
+    getDisabledPromptFiles(type) {
+        const disabledKey = this.disabledPromptsStorageKeyPrefix + type;
+        const value = this.storageService.get(disabledKey, StorageScope.PROFILE, '[]');
+        const result = ( new ResourceSet());
+        try {
+            const arr = JSON.parse(value);
+            if (Array.isArray(arr)) {
+                for (const s of arr) {
+                    try {
+                        result.add(URI.revive(s));
+                    }
+                    catch {
+                    }
+                }
+            }
+        }
+        catch {
+        }
+        return result;
+    }
+    setDisabledPromptFiles(type, uris) {
+        const disabled = ( Array.from(uris).map(uri => uri.toJSON()));
+        this.storageService.store(this.disabledPromptsStorageKeyPrefix + type, JSON.stringify(disabled), StorageScope.PROFILE, StorageTarget.USER);
+        if (type === PromptsType.agent) {
+            this.onDidChangeCustomAgentsEmitter?.fire();
+        }
+    }
 };
 PromptsService = ( __decorate([
     ( __param(0, ILogService)),
@@ -294,38 +406,36 @@ PromptsService = ( __decorate([
     ( __param(5, ILanguageService)),
     ( __param(6, IConfigurationService)),
     ( __param(7, IFileService)),
-    ( __param(8, IFilesConfigurationService))
+    ( __param(8, IFilesConfigurationService)),
+    ( __param(9, IStorageService))
 ], PromptsService));
-function getCommandNameFromPromptPath(promptPath) {
-    return promptPath.name ?? getCommandNameFromURI(promptPath.uri);
-}
 function getCommandNameFromURI(uri) {
     return basename(uri.fsPath, PROMPT_FILE_EXTENSION);
 }
-let ChatModeUpdateTracker = class ChatModeUpdateTracker extends Disposable {
-    static { ChatModeUpdateTracker_1 = this; }
-    static { this.CHAT_MODE_UPDATE_DELAY_MS = 200; }
-    get onDidChangeContent() {
-        return this.onDidChatModeModelChange.event;
+let UpdateTracker = class UpdateTracker extends Disposable {
+    static { UpdateTracker_1 = this; }
+    static { this.PROMPT_UPDATE_DELAY_MS = 200; }
+    get onDidPromptChange() {
+        return this.onDidPromptModelChange.event;
     }
-    constructor(fileLocator, modelService) {
+    constructor(fileLocator, promptType, modelService) {
         super();
         this.listeners = ( new ResourceMap());
-        this.onDidChatModeModelChange = this._register(( new Emitter()));
-        const delayer = this._register(( new Delayer(ChatModeUpdateTracker_1.CHAT_MODE_UPDATE_DELAY_MS)));
-        const trigger = () => delayer.trigger(() => this.onDidChatModeModelChange.fire());
-        const filesUpdatedEventRegistration = this._register(fileLocator.createFilesUpdatedEvent(PromptsType.mode));
-        this._register(filesUpdatedEventRegistration.event(() => trigger()));
+        this.onDidPromptModelChange = this._register(( new Emitter()));
+        const delayer = this._register(( new Delayer(UpdateTracker_1.PROMPT_UPDATE_DELAY_MS)));
+        const trigger = (event) => delayer.trigger(() => this.onDidPromptModelChange.fire(event));
+        const filesUpdatedEventRegistration = this._register(fileLocator.createFilesUpdatedEvent(promptType));
+        this._register(filesUpdatedEventRegistration.event(() => trigger({ kind: 'fileSystem' })));
         const onAdd = (model) => {
-            if (model.getLanguageId() === MODE_LANGUAGE_ID) {
-                this.listeners.set(model.uri, model.onDidChangeContent(() => trigger()));
+            if (model.getLanguageId() === getLanguageIdForPromptsType(promptType)) {
+                this.listeners.set(model.uri, model.onDidChangeContent(() => trigger({ kind: 'textModel', uri: model.uri })));
             }
         };
         const onRemove = (languageId, uri) => {
-            if (languageId === MODE_LANGUAGE_ID) {
+            if (languageId === getLanguageIdForPromptsType(promptType)) {
                 this.listeners.get(uri)?.dispose();
                 this.listeners.delete(uri);
-                trigger();
+                trigger({ kind: 'textModel', uri });
             }
         };
         this._register(modelService.onModelAdded(model => onAdd(model)));
@@ -341,8 +451,25 @@ let ChatModeUpdateTracker = class ChatModeUpdateTracker extends Disposable {
         this.listeners.clear();
     }
 };
-ChatModeUpdateTracker = ChatModeUpdateTracker_1 = ( __decorate([
-    ( __param(1, IModelService))
-], ChatModeUpdateTracker));
+UpdateTracker = UpdateTracker_1 = ( __decorate([
+    ( __param(2, IModelService))
+], UpdateTracker));
+var IAgentSource;
+(function (IAgentSource) {
+    function fromPromptPath(promptPath) {
+        if (promptPath.storage === PromptsStorage.extension) {
+            return {
+                storage: PromptsStorage.extension,
+                extensionId: promptPath.extension.identifier
+            };
+        }
+        else {
+            return {
+                storage: promptPath.storage
+            };
+        }
+    }
+    IAgentSource.fromPromptPath = fromPromptPath;
+})(IAgentSource || (IAgentSource = {}));
 
-export { ChatModeUpdateTracker, PromptsService };
+export { PromptsService, UpdateTracker };

package/vscode/src/vs/workbench/contrib/chat/common/promptSyntax/utils/promptFilesLocator.d.ts

@@ -2,11 +2,11 @@ import { URI } from "@codingame/monaco-vscode-api/vscode/vs/base/common/uri";
 import { IFileService } from "@codingame/monaco-vscode-api/vscode/vs/platform/files/common/files.service";
 import { IWorkspaceContextService } from "@codingame/monaco-vscode-api/vscode/vs/platform/workspace/common/workspace.service";
 import { IConfigurationService } from "@codingame/monaco-vscode-api/vscode/vs/platform/configuration/common/configuration.service";
-import { PromptsType } from "@codingame/monaco-vscode-e28ac690-06d5-5ee9-92d1-02df70296354-common/vscode/vs/workbench/contrib/chat/common/promptSyntax/promptTypes";
+import { PromptsType } from "@codingame/monaco-vscode-8d4ef243-d8c1-540d-9f11-38df46f54c9f-common/vscode/vs/workbench/contrib/chat/common/promptSyntax/promptTypes";
 import { IWorkbenchEnvironmentService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/environment/common/environmentService.service";
 import { ISearchService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/search/common/search.service";
 import { CancellationToken } from "@codingame/monaco-vscode-api/vscode/vs/base/common/cancellation";
-import { PromptsStorage } from "../service/promptsService.js";
+import { PromptsStorage } from "@codingame/monaco-vscode-2339d6ac-d6bb-53cd-95ee-81911735d1c7-common/vscode/vs/workbench/contrib/chat/common/promptSyntax/service/promptsService";
 import { IUserDataProfileService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/userDataProfile/common/userDataProfile.service";
 import { Event } from "@codingame/monaco-vscode-api/vscode/vs/base/common/event";
 import { Disposable } from "@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle";
@@ -34,6 +34,7 @@ export declare class PromptFilesLocator extends Disposable {
         readonly event: Event<void>;
         dispose: () => void;
     };
+    getAgentSourceFolder(): readonly URI[];
     /**
      * Get all possible unambiguous prompt file source folders based on
      * the current workspace folder structure.
@@ -68,8 +69,17 @@ export declare class PromptFilesLocator extends Disposable {
      * Uses the search service to find all files at the provided location
      */
     private searchFilesInLocation;
+    findCopilotInstructionsMDsInWorkspace(token: CancellationToken): Promise<URI[]>;
+    /**
+     * Gets list of `AGENTS.md` files anywhere in the workspace.
+     */
     findAgentMDsInWorkspace(token: CancellationToken): Promise<URI[]>;
     private findAgentMDsInFolder;
+    /**
+     * Gets list of `AGENTS.md` files only at the root workspace folder(s).
+     */
+    findAgentMDsInWorkspaceRoots(token: CancellationToken): Promise<URI[]>;
+    getAgentFileURIFromModeFile(oldURI: URI): URI | undefined;
 }
 /**
  * Checks if the provided `pattern` could be a valid glob pattern.