update cli

Author: 0x0f0f0f

openevolve/cli.py

@@ -22,10 +22,11 @@ def parse_args() -> argparse.Namespace:
     parser.add_argument(
         "evaluation_file", help="Path to the evaluation file containing an 'evaluate' function"
     )
+
     parser.add_argument(
-        "initial_program",
-        nargs="?",
-        help="Path to the initial program file",
+        "initial_programs",
+        nargs="+",
+        help="Path(s) to one or more initial program files",
         default=None,
     )
 
@@ -61,8 +62,6 @@ 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()
 
 
@@ -76,39 +75,12 @@ async def main_async() -> int:
     args = parse_args()
 
     # 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")
+
+    for program in args.initial_programs:
+        if not os.path.isfile(program):
+            print(f"Error: Initial program file '{program}' does not exist")
             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):
         print(f"Error: Evaluation file '{args.evaluation_file}' not found")
         return 1
@@ -135,7 +107,7 @@ async def main_async() -> int:
     # Initialize OpenEvolve
     try:
         openevolve = OpenEvolve(
-            initial_programs_paths=initial_programs_paths,
+            initial_programs_paths=args.initial_programs,
             evaluation_file=args.evaluation_file,
             config=config,
             config_path=args.config if config is None else None,