Function evolve_function

api.py:258–345

Evolve a specific Python function based on a set of test cases. Args: func: The initial function to evolve. test_cases: A list of (input, expected_output) tuples for validation. iterations: The number of evolution iterations to run. **kwargs: Additional argu

(
    func: Callable, test_cases: List[Tuple[Any, Any]], iterations: int = 100, **kwargs
)

# --- Additional Helper Functions for Common Use Cases ---

def evolve_function(
    func: Callable, test_cases: List[Tuple[Any, Any]], iterations: int = 100, **kwargs
) -> EvolutionResult:
    """
    Evolve a specific Python function based on a set of test cases.

    Args:
        func: The initial function to evolve.
        test_cases: A list of (input, expected_output) tuples for validation.
        iterations: The number of evolution iterations to run.
        **kwargs: Additional arguments to pass to the main `run_evolution` function.

    Returns:
        An EvolutionResult object with the optimized function.
    """

    # Get the source code of the provided function.
    func_source = inspect.getsource(func)
    func_name = func.__name__

    # Ensure the function source has evolution markers. If not, add them around the function body.
    if "EVOLVE-BLOCK-START" not in func_source:
        lines = func_source.split("\n")
        # Find the line where the function is defined.
        func_def_line = next(i for i, line in enumerate(lines) if line.strip().startswith("def "))

        # A simplified approach to find the end of the function by looking for a change in indentation.
        indent = len(lines[func_def_line]) - len(lines[func_def_line].lstrip())
        func_end = len(lines)
        for i in range(func_def_line + 1, len(lines)):
            if lines[i].strip() and (len(lines[i]) - len(lines[i].lstrip())) <= indent:
                func_end = i
                break

        # Insert evolution markers into the source code.
        lines.insert(func_def_line + 1, " " * (indent + 4) + "# EVOLVE-BLOCK-START")
        lines.insert(func_end + 1, " " * (indent + 4) + "# EVOLVE-BLOCK-END")
        func_source = "\n".join(lines)

    # Create a custom evaluator function that runs the test cases against the evolved code.
    def evaluator(program_path):
        import importlib.util
        
        # Load the evolved program from its file path.
        spec = importlib.util.spec_from_file_location("evolved", program_path)
        if spec is None or spec.loader is None:
            return {"score": 0.0, "error": "Failed to load program"}

        module = importlib.util.module_from_spec(spec)
        try:
            spec.loader.exec_module(module)
        except Exception as e:
            return {"score": 0.0, "error": f"Failed to execute program: {str(e)}"}

        if not hasattr(module, func_name):
            return {"score": 0.0, "error": f"Function '{func_name}' not found"}

        evolved_func = getattr(module, func_name)