fix configs

Author: codelion

configs/default_config.yaml

@@ -25,7 +25,7 @@ llm:
   secondary_model_weight: 0.2         # Sampling weight for secondary model
 
   # API configuration
-  api_base: "https://api.openai.com/v1"  # Base URL for API (change for non-OpenAI models)
+  api_base: "https://generativelanguage.googleapis.com/v1beta/openai/"  # Base URL for API (change for non-OpenAI models)
   api_key: null                       # API key (defaults to OPENAI_API_KEY env variable)
 
   # Generation parameters

examples/matrix_multiplication/optimize.py

@@ -23,6 +23,7 @@ async def main():
     parser = argparse.ArgumentParser(description="Matrix multiplication optimization example")
     parser.add_argument("--iterations", type=int, default=100, help="Number of iterations")
     parser.add_argument("--output", type=str, default="output", help="Output directory")
+    parser.add_argument("--config", type=str, help="Path to configuration file")
     args = parser.parse_args()
 
     # Set up paths
@@ -49,13 +50,16 @@ async def main():
     config.max_iterations = args.iterations
     config.llm.primary_model = "gemini-2.0-flash-lite"
     config.llm.secondary_model = "gemini-2.0-flash"
+    config.llm.api_base = "https://generativelanguage.googleapis.com/v1beta/openai/"
     config.diff_based_evolution = True
     config.allow_full_rewrites = False
 
-    # Initialize OpenEvolve
+    # Initialize OpenEvolve with the custom config
     openevolve = OpenEvolve(
         initial_program_path=str(initial_program_path),
         evaluation_file=str(evaluation_file),
+        config=config,  # Pass the config object directly
+        config_path=args.config,  # Also pass config_path if provided (lower priority)
         output_dir=str(output_dir),
     )
 

examples/min_max_distance/optimize.py

@@ -26,6 +26,7 @@ async def main():
     parser.add_argument("--iterations", type=int, default=100, help="Number of iterations")
     parser.add_argument("--output", type=str, default="output", help="Output directory")
     parser.add_argument("--num-points", type=int, default=16, help="Number of points")
+    parser.add_argument("--config", type=str, help="Path to configuration file")
     args = parser.parse_args()
 
     # Set up paths
@@ -52,6 +53,7 @@ async def main():
     config.max_iterations = args.iterations
     config.llm.primary_model = "gemini-2.0-flash-lite"
     config.llm.secondary_model = "gemini-2.0-flash"
+    config.llm.api_base = "https://generativelanguage.googleapis.com/v1beta/openai/"
     config.diff_based_evolution = True
     config.allow_full_rewrites = False
 
@@ -67,10 +69,12 @@ async def main():
     with open(evaluation_file, "w") as f:
         f.write(eval_content)
 
-    # Initialize OpenEvolve
+    # Initialize OpenEvolve with the custom config
     openevolve = OpenEvolve(
         initial_program_path=str(initial_program_path),
         evaluation_file=str(evaluation_file),
+        config=config,  # Pass config object directly
+        config_path=args.config,  # Also pass config_path if provided (lower priority)
         output_dir=str(output_dir),
     )
 

openevolve/cli.py

@@ -66,6 +66,24 @@ def parse_args() -> argparse.Namespace:
         default="INFO"
     )
 
+    parser.add_argument(
+        "--api-base",
+        help="Base URL for the LLM API",
+        default=None
+    )
+    
+    parser.add_argument(
+        "--primary-model",
+        help="Primary LLM model name",
+        default=None
+    )
+    
+    parser.add_argument(
+        "--secondary-model",
+        help="Secondary LLM model name",
+        default=None
+    )
+    
     return parser.parse_args()
 
 
@@ -87,12 +105,32 @@ async def main_async() -> int:
         print(f"Error: Evaluation file '{args.evaluation_file}' not found")
         return 1
 
+    # Create config object with command-line overrides
+    config = None
+    if args.api_base or args.primary_model or args.secondary_model:
+        # Load base config from file or defaults
+        config = load_config(args.config)
+        
+        # Apply command-line overrides
+        if args.api_base:
+            config.llm.api_base = args.api_base
+            print(f"Using API base: {config.llm.api_base}")
+        
+        if args.primary_model:
+            config.llm.primary_model = args.primary_model
+            print(f"Using primary model: {config.llm.primary_model}")
+        
+        if args.secondary_model:
+            config.llm.secondary_model = args.secondary_model
+            print(f"Using secondary model: {config.llm.secondary_model}")
+    
     # Initialize OpenEvolve
     try:
         openevolve = OpenEvolve(
             initial_program_path=args.initial_program,
             evaluation_file=args.evaluation_file,
-            config_path=args.config,
+            config=config,
+            config_path=args.config if config is None else None,
             output_dir=args.output,
         )
 

openevolve/controller.py

@@ -40,10 +40,16 @@ def __init__(
         initial_program_path: str,
         evaluation_file: str,
         config_path: Optional[str] = None,
+        config: Optional[Config] = None,
         output_dir: Optional[str] = None,
     ):
         # Load configuration
-        self.config = load_config(config_path)
+        if config is not None:
+            # Use provided Config object directly
+            self.config = config
+        else:
+            # Load from file or use defaults
+            self.config = load_config(config_path)
 
         # Set up output directory
         self.output_dir = output_dir or os.path.join(