CalvinBackup/agentic_security
1
0
Fork 0
mirror of https://github.com/msoedov/agentic_security.git synced 2026-07-09 21:18:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat(add form highlight):

Browse Source
This commit is contained in:
Alexander Myasoedov
2025-01-07 12:12:05 +02:00
parent 27f7ed693b
commit 8857842e40
5 changed files with 132 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agentic_security/probe_actor/fuzzer.py
+7 -3
View File
@@ -66,7 +66,7 @@ async def perform_single_shot_scan(
stop_event: asyncio.Event = None,
) -> AsyncGenerator[str, None]:
"""Perform a standard security scan."""
max_budget = max_budget * 100_000_000
selected_datasets = [m for m in datasets if m["selected"]]
try:
yield ScanResult.status_msg("Loading datasets...")
@@ -148,8 +148,12 @@ async def perform_single_shot_scan(
should_stop = True
break
if total_tokens > max_budget:
logger.info("Scan ran out of budget and stopped.")
yield ScanResult.status_msg("Scan ran out of budget and stopped.")
logger.info(
f"Scan ran out of budget and stopped. {total_tokens=} {max_budget=}"
)
yield ScanResult.status_msg(
f"Scan ran out of budget and stopped. {total_tokens=} {max_budget=}"
)
should_stop = True
break