diff --git a/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/config.py b/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/config.py index 76f79632c6..332e5a7108 100644 --- a/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/config.py +++ b/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/config.py @@ -34,8 +34,10 @@ def add_extensions(self, extensions: Union[str, List[str]]): COPILOT_EXTENSION = ( "https://raw.githubusercontent.com/flyteorg/flytetools/master/flytekitplugins/flyin/GitHub.copilot-1.138.563.vsix" ) -VIM_EXTENSION = "https://open-vsx.org/api/vscodevim/vim/1.27.0/file/vscodevim.vim-1.27.0.vsix" -CODE_TOGETHER_EXTENSION = "https://openvsxorg.blob.core.windows.net/resources/genuitecllc/codetogether/2023.2.0/genuitecllc.codetogether-2023.2.0.vsix" +VIM_EXTENSION = ( + "https://raw.githubusercontent.com/flyteorg/flytetools/master/flytekitplugins/flyin/vscodevim.vim-1.27.0.vsix" +) +CODE_TOGETHER_EXTENSION = "https://raw.githubusercontent.com/flyteorg/flytetools/master/flytekitplugins/flyin/genuitecllc.codetogether-2023.2.0.vsix" # Predefined VSCode config with extensions VIM_CONFIG = VscodeConfig( diff --git a/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/constants.py b/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/constants.py index 8de8348a0a..faabbfcaee 100644 --- a/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/constants.py +++ b/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/constants.py @@ -10,8 +10,8 @@ "arm64": "https://github.com/coder/code-server/releases/download/v4.18.0/code-server-4.18.0-linux-arm64.tar.gz", } DEFAULT_CODE_SERVER_EXTENSIONS = [ - "https://open-vsx.org/api/ms-python/python/2023.20.0/file/ms-python.python-2023.20.0.vsix", - "https://open-vsx.org/api/ms-toolsai/jupyter/2023.9.100/file/ms-toolsai.jupyter-2023.9.100.vsix", + "https://raw.githubusercontent.com/flyteorg/flytetools/master/flytekitplugins/flyin/ms-python.python-2023.20.0.vsix", + "https://raw.githubusercontent.com/flyteorg/flytetools/master/flytekitplugins/flyin/ms-toolsai.jupyter-2023.9.100.vsix", ] DEFAULT_CODE_SERVER_DIR_NAMES = { "amd64": "code-server-4.18.0-linux-amd64", diff --git a/plugins/flytekit-flyin/tests/test_flyin_plugin.py b/plugins/flytekit-flyin/tests/test_flyin_plugin.py index 7eb937394d..085532e8fc 100644 --- a/plugins/flytekit-flyin/tests/test_flyin_plugin.py +++ b/plugins/flytekit-flyin/tests/test_flyin_plugin.py @@ -237,6 +237,12 @@ def test_vscode_config(): assert vim_config.code_server_dir_names == DEFAULT_CODE_SERVER_DIR_NAMES assert vim_config.extension_remote_paths == DEFAULT_CODE_SERVER_EXTENSIONS + [VIM_EXTENSION] + all_extensions_config = VscodeConfig() + all_extensions_config.add_extensions([CODE_TOGETHER_EXTENSION, COPILOT_EXTENSION, VIM_EXTENSION]) + assert CODE_TOGETHER_EXTENSION in all_extensions_config.extension_remote_paths + assert COPILOT_EXTENSION in all_extensions_config.extension_remote_paths + assert VIM_EXTENSION in all_extensions_config.extension_remote_paths + def test_vscode_config_add_extensions(): additional_extensions = [COPILOT_EXTENSION, VIM_EXTENSION, CODE_TOGETHER_EXTENSION]