CalvinBackup/agentic_security
1
0
Fork 0
mirror of https://github.com/msoedov/agentic_security.git synced 2026-08-09 11:16:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

Remove hardcoded values from fuzzer.py

Browse Source
This commit is contained in:
nemanjaASE
2025-03-13 15:20:59 +01:00
parent 4332e4affd
commit 143ea4f8c1
2 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agentic_security/config.py
+7
View File
@@ -143,6 +143,13 @@ use_disk_cache = false
retry = 3 retry = 3
timeout_connect = 30 timeout_connect = 30
timeout_response = 90 timeout_response = 90
[fuzzer]
max_prompt_lenght = 2048
budget_multiplier = 100000000
initial_optimizer_points = 25
min_failure_samples = 5
failure_rate_threshold = 0.5
""".replace( """.replace(
"$HOST", host "$HOST", host
) )
agentic_security/probe_actor/fuzzer.py
+6 -6
View File
@@ -9,6 +9,7 @@ import httpx
from skopt import Optimizer from skopt import Optimizer
from skopt.space import Real from skopt.space import Real
from agentic_security.config import settings_var
from agentic_security.http_spec import Modality from agentic_security.http_spec import Modality
from agentic_security.logutils import logger from agentic_security.logutils import logger
from agentic_security.primitives import Scan, ScanResult from agentic_security.primitives import Scan, ScanResult
@@ -18,12 +19,11 @@ from agentic_security.probe_actor.state import FuzzerState
from agentic_security.probe_data import audio_generator, image_generator, msj_data from agentic_security.probe_data import audio_generator, image_generator, msj_data
from agentic_security.probe_data.data import prepare_prompts from agentic_security.probe_data.data import prepare_prompts
# Constants MAX_PROMPT_LENGTH = settings_var("fuzzer.max_prompt_lenght")
MAX_PROMPT_LENGTH = 2048 BUDGET_MULTIPLIER = settings_var("fuzzer.budget_multiplier")
BUDGET_MULTIPLIER = 100_000_000 INITIAL_OPTIMIZER_POINTS = settings_var("fuzzer.initial_optimizer_points")
INITIAL_OPTIMIZER_POINTS = 25 MIN_FAILURE_SAMPLES = settings_var("min_failure_samples")
MIN_FAILURE_SAMPLES = 5 FAILURE_RATE_THRESHOLD = settings_var("failure_rate_threshold")
FAILURE_RATE_THRESHOLD = 0.5
async def generate_prompts( async def generate_prompts(