change cli

Author: 0x0f0f0f

openevolve/cli.py

@@ -19,7 +19,7 @@ def parse_args() -> argparse.Namespace:
     """Parse command-line arguments"""
     parser = argparse.ArgumentParser(description="OpenEvolve - Evolutionary coding agent")
 
-    parser.add_argument("initial_program", help="Path to the initial program file")
+    parser.add_argument("initial_program", help="Path to the initial program file", default=None)
 
     parser.add_argument(
         "evaluation_file", help="Path to the evaluation file containing an 'evaluate' function"
@@ -57,6 +57,8 @@ def parse_args() -> argparse.Namespace:
 
     parser.add_argument("--secondary-model", help="Secondary LLM model name", default=None)
 
+    parser.add_argument("--initial-programs-dir", help="Path to initial programs directory", default=None)
+
     return parser.parse_args()
 
 
@@ -69,9 +71,38 @@ async def main_async() -> int:
     """
     args = parse_args()
 
-    # Check if files exist
-    if not os.path.exists(args.initial_program):
-        print(f"Error: Initial program file '{args.initial_program}' not found")
+    # Check if files exist.
+    # If args.initial_program is present, it should be a file.
+    # If args.initial_programs_dir is present, it should be a directory, and args.initial_program should be `None`.
+    if args.initial_programs_dir:
+        if args.initial_program:
+            print("Error: Cannot specify both initial-program and --initial-programs-dir")
+            return 1
+        if not os.path.isdir(args.initial_programs_dir):
+            print(f"Error: Initial programs path '{args.initial_programs_dir}' is not a directory")
+            return 1
+    elif args.initial_program:
+        if args.initial_programs_dir:
+            print("Error: Cannot specify both --initial-programs-dir and initial_program")
+            return 1
+        if not os.path.isfile(args.initial_program):
+            print(f"Error: Initial program file '{args.initial_program}' is not a file")
+            return 1
+        
+    # Populate the initial_programs_paths vector.
+    initial_programs_paths = []
+    if args.initial_programs_dir:
+        initial_programs_paths = [
+            os.path.join(args.initial_programs_dir, f)
+            for f in os.listdir(args.initial_programs_dir)
+            if f.endswith(".py")
+        ]
+    elif args.initial_program:
+        initial_programs_paths = [args.initial_program]
+
+    # Check that initial_programs_paths is not empty
+    if not initial_programs_paths:
+        print("Error: No initial programs found. Please provide a valid initial program or directory.")
         return 1
 
     if not os.path.exists(args.evaluation_file):
@@ -100,7 +131,7 @@ async def main_async() -> int:
     # Initialize OpenEvolve
     try:
         openevolve = OpenEvolve(
-            initial_program_path=args.initial_program,
+            initial_programs_paths=initial_programs_paths,
             evaluation_file=args.evaluation_file,
             config=config,
             config_path=args.config if config is None else None,