wip: fixing more path resolution in tests

This commit is contained in:
2026-03-12 19:28:15 -04:00
parent 1f8bb58219
commit 8bc6eae101
11 changed files with 32 additions and 7 deletions

View File

@@ -180,7 +180,8 @@ class AppController:
"cache_read_input_tokens": 0,
"cache_creation_input_tokens": 0,
"total_tokens": 0,
"last_latency": 0.0
"last_latency": 0.0,
"percentage": 0.0
}
self.mma_tier_usage: Dict[str, Dict[str, Any]] = {
"Tier 1": {"input": 0, "output": 0, "provider": "gemini", "model": "gemini-3.1-pro-preview", "tool_preset": None},
@@ -2225,7 +2226,7 @@ class AppController:
threading.Thread(target=worker, daemon=True).start()
def _recalculate_session_usage(self) -> None:
usage = {"input_tokens": 0, "output_tokens": 0, "cache_read_input_tokens": 0, "cache_creation_input_tokens": 0, "total_tokens": 0, "last_latency": 0.0}
usage = {"input_tokens": 0, "output_tokens": 0, "cache_read_input_tokens": 0, "cache_creation_input_tokens": 0, "total_tokens": 0, "last_latency": 0.0, "percentage": self.session_usage.get("percentage", 0.0)}
for entry in ai_client.get_comms_log():
if entry.get("kind") == "response" and "usage" in entry.get("payload", {}):
u = entry["payload"]["usage"]
@@ -2240,6 +2241,8 @@ class AppController:
def _refresh_api_metrics(self, payload: dict[str, Any], md_content: str | None = None) -> None:
if "latency" in payload:
self.session_usage["last_latency"] = payload["latency"]
if "usage" in payload and "percentage" in payload["usage"]:
self.session_usage["percentage"] = payload["usage"]["percentage"]
self._recalculate_session_usage()
if md_content is not None:
stats = ai_client.get_token_stats(md_content)