diff --git a/src/ai_client.py b/src/ai_client.py index 7cc3897..557d3b1 100644 --- a/src/ai_client.py +++ b/src/ai_client.py @@ -2089,6 +2089,7 @@ def send( ) -> str: monitor = performance_monitor.get_monitor() if monitor.enabled: monitor.start_component("ai_client.send") + _append_comms("OUT", "request", {"prompt": user_message, "system": _get_combined_system_prompt()}) with _send_lock: if _provider == "gemini": res = _send_gemini( diff --git a/src/app_controller.py b/src/app_controller.py index 2d3c84b..74ea325 100644 --- a/src/app_controller.py +++ b/src/app_controller.py @@ -159,6 +159,7 @@ class AppController: "output_tokens": 0, "cache_read_input_tokens": 0, "cache_creation_input_tokens": 0, + "total_tokens": 0, "last_latency": 0.0 } self.mma_tier_usage: Dict[str, Dict[str, Any]] = {