I have my command set up to open this URL:
vscode://file/<ABSOLUTE_PATH_TO_YOUR_DIR>
The "file" above is important, first time I set it up I missed it and it didn't work correctly.