refactor: improve stats panel styling and add version display
This commit is contained in:
@@ -60,7 +60,7 @@ Screen {
|
|||||||
max-height: 15;
|
max-height: 15;
|
||||||
background: transparent;
|
background: transparent;
|
||||||
padding: 0;
|
padding: 0;
|
||||||
margin: 0;
|
margin-top: 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
#vulnerabilities_panel {
|
#vulnerabilities_panel {
|
||||||
|
|||||||
@@ -1228,6 +1228,9 @@ class StrixTUIApp(App): # type: ignore[misc]
|
|||||||
if stats_text:
|
if stats_text:
|
||||||
stats_content.append(stats_text)
|
stats_content.append(stats_text)
|
||||||
|
|
||||||
|
version = get_package_version()
|
||||||
|
stats_content.append(f"\n v{version}", style="white")
|
||||||
|
|
||||||
from rich.panel import Panel
|
from rich.panel import Panel
|
||||||
|
|
||||||
stats_panel = Panel(
|
stats_panel = Panel(
|
||||||
|
|||||||
@@ -371,19 +371,20 @@ def build_tui_stats_text(tracer: Any, agent_config: dict[str, Any] | None = None
|
|||||||
if agent_config:
|
if agent_config:
|
||||||
llm_config = agent_config["llm_config"]
|
llm_config = agent_config["llm_config"]
|
||||||
model = getattr(llm_config, "model_name", "Unknown")
|
model = getattr(llm_config, "model_name", "Unknown")
|
||||||
stats_text.append(model, style="dim")
|
stats_text.append("◈ ", style="#22c55e")
|
||||||
|
stats_text.append(model, style="white")
|
||||||
|
|
||||||
llm_stats = tracer.get_total_llm_stats()
|
llm_stats = tracer.get_total_llm_stats()
|
||||||
total_stats = llm_stats["total"]
|
total_stats = llm_stats["total"]
|
||||||
|
|
||||||
total_tokens = total_stats["input_tokens"] + total_stats["output_tokens"]
|
total_tokens = total_stats["input_tokens"] + total_stats["output_tokens"]
|
||||||
if total_tokens > 0:
|
if total_tokens > 0:
|
||||||
stats_text.append("\n")
|
stats_text.append("\n ")
|
||||||
stats_text.append(f"{format_token_count(total_tokens)} tokens", style="dim")
|
stats_text.append(f"{format_token_count(total_tokens)} tokens", style="white")
|
||||||
|
|
||||||
if total_stats["cost"] > 0:
|
if total_stats["cost"] > 0:
|
||||||
stats_text.append("\n")
|
stats_text.append(" · ", style="white")
|
||||||
stats_text.append(f"${total_stats['cost']:.2f} spent", style="dim")
|
stats_text.append(f"${total_stats['cost']:.2f}", style="white")
|
||||||
|
|
||||||
return stats_text
|
return stats_text
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user