Merge pull request microsoft#242594 from microsoft/roblou/thoughtless-mastodon

Prevent using an invalid model with agent mode

Author: connor4312

src/vs/workbench/contrib/chat/browser/chatInputPart.ts

@@ -415,6 +415,7 @@ export class ChatInputPart extends Disposable implements IHistoryNavigationWidge
 			if (model) {
 				this._currentLanguageModel = { metadata: model, identifier: persistedSelection };
 				this._onDidChangeCurrentLanguageModel.fire(this._currentLanguageModel);
+				this.checkModelSupported();
 			} else {
 				this._waitForPersistedLanguageModel.value = this.languageModelsService.onDidChangeLanguageModels(e => {
 					const persistedModel = e.added?.find(m => m.identifier === persistedSelection);
@@ -424,19 +425,24 @@ export class ChatInputPart extends Disposable implements IHistoryNavigationWidge
 						if (persistedModel.metadata.isUserSelectable) {
 							this._currentLanguageModel = { metadata: persistedModel.metadata, identifier: persistedSelection };
 							this._onDidChangeCurrentLanguageModel.fire(this._currentLanguageModel!);
+							this.checkModelSupported();
 						}
 					}
 				});
 			}
 		}
 
 		this._register(this.chatAgentService.onDidChangeToolsAgentModeEnabled(() => {
-			if (this._currentLanguageModel && !this.modelSupportedForDefaultAgent(this._currentLanguageModel)) {
-				this.setCurrentLanguageModelToDefault();
-			}
+			this.checkModelSupported();
 		}));
 	}
 
+	private checkModelSupported(): void {
+		if (this._currentLanguageModel && !this.modelSupportedForDefaultAgent(this._currentLanguageModel)) {
+			this.setCurrentLanguageModelToDefault();
+		}
+	}
+
 	private supportsVision(): boolean {
 		return this._currentLanguageModel?.metadata.capabilities?.vision ?? false;
 	}
@@ -475,6 +481,9 @@ export class ChatInputPart extends Disposable implements IHistoryNavigationWidge
 		this._currentLanguageModel = hasUserSelectableLanguageModels && defaultLanguageModelId ?
 			{ metadata: this.languageModelsService.lookupLanguageModel(defaultLanguageModelId)!, identifier: defaultLanguageModelId } :
 			undefined;
+		if (this._currentLanguageModel) {
+			this._onDidChangeCurrentLanguageModel.fire(this._currentLanguageModel);
+		}
 	}
 
 	private setCurrentLanguageModelByUser(model: ILanguageModelChatMetadataAndIdentifier) {