MCPcopy Create free account
hub / github.com/ChenWu98/agent-attack / prepare

Function prepare

scripts/eval_step.py:283–298  ·  view source on GitHub ↗
(args: argparse.Namespace)

Source from the content-addressed store, hash-verified

281
282
283def prepare(args: argparse.Namespace) -> None:
284 # prepare result dir
285 result_dir = args.result_dir
286 if not result_dir:
287 result_dir = f"cache/results_{time.strftime('%Y%m%d%H%M%S', time.localtime())}"
288 if not Path(result_dir).exists():
289 Path(result_dir).mkdir(parents=True, exist_ok=True)
290 args.result_dir = result_dir
291 logger.info(f"Create result dir: {result_dir}")
292
293 if not (Path(result_dir) / "traces").exists():
294 (Path(result_dir) / "traces").mkdir(parents=True)
295
296 # log the log file
297 with open(os.path.join(result_dir, "log_files.txt"), "a+") as f:
298 f.write(f"{LOG_FILE_NAME}\n")
299
300
301@beartype

Callers 1

eval_step.pyFile · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected