From ab3351356144bac6aef78606d6251098ed76e967 Mon Sep 17 00:00:00 2001 From: zhanz5 <1721123209@qq.com> Date: Fri, 5 Jun 2026 14:19:33 +0800 Subject: [PATCH] style: apply black formatting to fuzzer.py --- agentic_security/probe_actor/fuzzer.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/agentic_security/probe_actor/fuzzer.py b/agentic_security/probe_actor/fuzzer.py index 95630e4..9804343 100644 --- a/agentic_security/probe_actor/fuzzer.py +++ b/agentic_security/probe_actor/fuzzer.py @@ -273,7 +273,9 @@ async def scan_module( failure_rate = module_failures / max(module_prompts, 1) failure_rates.append(failure_rate) - cost = calculate_cost(tokens, model=getattr(request_factory, 'model_name', 'unknown')) + cost = calculate_cost( + tokens, model=getattr(request_factory, "model_name", "unknown") + ) response_text = fuzzer_state.get_last_output(prompt) or "" @@ -557,7 +559,9 @@ async def perform_many_shot_scan( failure_rate = module_failures / max(processed_prompts, 1) failure_rates.append(failure_rate) - cost = calculate_cost(tokens, model=getattr(request_factory, 'model_name', 'unknown')) + cost = calculate_cost( + tokens, model=getattr(request_factory, "model_name", "unknown") + ) yield ScanResult( module=module.dataset_name,