diff --git a/cucyuqing/cmd/risk-analyze.py b/cucyuqing/cmd/risk-analyze.py index d431e01..749b1f4 100644 --- a/cucyuqing/cmd/risk-analyze.py +++ b/cucyuqing/cmd/risk-analyze.py @@ -118,6 +118,8 @@ async def batch_risk_analyze( task.response = await risk_analyze(task, model) queue.task_done() bar.update(1) + if bar.n % 100 == 0: + print(f"已完成 {bar.n} 条风险分析") async def producer(): for task in tasks: @@ -131,6 +133,8 @@ async def batch_risk_analyze( await queue.put(None) await asyncio.gather(*workers) + print("风险分析完成") + return tasks