skypilot-nightly 1.0.0.dev20250903__py3-none-any.whl → 1.0.0.dev20250904__py3-none-any.whl

sky/server/requests/requests.py

@@ -1,5 +1,6 @@
 """Utilities for REST API."""
 import asyncio
+import atexit
 import contextlib
 import dataclasses
 import enum
@@ -842,3 +843,11 @@ async def requests_gc_daemon():
         # Run the daemon at most once every hour to avoid too frequent
         # cleanup.
         await asyncio.sleep(max(retention_seconds, 3600))
+
+
+def _cleanup():
+    if _DB is not None:
+        asyncio.run(_DB.close())
+
+
+atexit.register(_cleanup)

sky/server/server.py

@@ -24,7 +24,6 @@ import aiofiles
 import anyio
 import fastapi
 from fastapi.middleware import cors
-from sqlalchemy import pool
 import starlette.middleware.base
 import uvloop
 
@@ -72,6 +71,7 @@ from sky.utils import dag_utils
 from sky.utils import perf_utils
 from sky.utils import status_lib
 from sky.utils import subprocess_utils
+from sky.utils.db import db_utils
 from sky.volumes.server import server as volumes_rest
 from sky.workspaces import server as workspaces_rest
 
@@ -1322,18 +1322,17 @@ async def download(download_body: payloads.DownloadBody,
                                     detail=f'Error creating zip file: {str(e)}')
 
 
+# TODO(aylei): run it asynchronously after global_user_state support async op
 @app.post('/provision_logs')
-async def provision_logs(cluster_body: payloads.ClusterNameBody,
-                         follow: bool = True,
-                         tail: int = 0) -> fastapi.responses.StreamingResponse:
+def provision_logs(cluster_body: payloads.ClusterNameBody,
+                   follow: bool = True,
+                   tail: int = 0) -> fastapi.responses.StreamingResponse:
     """Streams the provision.log for the latest launch request of a cluster."""
     # Prefer clusters table first, then cluster_history as fallback.
-    log_path_str = await context_utils.to_thread(
-        global_user_state.get_cluster_provision_log_path,
+    log_path_str = global_user_state.get_cluster_provision_log_path(
         cluster_body.cluster_name)
     if not log_path_str:
-        log_path_str = await context_utils.to_thread(
-            global_user_state.get_cluster_history_provision_log_path,
+        log_path_str = global_user_state.get_cluster_history_provision_log_path(
             cluster_body.cluster_name)
     if not log_path_str:
         raise fastapi.HTTPException(
@@ -1908,13 +1907,6 @@ if __name__ == '__main__':
 
     skyuvicorn.add_timestamp_prefix_for_server_logs()
 
-    # Initialize global user state db
-    global_user_state.initialize_and_get_db(pool.QueuePool)
-    # Initialize request db
-    requests_lib.reset_db_and_logs()
-    # Restore the server user hash
-    _init_or_restore_server_user_hash()
-
     parser = argparse.ArgumentParser()
     parser.add_argument('--host', default='127.0.0.1')
     parser.add_argument('--port', default=46580, type=int)
@@ -1930,7 +1922,17 @@ if __name__ == '__main__':
     # that it is shown only when the API server is started.
     usage_lib.maybe_show_privacy_policy()
 
-    config = server_config.compute_server_config(cmd_args.deploy)
+    # Initialize global user state db
+    db_utils.set_max_connections(1)
+    global_user_state.initialize_and_get_db()
+    # Initialize request db
+    requests_lib.reset_db_and_logs()
+    # Restore the server user hash
+    _init_or_restore_server_user_hash()
+    max_db_connections = global_user_state.get_max_db_connections()
+    config = server_config.compute_server_config(cmd_args.deploy,
+                                                 max_db_connections)
+
     num_workers = config.num_server_workers
 
     queue_server: Optional[multiprocessing.Process] = None
@@ -1955,11 +1957,12 @@ if __name__ == '__main__':
         logger.info(f'Starting SkyPilot API server, workers={num_workers}')
         # We don't support reload for now, since it may cause leakage of request
         # workers or interrupt running requests.
-        config = uvicorn.Config('sky.server.server:app',
-                                host=cmd_args.host,
-                                port=cmd_args.port,
-                                workers=num_workers)
-        skyuvicorn.run(config)
+        uvicorn_config = uvicorn.Config('sky.server.server:app',
+                                        host=cmd_args.host,
+                                        port=cmd_args.port,
+                                        workers=num_workers)
+        skyuvicorn.run(uvicorn_config,
+                       max_db_connections=config.num_db_connections_per_worker)
     except Exception as exc:  # pylint: disable=broad-except
         logger.error(f'Failed to start SkyPilot API server: '
                      f'{common_utils.format_exception(exc, use_bracket=True)}')

sky/server/uvicorn.py

@@ -26,6 +26,7 @@ from sky.utils import context_utils
 from sky.utils import env_options
 from sky.utils import perf_utils
 from sky.utils import subprocess_utils
+from sky.utils.db import db_utils
 
 logger = sky_logging.init_logger(__name__)
 
@@ -88,9 +89,12 @@ class Server(uvicorn.Server):
     - Run the server process with contextually aware.
     """
 
-    def __init__(self, config: uvicorn.Config):
+    def __init__(self,
+                 config: uvicorn.Config,
+                 max_db_connections: Optional[int] = None):
         super().__init__(config=config)
         self.exiting: bool = False
+        self.max_db_connections = max_db_connections
 
     def handle_exit(self, sig: int, frame: Union[FrameType, None]) -> None:
         """Handle exit signal.
@@ -196,6 +200,8 @@ class Server(uvicorn.Server):
 
     def run(self, *args, **kwargs):
         """Run the server process."""
+        if self.max_db_connections is not None:
+            db_utils.set_max_connections(self.max_db_connections)
         add_timestamp_prefix_for_server_logs()
         context_utils.hijack_sys_attrs()
         # Use default loop policy of uvicorn (use uvloop if available).
@@ -210,14 +216,14 @@ class Server(uvicorn.Server):
             asyncio.run(self.serve(*args, **kwargs))
 
 
-def run(config: uvicorn.Config):
+def run(config: uvicorn.Config, max_db_connections: Optional[int] = None):
     """Run unvicorn server."""
     if config.reload:
         # Reload and multi-workers are mutually exclusive
         # in uvicorn. Since we do not use reload now, simply
         # guard by an exception.
         raise ValueError('Reload is not supported yet.')
-    server = Server(config=config)
+    server = Server(config=config, max_db_connections=max_db_connections)
     try:
         if config.workers is not None and config.workers > 1:
             sock = config.bind_socket()

sky/skylet/constants.py

@@ -362,7 +362,7 @@ SKY_SSH_USER_PLACEHOLDER = 'skypilot:ssh_user'
 
 RCLONE_CONFIG_DIR = '~/.config/rclone'
 RCLONE_CONFIG_PATH = f'{RCLONE_CONFIG_DIR}/rclone.conf'
-RCLONE_LOG_DIR = '~/.sky/rclone_log'
+RCLONE_MOUNT_CACHED_LOG_DIR = '~/.sky/rclone_log'
 RCLONE_CACHE_DIR = '~/.cache/rclone'
 RCLONE_CACHE_REFRESH_INTERVAL = 10
 

sky/skypilot_config.py

@@ -227,7 +227,7 @@ def _get_config_from_path(path: Optional[str]) -> config_utils.Config:
     return parse_and_validate_config_file(path)
 
 
-def _resolve_user_config_path() -> Optional[str]:
+def resolve_user_config_path() -> Optional[str]:
     # find the user config file path, None if not resolved.
     user_config_path = _get_config_file_path(ENV_VAR_GLOBAL_CONFIG)
     if user_config_path:
@@ -252,7 +252,7 @@ def _resolve_user_config_path() -> Optional[str]:
 
 def get_user_config() -> config_utils.Config:
     """Returns the user config."""
-    return _get_config_from_path(_resolve_user_config_path())
+    return _get_config_from_path(resolve_user_config_path())
 
 
 def _resolve_project_config_path() -> Optional[str]:
@@ -574,8 +574,13 @@ def _reload_config_as_server() -> None:
                 'If db config is specified, no other config is allowed')
         logger.debug('retrieving config from database')
         with _DB_USE_LOCK:
-            sqlalchemy_engine = sqlalchemy.create_engine(db_url,
-                                                         poolclass=NullPool)
+            dispose_engine = False
+            if db_utils.get_max_connections() == 0:
+                dispose_engine = True
+                sqlalchemy_engine = sqlalchemy.create_engine(db_url,
+                                                             poolclass=NullPool)
+            else:
+                sqlalchemy_engine = db_utils.get_engine('config')
             db_utils.add_all_tables_to_db_sqlalchemy(Base.metadata,
                                                      sqlalchemy_engine)
 
@@ -597,7 +602,8 @@ def _reload_config_as_server() -> None:
                 server_config = overlay_skypilot_config(server_config,
                                                         db_config)
             # Close the engine to avoid connection leaks
-            sqlalchemy_engine.dispose()
+            if dispose_engine:
+                sqlalchemy_engine.dispose()
     if sky_logging.logging_enabled(logger, sky_logging.DEBUG):
         logger.debug(f'server config: \n'
                      f'{yaml_utils.dump_yaml_str(dict(server_config))}')
@@ -611,7 +617,7 @@ def _reload_config_as_client() -> None:
     _set_loaded_config_path(None)
 
     overrides: List[config_utils.Config] = []
-    user_config_path = _resolve_user_config_path()
+    user_config_path = resolve_user_config_path()
     user_config = _get_config_from_path(user_config_path)
     if user_config:
         overrides.append(user_config)
@@ -867,8 +873,13 @@ def update_api_server_config_no_lock(config: config_utils.Config) -> None:
             raise ValueError('Cannot change db url while server is running')
         if existing_db_url:
             with _DB_USE_LOCK:
-                sqlalchemy_engine = sqlalchemy.create_engine(existing_db_url,
-                                                             poolclass=NullPool)
+                dispose_engine = False
+                if db_utils.get_max_connections() == 0:
+                    dispose_engine = True
+                    sqlalchemy_engine = sqlalchemy.create_engine(
+                        existing_db_url, poolclass=NullPool)
+                else:
+                    sqlalchemy_engine = db_utils.get_engine('config')
                 db_utils.add_all_tables_to_db_sqlalchemy(
                     Base.metadata, sqlalchemy_engine)
 
@@ -897,7 +908,8 @@ def update_api_server_config_no_lock(config: config_utils.Config) -> None:
                 _set_config_yaml_to_db(API_SERVER_CONFIG_KEY, config)
                 db_updated = True
                 # Close the engine to avoid connection leaks
-                sqlalchemy_engine.dispose()
+                if dispose_engine:
+                    sqlalchemy_engine.dispose()
 
     if not db_updated:
         # save to the local file (PVC in Kubernetes, local file otherwise)

sky/ssh_node_pools/server.py

@@ -15,7 +15,7 @@ router = fastapi.APIRouter()
 
 
 @router.get('')
-async def get_ssh_node_pools() -> Dict[str, Any]:
+def get_ssh_node_pools() -> Dict[str, Any]:
     """Get all SSH Node Pool configurations."""
     try:
         return ssh_node_pools_core.get_all_pools()
@@ -27,7 +27,7 @@ async def get_ssh_node_pools() -> Dict[str, Any]:
 
 
 @router.post('')
-async def update_ssh_node_pools(pools_config: Dict[str, Any]) -> Dict[str, str]:
+def update_ssh_node_pools(pools_config: Dict[str, Any]) -> Dict[str, str]:
     """Update SSH Node Pool configurations."""
     try:
         ssh_node_pools_core.update_pools(pools_config)
@@ -39,7 +39,7 @@ async def update_ssh_node_pools(pools_config: Dict[str, Any]) -> Dict[str, str]:
 
 
 @router.delete('/{pool_name}')
-async def delete_ssh_node_pool(pool_name: str) -> Dict[str, str]:
+def delete_ssh_node_pool(pool_name: str) -> Dict[str, str]:
     """Delete a SSH Node Pool configuration."""
     try:
         if ssh_node_pools_core.delete_pool(pool_name):
@@ -83,7 +83,7 @@ async def upload_ssh_key(request: fastapi.Request) -> Dict[str, str]:
 
 
 @router.get('/keys')
-async def list_ssh_keys() -> List[str]:
+def list_ssh_keys() -> List[str]:
     """List available SSH keys."""
     try:
         return ssh_node_pools_core.list_ssh_keys()
@@ -200,7 +200,7 @@ async def down_ssh_node_pool_general(
 
 
 @router.get('/{pool_name}/status')
-async def get_ssh_node_pool_status(pool_name: str) -> Dict[str, str]:
+def get_ssh_node_pool_status(pool_name: str) -> Dict[str, str]:
     """Get the status of a specific SSH Node Pool."""
     try:
         # Call ssh_status to check the context

sky/users/server.py

@@ -33,8 +33,12 @@ USER_LOCK_TIMEOUT_SECONDS = 20
 router = fastapi.APIRouter()
 
 
+# All handlers in user handler are sync to get fastAPI run it in a
+# ThreadPoolExecutor to avoid blocking the async event loop.
+# TODO(aylei): make these async once we have the global_user_state async
+# support.
 @router.get('')
-async def users() -> List[Dict[str, Any]]:
+def users() -> List[Dict[str, Any]]:
     """Gets all users."""
     all_users = []
     user_list = global_user_state.get_all_users()
@@ -54,7 +58,7 @@ async def users() -> List[Dict[str, Any]]:
 
 
 @router.get('/role')
-async def get_current_user_role(request: fastapi.Request):
+def get_current_user_role(request: fastapi.Request):
     """Get current user's role."""
     # TODO(hailong): is there a reliable way to get the user
     # hash for the request without 'X-Auth-Request-Email' header?
@@ -70,7 +74,7 @@ async def get_current_user_role(request: fastapi.Request):
 
 
 @router.post('/create')
-async def user_create(user_create_body: payloads.UserCreateBody) -> None:
+def user_create(user_create_body: payloads.UserCreateBody) -> None:
     username = user_create_body.username
     password = user_create_body.password
     role = user_create_body.role
@@ -100,8 +104,8 @@ async def user_create(user_create_body: payloads.UserCreateBody) -> None:
 
 
 @router.post('/update')
-async def user_update(request: fastapi.Request,
-                      user_update_body: payloads.UserUpdateBody) -> None:
+def user_update(request: fastapi.Request,
+                user_update_body: payloads.UserUpdateBody) -> None:
     """Updates the user role."""
     user_id = user_update_body.user_id
     role = user_update_body.role
@@ -181,14 +185,13 @@ def _delete_user(user_id: str) -> None:
 
 
 @router.post('/delete')
-async def user_delete(user_delete_body: payloads.UserDeleteBody) -> None:
+def user_delete(user_delete_body: payloads.UserDeleteBody) -> None:
     user_id = user_delete_body.user_id
     _delete_user(user_id)
 
 
 @router.post('/import')
-async def user_import(
-        user_import_body: payloads.UserImportBody) -> Dict[str, Any]:
+def user_import(user_import_body: payloads.UserImportBody) -> Dict[str, Any]:
     """Import users from CSV content."""
     csv_content = user_import_body.csv_content
 
@@ -305,7 +308,7 @@ async def user_import(
 
 
 @router.get('/export')
-async def user_export() -> Dict[str, Any]:
+def user_export() -> Dict[str, Any]:
     """Export all users as CSV content."""
     try:
         # Get all users
@@ -369,7 +372,7 @@ def _user_lock(user_id: str) -> Generator[None, None, None]:
 
 
 @router.get('/service-account-tokens')
-async def get_service_account_tokens(
+def get_service_account_tokens(
         request: fastapi.Request) -> List[Dict[str, Any]]:
     """Get service account tokens. All users can see all tokens."""
     auth_user = request.state.auth_user
@@ -420,7 +423,7 @@ def _generate_service_account_user_id() -> str:
 
 
 @router.post('/service-account-tokens')
-async def create_service_account_token(
+def create_service_account_token(
         request: fastapi.Request,
         token_body: payloads.ServiceAccountTokenCreateBody) -> Dict[str, Any]:
     """Create a new service account token."""
@@ -508,7 +511,7 @@ async def create_service_account_token(
 
 
 @router.post('/service-account-tokens/delete')
-async def delete_service_account_token(
+def delete_service_account_token(
         request: fastapi.Request,
         token_body: payloads.ServiceAccountTokenDeleteBody) -> Dict[str, str]:
     """Delete a service account token.
@@ -549,7 +552,7 @@ async def delete_service_account_token(
 
 
 @router.post('/service-account-tokens/get-role')
-async def get_service_account_role(
+def get_service_account_role(
         request: fastapi.Request,
         role_body: payloads.ServiceAccountTokenRoleBody) -> Dict[str, Any]:
     """Get the role of a service account."""
@@ -585,7 +588,7 @@ async def get_service_account_role(
 
 
 @router.post('/service-account-tokens/update-role')
-async def update_service_account_role(
+def update_service_account_role(
         request: fastapi.Request,
         role_body: payloads.ServiceAccountTokenUpdateRoleBody
 ) -> Dict[str, str]:
@@ -628,7 +631,7 @@ async def update_service_account_role(
 
 
 @router.post('/service-account-tokens/rotate')
-async def rotate_service_account_token(
+def rotate_service_account_token(
         request: fastapi.Request,
         token_body: payloads.ServiceAccountTokenRotateBody) -> Dict[str, Any]:
     """Rotate a service account token.

sky/utils/db/db_utils.py

@@ -2,10 +2,12 @@
 import asyncio
 import contextlib
 import enum
+import os
+import pathlib
 import sqlite3
 import threading
 import typing
-from typing import Any, Callable, Iterable, Optional
+from typing import Any, Callable, Dict, Iterable, Optional
 
 import aiosqlite
 import aiosqlite.context
@@ -13,6 +15,7 @@ import sqlalchemy
 from sqlalchemy import exc as sqlalchemy_exc
 
 from sky import sky_logging
+from sky.skylet import constants
 
 logger = sky_logging.init_logger(__name__)
 if typing.TYPE_CHECKING:
@@ -346,3 +349,57 @@ class SQLiteConn(threading.local):
                                     ) -> Iterable[sqlite3.Row]:
         conn = await self._get_async_conn()
         return await conn.execute_fetchall(sql, parameters)
+
+    async def close(self):
+        if self._async_conn is not None:
+            await self._async_conn.close()
+        self.conn.close()
+
+
+_max_connections = 0
+_postgres_engine_cache: Dict[str, sqlalchemy.engine.Engine] = {}
+_sqlite_engine_cache: Dict[str, sqlalchemy.engine.Engine] = {}
+
+_db_creation_lock = threading.Lock()
+
+
+def set_max_connections(max_connections: int):
+    global _max_connections
+    _max_connections = max_connections
+
+
+def get_max_connections():
+    return _max_connections
+
+
+def get_engine(db_name: str):
+    conn_string = None
+    if os.environ.get(constants.ENV_VAR_IS_SKYPILOT_SERVER) is not None:
+        conn_string = os.environ.get(constants.ENV_VAR_DB_CONNECTION_URI)
+    if conn_string:
+        with _db_creation_lock:
+            if conn_string not in _postgres_engine_cache:
+                if _max_connections == 0:
+                    _postgres_engine_cache[conn_string] = (
+                        sqlalchemy.create_engine(
+                            conn_string, poolclass=sqlalchemy.pool.NullPool))
+                elif _max_connections == 1:
+                    _postgres_engine_cache[conn_string] = (
+                        sqlalchemy.create_engine(
+                            conn_string, poolclass=sqlalchemy.pool.StaticPool))
+                else:
+                    _postgres_engine_cache[conn_string] = (
+                        sqlalchemy.create_engine(
+                            conn_string,
+                            poolclass=sqlalchemy.pool.QueuePool,
+                            size=_max_connections,
+                            max_overflow=0))
+            engine = _postgres_engine_cache[conn_string]
+    else:
+        db_path = os.path.expanduser(f'~/.sky/{db_name}.db')
+        pathlib.Path(db_path).parents[0].mkdir(parents=True, exist_ok=True)
+        if db_path not in _sqlite_engine_cache:
+            _sqlite_engine_cache[db_path] = sqlalchemy.create_engine(
+                'sqlite:///' + db_path)
+        engine = _sqlite_engine_cache[db_path]
+    return engine

sky/utils/db/migration_utils.py

@@ -3,9 +3,6 @@
 import contextlib
 import logging
 import os
-import pathlib
-import threading
-from typing import Dict, Optional
 
 from alembic import command as alembic_command
 from alembic.config import Config
@@ -14,7 +11,6 @@ import filelock
 import sqlalchemy
 
 from sky import sky_logging
-from sky.skylet import constants
 
 logger = sky_logging.init_logger(__name__)
 
@@ -32,34 +28,6 @@ SERVE_DB_NAME = 'serve_db'
 SERVE_VERSION = '001'
 SERVE_LOCK_PATH = '~/.sky/locks/.serve_db.lock'
 
-_postgres_engine_cache: Dict[str, sqlalchemy.engine.Engine] = {}
-_sqlite_engine_cache: Dict[str, sqlalchemy.engine.Engine] = {}
-
-_db_creation_lock = threading.Lock()
-
-
-def get_engine(db_name: str,
-               pg_pool_class: Optional[sqlalchemy.pool.Pool] = None):
-    conn_string = None
-    if os.environ.get(constants.ENV_VAR_IS_SKYPILOT_SERVER) is not None:
-        conn_string = os.environ.get(constants.ENV_VAR_DB_CONNECTION_URI)
-    if conn_string:
-        if pg_pool_class is None:
-            pg_pool_class = sqlalchemy.NullPool
-        with _db_creation_lock:
-            if conn_string not in _postgres_engine_cache:
-                _postgres_engine_cache[conn_string] = sqlalchemy.create_engine(
-                    conn_string, poolclass=pg_pool_class)
-            engine = _postgres_engine_cache[conn_string]
-    else:
-        db_path = os.path.expanduser(f'~/.sky/{db_name}.db')
-        pathlib.Path(db_path).parents[0].mkdir(parents=True, exist_ok=True)
-        if db_path not in _sqlite_engine_cache:
-            _sqlite_engine_cache[db_path] = sqlalchemy.create_engine(
-                'sqlite:///' + db_path)
-        engine = _sqlite_engine_cache[db_path]
-    return engine
-
 
 @contextlib.contextmanager
 def db_lock(db_name: str):

{skypilot_nightly-1.0.0.dev20250903.dist-info → skypilot_nightly-1.0.0.dev20250904.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: skypilot-nightly
-Version: 1.0.0.dev20250903
+Version: 1.0.0.dev20250904
 Summary: SkyPilot: Run AI on Any Infra — Unified, Faster, Cheaper.
 Author: SkyPilot Team
 License: Apache 2.0
@@ -145,48 +145,48 @@ Requires-Dist: grpcio>=1.63.0; extra == "server"
 Requires-Dist: protobuf<7.0.0,>=5.26.1; extra == "server"
 Requires-Dist: aiosqlite; extra == "server"
 Provides-Extra: all
-Requires-Dist: ibm-cos-sdk; extra == "all"
-Requires-Dist: ray[default]!=2.6.0,>=2.2.0; extra == "all"
-Requires-Dist: vastai-sdk>=0.1.12; extra == "all"
-Requires-Dist: aiohttp; extra == "all"
-Requires-Dist: google-cloud-storage; extra == "all"
-Requires-Dist: websockets; extra == "all"
-Requires-Dist: azure-mgmt-compute>=33.0.0; extra == "all"
+Requires-Dist: runpod>=1.6.1; extra == "all"
+Requires-Dist: azure-common; extra == "all"
 Requires-Dist: kubernetes!=32.0.0,>=20.0.0; extra == "all"
-Requires-Dist: msgraph-sdk; extra == "all"
-Requires-Dist: azure-storage-blob>=12.23.1; extra == "all"
-Requires-Dist: anyio; extra == "all"
-Requires-Dist: boto3>=1.26.1; extra == "all"
-Requires-Dist: google-api-python-client>=2.69.0; extra == "all"
-Requires-Dist: pyopenssl<24.3.0,>=23.2.0; extra == "all"
-Requires-Dist: cudo-compute>=0.1.10; extra == "all"
-Requires-Dist: azure-mgmt-network>=27.0.0; extra == "all"
+Requires-Dist: aiohttp; extra == "all"
 Requires-Dist: pyvmomi==8.0.1.0.2; extra == "all"
-Requires-Dist: passlib; extra == "all"
-Requires-Dist: azure-identity>=1.19.0; extra == "all"
-Requires-Dist: botocore>=1.29.10; extra == "all"
-Requires-Dist: awscli>=1.27.10; extra == "all"
-Requires-Dist: python-dateutil; extra == "all"
-Requires-Dist: pyjwt; extra == "all"
 Requires-Dist: oci; extra == "all"
-Requires-Dist: ibm-cloud-sdk-core; extra == "all"
+Requires-Dist: ray[default]!=2.6.0,>=2.2.0; extra == "all"
+Requires-Dist: ibm-platform-services>=0.48.0; extra == "all"
+Requires-Dist: pyjwt; extra == "all"
 Requires-Dist: azure-cli>=2.65.0; extra == "all"
-Requires-Dist: aiosqlite; extra == "all"
-Requires-Dist: pydo>=0.3.0; extra == "all"
-Requires-Dist: docker; extra == "all"
-Requires-Dist: azure-common; extra == "all"
-Requires-Dist: colorama<0.4.5; extra == "all"
-Requires-Dist: msrestazure; extra == "all"
+Requires-Dist: ibm-cos-sdk; extra == "all"
+Requires-Dist: python-dateutil; extra == "all"
 Requires-Dist: ibm-vpc; extra == "all"
-Requires-Dist: ibm-platform-services>=0.48.0; extra == "all"
+Requires-Dist: pyopenssl<24.3.0,>=23.2.0; extra == "all"
+Requires-Dist: grpcio>=1.63.0; extra == "all"
+Requires-Dist: boto3>=1.26.1; extra == "all"
+Requires-Dist: colorama<0.4.5; extra == "all"
+Requires-Dist: azure-storage-blob>=12.23.1; extra == "all"
 Requires-Dist: azure-core>=1.24.0; extra == "all"
+Requires-Dist: azure-mgmt-compute>=33.0.0; extra == "all"
+Requires-Dist: anyio; extra == "all"
 Requires-Dist: sqlalchemy_adapter; extra == "all"
+Requires-Dist: pydo>=0.3.0; extra == "all"
+Requires-Dist: azure-mgmt-network>=27.0.0; extra == "all"
+Requires-Dist: casbin; extra == "all"
+Requires-Dist: websockets; extra == "all"
+Requires-Dist: aiosqlite; extra == "all"
+Requires-Dist: botocore>=1.29.10; extra == "all"
 Requires-Dist: azure-core>=1.31.0; extra == "all"
+Requires-Dist: ibm-cloud-sdk-core; extra == "all"
+Requires-Dist: vastai-sdk>=0.1.12; extra == "all"
+Requires-Dist: docker; extra == "all"
 Requires-Dist: nebius>=0.2.47; extra == "all"
-Requires-Dist: casbin; extra == "all"
+Requires-Dist: google-cloud-storage; extra == "all"
+Requires-Dist: msgraph-sdk; extra == "all"
+Requires-Dist: awscli>=1.27.10; extra == "all"
 Requires-Dist: protobuf<7.0.0,>=5.26.1; extra == "all"
-Requires-Dist: grpcio>=1.63.0; extra == "all"
-Requires-Dist: runpod>=1.6.1; extra == "all"
+Requires-Dist: google-api-python-client>=2.69.0; extra == "all"
+Requires-Dist: msrestazure; extra == "all"
+Requires-Dist: cudo-compute>=0.1.10; extra == "all"
+Requires-Dist: azure-identity>=1.19.0; extra == "all"
+Requires-Dist: passlib; extra == "all"
 Dynamic: author
 Dynamic: classifier
 Dynamic: description