Skip to content

Feat/verbose filter #14

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
meditationduck wants to merge 4 commits into
base: main
Choose a base branch
from
Open

Feat/verbose filter #14

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
bd21d99
✨ Add verbose output filtering and enhance logging in CLI and ClaudeC…
meditationduck Sep 11, 2025
9c3b492
✅ format for -v
meditationduck Sep 12, 2025
5fb9a3c
🔧 Refactor verbose output handling and improve console display
meditationduck Sep 12, 2025
96efed1
🔧 Update verbose filter help text and remove unused Grob class
meditationduck Sep 12, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
32 changes: 24 additions & 8 deletions wake_ai/cli.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
import rich.traceback
from rich.console import Console
from rich.logging import RichHandler
from wake_ai.utils.logging import get_logger, set_debug
from wake_ai.utils.logging import get_logger, set_verbose_level

if TYPE_CHECKING:
from wake_ai import AIWorkflow
Expand Down Expand Up @@ -256,8 +256,13 @@ def get_class_that_defined_method(meth):
@click.option(
"--verbose",
"-v",
is_flag=True,
help="Enable verbose logging (debug level)"
count=True,
help="Enable verbose logging (-v: info, -vv: debug, -vvv: trace)"
)
@click.option(
"--verbose-filter",
type=str,
help="Filter verbose output by tool names (comma-separated, e.g., 'Bash,Edit,Glob,Grep,MultiEdit,NotebookEdit,NotebookRead,Read,Task,TodoWrite,WebFetch,WebSearch,Write,mcp__wake,mcp__wake__function_name')"
)
@click.option(
"--no-progress",
Expand All @@ -271,18 +276,29 @@ def get_class_that_defined_method(meth):
help="List all available workflows"
)
@click.pass_context
def main(ctx: click.Context, working_dir: str | None, model: str, resume: bool, execution_dir: str | None, export: str | None, no_cleanup: bool, verbose: bool, no_progress: bool, list: bool):
def main(ctx: click.Context, working_dir: str | None, model: str, resume: bool, execution_dir: str | None, export: str | None, no_cleanup: bool, verbose: int, verbose_filter: str | None, no_progress: bool, list: bool):
"""AI-powered smart contract security analysis.

This command runs various AI workflows for smart contract analysis
using Claude to identify potential vulnerabilities and issues.
"""
rich.traceback.install(console=console)

# Set logging level based on verbose flag
if verbose:
set_debug(True)
console.print("[dim]Debug logging enabled[/dim]")
# Set logging level based on verbose count
if verbose >= 1:
set_verbose_level(verbose)
if verbose == 1:
console.print("[dim]Verbose logging enabled (info level)[/dim]")
elif verbose == 2:
console.print("[dim]Debug logging enabled[/dim]")
elif verbose >= 3:
console.print("[dim]Trace logging enabled (maximum verbosity)[/dim]")
# TODO: Implement trace level logging if needed

if verbose_filter:
from wake_ai.utils.logging import set_verbose_filters
set_verbose_filters(verbose_filter)
console.print(f"[dim]Verbose output filtered to: {verbose_filter}[/dim]")

# Handle list flag
if list:
Expand Down
Loading