Add iteration_found field to track when best program was discovered

Author: hafeezhmhahk

openevolve/controller.py

@@ -149,6 +149,7 @@ async def run(
             code=self.initial_program_code,
             language=self.language,
             metrics=initial_metrics,
+            iteration_found=start_iteration
         )
 
         self.database.add(initial_program)
@@ -235,7 +236,7 @@ async def run(
                 )
 
                 # Add to database
-                self.database.add(child_program)
+                self.database.add(child_program, iteration=i+1)
 
                 # Log progress
                 iteration_time = time.time() - iteration_start
@@ -380,6 +381,7 @@ def _save_checkpoint(self, iteration: int) -> None:
                     {
                         "id": best_program.id,
                         "generation": best_program.generation,
+                        "iteration_found": best_program.iteration_found,
                         "iteration": iteration,
                         "metrics": best_program.metrics,
                         "language": best_program.language,
@@ -435,6 +437,7 @@ def _save_best_program(self, program: Optional[Program] = None) -> None:
                 {
                     "id": program.id,
                     "generation": program.generation,
+                    "iteration_found": program.iteration_found,
                     "timestamp": program.timestamp,
                     "parent_id": program.parent_id,
                     "metrics": program.metrics,

openevolve/database.py

@@ -32,6 +32,7 @@ class Program:
     parent_id: Optional[str] = None
     generation: int = 0
     timestamp: float = field(default_factory=time.time)
+    iteration_found: int = 0  # Track which iteration this program was found
 
     # Performance metrics
     metrics: Dict[str, float] = field(default_factory=dict)
@@ -90,17 +91,24 @@ def __init__(self, config: DatabaseConfig):
 
         logger.info(f"Initialized program database with {len(self.programs)} programs")
 
-    def add(self, program: Program) -> str:
+    def add(self, program: Program, iteration: int = None) -> str:
         """
         Add a program to the database
 
         Args:
             program: Program to add
+            iteration: Current iteration (defaults to last_iteration)
 
         Returns:
             Program ID
         """
         # Store the program
+        # If iteration is provided, update the program's iteration_found
+        if iteration is not None:
+            program.iteration_found = iteration
+            # Update last_iteration if needed
+            self.last_iteration = max(self.last_iteration, iteration)
+            
         self.programs[program.id] = program
 
         # Calculate feature coordinates for MAP-Elites