salvacybersec/strix
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(tui): display agent vulnerability count in TUI

Browse Source
This commit is contained in:
0xallam
2026-01-06 15:15:58 -08:00
committed by Ahmed Allam
parent 16c9b05121
commit af8eeef4ac
1 changed files with 22 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
strix/interface/tui.py
View File

@@ -555,7 +555,9 @@ class StrixTUIApp(App): # type: ignore[misc]
} }
status_icon = status_indicators.get(status, "🔵") status_icon = status_indicators.get(status, "🔵")
agent_name = f"{status_icon} {agent_name_raw}" vuln_count = self._agent_vulnerability_count(agent_id)
vuln_indicator = f" ({vuln_count})" if vuln_count > 0 else ""
agent_name = f"{status_icon} {agent_name_raw}{vuln_indicator}"
if status == "running": if status == "running":
self._start_agent_verb_timer(agent_id) self._start_agent_verb_timer(agent_id)
@@ -955,6 +957,19 @@ class StrixTUIApp(App): # type: ignore[misc]
return True return True
return False return False
def _agent_vulnerability_count(self, agent_id: str) -> int:
count = 0
for _exec_id, tool_data in list(self.tracer.tool_executions.items()):
if tool_data.get("agent_id") == agent_id:
tool_name = tool_data.get("tool_name", "")
if tool_name == "create_vulnerability_report":
status = tool_data.get("status", "")
if status == "completed":
result = tool_data.get("result", {})
if isinstance(result, dict) and result.get("success"):
count += 1
return count
def _gather_agent_events(self, agent_id: str) -> list[dict[str, Any]]: def _gather_agent_events(self, agent_id: str) -> list[dict[str, Any]]:
chat_events = [ chat_events = [
{ {
@@ -1053,7 +1068,9 @@ class StrixTUIApp(App): # type: ignore[misc]
} }
status_icon = status_indicators.get(status, "🔵") status_icon = status_indicators.get(status, "🔵")
agent_name = f"{status_icon} {agent_name_raw}" vuln_count = self._agent_vulnerability_count(agent_id)
vuln_indicator = f" ({vuln_count})" if vuln_count > 0 else ""
agent_name = f"{status_icon} {agent_name_raw}{vuln_indicator}"
if status in ["running", "waiting"]: if status in ["running", "waiting"]:
self._start_agent_verb_timer(agent_id) self._start_agent_verb_timer(agent_id)
@@ -1128,7 +1145,9 @@ class StrixTUIApp(App): # type: ignore[misc]
} }
status_icon = status_indicators.get(status, "🔵") status_icon = status_indicators.get(status, "🔵")
agent_name = f"{status_icon} {agent_name_raw}" vuln_count = self._agent_vulnerability_count(agent_id)
vuln_indicator = f" ({vuln_count})" if vuln_count > 0 else ""
agent_name = f"{status_icon} {agent_name_raw}{vuln_indicator}"
new_node = new_parent.add( new_node = new_parent.add(
agent_name, agent_name,