scriptoriumadmin/strix
1
0
Fork 0
mirror of https://github.com/usestrix/strix.git synced 2026-09-06 08:15:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: Show Model Name in Live Stats Panel (#169)

Browse source 
Co-authored-by: Ahmed Allam <ahmed39652003@gmail.com>
This commit is contained in:
Vincent Yang 2025-12-03 13:45:01 -05:00 • committed by GitHub
parent 598791452c
commit 2faf23d79b
3 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
strix/interface/cli.py
View file

@ -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)

2
strix/interface/tui.py
View file

@ -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)

8
strix/interface/utils.py
View file

@ -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")