64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
"""前台（实时会商）诊断日志助手。
|
||
|
||
后台挂起路径用 ``DiagnosticsRecorder``（绑定作业上下文）；前台 multi_agent 路由用本助手
|
||
从 ``request`` 取 store + 当前用户，best-effort 落一条 ``scope="foreground"`` 的诊断事件。
|
||
任何异常静默吞掉——诊断日志绝不能反过来打断会商。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
from typing import Any
|
||
|
||
from fastapi import Request
|
||
|
||
from deerflow.persistence.roundtable_diagnostics import record_diagnostic
|
||
from deerflow.runtime.user_context import get_effective_user_id
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
def _resolve_user_id(request: Request) -> str | None:
|
||
try:
|
||
user = getattr(request.state, "user", None)
|
||
if user is not None:
|
||
return str(user.id)
|
||
return get_effective_user_id()
|
||
except Exception: # noqa: BLE001
|
||
return None
|
||
|
||
|
||
async def record_foreground_diag(
|
||
request: Request,
|
||
*,
|
||
stage: str,
|
||
level: str,
|
||
event: str,
|
||
message: str | None = None,
|
||
detail: Any = None,
|
||
thread_id: str | None = None,
|
||
draft_id: str | None = None,
|
||
task_id: str | None = None,
|
||
agent_id: str | None = None,
|
||
agent_name: str | None = None,
|
||
) -> None:
|
||
"""从 request 落一条前台诊断事件（best-effort）。"""
|
||
try:
|
||
store = getattr(request.app.state, "roundtable_diagnostic_store", None)
|
||
await record_diagnostic(
|
||
store,
|
||
scope="foreground",
|
||
stage=stage,
|
||
level=level,
|
||
event=event,
|
||
message=message,
|
||
detail=detail,
|
||
draft_id=draft_id,
|
||
task_id=task_id,
|
||
user_id=_resolve_user_id(request),
|
||
agent_id=agent_id,
|
||
agent_name=agent_name,
|
||
)
|
||
except Exception: # noqa: BLE001 — 诊断日志绝不打断会商
|
||
logger.debug("foreground roundtable diag failed", exc_info=True)
|