71 lines
3.0 KiB
Python
71 lines
3.0 KiB
Python
from __future__ import annotations
|
||
|
||
from review_agent.regulatory_review.schemas import Finding
|
||
|
||
|
||
def run_structure_check(document_texts: dict[str, str], rule_set: dict) -> list[Finding]:
|
||
findings: list[Finding] = []
|
||
combined_all_text = "\n".join(document_texts.values())
|
||
for requirement in rule_set.get("requirements", []):
|
||
if requirement.get("structure_required") and not _contains_any(
|
||
combined_all_text,
|
||
[requirement.get("title", ""), *requirement.get("aliases", [])],
|
||
):
|
||
findings.append(
|
||
Finding(
|
||
rule_code=requirement["code"],
|
||
category="structure",
|
||
severity=requirement.get("severity", "medium"),
|
||
title=f"申报资料目录缺少{requirement['title']}章节",
|
||
detail=f"未在申报资料目录或章节标题候选中发现{requirement['title']}。",
|
||
suggestion=requirement.get("suggestion", ""),
|
||
evidence={
|
||
"attachment4_code": requirement.get("attachment4_code"),
|
||
"expected_title": requirement["title"],
|
||
"aliases": requirement.get("aliases", []),
|
||
},
|
||
)
|
||
)
|
||
required_sections = requirement.get("required_sections") or []
|
||
if not required_sections:
|
||
continue
|
||
matching_docs = _matching_documents(document_texts, requirement.get("file_keywords", []))
|
||
if not matching_docs:
|
||
continue
|
||
combined_text = "\n".join(matching_docs.values())
|
||
for section in required_sections:
|
||
if _contains_any(combined_text, [section]):
|
||
continue
|
||
findings.append(
|
||
Finding(
|
||
rule_code=f"{requirement['code']}:{section}",
|
||
category="structure",
|
||
severity=requirement.get("severity", "medium"),
|
||
title=f"{requirement['title']}缺少{section}章节",
|
||
detail=f"已匹配{requirement['title']}文件，但未发现{section}相关内容。",
|
||
suggestion=requirement.get("suggestion", ""),
|
||
evidence={"section": section, "files": list(matching_docs)},
|
||
)
|
||
)
|
||
return findings
|
||
|
||
|
||
def _matching_documents(document_texts: dict[str, str], keywords: list[str]) -> dict[str, str]:
|
||
if not keywords:
|
||
return document_texts
|
||
result = {}
|
||
for name, text in document_texts.items():
|
||
haystack = f"{name}\n{text}".lower()
|
||
if any(str(keyword).lower() in haystack for keyword in keywords):
|
||
result[name] = text
|
||
return result
|
||
|
||
|
||
def _contains_any(text: str, needles: list[str]) -> bool:
|
||
normalized = _normalize_title(text)
|
||
return any(_normalize_title(needle) in normalized for needle in needles if needle)
|
||
|
||
|
||
def _normalize_title(value: str) -> str:
|
||
return "".join(str(value).lower().replace("/", "").replace("／", "").split())
|