[Fix] Mode profile choices leak between VS Code windows

src/__tests__/ClineProvider.delegation.spec.ts

@@ -294,6 +294,8 @@ describe("ClineProvider.delegateParentAndOpenChild()", () => {
 			isViewLaunched: false,
 			taskHistoryStore: makeStoreStub(),
 			providerSettingsManager,
+			// No window-local choice is recorded, so the lookup reaches the shared manager.
+			getModeConfigId: (mode: string) => providerSettingsManager.getModeConfigId(mode),
 			context: { workspaceState: { get: workspaceGet } },
 		} as unknown as ClineProvider
 
@@ -325,6 +327,53 @@ describe("ClineProvider.delegateParentAndOpenChild()", () => {
 		)
 	})
 
+	it("uses this window's mode profile instead of one another window saved in shared storage", async () => {
+		const parentTask = makeParentTask()
+		const createTask = vi
+			.fn()
+			.mockResolvedValue({ taskId: "child-local", run: vi.fn().mockResolvedValue(undefined) })
+		const getProfile = vi.fn().mockResolvedValue({
+			name: "local-profile",
+			id: "local-profile-id",
+			apiProvider: providerIdentifiers.openrouter,
+			openRouterModelId: "openai/gpt-4.1-mini",
+		})
+		const sharedGetModeConfigId = vi.fn().mockResolvedValue("other-window-profile-id")
+		const provider = {
+			taskScheduler: new TaskScheduler(),
+			emit: vi.fn(),
+			getCurrentTask: vi.fn(() => parentTask),
+			removeClineFromStack: vi.fn().mockResolvedValue(undefined),
+			createTask,
+			log: vi.fn(),
+			isViewLaunched: false,
+			taskHistoryStore: makeStoreStub(),
+			providerSettingsManager: { getModeConfigId: sharedGetModeConfigId, getProfile },
+			// This window's own choice differs from the shared value another window wrote.
+			getModeConfigId: vi.fn().mockResolvedValue("local-profile-id"),
+			context: { workspaceState: { get: vi.fn().mockReturnValue(false) } },
+		} as unknown as ClineProvider
+
+		await ClineProvider.prototype.delegateParentAndOpenChild.call(provider, {
+			parentTaskId: "parent-1",
+			message: "Local child",
+			initialTodos: [],
+			mode: "ask",
+		})
+
+		expect(getProfile).toHaveBeenCalledOnce()
+		expect(getProfile).toHaveBeenCalledWith({ id: "local-profile-id" })
+		expect(sharedGetModeConfigId).not.toHaveBeenCalled()
+		expect(createTask).toHaveBeenCalledWith(
+			"Local child",
+			undefined,
+			parentTask,
+			expect.objectContaining({
+				handoffExecutionContext: expect.objectContaining({ mode: "ask", apiConfigName: "local-profile" }),
+			}),
+		)
+	})
+
 	it.each([
 		{ name: "has no saved mode profile", savedConfigId: undefined, savedProfile: undefined },
 		{
@@ -346,6 +395,7 @@ describe("ClineProvider.delegateParentAndOpenChild()", () => {
 				? vi.fn().mockRejectedValue(savedProfile)
 				: vi.fn().mockResolvedValue(savedProfile)
 		const log = vi.fn()
+		const getModeConfigId = vi.fn().mockResolvedValue(savedConfigId)
 		const provider = {
 			taskScheduler: new TaskScheduler(),
 			emit: vi.fn(),
@@ -355,10 +405,9 @@ describe("ClineProvider.delegateParentAndOpenChild()", () => {
 			log,
 			isViewLaunched: false,
 			taskHistoryStore: makeStoreStub(),
-			providerSettingsManager: {
-				getModeConfigId: vi.fn().mockResolvedValue(savedConfigId),
-				getProfile,
-			},
+			providerSettingsManager: { getModeConfigId, getProfile },
+			// No window-local choice is recorded, so the lookup reaches the shared manager.
+			getModeConfigId,
 			context: { workspaceState: { get: vi.fn().mockReturnValue(false) } },
 		} as unknown as ClineProvider
 

src/__tests__/extension.spec.ts

@@ -202,6 +202,7 @@ vi.mock("../core/webview/ClineProvider", async () => {
 		contextProxy: { getGlobalState: vi.fn() },
 		customModesManager: {},
 		upsertProviderProfile: vi.fn().mockResolvedValue(undefined),
+		adoptModeConfigIds: vi.fn(),
 	}
 	return {
 		ClineProvider: Object.assign(
@@ -245,6 +246,21 @@ describe("extension.ts", () => {
 		settingsUpdatedHandler = undefined
 	})
 
+	test("adopts mode profile mappings from the activation auto-import callback", async () => {
+		const { activate } = await import("../extension")
+		const { autoImportSettings } = await import("../utils/autoImportSettings")
+		const { ClineProvider } = await import("../core/webview/ClineProvider")
+		const ids = { code: "imported-id" }
+		vi.mocked(autoImportSettings).mockImplementationOnce(async (_output, options) => {
+			expect(options.onModeConfigIdsImported).toBeTypeOf("function")
+			options.onModeConfigIdsImported?.(ids)
+		})
+
+		await activate(mockContext)
+
+		expect(ClineProvider.getVisibleInstance()!.adoptModeConfigIds).toHaveBeenCalledExactlyOnceWith(ids)
+	})
+
 	test("shares the extension-owned focus tracker with the sidebar and code actions and disposes its listeners", async () => {
 		vi.resetModules()
 		const { WebviewFocusTracker } = await import("../core/webview/WebviewFocusTracker")

src/core/config/ProviderSettingsManager.ts

@@ -543,7 +543,7 @@
 	/**
 	 * Set the API config for a specific mode.
 	 */
-	public async setModeConfig(mode: Mode, configId: string) {
+	public async setModeConfig(mode: Mode, configId: string, onPersisted?: () => void) {
 		try {
 			return await this.lock(async () => {
 				const providerProfiles = await this.load()
@@ -554,6 +554,7 @@
 				// Assign the chosen config ID to this mode
 				providerProfiles.modeApiConfigs[mode] = configId
 				await this.store(providerProfiles)
+				onPersisted?.()
 			})
 		} catch (error) {
 			throw new Error(`Failed to set mode config: ${error}`)
@@ -574,6 +575,21 @@
 		}
 	}
 
+	/**
+	 * Get a copy of the persisted mode -> API config ID map. The map is shared by every VS Code
+	 * window, so callers that need a stable view must keep their own copy.
+	 */
+	public async getModeConfigIds(): Promise<Record<string, string>> {
+		try {
+			return await this.lock(async () => {
+				const { modeApiConfigs } = await this.load()
+				return { ...modeApiConfigs }
+			})
+		} catch (error) {
+			throw new Error(`Failed to get mode configs: ${error}`)
+		}
+	}
+
 	public async export() {
 		try {
 			return await this.lock(async () => {
@@ -636,9 +652,13 @@
 		}
 	}
 
-	public async import(providerProfiles: ProviderProfiles) {
+	public async import(providerProfiles: ProviderProfiles, onPersisted?: () => void) {
 		try {
-			return await this.lock(() => this.store(providerProfiles))
+			return await this.lock(async () => {
+				await this.store(providerProfiles)
+				// Publish local choices before a later writer can acquire the storage lock.
+				onPersisted?.()
+			})
 		} catch (error) {
 			throw new Error(`Failed to import provider profiles: ${error}`)
 		}

src/core/config/__tests__/ProviderSettingsManager.spec.ts

@@ -89,6 +89,36 @@ describe("ProviderSettingsManager", () => {
 		providerSettingsManager = new ProviderSettingsManager(mockContext)
 	})
 
+	it.each([
+		{ operation: "import", fails: false },
+		{ operation: "import", fails: true },
+		{ operation: "mode write", fails: false },
+		{ operation: "mode write", fails: true },
+	])("publishes $operation choices only after persistence (failure: $fails)", async ({ operation, fails }) => {
+		let release = () => {}
+		const pending = new Promise<void>((resolve) => {
+			release = resolve
+		})
+		mockSecrets.store.mockImplementationOnce(async () => {
+			await pending
+			if (fails) throw new Error("write failed")
+		})
+		const published = vi.fn()
+		const write =
+			operation === "import"
+				? providerSettingsManager.import(
+						{ currentApiConfigName: "default", apiConfigs: {}, modeApiConfigs: { code: "imported-id" } },
+						published,
+					)
+				: providerSettingsManager.setModeConfig("code", "chosen-id", published)
+		await vi.waitFor(() => expect(mockSecrets.store).toHaveBeenCalledOnce())
+		expect(published).not.toHaveBeenCalled()
+		const result = fails ? expect(write).rejects.toThrow("write failed") : expect(write).resolves.toBeUndefined()
+		release()
+		await result
+		expect(published).toHaveBeenCalledTimes(fails ? 0 : 1)
+	})
+
 	describe("initialize", () => {
 		it("should not write to storage when secrets.get returns null", async () => {
 			// Mock readConfig to return null
@@ -1352,6 +1382,41 @@ describe("ProviderSettingsManager", () => {
 		})
 	})
 
+	describe("GetModeConfigIds", () => {
+		const profilesWithModes = (modeApiConfigs?: Record<string, string>) =>
+			JSON.stringify({
+				currentApiConfigName: "default",
+				apiConfigs: { default: { id: "default" } },
+				modeApiConfigs,
+				migrations: { rateLimitSecondsMigrated: true },
+			})
+
+		it("returns a copy of the stored mode map", async () => {
+			mockSecrets.get.mockResolvedValue(profilesWithModes({ code: "code-id", architect: "architect-id" }))
+
+			const ids = await providerSettingsManager.getModeConfigIds()
+			ids.code = "changed"
+
+			expect(ids).toEqual({ code: "changed", architect: "architect-id" })
+			await expect(providerSettingsManager.getModeConfigIds()).resolves.toEqual({
+				code: "code-id",
+				architect: "architect-id",
+			})
+		})
+
+		it("returns an empty map when none is stored", async () => {
+			mockSecrets.get.mockResolvedValue(profilesWithModes(undefined))
+
+			await expect(providerSettingsManager.getModeConfigIds()).resolves.toEqual({})
+		})
+
+		it("should throw error if secrets storage fails", async () => {
+			mockSecrets.get.mockRejectedValue(new Error("Storage failed"))
+
+			await expect(providerSettingsManager.getModeConfigIds()).rejects.toThrow("Failed to get mode configs")
+		})
+	})
+
 	describe("HasConfig", () => {
 		it("should return true for existing config", async () => {
 			const existingConfig: ProviderProfiles = {

src/core/config/__tests__/importExport.spec.ts

@@ -187,13 +187,38 @@ describe("importExport", () => {
 		beforeEach(() => {
 			;(fs.readFile as Mock).mockResolvedValue(validFileContent)
 			mockProviderSettingsManager.export.mockResolvedValue(previousProviderProfiles)
-			mockProviderSettingsManager.import.mockResolvedValue(undefined)
+			mockProviderSettingsManager.import.mockImplementation(async (_profiles, onPersisted) => {
+				onPersisted?.()
+			})
 			mockProviderSettingsManager.listConfig.mockResolvedValue(profileMetadata)
 			mockContextProxy.setValues.mockResolvedValue(undefined)
 			mockContextProxy.setValue.mockResolvedValue(undefined)
 			mockContextProxy.setProviderSettings.mockResolvedValue(undefined)
 		})
 
+		it.each([false, true])(
+			"updates the importing window only after successful persistence (failure: %s)",
+			async (fails) => {
+				const modeApiConfigs = { "new-custom-mode": "test-id" }
+				const file = JSON.parse(validFileContent)
+				file.providerProfiles.modeApiConfigs = modeApiConfigs
+				vi.mocked(fs.readFile).mockResolvedValue(JSON.stringify(file))
+				const onModeConfigIdsImported = vi.fn()
+				if (fails) mockProviderSettingsManager.import.mockRejectedValueOnce(new Error("write failed"))
+
+				const result = await importSettingsFromPath("/mock/path/settings.json", {
+					providerSettingsManager: mockProviderSettingsManager,
+					contextProxy: mockContextProxy,
+					customModesManager: mockCustomModesManager,
+					onModeConfigIdsImported,
+				})
+
+				expect(result.success).toBe(!fails)
+				if (fails) expect(onModeConfigIdsImported).not.toHaveBeenCalled()
+				else expect(onModeConfigIdsImported).toHaveBeenCalledExactlyOnceWith(modeApiConfigs)
+			},
+		)
+
 		it("waits for each provider-state write before reporting success", async () => {
 			const currentProfileWrite = createDeferred<void>()
 			const providerSettingsWrite = createDeferred<void>()
@@ -638,6 +663,7 @@ describe("importExport", () => {
 			// Create a mock provider for the test
 			const mockProvider = {
 				settingsImportedAt: 0,
+				adoptModeConfigIds: vi.fn(),
 				postStateToWebview: vi.fn().mockResolvedValue(undefined),
 				postStateToWebviewWithoutTaskHistory: vi.fn().mockResolvedValue(undefined),
 			}
@@ -921,6 +947,7 @@ describe("importExport", () => {
 				const seenImportedAt: Array<number | undefined> = []
 				const mockProvider = {
 					settingsImportedAt: undefined as number | undefined,
+					adoptModeConfigIds: vi.fn(),
 					postStateToWebview: vi.fn().mockImplementation(async () => {
 						seenImportedAt.push(mockProvider.settingsImportedAt)
 					}),
@@ -997,6 +1024,7 @@ describe("importExport", () => {
 				const seenImportedAt: Array<number | undefined> = []
 				const mockProvider = {
 					settingsImportedAt: undefined as number | undefined,
+					adoptModeConfigIds: vi.fn(),
 					postStateToWebview: vi.fn().mockImplementation(async () => {
 						seenImportedAt.push(mockProvider.settingsImportedAt)
 					}),
@@ -1268,6 +1296,7 @@ describe("importExport", () => {
 
 				const mockProvider = {
 					settingsImportedAt: 0,
+					adoptModeConfigIds: vi.fn(),
 					postStateToWebview: vi.fn().mockResolvedValue(undefined),
 				}
 
@@ -1537,6 +1566,7 @@ describe("importExport", () => {
 
 				const mockProvider = {
 					settingsImportedAt: 0,
+					adoptModeConfigIds: vi.fn(),
 					postStateToWebview: vi.fn().mockResolvedValue(undefined),
 				}