@theia/ai-chat 1.74.0-next.0 → 1.74.0-next.6

package/lib/browser/custom-agent-frontend-application-contribution.d.ts

@@ -1,7 +1,8 @@
 import { PromptFragmentCustomizationService } from '@theia/ai-core';
 import { Command, CommandContribution, CommandRegistry, ILogger, MessageService } from '@theia/core';
-import { FrontendApplicationContribution } from '@theia/core/lib/browser';
+import { FrontendApplicationContribution, LocalStorageService } from '@theia/core/lib/browser';
 import { CustomAgentFactory } from './custom-agent-factory';
+type CustomAgentMigrationReports = Awaited<ReturnType<PromptFragmentCustomizationService['migrateCustomAgentsYaml']>>;
 export declare const RERUN_CUSTOM_AGENT_MIGRATION_COMMAND: Command;
 export declare class AICustomAgentsFrontendApplicationContribution implements FrontendApplicationContribution, CommandContribution {
     protected readonly customAgentFactory: CustomAgentFactory;
@@ -9,11 +10,31 @@ export declare class AICustomAgentsFrontendApplicationContribution implements Fr
     private readonly agentService;
     private readonly chatAgentService;
     protected readonly messageService?: MessageService;
+    protected readonly storageService: LocalStorageService;
     protected readonly logger: ILogger;
     private knownCustomAgents;
+    /** Ensures the migration consent prompt is shown at most once per session. */
+    protected migrationPromptShown: boolean;
+    /** Guards against concurrent pending-migration checks from re-entrant refreshes. */
+    protected migrationCheckInFlight: boolean;
     registerCommands(commands: CommandRegistry): void;
     onStart(): void;
     protected refreshCustomAgents(): Promise<void>;
+    /**
+     * Prompts the user once per session before migrating legacy `customAgents.yml` files (and before
+     * correcting agents migrated with merged headings by an earlier Theia version). Nothing is written
+     * until the user confirms. Declining is safe: legacy `customAgents.yml` files keep being loaded, so
+     * agents continue to work unmigrated, and the prompt reappears next session until they migrate or
+     * pick "Don't Show Again".
+     *
+     * This is intentionally not backed by a preference: the whole migration path is transitional and
+     * will be removed once the legacy format is no longer supported. The "don't show again" choice is
+     * persisted in local storage (like the dev-container suggestion), so there is no setting to retire.
+     */
+    protected maybeMigrate(): Promise<void>;
+    protected promptAndMigrate(): Promise<void>;
+    protected showMigrationSummary(reports: CustomAgentMigrationReports): void;
     onStop(): void;
 }
+export {};
 //# sourceMappingURL=custom-agent-frontend-application-contribution.d.ts.map

package/lib/browser/custom-agent-frontend-application-contribution.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"custom-agent-frontend-application-contribution.d.ts","sourceRoot":"","sources":["../../src/browser/custom-agent-frontend-application-contribution.ts"],"names":[],"mappings":"AAgBA,OAAO,EAAwC,kCAAkC,EAAE,MAAM,gBAAgB,CAAC;AAC1G,OAAO,EAAE,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,OAAO,EAAE,cAAc,EAAO,MAAM,aAAa,CAAC;AAC1G,OAAO,EAAE,+BAA+B,EAAE,MAAM,yBAAyB,CAAC;AAG1E,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAE5D,eAAO,MAAM,oCAAoC,EAAE,OAQlD,CAAC;AAEF,qBACa,6CAA8C,YAAW,+BAA+B,EAAE,mBAAmB;IAEtH,SAAS,CAAC,QAAQ,CAAC,kBAAkB,EAAE,kBAAkB,CAAC;IAG1D,SAAS,CAAC,QAAQ,CAAC,oBAAoB,EAAE,kCAAkC,CAAC;IAG5E,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAe;IAG5C,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAmB;IAGpD,SAAS,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,cAAc,CAAC;IAGnD,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;IAEnC,OAAO,CAAC,iBAAiB,CAAkD;IAE3E,gBAAgB,CAAC,QAAQ,EAAE,eAAe,GAAG,IAAI;IAoBjD,OAAO,IAAI,IAAI;cAKC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAgCpD,MAAM,IAAI,IAAI;CAEjB"}
+{"version":3,"file":"custom-agent-frontend-application-contribution.d.ts","sourceRoot":"","sources":["../../src/browser/custom-agent-frontend-application-contribution.ts"],"names":[],"mappings":"AAgBA,OAAO,EAAwC,kCAAkC,EAAE,MAAM,gBAAgB,CAAC;AAC1G,OAAO,EAAE,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,OAAO,EAAE,cAAc,EAAO,MAAM,aAAa,CAAC;AAC1G,OAAO,EAAE,+BAA+B,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAI/F,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAE5D,KAAK,2BAA2B,GAAG,OAAO,CAAC,UAAU,CAAC,kCAAkC,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC;AAKtH,eAAO,MAAM,oCAAoC,EAAE,OAQlD,CAAC;AAEF,qBACa,6CAA8C,YAAW,+BAA+B,EAAE,mBAAmB;IAEtH,SAAS,CAAC,QAAQ,CAAC,kBAAkB,EAAE,kBAAkB,CAAC;IAG1D,SAAS,CAAC,QAAQ,CAAC,oBAAoB,EAAE,kCAAkC,CAAC;IAG5E,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAe;IAG5C,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAmB;IAGpD,SAAS,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,cAAc,CAAC;IAGnD,SAAS,CAAC,QAAQ,CAAC,cAAc,EAAE,mBAAmB,CAAC;IAGvD,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;IAEnC,OAAO,CAAC,iBAAiB,CAAkD;IAE3E,8EAA8E;IAC9E,SAAS,CAAC,oBAAoB,UAAS;IACvC,oFAAoF;IACpF,SAAS,CAAC,sBAAsB,UAAS;IAEzC,gBAAgB,CAAC,QAAQ,EAAE,eAAe,GAAG,IAAI;IAWjD,OAAO,IAAI,IAAI;cAKC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IA2BpD;;;;;;;;;;OAUG;cACa,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;cAyB7B,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;IAgCjD,SAAS,CAAC,oBAAoB,CAAC,OAAO,EAAE,2BAA2B,GAAG,IAAI;IAmB1E,MAAM,IAAI,IAAI;CAEjB"}

package/lib/browser/custom-agent-frontend-application-contribution.js

@@ -19,9 +19,13 @@ exports.AICustomAgentsFrontendApplicationContribution = exports.RERUN_CUSTOM_AGE
 const tslib_1 = require("tslib");
 const ai_core_1 = require("@theia/ai-core");
 const core_1 = require("@theia/core");
+const browser_1 = require("@theia/core/lib/browser");
+const frontend_application_config_provider_1 = require("@theia/core/lib/browser/frontend-application-config-provider");
 const inversify_1 = require("@theia/core/shared/inversify");
 const common_1 = require("../common");
 const custom_agent_factory_1 = require("./custom-agent-factory");
+/** Local-storage key recording that the user dismissed the custom-agent migration prompt for good. */
+const MIGRATION_DONT_SHOW_AGAIN_KEY = 'ai-chat.customAgents.migration.dontShowAgain';
 exports.RERUN_CUSTOM_AGENT_MIGRATION_COMMAND = core_1.Command.toLocalizedCommand({
     id: 'ai-chat.customAgents.rerunMigration',
     label: 'Re-run custom-agent migration',
@@ -30,6 +34,10 @@ exports.RERUN_CUSTOM_AGENT_MIGRATION_COMMAND = core_1.Command.toLocalizedCommand
 let AICustomAgentsFrontendApplicationContribution = class AICustomAgentsFrontendApplicationContribution {
     constructor() {
         this.knownCustomAgents = new Map();
+        /** Ensures the migration consent prompt is shown at most once per session. */
+        this.migrationPromptShown = false;
+        /** Guards against concurrent pending-migration checks from re-entrant refreshes. */
+        this.migrationCheckInFlight = false;
     }
     registerCommands(commands) {
         commands.registerCommand(exports.RERUN_CUSTOM_AGENT_MIGRATION_COMMAND, {
@@ -39,12 +47,7 @@ let AICustomAgentsFrontendApplicationContribution = class AICustomAgentsFrontend
                     return;
                 }
                 const reports = await this.customizationService.migrateCustomAgentsYaml();
-                const migrated = reports.reduce((sum, r) => sum + r.migrated, 0);
-                const failed = reports.reduce((sum, r) => sum + r.failed, 0);
-                const backedUp = reports.filter(r => r.yamlBackedUp).length;
-                const overrides = reports.reduce((sum, r) => sum + r.promptOverridesMigrated, 0);
-                const message = core_1.nls.localize('theia/ai/chat/customAgents/migrationResult', 'Custom-agent migration: {0} migrated, {1} failed, {2} legacy customAgents.yml backed up to .bak, {3} prompt overrides folded into agent.md.', migrated, failed, backedUp, overrides);
-                this.messageService?.info(message);
+                this.showMigrationSummary(reports);
             }
         });
     }
@@ -58,14 +61,8 @@ let AICustomAgentsFrontendApplicationContribution = class AICustomAgentsFrontend
         }
         // Migration must run on the refresh path (not just at onStart) because the workspace's
         // additional template directories are populated asynchronously by TemplatePreferenceContribution
-        // after preferences/workspace/trust are ready. Migration is idempotent: if a customAgents.yml has
-        // already been migrated (or never existed), the call is a no-op.
-        try {
-            await this.customizationService.migrateCustomAgentsYaml();
-        }
-        catch (e) {
-            this.logger.warn('Custom-agent auto-migration failed; continuing without migration', e);
-        }
+        // after preferences/workspace/trust are ready.
+        await this.maybeMigrate();
         const customAgents = await this.customizationService.getCustomAgents();
         const customAgentsToAdd = customAgents.filter(agent => !this.knownCustomAgents.has(agent.id) || !ai_core_1.CustomAgentDescription.equals(this.knownCustomAgents.get(agent.id), agent));
         const customAgentIdsToRemove = [...this.knownCustomAgents.values()].filter(agent => !customAgents.find(a => ai_core_1.CustomAgentDescription.equals(a, agent))).map(a => a.id);
@@ -80,6 +77,87 @@ let AICustomAgentsFrontendApplicationContribution = class AICustomAgentsFrontend
             this.knownCustomAgents.set(agent.id, agent);
         });
     }
+    /**
+     * Prompts the user once per session before migrating legacy `customAgents.yml` files (and before
+     * correcting agents migrated with merged headings by an earlier Theia version). Nothing is written
+     * until the user confirms. Declining is safe: legacy `customAgents.yml` files keep being loaded, so
+     * agents continue to work unmigrated, and the prompt reappears next session until they migrate or
+     * pick "Don't Show Again".
+     *
+     * This is intentionally not backed by a preference: the whole migration path is transitional and
+     * will be removed once the legacy format is no longer supported. The "don't show again" choice is
+     * persisted in local storage (like the dev-container suggestion), so there is no setting to retire.
+     */
+    async maybeMigrate() {
+        if (!this.customizationService || !this.messageService) {
+            return;
+        }
+        if (this.migrationPromptShown || this.migrationCheckInFlight) {
+            return;
+        }
+        this.migrationCheckInFlight = true;
+        try {
+            if (await this.storageService.getData(MIGRATION_DONT_SHOW_AGAIN_KEY)) {
+                this.migrationPromptShown = true; // suppress further checks for this session too
+                return;
+            }
+            if (!(await this.customizationService.hasPendingCustomAgentMigration())) {
+                return;
+            }
+            this.migrationPromptShown = true;
+            await this.promptAndMigrate();
+        }
+        catch (e) {
+            this.logger.warn('Custom-agent migration check failed; continuing without migration', e);
+        }
+        finally {
+            this.migrationCheckInFlight = false;
+        }
+    }
+    async promptAndMigrate() {
+        if (!this.customizationService || !this.messageService) {
+            return;
+        }
+        const migrate = core_1.nls.localizeByDefault('Migrate');
+        const notNow = core_1.nls.localizeByDefault('Not Now');
+        const dontShowAgain = core_1.nls.localizeByDefault("Don't Show Again");
+        const message = core_1.nls.localize('theia/ai/chat/customAgents/migratePrompt/message', 'Your custom agents still use the older `customAgents.yml` format. '
+            + 'We recommend migrating them to the new per-agent format (`agents/<id>/agent.md`), which {0} uses going forward. '
+            + 'Each agent gets its own file next to your `customAgents.yml`, and the original is kept as a `customAgents.yml.bak` backup.', frontend_application_config_provider_1.FrontendApplicationConfigProvider.get().applicationName);
+        const choice = await this.messageService.info(message, migrate, notNow, dontShowAgain);
+        if (choice === migrate) {
+            try {
+                const reports = await this.customizationService.migrateCustomAgentsYaml();
+                this.showMigrationSummary(reports);
+            }
+            catch (e) {
+                this.logger.warn('Custom-agent migration failed', e);
+            }
+        }
+        else if (choice === dontShowAgain) {
+            try {
+                await this.storageService.setData(MIGRATION_DONT_SHOW_AGAIN_KEY, true);
+            }
+            catch (e) {
+                this.logger.warn('Failed to persist custom-agent migration prompt state', e);
+            }
+        }
+        // 'Not Now' or dismissal: do nothing; the prompt reappears next session until migrated or dismissed for good.
+    }
+    showMigrationSummary(reports) {
+        if (reports.length === 0) {
+            this.messageService?.info(core_1.nls.localize('theia/ai/chat/customAgents/migrationResult/empty', 'Custom-agent migration: nothing to migrate.'));
+            return;
+        }
+        const migrated = reports.reduce((sum, r) => sum + r.migrated, 0);
+        const corrected = reports.reduce((sum, r) => sum + r.corrected, 0);
+        const failed = reports.reduce((sum, r) => sum + r.failed, 0);
+        const backedUp = reports.filter(r => r.yamlBackedUp).length;
+        const overrides = reports.reduce((sum, r) => sum + r.promptOverridesMigrated, 0);
+        const message = core_1.nls.localize('theia/ai/chat/customAgents/migrationResult', 'Custom-agent migration: {0} migrated, {1} corrected, {2} failed, {3} legacy customAgents.yml backed up to .bak, {4} prompt overrides folded into agent.md.', migrated, corrected, failed, backedUp, overrides);
+        // Keep the result visible until the user dismisses it (no auto-hide timeout).
+        this.messageService?.info(message, { timeout: 0 });
+    }
     onStop() {
     }
 };
@@ -106,6 +184,10 @@ tslib_1.__decorate([
     (0, inversify_1.optional)(),
     tslib_1.__metadata("design:type", core_1.MessageService)
 ], AICustomAgentsFrontendApplicationContribution.prototype, "messageService", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(browser_1.LocalStorageService),
+    tslib_1.__metadata("design:type", browser_1.LocalStorageService)
+], AICustomAgentsFrontendApplicationContribution.prototype, "storageService", void 0);
 tslib_1.__decorate([
     (0, inversify_1.inject)(core_1.ILogger),
     (0, inversify_1.named)('ai-chat:AICustomAgentsFrontendApplicationContribution'),

package/lib/browser/custom-agent-frontend-application-contribution.js.map

@@ -1 +1 @@
-{"version":3,"file":"custom-agent-frontend-application-contribution.js","sourceRoot":"","sources":["../../src/browser/custom-agent-frontend-application-contribution.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;;AAEhF,4CAA0G;AAC1G,sCAA0G;AAE1G,4DAAmF;AACnF,sCAA6C;AAC7C,iEAA4D;AAE/C,QAAA,oCAAoC,GAAY,cAAO,CAAC,kBAAkB,CACnF;IACI,EAAE,EAAE,qCAAqC;IACzC,KAAK,EAAE,+BAA+B;IACtC,QAAQ,EAAE,IAAI;CACjB,EACD,2CAA2C,EAC3C,wBAAwB,CAC3B,CAAC;AAGK,IAAM,6CAA6C,GAAnD,MAAM,6CAA6C;IAAnD;QAmBK,sBAAiB,GAAwC,IAAI,GAAG,EAAE,CAAC;IA6D/E,CAAC;IA3DG,gBAAgB,CAAC,QAAyB;QACtC,QAAQ,CAAC,eAAe,CAAC,4CAAoC,EAAE;YAC3D,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB;YAC5C,OAAO,EAAE,KAAK,IAAI,EAAE;gBAChB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;oBAAC,OAAO;gBAAC,CAAC;gBAC3C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,uBAAuB,EAAE,CAAC;gBAC1E,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;gBACjE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;gBAC7D,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC;gBAC5D,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC;gBACjF,MAAM,OAAO,GAAG,UAAG,CAAC,QAAQ,CACxB,4CAA4C,EAC5C,6IAA6I,EAC7I,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,CACxC,CAAC;gBACF,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YACvC,CAAC;SACJ,CAAC,CAAC;IACP,CAAC;IAED,OAAO;QACH,IAAI,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC,CAAC;QACrF,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,CAAC,CAAC,CAAC,CAAC;IAChG,CAAC;IAES,KAAK,CAAC,mBAAmB;QAC/B,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC7B,OAAO;QACX,CAAC;QACD,uFAAuF;QACvF,iGAAiG;QACjG,kGAAkG;QAClG,iEAAiE;QACjE,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,oBAAoB,CAAC,uBAAuB,EAAE,CAAC;QAC9D,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kEAAkE,EAAE,CAAC,CAAC,CAAC;QAC5F,CAAC;QAED,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,eAAe,EAAE,CAAC;QACvE,MAAM,iBAAiB,GAAG,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAClD,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,gCAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAE,EAAE,KAAK,CAAC,CAAC,CAAC;QAC3H,MAAM,sBAAsB,GAAG,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAC/E,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,gCAAsB,CAAC,MAAM,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAErF,+FAA+F;QAC/F,sBAAsB,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE;YAChC,IAAI,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC;YAC9C,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;YACtC,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QACH,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;YAC9B,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC;YACzI,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACP,CAAC;IAED,MAAM;IACN,CAAC;CACJ,CAAA;AAhFY,sGAA6C;AAEnC;IADlB,IAAA,kBAAM,EAAC,yCAAkB,CAAC;;yFAC+B;AAGvC;IADlB,IAAA,kBAAM,EAAC,4CAAkC,CAAC;IAAE,IAAA,oBAAQ,GAAE;;2FACqB;AAG3D;IADhB,IAAA,kBAAM,EAAC,sBAAY,CAAC;;mFACuB;AAG3B;IADhB,IAAA,kBAAM,EAAC,yBAAgB,CAAC;;uFAC2B;AAGjC;IADlB,IAAA,kBAAM,EAAC,qBAAc,CAAC;IAAE,IAAA,oBAAQ,GAAE;sCACC,qBAAc;qFAAC;AAGhC;IADlB,IAAA,kBAAM,EAAC,cAAO,CAAC;IAAE,IAAA,iBAAK,EAAC,uDAAuD,CAAC;;6EAC7C;wDAjB1B,6CAA6C;IADzD,IAAA,sBAAU,GAAE;GACA,6CAA6C,CAgFzD"}
+{"version":3,"file":"custom-agent-frontend-application-contribution.js","sourceRoot":"","sources":["../../src/browser/custom-agent-frontend-application-contribution.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;;AAEhF,4CAA0G;AAC1G,sCAA0G;AAC1G,qDAA+F;AAC/F,uHAAiH;AACjH,4DAAmF;AACnF,sCAA6C;AAC7C,iEAA4D;AAI5D,sGAAsG;AACtG,MAAM,6BAA6B,GAAG,8CAA8C,CAAC;AAExE,QAAA,oCAAoC,GAAY,cAAO,CAAC,kBAAkB,CACnF;IACI,EAAE,EAAE,qCAAqC;IACzC,KAAK,EAAE,+BAA+B;IACtC,QAAQ,EAAE,IAAI;CACjB,EACD,2CAA2C,EAC3C,wBAAwB,CAC3B,CAAC;AAGK,IAAM,6CAA6C,GAAnD,MAAM,6CAA6C;IAAnD;QAsBK,sBAAiB,GAAwC,IAAI,GAAG,EAAE,CAAC;QAE3E,8EAA8E;QACpE,yBAAoB,GAAG,KAAK,CAAC;QACvC,oFAAoF;QAC1E,2BAAsB,GAAG,KAAK,CAAC;IAsI7C,CAAC;IApIG,gBAAgB,CAAC,QAAyB;QACtC,QAAQ,CAAC,eAAe,CAAC,4CAAoC,EAAE;YAC3D,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB;YAC5C,OAAO,EAAE,KAAK,IAAI,EAAE;gBAChB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;oBAAC,OAAO;gBAAC,CAAC;gBAC3C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,uBAAuB,EAAE,CAAC;gBAC1E,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;YACvC,CAAC;SACJ,CAAC,CAAC;IACP,CAAC;IAED,OAAO;QACH,IAAI,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC,CAAC;QACrF,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,CAAC,CAAC,CAAC,CAAC;IAChG,CAAC;IAES,KAAK,CAAC,mBAAmB;QAC/B,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC7B,OAAO;QACX,CAAC;QACD,uFAAuF;QACvF,iGAAiG;QACjG,+CAA+C;QAC/C,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAE1B,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,eAAe,EAAE,CAAC;QACvE,MAAM,iBAAiB,GAAG,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAClD,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,gCAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAE,EAAE,KAAK,CAAC,CAAC,CAAC;QAC3H,MAAM,sBAAsB,GAAG,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAC/E,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,gCAAsB,CAAC,MAAM,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAErF,+FAA+F;QAC/F,sBAAsB,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE;YAChC,IAAI,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC;YAC9C,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;YACtC,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QACH,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;YAC9B,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC;YACzI,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACP,CAAC;IAED;;;;;;;;;;OAUG;IACO,KAAK,CAAC,YAAY;QACxB,IAAI,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACrD,OAAO;QACX,CAAC;QACD,IAAI,IAAI,CAAC,oBAAoB,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YAC3D,OAAO;QACX,CAAC;QACD,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC;QACnC,IAAI,CAAC;YACD,IAAI,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAAU,6BAA6B,CAAC,EAAE,CAAC;gBAC5E,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC,CAAC,+CAA+C;gBACjF,OAAO;YACX,CAAC;YACD,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,oBAAoB,CAAC,8BAA8B,EAAE,CAAC,EAAE,CAAC;gBACtE,OAAO;YACX,CAAC;YACD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACjC,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAClC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mEAAmE,EAAE,CAAC,CAAC,CAAC;QAC7F,CAAC;gBAAS,CAAC;YACP,IAAI,CAAC,sBAAsB,GAAG,KAAK,CAAC;QACxC,CAAC;IACL,CAAC;IAES,KAAK,CAAC,gBAAgB;QAC5B,IAAI,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACrD,OAAO;QACX,CAAC;QACD,MAAM,OAAO,GAAG,UAAG,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,UAAG,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAChD,MAAM,aAAa,GAAG,UAAG,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;QAChE,MAAM,OAAO,GAAG,UAAG,CAAC,QAAQ,CACxB,kDAAkD,EAClD,oEAAoE;cAClE,kHAAkH;cAClH,4HAA4H,EAC9H,wEAAiC,CAAC,GAAG,EAAE,CAAC,eAAe,CAC1D,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;QACvF,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YACrB,IAAI,CAAC;gBACD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,uBAAuB,EAAE,CAAC;gBAC1E,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;YACvC,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,EAAE,CAAC,CAAC,CAAC;YACzD,CAAC;QACL,CAAC;aAAM,IAAI,MAAM,KAAK,aAAa,EAAE,CAAC;YAClC,IAAI,CAAC;gBACD,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,6BAA6B,EAAE,IAAI,CAAC,CAAC;YAC3E,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uDAAuD,EAAE,CAAC,CAAC,CAAC;YACjF,CAAC;QACL,CAAC;QACD,8GAA8G;IAClH,CAAC;IAES,oBAAoB,CAAC,OAAoC;QAC/D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,UAAG,CAAC,QAAQ,CAAC,kDAAkD,EAAE,6CAA6C,CAAC,CAAC,CAAC;YAC3I,OAAO;QACX,CAAC;QACD,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACjE,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;QACnE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC7D,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC;QAC5D,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC;QACjF,MAAM,OAAO,GAAG,UAAG,CAAC,QAAQ,CACxB,4CAA4C,EAC5C,4JAA4J,EAC5J,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,CACnD,CAAC;QACF,8EAA8E;QAC9E,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,MAAM;IACN,CAAC;CACJ,CAAA;AAjKY,sGAA6C;AAEnC;IADlB,IAAA,kBAAM,EAAC,yCAAkB,CAAC;;yFAC+B;AAGvC;IADlB,IAAA,kBAAM,EAAC,4CAAkC,CAAC;IAAE,IAAA,oBAAQ,GAAE;;2FACqB;AAG3D;IADhB,IAAA,kBAAM,EAAC,sBAAY,CAAC;;mFACuB;AAG3B;IADhB,IAAA,kBAAM,EAAC,yBAAgB,CAAC;;uFAC2B;AAGjC;IADlB,IAAA,kBAAM,EAAC,qBAAc,CAAC;IAAE,IAAA,oBAAQ,GAAE;sCACC,qBAAc;qFAAC;AAGhC;IADlB,IAAA,kBAAM,EAAC,6BAAmB,CAAC;sCACO,6BAAmB;qFAAC;AAGpC;IADlB,IAAA,kBAAM,EAAC,cAAO,CAAC;IAAE,IAAA,iBAAK,EAAC,uDAAuD,CAAC;;6EAC7C;wDApB1B,6CAA6C;IADzD,IAAA,sBAAU,GAAE;GACA,6CAA6C,CAiKzD"}

package/package.json

@@ -1,16 +1,16 @@
 {
   "name": "@theia/ai-chat",
-  "version": "1.74.0-next.0+72573426f",
+  "version": "1.74.0-next.6+04cb275b0",
   "description": "Theia - AI Chat Extension",
   "dependencies": {
-    "@theia/ai-core": "1.74.0-next.0+72573426f",
-    "@theia/core": "1.74.0-next.0+72573426f",
-    "@theia/editor": "1.74.0-next.0+72573426f",
-    "@theia/file-search": "1.74.0-next.0+72573426f",
-    "@theia/filesystem": "1.74.0-next.0+72573426f",
-    "@theia/monaco": "1.74.0-next.0+72573426f",
+    "@theia/ai-core": "1.74.0-next.6+04cb275b0",
+    "@theia/core": "1.74.0-next.6+04cb275b0",
+    "@theia/editor": "1.74.0-next.6+04cb275b0",
+    "@theia/file-search": "1.74.0-next.6+04cb275b0",
+    "@theia/filesystem": "1.74.0-next.6+04cb275b0",
+    "@theia/monaco": "1.74.0-next.6+04cb275b0",
     "@theia/monaco-editor-core": "1.108.201",
-    "@theia/workspace": "1.74.0-next.0+72573426f",
+    "@theia/workspace": "1.74.0-next.6+04cb275b0",
     "js-yaml": "^4.2.0",
     "tslib": "^2.8.1"
   },
@@ -54,5 +54,5 @@
   "nyc": {
     "extends": "../../configs/nyc.json"
   },
-  "gitHead": "72573426f320fefc89a96ba55ce05c752d182123"
+  "gitHead": "04cb275b0eb41688cdf025c43555938fea5bc1ef"
 }

package/src/browser/custom-agent-frontend-application-contribution.ts

@@ -16,11 +16,17 @@
 
 import { AgentService, CustomAgentDescription, PromptFragmentCustomizationService } from '@theia/ai-core';
 import { Command, CommandContribution, CommandRegistry, ILogger, MessageService, nls } from '@theia/core';
-import { FrontendApplicationContribution } from '@theia/core/lib/browser';
+import { FrontendApplicationContribution, LocalStorageService } from '@theia/core/lib/browser';
+import { FrontendApplicationConfigProvider } from '@theia/core/lib/browser/frontend-application-config-provider';
 import { inject, injectable, optional, named } from '@theia/core/shared/inversify';
 import { ChatAgentService } from '../common';
 import { CustomAgentFactory } from './custom-agent-factory';
 
+type CustomAgentMigrationReports = Awaited<ReturnType<PromptFragmentCustomizationService['migrateCustomAgentsYaml']>>;
+
+/** Local-storage key recording that the user dismissed the custom-agent migration prompt for good. */
+const MIGRATION_DONT_SHOW_AGAIN_KEY = 'ai-chat.customAgents.migration.dontShowAgain';
+
 export const RERUN_CUSTOM_AGENT_MIGRATION_COMMAND: Command = Command.toLocalizedCommand(
     {
         id: 'ai-chat.customAgents.rerunMigration',
@@ -48,27 +54,26 @@ export class AICustomAgentsFrontendApplicationContribution implements FrontendAp
     @inject(MessageService) @optional()
     protected readonly messageService?: MessageService;
 
+    @inject(LocalStorageService)
+    protected readonly storageService: LocalStorageService;
+
     @inject(ILogger) @named('ai-chat:AICustomAgentsFrontendApplicationContribution')
     protected readonly logger: ILogger;
 
     private knownCustomAgents: Map<string, CustomAgentDescription> = new Map();
 
+    /** Ensures the migration consent prompt is shown at most once per session. */
+    protected migrationPromptShown = false;
+    /** Guards against concurrent pending-migration checks from re-entrant refreshes. */
+    protected migrationCheckInFlight = false;
+
     registerCommands(commands: CommandRegistry): void {
         commands.registerCommand(RERUN_CUSTOM_AGENT_MIGRATION_COMMAND, {
             isEnabled: () => !!this.customizationService,
             execute: async () => {
                 if (!this.customizationService) { return; }
                 const reports = await this.customizationService.migrateCustomAgentsYaml();
-                const migrated = reports.reduce((sum, r) => sum + r.migrated, 0);
-                const failed = reports.reduce((sum, r) => sum + r.failed, 0);
-                const backedUp = reports.filter(r => r.yamlBackedUp).length;
-                const overrides = reports.reduce((sum, r) => sum + r.promptOverridesMigrated, 0);
-                const message = nls.localize(
-                    'theia/ai/chat/customAgents/migrationResult',
-                    'Custom-agent migration: {0} migrated, {1} failed, {2} legacy customAgents.yml backed up to .bak, {3} prompt overrides folded into agent.md.',
-                    migrated, failed, backedUp, overrides
-                );
-                this.messageService?.info(message);
+                this.showMigrationSummary(reports);
             }
         });
     }
@@ -84,13 +89,8 @@ export class AICustomAgentsFrontendApplicationContribution implements FrontendAp
         }
         // Migration must run on the refresh path (not just at onStart) because the workspace's
         // additional template directories are populated asynchronously by TemplatePreferenceContribution
-        // after preferences/workspace/trust are ready. Migration is idempotent: if a customAgents.yml has
-        // already been migrated (or never existed), the call is a no-op.
-        try {
-            await this.customizationService.migrateCustomAgentsYaml();
-        } catch (e) {
-            this.logger.warn('Custom-agent auto-migration failed; continuing without migration', e);
-        }
+        // after preferences/workspace/trust are ready.
+        await this.maybeMigrate();
 
         const customAgents = await this.customizationService.getCustomAgents();
         const customAgentsToAdd = customAgents.filter(agent =>
@@ -110,6 +110,93 @@ export class AICustomAgentsFrontendApplicationContribution implements FrontendAp
         });
     }
 
+    /**
+     * Prompts the user once per session before migrating legacy `customAgents.yml` files (and before
+     * correcting agents migrated with merged headings by an earlier Theia version). Nothing is written
+     * until the user confirms. Declining is safe: legacy `customAgents.yml` files keep being loaded, so
+     * agents continue to work unmigrated, and the prompt reappears next session until they migrate or
+     * pick "Don't Show Again".
+     *
+     * This is intentionally not backed by a preference: the whole migration path is transitional and
+     * will be removed once the legacy format is no longer supported. The "don't show again" choice is
+     * persisted in local storage (like the dev-container suggestion), so there is no setting to retire.
+     */
+    protected async maybeMigrate(): Promise<void> {
+        if (!this.customizationService || !this.messageService) {
+            return;
+        }
+        if (this.migrationPromptShown || this.migrationCheckInFlight) {
+            return;
+        }
+        this.migrationCheckInFlight = true;
+        try {
+            if (await this.storageService.getData<boolean>(MIGRATION_DONT_SHOW_AGAIN_KEY)) {
+                this.migrationPromptShown = true; // suppress further checks for this session too
+                return;
+            }
+            if (!(await this.customizationService.hasPendingCustomAgentMigration())) {
+                return;
+            }
+            this.migrationPromptShown = true;
+            await this.promptAndMigrate();
+        } catch (e) {
+            this.logger.warn('Custom-agent migration check failed; continuing without migration', e);
+        } finally {
+            this.migrationCheckInFlight = false;
+        }
+    }
+
+    protected async promptAndMigrate(): Promise<void> {
+        if (!this.customizationService || !this.messageService) {
+            return;
+        }
+        const migrate = nls.localizeByDefault('Migrate');
+        const notNow = nls.localizeByDefault('Not Now');
+        const dontShowAgain = nls.localizeByDefault("Don't Show Again");
+        const message = nls.localize(
+            'theia/ai/chat/customAgents/migratePrompt/message',
+            'Your custom agents still use the older `customAgents.yml` format. '
+            + 'We recommend migrating them to the new per-agent format (`agents/<id>/agent.md`), which {0} uses going forward. '
+            + 'Each agent gets its own file next to your `customAgents.yml`, and the original is kept as a `customAgents.yml.bak` backup.',
+            FrontendApplicationConfigProvider.get().applicationName
+        );
+        const choice = await this.messageService.info(message, migrate, notNow, dontShowAgain);
+        if (choice === migrate) {
+            try {
+                const reports = await this.customizationService.migrateCustomAgentsYaml();
+                this.showMigrationSummary(reports);
+            } catch (e) {
+                this.logger.warn('Custom-agent migration failed', e);
+            }
+        } else if (choice === dontShowAgain) {
+            try {
+                await this.storageService.setData(MIGRATION_DONT_SHOW_AGAIN_KEY, true);
+            } catch (e) {
+                this.logger.warn('Failed to persist custom-agent migration prompt state', e);
+            }
+        }
+        // 'Not Now' or dismissal: do nothing; the prompt reappears next session until migrated or dismissed for good.
+    }
+
+    protected showMigrationSummary(reports: CustomAgentMigrationReports): void {
+        if (reports.length === 0) {
+            this.messageService?.info(nls.localize('theia/ai/chat/customAgents/migrationResult/empty', 'Custom-agent migration: nothing to migrate.'));
+            return;
+        }
+        const migrated = reports.reduce((sum, r) => sum + r.migrated, 0);
+        const corrected = reports.reduce((sum, r) => sum + r.corrected, 0);
+        const failed = reports.reduce((sum, r) => sum + r.failed, 0);
+        const backedUp = reports.filter(r => r.yamlBackedUp).length;
+        const overrides = reports.reduce((sum, r) => sum + r.promptOverridesMigrated, 0);
+        const message = nls.localize(
+            'theia/ai/chat/customAgents/migrationResult',
+            'Custom-agent migration: {0} migrated, {1} corrected, {2} failed, {3} legacy customAgents.yml backed up to .bak, {4} prompt overrides folded into agent.md.',
+            migrated, corrected, failed, backedUp, overrides
+        );
+        // Keep the result visible until the user dismisses it (no auto-hide timeout).
+        this.messageService?.info(message, { timeout: 0 });
+    }
+
     onStop(): void {
     }
 }