Fix configs

Author: codelion

openevolve/config.py

@@ -83,6 +83,9 @@ class DatabaseConfig:
     # Migration parameters for island-based evolution
     migration_interval: int = 50  # Migrate every N generations
     migration_rate: float = 0.1  # Fraction of population to migrate
+    
+    # Random seed for reproducible sampling
+    random_seed: Optional[int] = None
 
 
 @dataclass
@@ -209,6 +212,7 @@ def to_dict(self) -> Dict[str, Any]:
                 "feature_bins": self.database.feature_bins,
                 "migration_interval": self.database.migration_interval,
                 "migration_rate": self.database.migration_rate,
+                "random_seed": self.database.random_seed,
             },
             "evaluator": {
                 "timeout": self.evaluator.timeout,

openevolve/controller.py

@@ -66,6 +66,14 @@ def __init__(
 
         # Set up logging
         self._setup_logging()
+        
+        # Set random seed for reproducibility if specified
+        if self.config.random_seed is not None:
+            import random
+            import numpy as np
+            random.seed(self.config.random_seed)
+            np.random.seed(self.config.random_seed)
+            logger.info(f"Set random seed to {self.config.random_seed} for reproducibility")
 
         # Load initial program
         self.initial_program_path = initial_program_path
@@ -85,6 +93,11 @@ def __init__(
         # Initialize components
         self.llm_ensemble = LLMEnsemble(self.config.llm)
         self.prompt_sampler = PromptSampler(self.config.prompt)
+        
+        # Pass random seed to database if specified
+        if self.config.random_seed is not None:
+            self.config.database.random_seed = self.config.random_seed
+            
         self.database = ProgramDatabase(self.config.database)
         self.evaluator = Evaluator(self.config.evaluator, evaluation_file, self.llm_ensemble)
 

openevolve/database.py

@@ -97,6 +97,12 @@ def __init__(self, config: DatabaseConfig):
         # Load database from disk if path is provided
         if config.db_path and os.path.exists(config.db_path):
             self.load(config.db_path)
+            
+        # Set random seed for reproducible sampling if specified
+        if config.random_seed is not None:
+            import random
+            random.seed(config.random_seed)
+            logger.debug(f"Database: Set random seed to {config.random_seed}")
 
         logger.info(f"Initialized program database with {len(self.programs)} programs")
 

openevolve/llm/openai.py

@@ -36,8 +36,10 @@ def __init__(
 
     async def generate(self, prompt: str, **kwargs) -> str:
         """Generate text from a prompt"""
+        # Use default system message if not provided in kwargs
+        system_message = kwargs.pop('system_message', "You are a helpful assistant.")
         return await self.generate_with_context(
-            system_message=self.config.system_message,
+            system_message=system_message,
             messages=[{"role": "user", "content": prompt}],
             **kwargs,
         )