Skip to content

Commit 59c3f0b

Browse files
authored
Fix model storage not updating properly (microsoft#259624)
1 parent 6ccc57f commit 59c3f0b

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

src/vs/workbench/contrib/chat/common/languageModels.ts

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -363,13 +363,16 @@ export class LanguageModelsService implements ILanguageModelsService {
363363
this._logService.warn(`[LM] Cannot update model picker preference for unknown model ${modelIdentifier}`);
364364
return;
365365
}
366-
delete this._modelPickerUserPreferences[modelIdentifier];
367-
if (model.isUserSelectable !== showInModelPicker) {
368-
this._modelPickerUserPreferences[modelIdentifier] = showInModelPicker;
366+
367+
this._modelPickerUserPreferences[modelIdentifier] = showInModelPicker;
368+
if (showInModelPicker === model.isUserSelectable) {
369+
delete this._modelPickerUserPreferences[modelIdentifier];
370+
this._storageService.store('chatModelPickerPreferences', this._modelPickerUserPreferences, StorageScope.PROFILE, StorageTarget.USER);
371+
} else if (model.isUserSelectable !== showInModelPicker) {
369372
this._storageService.store('chatModelPickerPreferences', this._modelPickerUserPreferences, StorageScope.PROFILE, StorageTarget.USER);
370-
this._onLanguageModelChange.fire();
371-
this._logService.trace(`[LM] Updated model picker preference for ${modelIdentifier} to ${showInModelPicker}`);
372373
}
374+
this._onLanguageModelChange.fire();
375+
this._logService.trace(`[LM] Updated model picker preference for ${modelIdentifier} to ${showInModelPicker}`);
373376
}
374377

375378
getVendors(): IUserFriendlyLanguageModel[] {

0 commit comments

Comments
 (0)