输出风险分析

This commit is contained in:
2024-10-18 19:35:26 +08:00
parent 115d95bbef
commit 1545a85b09

View File

@@ -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