skypilot-nightly 1.0.0.dev20250912__py3-none-any.whl → 1.0.0.dev20250913__py3-none-any.whl

sky/__init__.py

@@ -7,7 +7,7 @@ import urllib.request
 from sky.utils import directory_utils
 
 # Replaced with the current commit when building the wheels.
-_SKYPILOT_COMMIT_SHA = '16e9ad33e647ab3859a2b5624be7386721c9ef8b'
+_SKYPILOT_COMMIT_SHA = 'bf9b3c4e09e97cf2dafed9f351d0b36438adf4ec'
 
 
 def _get_git_commit():
@@ -37,7 +37,7 @@ def _get_git_commit():
 
 
 __commit__ = _get_git_commit()
-__version__ = '1.0.0.dev20250912'
+__version__ = '1.0.0.dev20250913'
 __root_dir__ = directory_utils.get_sky_dir()
 
 
@@ -149,6 +149,7 @@ Vsphere = clouds.Vsphere
 Fluidstack = clouds.Fluidstack
 Nebius = clouds.Nebius
 Hyperbolic = clouds.Hyperbolic
+Seeweb = clouds.Seeweb
 
 __all__ = [
     '__version__',
@@ -169,6 +170,7 @@ __all__ = [
     'Fluidstack',
     'Nebius',
     'Hyperbolic',
+    'Seeweb',
     'Optimizer',
     'OptimizeTarget',
     'backends',

sky/adaptors/seeweb.py

@@ -0,0 +1,103 @@
+""" Seeweb Adaptor """
+import configparser
+from pathlib import Path
+
+from sky.adaptors import common
+from sky.utils import annotations
+
+
+class SeewebError(Exception):
+    """Base exception for Seeweb adaptor errors."""
+
+
+class SeewebCredentialsFileNotFound(SeewebError):
+    """Raised when the Seeweb credentials file is missing."""
+
+
+class SeewebApiKeyMissing(SeewebError):
+    """Raised when the Seeweb API key is missing or empty."""
+
+
+class SeewebAuthenticationError(SeewebError):
+    """Raised when authenticating with Seeweb API fails."""
+
+
+_IMPORT_ERROR_MESSAGE = ('Failed to import dependencies for Seeweb.'
+                         'Try pip install "skypilot[seeweb]"')
+
+ecsapi = common.LazyImport(
+    'ecsapi',
+    import_error_message=_IMPORT_ERROR_MESSAGE,
+)
+boto3 = common.LazyImport('boto3', import_error_message=_IMPORT_ERROR_MESSAGE)
+botocore = common.LazyImport('botocore',
+                             import_error_message=_IMPORT_ERROR_MESSAGE)
+
+_LAZY_MODULES = (ecsapi, boto3, botocore)
+
+
+@common.load_lazy_modules(_LAZY_MODULES)
+def check_compute_credentials() -> bool:
+    """Checks if the user has access credentials to Seeweb's compute service.
+
+    Returns True if credentials are valid; otherwise raises a SeewebError.
+    """
+    # Read API key from standard Seeweb configuration file
+    key_path = Path('~/.seeweb_cloud/seeweb_keys').expanduser()
+    if not key_path.exists():
+        raise SeewebCredentialsFileNotFound(
+            'Missing Seeweb API key file ~/.seeweb_cloud/seeweb_keys')
+
+    parser = configparser.ConfigParser()
+    parser.read(key_path)
+    try:
+        api_key = parser['DEFAULT']['api_key'].strip()
+    except KeyError as e:
+        raise SeewebApiKeyMissing(
+            'Missing api_key in ~/.seeweb_cloud/seeweb_keys') from e
+    if not api_key:
+        raise SeewebApiKeyMissing(
+            'Empty api_key in ~/.seeweb_cloud/seeweb_keys')
+
+    # Test connection by fetching servers list to validate the key
+    try:
+        seeweb_client = ecsapi.Api(token=api_key)
+        seeweb_client.fetch_servers()
+    except Exception as e:  # pylint: disable=broad-except
+        raise SeewebAuthenticationError(
+            f'Unable to authenticate with Seeweb API: {e}') from e
+
+    return True
+
+
+@common.load_lazy_modules(_LAZY_MODULES)
+def check_storage_credentials() -> bool:
+    """Checks if the user has access credentials to Seeweb's storage service.
+
+    Mirrors compute credentials validation.
+    """
+    return check_compute_credentials()
+
+
+@common.load_lazy_modules(_LAZY_MODULES)
+@annotations.lru_cache(scope='global', maxsize=1)
+def client():
+    """Returns an authenticated ecsapi.Api object."""
+    # Create authenticated client using the same credential pattern
+    key_path = Path('~/.seeweb_cloud/seeweb_keys').expanduser()
+    if not key_path.exists():
+        raise SeewebCredentialsFileNotFound(
+            'Missing Seeweb API key file ~/.seeweb_cloud/seeweb_keys')
+
+    parser = configparser.ConfigParser()
+    parser.read(key_path)
+    try:
+        api_key = parser['DEFAULT']['api_key'].strip()
+    except KeyError as e:
+        raise SeewebApiKeyMissing(
+            'Missing api_key in ~/.seeweb_cloud/seeweb_keys') from e
+    if not api_key:
+        raise SeewebApiKeyMissing(
+            'Empty api_key in ~/.seeweb_cloud/seeweb_keys')
+
+    return ecsapi.Api(token=api_key)

sky/authentication.py

@@ -40,6 +40,7 @@ from sky.adaptors import gcp
 from sky.adaptors import ibm
 from sky.adaptors import kubernetes
 from sky.adaptors import runpod
+from sky.adaptors import seeweb as seeweb_adaptor
 from sky.adaptors import vast
 from sky.provision.fluidstack import fluidstack_utils
 from sky.provision.kubernetes import utils as kubernetes_utils
@@ -601,3 +602,40 @@ def setup_hyperbolic_authentication(config: Dict[str, Any]) -> Dict[str, Any]:
     config['auth']['ssh_public_key'] = public_key_path
 
     return configure_ssh_info(config)
+
+
+def setup_seeweb_authentication(config: Dict[str, Any]) -> Dict[str, Any]:
+    """Registers the public key with Seeweb and notes the remote name."""
+    # 1. local key pair
+    get_or_generate_keys()
+
+    # 2. public key
+    _, public_key_path = get_or_generate_keys()
+    with open(public_key_path, 'r', encoding='utf-8') as f:
+        public_key = f.read().strip()
+
+    # 3. Seeweb API client
+    client = seeweb_adaptor.client()
+
+    # 4. Check if key is already registered
+    prefix = f'sky-key-{common_utils.get_user_hash()}'
+    remote_name = None
+    for k in client.fetch_ssh_keys():
+        if k.key.strip() == public_key:
+            remote_name = k.label  # already present
+            break
+
+    # 5. doesn't exist, choose a unique name and create it
+    if remote_name is None:
+        suffix = 1
+        remote_name = prefix
+        existing_names = {k.label for k in client.fetch_ssh_keys()}
+        while remote_name in existing_names:
+            suffix += 1
+            remote_name = f'{prefix}-{suffix}'
+        client.create_ssh_key(label=remote_name, key=public_key)
+
+    # 6. Put the remote name in cluster-config (like for Lambda)
+    config['auth']['remote_key_name'] = remote_name
+
+    return config

sky/backends/backend_utils.py

@@ -1116,6 +1116,8 @@ def _add_auth_to_cluster_config(cloud: clouds.Cloud, tmp_yaml_path: str):
         config = auth.setup_fluidstack_authentication(config)
     elif isinstance(cloud, clouds.Hyperbolic):
         config = auth.setup_hyperbolic_authentication(config)
+    elif isinstance(cloud, clouds.Seeweb):
+        config = auth.setup_seeweb_authentication(config)
     else:
         assert False, cloud
     yaml_utils.dump_yaml(tmp_yaml_path, config)
@@ -2333,7 +2335,8 @@ def _update_cluster_status(cluster_name: str) -> Optional[Dict[str, Any]]:
                                                 handle,
                                                 requested_resources=None,
                                                 ready=True,
-                                                is_launch=False)
+                                                is_launch=False,
+                                                update_only=True)
         return global_user_state.get_cluster_from_name(cluster_name)
 
     # All cases below are transitioning the cluster to non-UP states.
@@ -2543,7 +2546,8 @@ def _update_cluster_status(cluster_name: str) -> Optional[Dict[str, Any]]:
                                                 handle,
                                                 requested_resources=None,
                                                 ready=False,
-                                                is_launch=False)
+                                                is_launch=False,
+                                                update_only=True)
         return global_user_state.get_cluster_from_name(cluster_name)
     # Now is_abnormal is False: either node_statuses is empty or all nodes are
     # STOPPED.
@@ -3719,13 +3723,24 @@ def invoke_skylet_with_retries(func: Callable[..., T]) -> T:
             return func()
         except grpc.RpcError as e:
             last_exception = e
-            if e.code() == grpc.StatusCode.INTERNAL:
-                with ux_utils.print_exception_no_traceback():
-                    raise exceptions.SkyletInternalError(e.details())
-            elif e.code() == grpc.StatusCode.UNAVAILABLE:
-                time.sleep(backoff.current_backoff())
-            else:
-                raise e
+            _handle_grpc_error(e, backoff.current_backoff())
+
     raise RuntimeError(
         f'Failed to invoke Skylet after {max_attempts} attempts: {last_exception}'
     ) from last_exception
+
+
+def _handle_grpc_error(e: 'grpc.RpcError', current_backoff: float) -> None:
+    if e.code() == grpc.StatusCode.INTERNAL:
+        with ux_utils.print_exception_no_traceback():
+            raise exceptions.SkyletInternalError(e.details())
+    elif e.code() == grpc.StatusCode.UNAVAILABLE:
+        time.sleep(current_backoff)
+    elif e.code() == grpc.StatusCode.UNIMPLEMENTED:
+        # Handle backwards compatibility: old server doesn't implement this RPC.
+        # Let the caller fall back to legacy execution.
+        raise exceptions.SkyletMethodNotImplementedError(
+            f'gRPC method not implemented on server, falling back to legacy execution: {e.details()}'
+        )
+    else:
+        raise e