fixes

Author: codelion

examples/r_robust_regression/config.yaml

@@ -4,6 +4,7 @@
 max_iterations: 100
 checkpoint_interval: 10
 log_level: "INFO"
+file_suffix: ".r"
 
 # LLM configuration
 llm:

examples/rust_adaptive_sort/config.yaml

@@ -4,6 +4,7 @@
 max_iterations: 150
 checkpoint_interval: 15
 log_level: "INFO"
+file_suffix: ".rs"
 
 # LLM configuration
 llm:

openevolve/config.py

@@ -337,6 +337,7 @@ class Config:
     log_dir: Optional[str] = None
     random_seed: Optional[int] = 42
     language: str = None
+    file_suffix: str = ".py"
 
     # Component configurations
     llm: LLMConfig = field(default_factory=LLMConfig)

openevolve/controller.py

@@ -138,6 +138,10 @@ def __init__(
             if not self.file_extension.startswith("."):
                 self.file_extension = f".{self.file_extension}"
 
+        # Set the file_suffix in config (can be overridden in YAML)
+        if not hasattr(self.config, 'file_suffix') or self.config.file_suffix == ".py":
+            self.config.file_suffix = self.file_extension
+
         # Initialize components
         self.llm_ensemble = LLMEnsemble(self.config.llm.models)
         self.llm_evaluator_ensemble = LLMEnsemble(self.config.llm.evaluator_models)
@@ -301,7 +305,7 @@ async def run(
         try:
             self.parallel_controller = ProcessParallelController(
                 self.config, self.evaluation_file, self.database, self.evolution_tracer,
-                file_suffix=self.file_extension
+                file_suffix=self.config.file_suffix
             )
 
             # Set up signal handlers for graceful shutdown