Add Quiet Install

guardrails/cli/hub/install.py

@@ -237,62 +237,69 @@ def install(
     package_uri: str = typer.Argument(
         help="URI to the package to install. Example: hub://guardrails/regex_match."
     ),
+    quiet: bool = typer.Option(
+        False,
+        "--quiet",
+        help="Run the command in quiet mode to reduce output verbosity.",
+    ),
 ):
     """Install a validator from the Hub."""
     if not package_uri.startswith("hub://"):
         logger.error("Invalid URI!")
         sys.exit(1)
 
-    console.print(f"\nInstalling {package_uri}...\n")
-    logger.log(
-        level=LEVELS.get("SPAM"),  # type: ignore
-        msg=f"Installing {package_uri}...",
-    )
+    if not quiet:
+        console.print(f"\nInstalling {package_uri}...\n")
+        logger.log(
+            level=LEVELS.get("SPAM"),  # type: ignore
+            msg=f"Installing {package_uri}...",
+        )
 
     # Validation
     module_name = package_uri.replace("hub://", "")
 
     # Prep
-    with console.status("Fetching manifest", spinner="bouncingBar"):
+    with console.status("Fetching manifest", spinner="bouncingBar") as status:
+        if not quiet:
+            status.update("Fetching manifest")
         module_manifest = get_validator_manifest(module_name)
         site_packages = get_site_packages_location()
 
     # Install
-    with console.status("Downloading dependencies", spinner="bouncingBar"):
+    with console.status("Downloading dependencies", spinner="bouncingBar") as status:
+        if not quiet:
+            status.update("Downloading dependencies")
         install_hub_module(module_manifest, site_packages)
 
     # Post-install
-    with console.status("Running post-install setup", spinner="bouncingBar"):
+    with console.status("Running post-install setup", spinner="bouncingBar") as status:
+        if not quiet:
+            status.update("Running post-install setup")
         run_post_install(module_manifest, site_packages)
         add_to_hub_inits(module_manifest, site_packages)
 
-    success_message_cli = Template(
-        """✅Successfully installed ${module_name}!
-
-[bold]Import validator:[/bold]
-from guardrails.hub import ${export}
-
-[bold]Get more info:[/bold]
-https://hub.guardrailsai.com/validator/${id}
-"""
-    ).safe_substitute(
-        module_name=package_uri,
-        id=module_manifest.id,
-        export=module_manifest.exports[0],
-    )
-    success_message_logger = Template(
-        """✅Successfully installed ${module_name}!
-
-Import validator:
-from guardrails.hub import ${export}
-
-Get more info:
-https://hub.guardrailsai.com/validator/${id}
-"""
-    ).safe_substitute(
-        module_name=package_uri,
-        id=module_manifest.id,
-        export=module_manifest.exports[0],
-    )
-    console.print(success_message_cli)  # type: ignore
-    logger.log(level=LEVELS.get("SPAM"), msg=success_message_logger)  # type: ignore
+    if not quiet:
+        success_message_cli = Template(
+            "✅Successfully installed ${module_name}!\n\n"
+            "[bold]Import validator:[/bold]\n"
+            "from guardrails.hub import ${export}\n\n"
+            "[bold]Get more info:[/bold]\n"
+            "https://hub.guardrailsai.com/validator/${id}\n"
+        ).safe_substitute(
+            module_name=package_uri,
+            id=module_manifest.id,
+            export=module_manifest.exports[0],
+        )
+        success_message_logger = Template(
+            "✅Successfully installed ${module_name}!\n\n"
+            "Import validator:\n"
+            "from guardrails.hub import ${export}\n\n"
+            "Get more info:\n"
+            "https://hub.guardrailsai.com/validator/${id}\n"
+        ).safe_substitute(
+            module_name=package_uri,
+            id=module_manifest.id,
+            export=module_manifest.exports[0],
+        )
+        console.print(success_message_cli)  # type: ignore
+        logger.log(level=LEVELS.get("SPAM"), msg=success_message_logger)  # type: ignore

tests/unit_tests/cli/hub/test_install.py

@@ -60,7 +60,7 @@ def test_happy_path(self, mocker):
 
         from guardrails.cli.hub.install import install
 
-        install("hub://guardrails/test-validator")
+        install("hub://guardrails/test-validator", quiet=False)
 
         log_calls = [
             call(level=5, msg="Installing hub://guardrails/test-validator..."),