productivity setting to add any missing imports when you save your file in vscode or with an explicit shortcut
{
"key": "ctrl+o",
"command": "editor.action.sourceAction",
"args": {
"kind": "source.addMissingImports",
"apply": "first"
}
},
"editor.codeActionsOnSave": { "source.addMissingImports": "always" },