feat(regulatory): 增加风险归并与核查报告导出

review_agent/regulatory_review/services/risk_assess.py

@@ -0,0 +1,50 @@
+from __future__ import annotations
+
+from collections import Counter
+
+from review_agent.models import RegulatoryIssue, RegulatoryReviewBatch
+from review_agent.regulatory_review.schemas import Finding
+
+from .rag_citation import retrieve_citations
+
+
+SEVERITY_ORDER = ["blocking", "high", "medium", "low", "info"]
+
+
+def persist_findings(batch: RegulatoryReviewBatch, findings: list[Finding]) -> list[RegulatoryIssue]:
+    RegulatoryIssue.objects.filter(batch=batch).delete()
+    unique = {}
+    for finding in findings:
+        unique.setdefault((finding.rule_code, finding.category, finding.title), finding)
+
+    issues = []
+    for finding in unique.values():
+        citations = finding.citations or _safe_citations(finding)
+        issues.append(
+            RegulatoryIssue.objects.create(
+                batch=batch,
+                rule_code=finding.rule_code,
+                category=finding.category,
+                severity=finding.severity,
+                title=finding.title,
+                detail=finding.detail,
+                suggestion=finding.suggestion,
+                evidence=finding.evidence,
+                citations=citations,
+            )
+        )
+    batch.risk_summary = _risk_summary(issues)
+    batch.save(update_fields=["risk_summary"])
+    return issues
+
+
+def _safe_citations(finding: Finding) -> list[dict[str, object]]:
+    try:
+        return retrieve_citations(finding.title)
+    except Exception:
+        return [{"source": "原文依据待补充", "text": "RAG 索引不可用或无命中", "score": None}]
+
+
+def _risk_summary(issues: list[RegulatoryIssue]) -> dict[str, int]:
+    counts = Counter(issue.severity for issue in issues)
+    return {severity: counts.get(severity, 0) for severity in SEVERITY_ORDER}