vscode-install command

Author: mohammedahmed18

codeflash/cli_cmds/cli.py

@@ -5,7 +5,7 @@
 
 from codeflash.cli_cmds import logging_config
 from codeflash.cli_cmds.cli_common import apologize_and_exit
-from codeflash.cli_cmds.cmd_init import init_codeflash, install_github_actions
+from codeflash.cli_cmds.cmd_init import init_codeflash, install_github_actions, install_vscode_extension
 from codeflash.cli_cmds.console import logger
 from codeflash.code_utils import env_utils
 from codeflash.code_utils.code_utils import exit_with_message
@@ -21,6 +21,9 @@ def parse_args() -> Namespace:
     init_parser = subparsers.add_parser("init", help="Initialize Codeflash for a Python project.")
     init_parser.set_defaults(func=init_codeflash)
 
+    install_vscode_parser = subparsers.add_parser("vscode-install", help="Install the Codeflash VSCode extension")
+    install_vscode_parser.set_defaults(func=install_vscode_extension)
+
     init_actions_parser = subparsers.add_parser("init-actions", help="Initialize GitHub Actions workflow")
     init_actions_parser.set_defaults(func=install_github_actions)