read verbose from env variables

deepsense-ai · Dec 19, 2024 · 76ef33c · 76ef33c
1 parent 4947470
commit 76ef33c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/packages/ragbits-cli/src/ragbits/cli/__init__.py b/packages/ragbits-cli/src/ragbits/cli/__init__.py
@@ -1,4 +1,5 @@
 import importlib.util
+import os
 import pkgutil
 from pathlib import Path
 from typing import Annotated
@@ -8,6 +9,7 @@
 from typer.main import get_command
 
 import ragbits
+from ragbits.core import audit
 
 from .state import OutputType, cli_state, print_output
 
@@ -28,15 +30,17 @@ def ragbits_cli(
     output: Annotated[
         OutputType, typer.Option("--output", "-o", help="Set the output type (text or json)")
     ] = OutputType.text.value,  # type: ignore
-    verbose: bool = typer.Option(False, "--verbose", "-v", help="Enable verbose mode"),
+    verbose: bool = typer.Option(
+        bool(int(os.getenv("RAGBITS_VERBOSE", "0"))), "--verbose", "-v", help="Enable verbose mode"
+    ),
 ) -> None:
     """Common CLI arguments for all ragbits commands."""
     cli_state.output_type = output
     cli_state.verbose = verbose
 
     if verbose:
         typer.echo("Verbose mode is enabled.")
-        ragbits.core.audit.set_trace_handlers('cli')
+        audit.set_trace_handlers("cli")
     else:
         typer.echo("Verbose mode is disabled.")
 

diff --git a/packages/ragbits-core/src/ragbits/core/audit/__init__.py b/packages/ragbits-core/src/ragbits/core/audit/__init__.py
@@ -6,7 +6,6 @@
 from types import SimpleNamespace
 from typing import Any, ParamSpec, TypeVar
 
-from ragbits.cli import cli_state
 from ragbits.core.audit.base import TraceHandler
 
 __all__ = ["TraceHandler", "set_trace_handlers", "trace", "traceable"]