I installed VS Code on Ubuntu focal (20.04) via apt and every time I upgrade, the CLI command is gone again. The "Install code command ..." in the UI also does not exist for me.

This is what solves it for me:

sudo ln -s /usr/share/code/bin/code /usr/bin/code