pytest

Author: aaravnavani

guardrails/cli/configure.py

@@ -55,66 +55,61 @@ def configure(
     ),
 ):
     """Set the global configuration for the Guardrails CLI and Hub."""
-    # Get the existing configuration if present
-
-    headless = token is not None or no_metrics is not None
     existing_config = get_existing_config()
 
-    existing_token = existing_config.get("token", "")
-    existing_no_metrics = existing_config.get("no_metrics", "false")
+    # Normalize no_metrics to bool
+    input_no_metrics = no_metrics.lower() if no_metrics else None
+    if input_no_metrics == 'yes':
+        no_metrics_bool = True
+    elif input_no_metrics == 'no':
+        no_metrics_bool = False
+    else:
+        no_metrics_bool = bool(existing_config.get("no_metrics", "false") == "true")
 
-    if not headless:
-        notice_message = """
+    # Default token to existing token if None provided
+    if token is None:
+        token = existing_config.get("token", "")
 
-    You can find your token at https://hub.guardrailsai.com/tokens
-    """
-        logger.log(level=LEVELS.get("NOTICE"), msg=notice_message)  # type: ignore
+    # Provide default logging level for NOTICE if not in LEVELS
+    notice_level = LEVELS.get("NOTICE", 20)  # Assuming 20 is a reasonable default
 
-    # Prompt for token if not provided
-    if not token and not headless:
-        token = typer.prompt(
-            "> Token (optional) [None]",
-            default=existing_token,
-            hide_input=True,
-        )
-
-    # Prompt for no_metrics if not provided and not running headless
-    if no_metrics is None and not headless:
-        no_metrics = typer.prompt(
-            "> Disable anonymous metrics reporting?",
-            default=existing_no_metrics,
-        )
+    if not token and not no_metrics:
+        notice_message = """
+        You can find your token at https://hub.guardrailsai.com/tokens
+        """
+        logger.log(level=notice_level, msg=notice_message)
 
     try:
         # If token or no_metrics was updated, save the configuration
-        token_was_updated = token and token != existing_token
-        no_metrics_was_updated = no_metrics != existing_no_metrics
+        token_was_updated = token and token != existing_config.get("token", "")
+        no_metrics_was_updated = no_metrics_bool != (existing_config.get("no_metrics", "false") == "true")
+
         if token_was_updated or no_metrics_was_updated:
             logger.info("Configuring...")
-            save_configuration_file(token, no_metrics)
+            save_configuration_file(token, no_metrics_bool)
 
             # Authenticate with the Hub if token was updated
-        if token_was_updated:
-            logger.info("Validating credentials...")
-            get_auth()
-            success_message = """
-
-            Login successful.
+            if token_was_updated:
+                logger.info("Validating credentials...")
+                get_auth()
+                success_message = """
+                Login successful.
 
-            Get started by installing our RegexMatch validator:
-            https://hub.guardrailsai.com/validator/guardrails_ai/regex_match
+                Get started by installing our RegexMatch validator:
+                https://hub.guardrailsai.com/validator/guardrails_ai/regex_match
 
-            You can install it by running:
-            guardrails hub install hub://guardrails/regex_match
+                You can install it by running:
+                guardrails hub install hub://guardrails/regex_match
 
-            Find more validators at https://hub.guardrailsai.com
-            """
-            logger.log(level=LEVELS.get("SUCCESS"), msg=success_message)  # type: ignore`
+                Find more validators at https://hub.guardrailsai.com
+                """
+                logger.log(level=LEVELS.get("SUCCESS", 25), msg=success_message)  # Assuming 25 is the SUCCESS level
         elif not token:
             print("No token provided. Skipping authentication.")
 
-        if not headless and not token_was_updated:
-            print("Existing token found. Skipping re-authentication.")
+        if not (token_was_updated or no_metrics_was_updated) and not (token and no_metrics):
+            print("Existing configuration found. Skipping re-authentication.")
+
     except AuthenticationError as auth_error:
         logger.error(auth_error)
         logger.error(