Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 9 additions & 0 deletions main.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
import asyncio
import os
from argparse import ArgumentParser
from datetime import datetime
from pathlib import Path
from config import get_config
from treesearch.search import TreeSearch
from utils.log import _ROOT_LOGGER, attach_file_handler, set_log_level
Expand Down Expand Up @@ -46,6 +48,13 @@ async def main():
config.agent.code = config.agent.code.model_copy(update={"model": args.model})


# Isolate this run's outputs in a timestamped subdirectory so artifacts from
# different runs are not mixed in the same out/ directory.
run_id = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
config = config.model_copy(update={"out_dir": str(Path(config.out_dir) / f"run_{run_id}")})
out_dir = mkdir(config.out_dir)


# Prepare to run AutoRecLab
attach_file_handler(out_dir)
cost_tracker.set_out_dir(out_dir)
Expand Down