This commit is contained in:
2026-03-07 14:13:25 -05:00
parent aaed011d9e
commit 91d7e2055f
3 changed files with 13 additions and 23 deletions

View File

@@ -159,7 +159,6 @@ 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]] = {
@@ -327,20 +326,8 @@ class AppController:
'_inject_preview': '_inject_preview',
'_show_inject_modal': '_show_inject_modal'
})
self.perf_monitor = performance_monitor.get_monitor()
self._perf_profiling_enabled = False
self._init_actions()
@property
def perf_profiling_enabled(self) -> bool:
return self._perf_profiling_enabled
@perf_profiling_enabled.setter
def perf_profiling_enabled(self, value: bool) -> None:
self._perf_profiling_enabled = value
if hasattr(self, 'perf_monitor'):
self.perf_monitor.enabled = value
def _update_inject_preview(self) -> None:
"""Updates the preview content based on the selected file and injection mode."""
if not self._inject_file_path:
@@ -428,9 +415,7 @@ class AppController:
})
def _process_pending_gui_tasks(self) -> None:
if getattr(self, 'perf_profiling_enabled', False): self.perf_monitor.start_component("controller._process_pending_gui_tasks")
if not self._pending_gui_tasks:
if getattr(self, 'perf_profiling_enabled', False): self.perf_monitor.end_component("controller._process_pending_gui_tasks")
return
sys.stderr.write(f"[DEBUG] _process_pending_gui_tasks: processing {len(self._pending_gui_tasks)} tasks\n")
sys.stderr.flush()
@@ -440,6 +425,11 @@ class AppController:
for task in tasks:
try:
action = task.get("action")
sys.stderr.write(f"[DEBUG] Processing GUI task: action={action}\n")
sys.stderr.flush()
if action:
session_logger.log_api_hook("PROCESS_TASK", action, str(task))
# ...
if action == "refresh_api_metrics":
self._refresh_api_metrics(task.get("payload", {}), md_content=self.last_md or None)
elif action == "set_ai_status":
@@ -942,11 +932,9 @@ class AppController:
while True:
event_name, payload = self.event_queue.get()
if getattr(self, 'perf_profiling_enabled', False): self.perf_monitor.start_component("controller._process_event")
sys.stderr.write(f"[DEBUG] _process_event_queue got event: {event_name} with payload: {str(payload)[:100]}\n")
sys.stderr.flush()
if event_name == "shutdown":
if getattr(self, 'perf_profiling_enabled', False): self.perf_monitor.end_component("controller._process_event")
break
if event_name == "user_request":
threading.Thread(target=self._handle_request_event, args=(payload,), daemon=True).start()
@@ -992,7 +980,6 @@ class AppController:
"action": "ticket_completed",
"payload": payload
})
if getattr(self, 'perf_profiling_enabled', False): self.perf_monitor.end_component("controller._process_event")
def _handle_request_event(self, event: events.UserRequestEvent) -> None:
"""Processes a UserRequestEvent by calling the AI client."""
@@ -1101,8 +1088,8 @@ class AppController:
def _on_api_event(self, event_name: str = "generic_event", **kwargs: Any) -> None:
payload = kwargs.get("payload", {})
# Push to background event queue for processing, NOT GUI queue
self.event_queue.put("refresh_api_metrics", payload)
with self._pending_gui_tasks_lock:
self._pending_gui_tasks.append({"action": "refresh_api_metrics", "payload": payload})
if self.test_hooks_enabled:
with self._api_event_queue_lock:
self._api_event_queue.append({"type": event_name, "payload": payload})
@@ -1849,6 +1836,7 @@ class AppController:
self._update_cached_stats()
def _update_cached_stats(self) -> None:
import ai_client
self._cached_cache_stats = ai_client.get_gemini_cache_stats()
self._cached_tool_stats = dict(self._tool_stats)