[SCP] Fix the code for openapi changes (#3826)

* fix the code for scp openapi change fix the code for scp openapi change * fix the code for scp openapi change fix the code for scp openapi change
skypilot-org · Aug 15, 2024 · df2a9b4 · df2a9b4
1 parent 6f89fa5
commit df2a9b4
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/sky/clouds/utils/scp_utils.py b/sky/clouds/utils/scp_utils.py
@@ -223,6 +223,7 @@ def _delete(self, url, request_body=None):
     def create_security_group(self, zone_id, vpc, sg_name):
         url = f'{API_ENDPOINT}/security-group/v3/security-groups'
         request_body = {
+            'loggable': False,
             'securityGroupName': sg_name,
             'serviceZoneId': zone_id,
             'vpcId': vpc,

diff --git a/sky/skylet/providers/scp/config.py b/sky/skylet/providers/scp/config.py
@@ -114,13 +114,21 @@ def get_vcp_subnets(self):
 
     def _get_vm_init_script(self, ssh_public_key):
 
+        import subprocess
         init_script_content = self._get_default_config_cmd(
         ) + self._get_ssh_key_gen_cmd(ssh_public_key)
+        init_script_content_string = f'"{init_script_content}"'
+        command = f'echo {init_script_content_string} | base64'
+        result = subprocess.run(command,
+                                shell=True,
+                                capture_output=True,
+                                text=True)
+        init_script_content_base64 = result.stdout
         return {
-            "encodingType": "plain",
+            "encodingType": "base64",
             "initialScriptShell": "bash",
             "initialScriptType": "text",
-            "initialScriptContent": init_script_content
+            "initialScriptContent": init_script_content_base64
         }
 
     def _get_ssh_key_gen_cmd(self, ssh_public_key):