刘金宝
49 lines
2.4 KiB
Python
49 lines
2.4 KiB
Python
class ReportAgent:
|
|
"""报告 Agent：整理评分结果为接口和 PDF 可复用的报告结构。"""
|
|
|
|
def build_report(self, scoring_result: dict) -> dict:
|
|
"""报告整理：校验评分结果字段并补齐展示默认值，不重新评分。"""
|
|
dimension_scores = self._normalize_dimension_scores(scoring_result.get("dimension_scores", []))
|
|
total_score = self._safe_float(scoring_result.get("total_score"), 0)
|
|
return {
|
|
"score_type": scoring_result.get("score_type", "percentage"),
|
|
"total_score": total_score,
|
|
"dimension_scores": dimension_scores,
|
|
"errors": self._ensure_list(scoring_result.get("errors")),
|
|
"improvement_plan": self._ensure_list(scoring_result.get("improvement_plan")),
|
|
"evidence_summary": self._ensure_list(scoring_result.get("evidence_summary")),
|
|
"guideline_refs": self._ensure_list(scoring_result.get("guideline_refs")),
|
|
"overall_comment": scoring_result.get("overall_comment", ""),
|
|
"_llm_model": scoring_result.get("_llm_model"),
|
|
"_latency_metrics": scoring_result.get("_latency_metrics", {}),
|
|
}
|
|
|
|
def _normalize_dimension_scores(self, raw_scores: object) -> list[dict]:
|
|
"""维度校验：把模型输出归一为前端和数据库可保存的评分列表。"""
|
|
if not isinstance(raw_scores, list):
|
|
return []
|
|
normalized: list[dict] = []
|
|
for item in raw_scores:
|
|
if not isinstance(item, dict):
|
|
continue
|
|
normalized.append(
|
|
{
|
|
"dimension": str(item.get("dimension", "未命名维度")),
|
|
"score": self._safe_float(item.get("score"), 0),
|
|
"max_score": self._safe_float(item.get("max_score"), 0),
|
|
"comment": str(item.get("comment", "")),
|
|
}
|
|
)
|
|
return normalized
|
|
|
|
def _ensure_list(self, value: object) -> list:
|
|
"""列表校验：保证报告中的数组字段稳定返回 list。"""
|
|
return value if isinstance(value, list) else []
|
|
|
|
def _safe_float(self, value: object, default: float) -> float:
|
|
"""数值校验：把模型输出中的分数安全转换为 float。"""
|
|
try:
|
|
return float(value)
|
|
except (TypeError, ValueError):
|
|
return default
|