Bruce/DEMO-AGENT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(regulatory): 为核查流程增加LLM复核记录

Browse Source
This commit is contained in:
bruce
2026-06-07 11:52:54 +08:00
parent 945669b9c2
commit 8f16675a92
4 changed files with 115 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
review_agent/regulatory_review/workflow.py
View File

@@ -23,6 +23,7 @@ from review_agent.regulatory_review.services.consistency_check import run_consis
from review_agent.regulatory_review.services.export import build_assistant_summary, export_review_results
from review_agent.regulatory_review.services.feishu_notifier import create_mock_notifications
from review_agent.regulatory_review.services.info_extract import detect_regulatory_condition_candidates
from review_agent.regulatory_review.services.llm_review import review_condition_fields, review_workflow_payload
from review_agent.regulatory_review.services.risk_assess import persist_findings
from review_agent.regulatory_review.services.rule_loader import load_rule_file
from review_agent.regulatory_review.services.structure_check import run_structure_check
@@ -121,6 +122,7 @@ class RegulatoryWorkflowExecutor:
self.findings = []
self.document_texts: dict[str, str] = {}
self.text_extract_status: dict[str, dict[str, object]] = {}
self.llm_reviews: dict[str, dict[str, object]] = {}
def run(self) -> None:
self.batch.status = RegulatoryReviewBatch.Status.RUNNING
@@ -188,10 +190,19 @@ class RegulatoryWorkflowExecutor:
self.rule_set = apply_rule_scope(load_rule_file(), self.batch.condition_json.get("rule_scope") or {})
return
if node_code == "completeness_check":
self.findings.extend(run_completeness_check(self.batch.source_summary_batch, self._rules()))
findings = run_completeness_check(self.batch.source_summary_batch, self._rules())
self.findings.extend(findings)
self._save_llm_review(
"completeness_check",
{
"findings": [finding.to_dict() for finding in findings],
"rules_count": len(self._rules().get("requirements", [])),
},
)
return
if node_code == "text_extract":
self.document_texts = self._extract_source_texts()
self._save_llm_review("text_extract", {"files": self.text_extract_status})
save_artifact(
self.batch,
name="text_extract_status.json",
@@ -201,12 +212,17 @@ class RegulatoryWorkflowExecutor:
)
return
if node_code == "structure_check":
self.findings.extend(run_structure_check(self.document_texts, self._rules()))
findings = run_structure_check(self.document_texts, self._rules())
self.findings.extend(findings)
self._save_llm_review("structure_check", {"findings": [finding.to_dict() for finding in findings]})
return
if node_code == "consistency_check":
self.findings.extend(run_consistency_check(self.document_texts))
findings = run_consistency_check(self.document_texts)
self.findings.extend(findings)
self._save_llm_review("consistency_check", {"findings": [finding.to_dict() for finding in findings]})
return
if node_code == "risk_assess":
self._save_llm_review("risk_assess", {"findings": [finding.to_dict() for finding in self.findings]})
issues = persist_findings(self.batch, self.findings)
create_mock_notifications(self.batch)
save_artifact(
@@ -225,6 +241,7 @@ class RegulatoryWorkflowExecutor:
}
for issue in issues
],
"llm_reviews": self.llm_reviews,
},
ensure_ascii=False,
indent=2,
@@ -290,12 +307,18 @@ class RegulatoryWorkflowExecutor:
}
continue
result = extract_text(path)
field_review = review_condition_fields(
text=result.front_text or result.text,
rule_fields=result.field_candidates or {},
file_context=f"{item.directory_level}\n{item.file_name}\n{item.relative_path}",
)
self.text_extract_status[item.file_name] = {
"status": result.status,
"path": str(path),
"content_hash": result.content_hash,
"section_candidates": result.section_candidates,
"field_candidates": result.field_candidates,
"field_candidates": field_review.get("selected_fields", result.field_candidates),
"field_review": field_review,
"front_text": result.front_text,
"error_message": result.error_message,
}
@@ -303,6 +326,18 @@ class RegulatoryWorkflowExecutor:
texts[item.file_name] = result.text
return texts
def _save_llm_review(self, stage: str, payload: dict[str, object]) -> dict[str, object]:
review = review_workflow_payload(stage=stage, payload=payload)
self.llm_reviews[stage] = review
save_artifact(
self.batch,
name=f"llm_review_{stage}.json",
artifact_type="json",
content=json.dumps(review, ensure_ascii=False, indent=2),
metadata={"artifact": "llm_review", "stage": stage},
)
return review
def start_regulatory_review_workflow(batch: RegulatoryReviewBatch, *, async_run: bool = True) -> None:
executor = RegulatoryWorkflowExecutor(batch)