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

feat: wire feishu notifications into workflows

Browse Source
This commit is contained in:
bruce
2026-06-07 22:09:47 +08:00
parent 820069f558
commit cbc7493df8
8 changed files with 301 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
review_agent/application_form_fill/services/notifier.py
View File

@@ -7,6 +7,8 @@ from review_agent.models import (
ApplicationFormFillNotificationRecord,
ExportedSummaryFile,
)
from review_agent.notifications.dispatcher import dispatch_workflow_notification
from review_agent.notifications.workflow_adapters import build_application_form_fill_context
def notify_completion(
@@ -33,6 +35,13 @@ def notify_completion(
retry_count=1,
error_message="mock notification failed",
)
unified_error = ""
try:
unified_record = dispatch_workflow_notification(build_application_form_fill_context(batch))
if unified_record.send_status == unified_record.SendStatus.FAILED:
unified_error = unified_record.error_message
except Exception as exc:
unified_error = str(exc)
return ApplicationFormFillNotificationRecord.objects.create(
batch=batch,
recipient=batch.user,
@@ -41,5 +50,6 @@ def notify_completion(
export_ids=export_ids,
message_summary=message_summary,
send_status=ApplicationFormFillNotificationRecord.SendStatus.SUCCESS,
error_message=unified_error,
sent_at=timezone.now(),
)

13
review_agent/file_summary/workflow.py
View File

@@ -17,6 +17,8 @@ from review_agent.models import (
Message,
WorkflowNodeRun,
)
from review_agent.notifications.dispatcher import dispatch_workflow_notification
from review_agent.notifications.workflow_adapters import build_file_summary_context
from .events import record_event
from .services.archive import ARCHIVE_EXTENSIONS
@@ -154,14 +156,25 @@ class WorkflowExecutor:
self.batch.finished_at = timezone.now()
self.batch.save(update_fields=["status", "error_message", "finished_at"])
record_event(self.batch, "workflow_failed", {"message": str(exc)})
self._dispatch_completion_notification()
return
self.batch.status = FileSummaryBatch.Status.SUCCESS
self.batch.finished_at = timezone.now()
self.batch.save(update_fields=["status", "finished_at"])
record_event(self.batch, "workflow_completed", {"batch_id": self.batch.pk})
self._dispatch_completion_notification()
logger.info("Workflow run completed", extra={"batch_id": self.batch.pk})
def _dispatch_completion_notification(self) -> None:
try:
dispatch_workflow_notification(build_file_summary_context(self.batch))
except Exception as exc:
logger.warning(
"File summary notification failed without blocking workflow",
extra={"batch_id": self.batch.pk, "error": str(exc)},
)
def _run_node(self, node: WorkflowNodeRun) -> None:
logger.info(
"Workflow node started",

102
review_agent/notifications/workflow_adapters.py Normal file
View File

@@ -0,0 +1,102 @@
from __future__ import annotations
from review_agent.application_form_fill.constants import WORKFLOW_TYPE as FORM_FILL_WORKFLOW_TYPE
from review_agent.models import (
ApplicationFormFillBatch,
ExportedSummaryFile,
FileSummaryBatch,
RegulatoryIssue,
RegulatoryReviewBatch,
)
from .context import NotificationContext
def build_file_summary_context(batch: FileSummaryBatch) -> NotificationContext:
status = batch.status
abnormal_count = int(batch.failed_files or 0) + int(batch.unsupported_files or 0) + int(batch.uncertain_files or 0)
return NotificationContext(
workflow_type="file_summary",
workflow_name="自动汇总",
workflow_batch_id=batch.pk,
workflow_batch_no=batch.batch_no,
workflow_status=status,
trigger_user_id=batch.user_id,
trigger_username=batch.user.get_username(),
title=f"自动汇总{_status_label(status)}",
summary_lines=(
f"文件总数 {batch.total_files} 个，成功 {batch.success_files} 个",
f"异常/不支持/不确定 {abnormal_count} 个，总页数 {batch.total_pages}",
_error_line(batch.error_message),
),
next_step="查看文件目录、页数统计和导出结果",
result_path=f"/api/review-agent/file-summary/{batch.pk}/status/",
)
def build_regulatory_review_context(batch: RegulatoryReviewBatch) -> NotificationContext:
summary = batch.risk_summary or _count_regulatory_issues(batch)
return NotificationContext(
workflow_type="regulatory_review",
workflow_name="法规核查",
workflow_batch_id=batch.pk,
workflow_batch_no=batch.batch_no,
workflow_status=batch.status,
trigger_user_id=batch.user_id,
trigger_username=batch.user.get_username(),
title=f"法规核查{_status_label(batch.status)}",
summary_lines=(
f"阻断项 {int(summary.get('blocking') or 0)} 个，高风险 {int(summary.get('high') or 0)} 个",
f"中风险 {int(summary.get('medium') or 0)} 个，低风险 {int(summary.get('low') or 0)} 个",
_error_line(batch.error_message),
),
next_step="查看风险报告并处理整改项",
result_path=f"/api/review-agent/regulatory-review/{batch.pk}/status/",
)
def build_application_form_fill_context(batch: ApplicationFormFillBatch) -> NotificationContext:
export_count = ExportedSummaryFile.objects.filter(
workflow_type=FORM_FILL_WORKFLOW_TYPE,
workflow_batch_id=batch.pk,
).count()
return NotificationContext(
workflow_type=FORM_FILL_WORKFLOW_TYPE,
workflow_name="自动填表",
workflow_batch_id=batch.pk,
workflow_batch_no=batch.batch_no,
workflow_status=batch.status,
trigger_user_id=batch.user_id,
trigger_username=batch.user.get_username(),
title=f"自动填表{_status_label(batch.status)}",
summary_lines=(
f"模板 {', '.join(batch.selected_templates or []) or '未识别'}",
f"导出文件 {export_count} 个，冲突字段 {len(batch.conflict_summary or [])} 个",
_error_line(batch.error_message),
),
next_step="下载生成文件并检查字段冲突",
result_path=f"/api/review-agent/application-form-fill/{batch.pk}/status/",
)
def _count_regulatory_issues(batch: RegulatoryReviewBatch) -> dict[str, int]:
return {
severity: RegulatoryIssue.objects.filter(batch=batch, severity=severity).count()
for severity in ["blocking", "high", "medium", "low", "info"]
}
def _status_label(status: str) -> str:
labels = {
"success": "完成",
"partial_success": "部分完成",
"failed": "失败",
"cancelled": "已取消",
}
return labels.get(status, status)
def _error_line(error_message: str) -> str:
if not error_message:
return ""
return f"失败原因：{error_message[:160]}"

13
review_agent/regulatory_review/workflow.py
View File

@@ -18,6 +18,8 @@ from review_agent.models import (
RegulatoryReviewBatch,
WorkflowNodeRun,
)
from review_agent.notifications.dispatcher import dispatch_workflow_notification
from review_agent.notifications.workflow_adapters import build_regulatory_review_context
from review_agent.regulatory_review.services.completeness_check import run_completeness_check
from review_agent.regulatory_review.services.consistency_check import run_consistency_check
from review_agent.regulatory_review.services.export import build_assistant_summary, export_review_results
@@ -146,14 +148,25 @@ class RegulatoryWorkflowExecutor:
self.batch.finished_at = timezone.now()
self.batch.save(update_fields=["status", "error_message", "finished_at"])
record_event(self.batch, "workflow_failed", {"message": str(exc)})
self._dispatch_completion_notification()
return
self.batch.status = RegulatoryReviewBatch.Status.SUCCESS
self.batch.finished_at = timezone.now()
self.batch.save(update_fields=["status", "finished_at"])
record_event(self.batch, "workflow_completed", {"batch_id": self.batch.pk})
self._dispatch_completion_notification()
logger.info("法规核查工作流完成 batch_no=%s findings=%s", self.batch.batch_no, len(self.findings))
def _dispatch_completion_notification(self) -> None:
try:
dispatch_workflow_notification(build_regulatory_review_context(self.batch))
except Exception as exc:
logger.warning(
"Regulatory review notification failed without blocking workflow",
extra={"batch_id": self.batch.pk, "error": str(exc)},
)
def _nodes(self):
return WorkflowNodeRun.objects.filter(
workflow_type="regulatory_review",