feat:新增投顾agent和nl2sqlagent

nl2sql/diagnostics.py

@@ -0,0 +1,50 @@
+"""NL2SQL 管理员诊断数据构造。"""
+from __future__ import annotations
+
+from collections import deque
+from threading import Lock
+
+def build_diagnostic_snapshot(
+    *,
+    query_id: str,
+    status: str,
+    access_tables: set[str] | list[str],
+    security_rule: str | None = None,
+    model_elapsed_ms: float | None = None,
+    sql: str | None = None,
+) -> dict:
+    """构造不含原始 SQL 和敏感结果的诊断快照。"""
+    return {
+        "query_id": query_id,
+        "status": status,
+        "access_tables": sorted(set(access_tables)),
+        "security_rule": security_rule,
+        "model_elapsed_ms": model_elapsed_ms,
+        "has_sql": bool(sql),
+    }
+
+
+class DiagnosticRegistry:
+    """保存最近的脱敏诊断摘要，进程重启后自动清空。"""
+
+    def __init__(self, *, max_items: int = 100):
+        self._items = deque(maxlen=max_items)
+        self._lock = Lock()
+
+    def record(self, **kwargs) -> None:
+        """记录诊断摘要并丢弃未定义的敏感字段。"""
+        allowed = {
+            "query_id", "status", "access_tables", "security_rule",
+            "model_elapsed_ms", "sql",
+        }
+        snapshot = build_diagnostic_snapshot(**{key: value for key, value in kwargs.items() if key in allowed})
+        with self._lock:
+            self._items.appendleft(snapshot)
+
+    def list_recent(self) -> list[dict]:
+        """返回最近诊断摘要的副本。"""
+        with self._lock:
+            return [dict(item) for item in self._items]
+
+
+diagnostic_registry = DiagnosticRegistry()