diff --git a/strix/interface/cli.py b/strix/interface/cli.py index 626cbde..5579f82 100644 --- a/strix/interface/cli.py +++ b/strix/interface/cli.py @@ -139,7 +139,7 @@ async def run_cli(args: Any) -> None: # noqa: PLR0915 status_text.append("Running penetration test...", style="bold #22c55e") status_text.append("\n\n") - stats_text = build_live_stats_text(tracer) + stats_text = build_live_stats_text(tracer, agent_config) if stats_text: status_text.append(stats_text) diff --git a/strix/interface/tui.py b/strix/interface/tui.py index 1b0bc37..58fa120 100644 --- a/strix/interface/tui.py +++ b/strix/interface/tui.py @@ -676,7 +676,7 @@ class StrixTUIApp(App): # type: ignore[misc] stats_content = Text() - stats_text = build_live_stats_text(self.tracer) + stats_text = build_live_stats_text(self.tracer, self.agent_config) if stats_text: stats_content.append(stats_text) diff --git a/strix/interface/utils.py b/strix/interface/utils.py index 6e3c56d..78e4023 100644 --- a/strix/interface/utils.py +++ b/strix/interface/utils.py @@ -129,7 +129,7 @@ def build_final_stats_text(tracer: Any) -> Text: return stats_text -def build_live_stats_text(tracer: Any) -> Text: +def build_live_stats_text(tracer: Any, agent_config: dict[str, Any] | None = None) -> Text: stats_text = Text() if not tracer: return stats_text @@ -165,6 +165,12 @@ def build_live_stats_text(tracer: Any) -> Text: stats_text.append("\n") + if agent_config: + llm_config = agent_config["llm_config"] + model = getattr(llm_config, "model_name", "Unknown") + stats_text.append(f"🧠 Model: {model}") + stats_text.append("\n") + stats_text.append("🤖 Agents: ", style="bold white") stats_text.append(str(agent_count), style="dim white") stats_text.append(" • ", style="dim white")