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

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 53c9da9213
commit c97ff94617
3 changed files with 9 additions and 3 deletions

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)

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)

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