From b78fb404b599be937d81e6949221905aa47c9c3e Mon Sep 17 00:00:00 2001 From: demoncoder-crypto Date: Thu, 3 Apr 2025 18:35:33 +0530 Subject: [PATCH] feat: Add LLM direct seed generation mode --- run_all_experiments.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/run_all_experiments.py b/run_all_experiments.py index 3a5c96de7b..cdb94501df 100755 --- a/run_all_experiments.py +++ b/run_all_experiments.py @@ -271,12 +271,20 @@ def parse_args() -> argparse.Namespace: '--agent', action='store_true', default=False, - help='Enables agent enhancement.') + help='Run experiment using the agent framework.') parser.add_argument('-mr', '--max-round', type=int, default=100, help='Max trial round for agents.') + parser.add_argument( + '--generate-seeds', action='store_true', default=False, + help='Use LLM to generate a seed corpus creation script for each target.') + parser.add_argument( + '--seed-generation-mode', + choices=['script', 'direct'], + default='script', + help="Method for LLM seed generation: 'script' (generates python script) or 'direct' (generates file content).") args = parser.parse_args() if args.num_samples: @@ -616,6 +624,14 @@ def main(): coverage_gain_dict = _process_total_coverage_gain() _print_experiment_results(experiment_results, coverage_gain_dict) + # Set the environment variable if the flag is present + if args.generate_seeds: + os.environ['LLM_GENERATE_CORPUS'] = '1' + else: + # Ensure it's unset if the flag is not provided + if 'LLM_GENERATE_CORPUS' in os.environ: + del os.environ['LLM_GENERATE_CORPUS'] + if __name__ == '__main__': sys.exit(main())