From 9dc23b714cbf27f8b774d4ced306f9b152ba2537 Mon Sep 17 00:00:00 2001 From: heimoshuiyu Date: Fri, 18 Oct 2024 19:59:02 +0800 Subject: [PATCH] =?UTF-8?q?=E6=97=A0=E9=A3=8E=E9=99=A9=E4=B9=9F=E6=9B=B4?= =?UTF-8?q?=E6=96=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cucyuqing/cmd/risk-analyze.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cucyuqing/cmd/risk-analyze.py b/cucyuqing/cmd/risk-analyze.py index 749b1f4..e2b1e54 100644 --- a/cucyuqing/cmd/risk-analyze.py +++ b/cucyuqing/cmd/risk-analyze.py @@ -40,6 +40,8 @@ async def do_analyze(): analyze_result = await batch_risk_analyze(docs, risk_types) for task in analyze_result: if "是" not in task.response: + if risks_to_update.get(task.doc.id) is None: + risks_to_update[task.doc.id] = set() continue print(f"风险: {task.risk_type.name} 标题: {task.doc.title} {task.doc.id}") @@ -58,7 +60,7 @@ async def do_analyze(): """, { "es_id": doc_id, - "risk_types": json.dumps(list(risks), ensure_ascii=False), + "risk_types": json.dumps(list(risks), ensure_ascii=False) if risks else None, }, )