heimoshuiyu/cucyuqing
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

无风险也更新

Browse Source
This commit is contained in:
黑墨水鱼
2024-10-18 19:59:02 +08:00
parent 1545a85b09
commit 9dc23b714c
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cucyuqing/cmd/risk-analyze.py
View File

@@ -40,6 +40,8 @@ async def do_analyze():
analyze_result = await batch_risk_analyze(docs, risk_types) analyze_result = await batch_risk_analyze(docs, risk_types)
for task in analyze_result: for task in analyze_result:
if "是" not in task.response: if "是" not in task.response:
if risks_to_update.get(task.doc.id) is None:
risks_to_update[task.doc.id] = set()
continue continue
print(f"风险: {task.risk_type.name} 标题: {task.doc.title} {task.doc.id}") print(f"风险: {task.risk_type.name} 标题: {task.doc.title} {task.doc.id}")
@@ -58,7 +60,7 @@ async def do_analyze():
""", """,
{ {
"es_id": doc_id, "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,
}, },
) )