Layout on content change & track some disposables (microsoft#258870)

Fixes microsoft#258432

Author: TylerLeonhardt

src/vs/platform/quickinput/browser/quickInputController.ts

@@ -266,6 +266,7 @@ export class QuickInputController extends Disposable {
 				tree.tree.setFocus([]);
 			}, 0);
 		}));
+		this._register(tree.tree.onDidChangeContentHeight(() => this.updateLayout()));
 
 		const focusTracker = dom.trackFocus(container);
 		this._register(focusTracker);

src/vs/workbench/contrib/chat/browser/actions/chatToolPicker.ts

@@ -533,7 +533,7 @@ async function showToolsPickerTree(
 	};
 	treePicker.title = localize('configureTools', "Configure Tools");
 	treePicker.buttons = [addMcpServerButton, installExtension, configureToolSets];
-	treePicker.onDidTriggerButton(button => {
+	store.add(treePicker.onDidTriggerButton(button => {
 		if (button === addMcpServerButton) {
 			commandService.executeCommand(McpCommandIds.AddConfiguration);
 		} else if (button === installExtension) {
@@ -542,11 +542,11 @@ async function showToolsPickerTree(
 			commandService.executeCommand(ConfigureToolSets.ID);
 		}
 		treePicker.hide();
-	});
+	}));
 
 	treePicker.show();
 
-	await Promise.race([Event.toPromise(Event.any(treePicker.onDidAccept, treePicker.onDidHide))]);
+	await Promise.race([Event.toPromise(Event.any(treePicker.onDidAccept, treePicker.onDidHide), store)]);
 
 	store.dispose();