diff --git a/guardrails/cli/hub/install.py b/guardrails/cli/hub/install.py index d6a3890d8..ff5e045a3 100644 --- a/guardrails/cli/hub/install.py +++ b/guardrails/cli/hub/install.py @@ -194,11 +194,11 @@ def install( "--install-local-models/--no-install-local-models", help="Install local models", ), - quiet: bool = typer.Option( + verbose: bool = typer.Option( False, - "-q", - "--quiet", - help="Run the command in quiet mode to reduce output verbosity.", + "-v", + "--verbose", + help="Run the command in verbose mode to increase output verbosity.", ), ): try: @@ -211,10 +211,11 @@ def confirm(): " local models for local inference?", ) + is_quiet = not verbose install( package_uri, install_local_models=local_models, - quiet=quiet, + quiet=is_quiet, install_local_models_confirm=confirm, ) except Exception as e: diff --git a/tests/unit_tests/cli/hub/test_install.py b/tests/unit_tests/cli/hub/test_install.py index a4387be3f..57837bd6d 100644 --- a/tests/unit_tests/cli/hub/test_install.py +++ b/tests/unit_tests/cli/hub/test_install.py @@ -30,7 +30,7 @@ def test_install_local_models__false(self, mocker): mock_install.assert_called_once_with( "hub://guardrails/test-validator", install_local_models=False, - quiet=False, + quiet=True, install_local_models_confirm=ANY, ) @@ -46,7 +46,7 @@ def test_install_local_models__true(self, mocker): mock_install.assert_called_once_with( "hub://guardrails/test-validator", install_local_models=True, - quiet=False, + quiet=True, install_local_models_confirm=ANY, ) @@ -62,23 +62,23 @@ def test_install_local_models__none(self, mocker): mock_install.assert_called_once_with( "hub://guardrails/test-validator", install_local_models=None, - quiet=False, + quiet=True, install_local_models_confirm=ANY, ) assert result.exit_code == 0 - def test_install_quiet(self, mocker): + def test_install_verbose(self, mocker): mock_install = mocker.patch("guardrails.hub.install.install") runner = CliRunner() result = runner.invoke( - hub_command, ["install", "hub://guardrails/test-validator", "--quiet"] + hub_command, ["install", "hub://guardrails/test-validator", "--verbose"] ) mock_install.assert_called_once_with( "hub://guardrails/test-validator", install_local_models=None, - quiet=True, + quiet=False, install_local_models_confirm=ANY, )