feat: add error handling for headless mode in agent execution and improve CLI on scan failures

This commit is contained in:
Ahmed Allam
2025-11-01 03:29:44 +02:00
parent cf1d43706a
commit 6a84ea94fa
2 changed files with 25 additions and 1 deletions
+9 -1
View File
@@ -134,8 +134,16 @@ async def run_cli(args: Any) -> None: # noqa: PLR0915
console.print()
with console.status("[bold cyan]Running penetration test...", spinner="dots") as status:
agent = StrixAgent(agent_config)
await agent.execute_scan(scan_config)
result = await agent.execute_scan(scan_config)
status.stop()
if isinstance(result, dict) and not result.get("success", True):
error_msg = result.get("error", "Unknown error")
console.print()
console.print(f"[bold red]❌ Penetration test failed:[/] {error_msg}")
console.print()
sys.exit(1)
except Exception as e:
console.print(f"[bold red]Error during penetration test:[/] {e}")
raise