strict checking for context-name in Node Pools to start with "ssh-"

sky/backends/cloud_vm_ray_backend.py

@@ -1480,6 +1480,7 @@ def _insufficient_resources_msg(
         elif to_provision.region is not None and to_provision.cloud is not None:
             # For public clouds, provision.region is always set.
             if clouds.SSH().is_same_cloud(to_provision.cloud):
+                assert to_provision.region.startswith("ssh-"), "SSH context must start with 'ssh-'"
                 message += (
                     f'in SSH Node Pool ({to_provision.region.lstrip("ssh-")}) '
                     f'for {requested_resources}. The SSH Node Pool may not '

sky/check.py

@@ -591,6 +591,7 @@ def _dim_color(str_to_format: str) -> str:
                                               'configuration.'))
                 else:
                     # Default case - not set up
+                    assert context.startswith("ssh-"), "SSH context must start with 'ssh-'"
                     text_suffix = (': ' + _red_color('disabled. ') +
                                    _dim_color('Reason: Not set up. Use '
                                               '`sky ssh up --infra '

sky/clouds/ssh.py

@@ -252,7 +252,7 @@ def check_single_context(cls, context: str) -> Tuple[bool, str]:
     @classmethod
     def expand_infras(cls) -> List[str]:
         return [
-            f'{cls.canonical_name()}/{c.lstrip("ssh-")}'
+            f'{cls.canonical_name()}/{c.lstrip("ssh-")}' if c.startswith("ssh-") else "."
             for c in cls.existing_allowed_contexts(silent=True)
         ]
 

sky/optimizer.py

@@ -1289,6 +1289,7 @@ def _check_specified_regions(task: task_lib.Task) -> None:
         msg = f'Task{task_name} requires '
         if region not in existing_contexts:
             if is_ssh:
+                assert region.startswith("ssh-"), "SSH context must start with 'ssh-'"
                 infra_str = f'SSH/{region.lstrip("ssh-")}'
             else:
                 infra_str = f'Kubernetes/{region}'