| 395 | |
| 396 | |
| 397 | def prepare(args: argparse.Namespace) -> None: |
| 398 | # convert prompt python files to json |
| 399 | from agent.prompts import to_json |
| 400 | |
| 401 | to_json.run() |
| 402 | |
| 403 | # prepare result dir |
| 404 | result_dir = args.result_dir |
| 405 | if not result_dir: |
| 406 | result_dir = ( |
| 407 | f"cache/results_{time.strftime('%Y%m%d%H%M%S', time.localtime())}" |
| 408 | ) |
| 409 | if not Path(result_dir).exists(): |
| 410 | Path(result_dir).mkdir(parents=True, exist_ok=True) |
| 411 | args.result_dir = result_dir |
| 412 | logger.info(f"Create result dir: {result_dir}") |
| 413 | |
| 414 | if not (Path(result_dir) / "traces").exists(): |
| 415 | (Path(result_dir) / "traces").mkdir(parents=True) |
| 416 | |
| 417 | # log the log file |
| 418 | with open(os.path.join(result_dir, "log_files.txt"), "a+") as f: |
| 419 | f.write(f"{LOG_FILE_NAME}\n") |
| 420 | |
| 421 | |
| 422 | # def get_unfinished(config_files: list[str], result_dir: str) -> list[str]: |