Drop dstack config, dstack stats, dstack gateway create (#3351)

* Drop dstack config

* Drop dstack stats

* Drop dstack gateway create

* Drop ProfileRetryPolicy

* Fix tests

Author: r4victor

docs/docs/guides/dstack-sky.md

@@ -5,7 +5,7 @@ sign up with [dstack Sky](../guides/dstack-sky.md).
 
 ### Set up the CLI
 
-If you've signed up, open your project settings, and copy the `dstack config` command to point the CLI to the project.
+If you've signed up, open your project settings, and copy the `dstack project add` command to point the CLI to the project.
 
 ![](https://raw.githubusercontent.com/dstackai/static-assets/main/static-assets/images/dstack-sky-project-config.png){ width=800 }
 

docs/docs/reference/cli/dstack/config.md

@@ -17,8 +17,6 @@
     print_gateways_table,
 )
 from dstack._internal.core.errors import CLIError
-from dstack._internal.core.models.backends.base import BackendType
-from dstack._internal.core.models.gateways import GatewayConfiguration
 from dstack._internal.utils.logging import get_logger
 
 logger = get_logger(__name__)
@@ -62,24 +60,6 @@ def _register(self):
                 help="Output in JSON format (equivalent to --format json)",
             )
 
-        create_parser = subparsers.add_parser(
-            "create",
-            help="Add a gateway. Deprecated in favor of `dstack apply` with gateway configuration.",
-            formatter_class=self._parser.formatter_class,
-        )
-        create_parser.set_defaults(subfunc=self._create)
-        create_parser.add_argument(
-            "--backend", choices=["aws", "azure", "gcp", "kubernetes"], required=True
-        )
-        create_parser.add_argument("--region", required=True)
-        create_parser.add_argument(
-            "--set-default", action="store_true", help="Set as default gateway for the project"
-        )
-        create_parser.add_argument("--name", help="Set a custom name for the gateway")
-        create_parser.add_argument(
-            "--domain", help="Set the domain for the gateway", required=True
-        )
-
         delete_parser = subparsers.add_parser(
             "delete", help="Delete a gateway", formatter_class=self._parser.formatter_class
         )
@@ -129,26 +109,6 @@ def _list(self, args: argparse.Namespace):
         except KeyboardInterrupt:
             pass
 
-    def _create(self, args: argparse.Namespace):
-        logger.warning(
-            "`dstack gateway create` is deperecated in favor of `dstack apply` with gateway configurations."
-        )
-        with console.status("Creating gateway..."):
-            configuration = GatewayConfiguration(
-                name=args.name,
-                backend=BackendType(args.backend),
-                region=args.region,
-            )
-            gateway = self.api.client.gateways.create(self.api.project, configuration)
-            if args.set_default:
-                self.api.client.gateways.set_default(self.api.project, gateway.name)
-            if args.domain:
-                self.api.client.gateways.set_wildcard_domain(
-                    self.api.project, gateway.name, args.domain
-                )
-        gateway = self.api.client.gateways.get(self.api.project, gateway.name)
-        print_gateways_table([gateway])
-
     def _delete(self, args: argparse.Namespace):
         gateway = self.api.client.gateways.get(self.api.project, args.name)
         print_gateways_table([gateway])

src/dstack/_internal/cli/commands/config.py

@@ -7,7 +7,6 @@
 from dstack._internal.cli.commands.apply import ApplyCommand
 from dstack._internal.cli.commands.attach import AttachCommand
 from dstack._internal.cli.commands.completion import CompletionCommand
-from dstack._internal.cli.commands.config import ConfigCommand
 from dstack._internal.cli.commands.delete import DeleteCommand
 from dstack._internal.cli.commands.fleet import FleetCommand
 from dstack._internal.cli.commands.gateway import GatewayCommand
@@ -19,7 +18,6 @@
 from dstack._internal.cli.commands.ps import PsCommand
 from dstack._internal.cli.commands.secrets import SecretCommand
 from dstack._internal.cli.commands.server import ServerCommand
-from dstack._internal.cli.commands.stats import StatsCommand
 from dstack._internal.cli.commands.stop import StopCommand
 from dstack._internal.cli.commands.volume import VolumeCommand
 from dstack._internal.cli.utils.common import _colors, console
@@ -63,7 +61,6 @@ def main():
     subparsers = parser.add_subparsers(metavar="COMMAND")
     ApplyCommand.register(subparsers)
     AttachCommand.register(subparsers)
-    ConfigCommand.register(subparsers)
     DeleteCommand.register(subparsers)
     FleetCommand.register(subparsers)
     GatewayCommand.register(subparsers)
@@ -75,7 +72,6 @@ def main():
     PsCommand.register(subparsers)
     SecretCommand.register(subparsers)
     ServerCommand.register(subparsers)
-    StatsCommand.register(subparsers)
     StopCommand.register(subparsers)
     VolumeCommand.register(subparsers)
     CompletionCommand.register(subparsers)

src/dstack/_internal/cli/commands/gateway.py

@@ -98,26 +98,6 @@ def parse_idle_duration(v: Optional[Union[int, str, bool]]) -> Optional[int]:
     return parse_duration(v)
 
 
-# Deprecated in favor of ProfileRetry().
-# TODO: Remove when no longer referenced.
-class ProfileRetryPolicy(CoreModel):
-    retry: Annotated[bool, Field(description="Whether to retry the run on failure or not")] = False
-    duration: Annotated[
-        Optional[Union[int, str]],
-        Field(description="The maximum period of retrying the run, e.g., `4h` or `1d`"),
-    ] = None
-
-    _validate_duration = validator("duration", pre=True, allow_reuse=True)(parse_duration)
-
-    @root_validator
-    def _validate_fields(cls, values):
-        if values["retry"] and "duration" not in values:
-            values["duration"] = DEFAULT_RETRY_DURATION
-        if values.get("duration") is not None:
-            values["retry"] = True
-        return values
-
-
 class RetryEvent(str, Enum):
     NO_CAPACITY = "no-capacity"
     INTERRUPTION = "interruption"

src/dstack/_internal/cli/commands/stats.py

@@ -0,0 +1,42 @@
+from pathlib import Path
+from unittest.mock import patch
+
+import yaml
+from pytest import CaptureFixture
+
+from tests._internal.cli.common import run_dstack_cli
+
+
+class TestProjectAdd:
+    def test_adds_project(self, capsys: CaptureFixture, tmp_path: Path):
+        cli_config_path = tmp_path / ".dstack" / "config.yml"
+        with patch("dstack.api.server.APIClient") as APIClientMock:
+            api_client_mock = APIClientMock.return_value
+            exit_code = run_dstack_cli(
+                [
+                    "project",
+                    "add",
+                    "--name",
+                    "project",
+                    "--url",
+                    "http://127.0.0.1:31313",
+                    "--token",
+                    "token",
+                    "-y",
+                ],
+                home_dir=tmp_path,
+            )
+            APIClientMock.assert_called_once_with(base_url="http://127.0.0.1:31313", token="token")
+            api_client_mock.projects.get.assert_called_with("project")
+        assert exit_code == 0
+        assert yaml.load(cli_config_path.read_text(), yaml.FullLoader) == {
+            "projects": [
+                {
+                    "default": True,
+                    "name": "project",
+                    "token": "token",
+                    "url": "http://127.0.0.1:31313",
+                }
+            ],
+            "repos": [],
+        }