Merge pull request #3 from dexhunter/feat/configurable_log_dir

feat: Make log-dir configurable

Author: DhruvSrikanth

weco/cli.py

@@ -50,6 +50,9 @@ def main() -> None:
     )
     parser.add_argument("--steps", type=int, required=True, help="Number of steps to run")
     parser.add_argument("--model", type=str, required=True, help="Model to use for optimization")
+    parser.add_argument(
+        "--log-dir", type=str, default=".runs", help="Directory to store logs and results"
+    )
     parser.add_argument(
         "--additional-instructions",
         default=None,
@@ -83,7 +86,13 @@ def main() -> None:
             timeout = 800
 
         # Initialize panels
-        summary_panel = SummaryPanel(maximize=maximize, metric_name=metric_name, total_steps=steps, model=args.model)
+        summary_panel = SummaryPanel(
+            maximize=maximize,
+            metric_name=metric_name,
+            total_steps=steps,
+            model=args.model,
+            runs_dir=args.log_dir,
+        )
         plan_panel = PlanPanel()
         solution_panels = SolutionPanels(metric_name=metric_name)
         eval_output_panel = EvaluationOutputPanel()
@@ -112,7 +121,7 @@ def main() -> None:
         with Live(layout, refresh_per_second=refresh_rate, screen=True) as live:
             # Define the runs directory (.runs/<session-id>)
             session_id = session_response["session_id"]
-            runs_dir = pathlib.Path(".runs") / session_id
+            runs_dir = pathlib.Path(args.log_dir) / session_id
             runs_dir.mkdir(parents=True, exist_ok=True)
 
             # Save the original code (.runs/<session-id>/original.py)

weco/panels.py

@@ -11,15 +11,16 @@
 class SummaryPanel:
     """Holds a summary of the optimization session."""
 
-    def __init__(self, maximize: bool, metric_name: str, total_steps: int, model: str, session_id: str = None):
+    def __init__(self, maximize: bool, metric_name: str, total_steps: int, model: str, runs_dir: str, session_id: str = None):
         self.maximize = maximize
         self.metric_name = metric_name
         self.goal = ("Maximizing" if self.maximize else "Minimizing") + f" {self.metric_name}..."
         self.total_input_tokens = 0
         self.total_output_tokens = 0
         self.total_steps = total_steps
         self.model = model
-        self.session_id = session_id or "N/A"
+        self.runs_dir = runs_dir
+        self.session_id = session_id if session_id is not None else "N/A"
         self.progress = Progress(
             TextColumn("[progress.description]{task.description}"),
             BarColumn(bar_width=20),
@@ -55,8 +56,7 @@ def get_display(self, final_message: Optional[str] = None) -> Panel:
         summary_table.add_row(f"[bold cyan]Model:[/] {self.model}")
         summary_table.add_row("")
         # Log directory
-        runs_dir = f".runs/{self.session_id}"
-        summary_table.add_row(f"[bold cyan]Logs:[/] [blue underline]{runs_dir}[/]")
+        summary_table.add_row(f"[bold cyan]Logs:[/] [blue underline]{self.runs_dir}/{self.session_id}[/]")
         summary_table.add_row("")
         # Token counts
         summary_table.add_row(