Archive generated scripts output if exists

Author: shuds13

agentic/libe_agent_basic.py

@@ -20,6 +20,7 @@
 import subprocess
 import argparse
 import shutil
+import time
 from pathlib import Path
 from langchain_openai import ChatOpenAI
 from langchain.agents import create_agent
@@ -52,6 +53,9 @@ def create_llm(model, temperature=0, base_url=None):
 # Show prompts flag (set by command line)
 SHOW_PROMPTS = False
 
+# Directory where existing generated_scripts runs are moved (create if missing)
+ARCHIVE_RUNS_DIR = "archive_runs"
+
 # Files and directories to archive after each run
 # Can include directory names and glob patterns (e.g., "*.npy", "ensemble/", "*.log")
 ARCHIVE_ITEMS = [
@@ -131,6 +135,20 @@ def archive_run_outputs(output_dir, archive_name, error_msg=""):
                 if filepath.is_file():
                     shutil.move(str(filepath), str(run_output_dir / filepath.name))
 
+def archive_existing_output_dir(output_dir, archive_parent=None):
+    """If output_dir exists, move it to archive_parent/output_dir_<unique>, then create fresh output_dir."""
+    output_dir = Path(output_dir)
+    archive_dir = Path(archive_parent or ARCHIVE_RUNS_DIR)
+    if not output_dir.exists():
+        output_dir.mkdir(parents=True, exist_ok=True)
+        return
+    archive_dir.mkdir(parents=True, exist_ok=True)
+    dest = archive_dir / f"{output_dir.name}_{hex(time.time_ns())[2:10]}"
+    shutil.move(str(output_dir), str(dest))
+    print(f"Moved existing {output_dir} to {dest}")
+    output_dir.mkdir(parents=True, exist_ok=True)
+
+
 def detect_run_script(directory):
     """Find the run script in directory (first run_*.py file)"""
     directory = Path(directory)
@@ -225,7 +243,10 @@ async def main():
 
     SHOW_PROMPTS = args.show_prompts
     output_dir = "generated_scripts"
-    
+
+    # If output_dir already exists, move it to archive_runs/generated_scripts_<hash>, then create fresh
+    archive_existing_output_dir(output_dir)
+
     # Copy existing scripts
     current_scripts = copy_existing_scripts(args.scripts, output_dir)
     run_script_name = detect_run_script(output_dir)

agentic/libe_agent_basic_auto.py

@@ -19,6 +19,7 @@
 import subprocess
 import argparse
 import shutil
+import time
 from pathlib import Path
 from typing import Optional
 from pydantic import BaseModel, Field
@@ -57,6 +58,9 @@ def create_llm(model, temperature=0, base_url=None):
 # Current archive name (scripts and their output go together)
 CURRENT_ARCHIVE = None
 
+# Directory where existing generated_scripts runs are moved (create if missing)
+ARCHIVE_RUNS_DIR = "archive_runs"
+
 # Files and directories to archive after each run
 ARCHIVE_ITEMS = [
     "ensemble",           # libEnsemble output directory
@@ -67,6 +71,20 @@ def create_llm(model, temperature=0, base_url=None):
 ]
 
 
+def archive_existing_output_dir(output_dir, archive_parent=None):
+    """If output_dir exists, move it to archive_parent/output_dir_<unique>, then create fresh output_dir."""
+    output_dir = Path(output_dir)
+    archive_dir = Path(archive_parent or ARCHIVE_RUNS_DIR)
+    if not output_dir.exists():
+        output_dir.mkdir(parents=True, exist_ok=True)
+        return
+    archive_dir.mkdir(parents=True, exist_ok=True)
+    dest = archive_dir / f"{output_dir.name}_{hex(time.time_ns())[2:10]}"
+    shutil.move(str(output_dir), str(dest))
+    print(f"Moved existing {output_dir} to {dest}")
+    output_dir.mkdir(parents=True, exist_ok=True)
+
+
 # Tool schemas
 class RunScriptInput(BaseModel):
     script_name: str = Field(description="Name of the Python script to run (e.g., 'run_libe.py')")
@@ -227,8 +245,8 @@ def setup_work_directory(scripts_dir: str) -> Path:
     """Copy scripts to working directory and archive initial version"""
     global WORK_DIR
     scripts_dir = Path(scripts_dir)
+    archive_existing_output_dir("generated_scripts")
     work_dir = Path("generated_scripts")
-    work_dir.mkdir(exist_ok=True)
     WORK_DIR = work_dir
 
     # Copy all Python files to work dir

agentic/libe_agent_interactive_llm_first.py

@@ -20,6 +20,7 @@
 import subprocess
 import argparse
 import shutil
+import time
 from pathlib import Path
 from pydantic import BaseModel, Field
 from langchain_openai import ChatOpenAI
@@ -39,6 +40,23 @@
     MODEL = DEFAULT_ANTHROPIC_MODEL
 SHOW_PROMPTS = False
 
+# Directory where existing generated_scripts runs are moved (create if missing)
+ARCHIVE_RUNS_DIR = "archive_runs"
+
+
+def archive_existing_output_dir(output_dir, archive_parent=None):
+    """If output_dir exists, move it to archive_parent/output_dir_<unique>, then create fresh output_dir."""
+    output_dir = Path(output_dir)
+    archive_dir = Path(archive_parent or ARCHIVE_RUNS_DIR)
+    if not output_dir.exists():
+        output_dir.mkdir(parents=True, exist_ok=True)
+        return
+    archive_dir.mkdir(parents=True, exist_ok=True)
+    dest = archive_dir / f"{output_dir.name}_{hex(time.time_ns())[2:10]}"
+    shutil.move(str(output_dir), str(dest))
+    print(f"Moved existing {output_dir} to {dest}")
+    output_dir.mkdir(parents=True, exist_ok=True)
+
 
 def create_llm(model, temperature=0, base_url=None):
     """Create LLM — ChatAnthropic for Claude models, ChatOpenAI otherwise."""
@@ -256,8 +274,8 @@ async def main():
 
     SHOW_PROMPTS = args.show_prompts
     interactive = args.interactive
+    archive_existing_output_dir("generated_scripts")
     WORK_DIR = Path("generated_scripts")
-    WORK_DIR.mkdir(exist_ok=True)
 
     # Connect to MCP server
     mcp_server = find_mcp_server(args.mcp_server)

agentic/libe_agent_with_script_generator.py

@@ -24,6 +24,7 @@
 import subprocess
 import argparse
 import shutil
+import time
 from pathlib import Path
 from datetime import datetime
 from langchain_openai import ChatOpenAI
@@ -36,6 +37,9 @@
 # Maximum retry attempts for fixing failed scripts
 MAX_RETRIES = 2
 
+# Directory where existing generated_scripts runs are moved (create if missing)
+ARCHIVE_RUNS_DIR = "archive_runs"
+
 # LLM model to use — default depends on which API key is available
 DEFAULT_OPENAI_MODEL = "gpt-4o-mini"
 DEFAULT_ANTHROPIC_MODEL = "claude-sonnet-4-20250514"
@@ -233,6 +237,20 @@ def detect_run_script(directory):
         return None
     return run_scripts[0].name
 
+def archive_existing_output_dir(output_dir, archive_parent=None):
+    """If output_dir exists, move it to archive_parent/output_dir_<unique>, then create fresh output_dir."""
+    output_dir = Path(output_dir)
+    archive_dir = Path(archive_parent or ARCHIVE_RUNS_DIR)
+    if not output_dir.exists():
+        output_dir.mkdir(parents=True, exist_ok=True)
+        return
+    archive_dir.mkdir(parents=True, exist_ok=True)
+    dest = archive_dir / f"{output_dir.name}_{hex(time.time_ns())[2:10]}"
+    shutil.move(str(output_dir), str(dest))
+    print(f"Moved existing {output_dir} to {dest}")
+    output_dir.mkdir(parents=True, exist_ok=True)
+
+
 def copy_existing_scripts(scripts_dir, output_dir):
     """Copy scripts from existing directory and return as formatted text"""
     print(f"Using existing scripts from: {scripts_dir}")
@@ -368,7 +386,8 @@ async def main():
     SHOW_PROMPTS = args.show_prompts
 
     output_dir = "generated_scripts"
-    
+    archive_existing_output_dir(output_dir)
+
     # Copy existing scripts if provided
     archive_counter = 1
     if args.scripts: