Hiep Ha VanandClaude Sonnet 5
40b12ecb15
- ui/monitoring_tab.py (1546 dong) tach thanh presentation/monitoring/** (container + 7 tab/card + shared helper), ui/monitoring_tab.py con lai re-export shim de app.py khong doi. - infrastructure/telemetry/audit_logger.py: CanonicalAuditLogger, core/audit_log.py thanh wrapper mong, tuong thich nguoc 100% voi schema .jsonl cu. - application/monitoring/monitoring_query_service.py: MonitoringQueryService read-only, filter/sort/pagination, khong import PySide6. - Go circular import model_pricing<->usage_tracker va agent_security<-> agent_security_alert (core/agent_security_types.py moi). - infrastructure/sandbox/sandbox_capabilities.py: SandboxCapabilityMatrix theo OS (Windows/Linux/macOS), chua dau noi vao core/sandbox_manager.py. - conftest.py: sua loi checkout khong ten cowork_local khien pytest import nham thu muc khac. - 77 test moi, 167/167 pass. QA da xac nhan UI/business logic khong doi (xem evidence/report/unified_report.html). Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
93 lines
4.4 KiB
Python
93 lines
4.4 KiB
Python
"""Agent Status tab — which agent roles are currently running, read from the
|
|
existing ``ChatPanel``/``TaskScheduler``/GraphRAG-ask-worker state (no new
|
|
runtime tracking of its own). Extracted from ``ui/monitoring_tab.py``'s
|
|
status-table wiring + ``_refresh_agent_status``.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from typing import Callable, Optional
|
|
|
|
from PySide6.QtCore import QSize
|
|
from PySide6.QtWidgets import QHeaderView, QTableWidget, QTableWidgetItem, QWidget
|
|
|
|
from ....core import agent_roles
|
|
from ....i18n import tr
|
|
from ....ui.widgets import badge_pill_widget
|
|
from ..shared.filter_scaffold import build_filter_scaffold
|
|
from ..shared.formatters import agent_avatar_icon
|
|
|
|
|
|
class AgentStatusTab(QWidget):
|
|
def __init__(self, ctx, on_refresh_all: Callable[[], None],
|
|
cowork=None, structure=None, task_scheduler=None):
|
|
super().__init__()
|
|
self._ctx = ctx
|
|
self._cowork = cowork
|
|
self._structure = structure
|
|
self._task_scheduler = task_scheduler
|
|
|
|
self.table = QTableWidget(0, 3)
|
|
self.table.setEditTriggers(QTableWidget.NoEditTriggers)
|
|
# No detail to open on click, no filtering — a plain read-only table,
|
|
# so selection is off rather than left dangling with no effect.
|
|
self.table.setSelectionMode(QTableWidget.NoSelection)
|
|
self.table.verticalHeader().setVisible(False)
|
|
self.table.horizontalHeader().setStretchLastSection(True)
|
|
self.table.horizontalHeader().setSectionResizeMode(0, QHeaderView.ResizeToContents)
|
|
self.table.setColumnWidth(1, 130) # status is a cell widget — size it explicitly
|
|
self.table.setIconSize(QSize(20, 20))
|
|
|
|
parts = build_filter_scaffold(
|
|
self, self.table, on_refresh=on_refresh_all,
|
|
title_key="monitoring.agent_status_title", with_search=False, with_detail=False)
|
|
self.title_lbl = parts["title_lbl"]
|
|
self.title_key = parts["title_key"]
|
|
self.title_refresh_btn = parts["title_refresh_btn"]
|
|
|
|
def retranslate(self) -> None:
|
|
self.title_lbl.setText(tr(self.title_key))
|
|
self.title_refresh_btn.setText(tr("monitoring.refresh"))
|
|
self.table.setHorizontalHeaderLabels([
|
|
tr("monitoring.col_agent"), tr("monitoring.detail_status"), tr("monitoring.col_source"),
|
|
])
|
|
|
|
def refresh(self) -> None:
|
|
cowork_n = len(self._cowork.active_workers()) if self._cowork is not None else 0
|
|
task_n = self._task_scheduler.running_count() if self._task_scheduler is not None else 0
|
|
ask_worker = getattr(self._structure, "_ask_worker", None)
|
|
knowledge_n = 1 if (ask_worker is not None and ask_worker.isRunning()) else 0
|
|
|
|
# Security is a system-management agent that runs INLINE on the
|
|
# active turn (agent_security prompt/command validation) — there is
|
|
# no separate worker to count, so its "active" cell shows On/Off
|
|
# from Settings instead of a live count.
|
|
sec_on = bool(self._ctx.config.agent_security.get("enabled"))
|
|
|
|
rows = [
|
|
(agent_roles.COWORK, cowork_n, tr("monitoring.source_cowork")),
|
|
(agent_roles.TASK, task_n, tr("monitoring.source_task")),
|
|
(agent_roles.KNOWLEDGE, knowledge_n, tr("monitoring.source_knowledge")),
|
|
(agent_roles.PLANNER, None, tr("monitoring.source_planner")),
|
|
(agent_roles.REASONING, None, tr("monitoring.source_reasoning")),
|
|
(agent_roles.SECURITY, None, tr("monitoring.source_security")),
|
|
]
|
|
self.table.setRowCount(len(rows))
|
|
for row, (role_key, count, source) in enumerate(rows):
|
|
label = agent_roles.label_for(role_key)
|
|
name_item = QTableWidgetItem(label)
|
|
name_item.setIcon(agent_avatar_icon(label))
|
|
self.table.setItem(row, 0, name_item)
|
|
|
|
if role_key == agent_roles.SECURITY:
|
|
running = sec_on
|
|
status_text = tr("monitoring.on") if sec_on else tr("monitoring.off")
|
|
elif count is not None:
|
|
running = count > 0
|
|
status_text = tr("monitoring.active_n", n=count) if running else tr("monitoring.idle")
|
|
else:
|
|
running = False
|
|
status_text = "—"
|
|
badge_tone = "badgeSuccess" if running else "badgeNeutral"
|
|
self.table.setCellWidget(row, 1, badge_pill_widget(status_text, badge_tone))
|
|
self.table.setItem(row, 2, QTableWidgetItem(source))
|