CalvinBackup/agentic_security
1
0
Fork 0
mirror of https://github.com/msoedov/agentic_security.git synced 2026-07-10 05:28:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat(add InMemorySecrets to fuzzer):

Browse Source
This commit is contained in:
Alexander Myasoedov
2025-02-20 16:24:52 +02:00
parent da50a48061
commit 1ce59151f3
4 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agentic_security/probe_actor/fuzzer.py
+4
View File
@@ -84,6 +84,7 @@ async def perform_single_shot_scan(
tools_inbox=None,
optimize=False,
stop_event: asyncio.Event = None,
secrets: dict[str, str] = {},
) -> AsyncGenerator[str, None]:
"""Perform a standard security scan."""
max_budget = max_budget * 100_000_000
@@ -218,6 +219,7 @@ async def perform_many_shot_scan(
stop_event: asyncio.Event = None,
probe_frequency: float = 0.2,
max_ctx_length: int = 10_000,
secrets: dict[str, str] = {},
) -> AsyncGenerator[str, None]:
"""Perform a multi-step security scan with probe injection."""
request_factory = multi_modality_spec(request_factory)
@@ -346,6 +348,7 @@ def scan_router(
tools_inbox=tools_inbox,
optimize=scan_parameters.optimize,
stop_event=stop_event,
secrets=scan_parameters.secrets,
)
else:
return perform_single_shot_scan(
@@ -355,4 +358,5 @@ def scan_router(
tools_inbox=tools_inbox,
optimize=scan_parameters.optimize,
stop_event=stop_event,
secrets=scan_parameters.secrets,
)