OpenAlpha_Evolve is an open-source Python framework inspired by the groundbreaking research on autonomous coding agents like DeepMind's AlphaEvolve. It's a regeneration of the core idea: an intelligent system that iteratively writes, tests, and improves code using Large Language Models (LLMs) like Google's Gemini, guided by the principles of evolution.
Our mission is to provide an accessible, understandable, and extensible platform for researchers, developers, and enthusiasts to explore the fascinating intersection of AI, code generation, and automated problem-solving.
Imagine an agent that can:
- Understand a complex problem description.
- Generate initial algorithmic solutions.
- Rigorously test its own code.
- Learn from failures and successes.
- Evolve increasingly sophisticated and efficient algorithms over time.
OpenAlpha_Evolve is a step towards this vision. It's not just about generating code; it's about creating a system that discovers and refines solutions autonomously.
OpenAlpha_Evolve employs a modular, agent-based architecture to orchestrate an evolutionary process:
- Task Definition: You, the user, define the algorithmic "quest" – the problem to be solved, including examples of inputs and expected outputs.
- Prompt Engineering (
PromptDesignerAgent
): This agent crafts intelligent prompts for the LLM. It designs:- Initial Prompts: To generate the first set of candidate solutions.
- Mutation Prompts: To introduce variations and improvements to existing solutions, often requesting changes in a "diff" format.
- Bug-Fix Prompts: To guide the LLM in correcting errors from previous attempts, also typically expecting a "diff".
- Code Generation (
CodeGeneratorAgent
): Powered by an LLM (currently configured for Gemini), this agent takes the prompts and generates Python code. If a "diff" is requested and received, it attempts to apply the changes to the parent code. - Evaluation (
EvaluatorAgent
): The generated code is put to the test!- Syntax Check: Is the code valid Python?
- Execution: The code is run in a temporary, isolated environment against the input/output examples defined in the task.
- Fitness Scoring: Programs are scored based on correctness (how many test cases pass), efficiency (runtime), and other potential metrics.
- Database (
DatabaseAgent
): All programs (code, fitness scores, generation, lineage) are stored, creating a record of the evolutionary history (currently in-memory). - Selection (
SelectionControllerAgent
): The "survival of the fittest" principle in action. This agent selects:- Parents: Promising programs from the current generation to produce offspring.
- Survivors: The best programs from both the current population and new offspring to advance to the next generation.
- Iteration: This cycle repeats for a defined number of generations, with each new generation aiming to produce better solutions than the last.
- Orchestration (
TaskManagerAgent
): The maestro of the operation, coordinating all other agents and managing the overall evolutionary loop.
- LLM-Powered Code Generation: Leverages state-of-the-art Large Language Models (Google Gemini integration included).
- Evolutionary Algorithm Core: Implements iterative improvement through selection, LLM-driven mutation/bug-fixing (via diffs), and survival.
- Modular Agent Architecture: Easily extend or replace individual components (e.g., use a different LLM, database, or evaluation strategy).
- Automated Program Evaluation: Syntax checking and functional testing against user-provided examples with timeout mechanisms.
- Configuration Management: Easily tweak parameters like population size, number of generations, LLM models, and API settings via
config/settings.py
. - Detailed Logging: Comprehensive logs provide insights into each step of the evolutionary process.
- Diff-based Mutations: The system is designed to use diffs for mutations and bug fixes, allowing for more targeted code modifications by the LLM.
- Open Source & Extensible: Built with Python, designed for experimentation and community contributions.
./
├── agents/ # Core intelligent agents (subdirectories for each)
│ ├── code_generator/
│ ├── database_agent/
│ ├── evaluator_agent/
│ ├── prompt_designer/
│ ├── selection_controller/
│ ├── task_manager/
│ ├── rl_finetuner/ # Placeholder for Reinforcement Learning Fine-Tuner
│ └── monitoring_agent/ # Placeholder for Monitoring Agent
├── config/ # Configuration files (settings.py)
├── core/ # Core interfaces, data models (Program, TaskDefinition)
├── utils/ # Utility functions (if any, currently minimal)
├── tests/ # Unit and integration tests (placeholders, to be expanded)
├── scripts/ # Helper scripts (e.g., diagram generation)
├── main.py # Main entry point to run the system
├── requirements.txt # Project dependencies
├── .env.example # Example for environment variables (copy to .env)
├── .gitignore # Specifies intentionally untracked files that Git should ignore
├── LICENSE.md # Project's license information (MIT License)
└── README.md # This file!
-
Prerequisites:
- Python 3.10+
pip
for package managementgit
for cloning
-
Clone the Repository:
git clone https://github.com/shyamsaktawat/OpenAlpha_Evolve.git cd OpenAlpha_Evolve
-
Set Up a Virtual Environment (recommended):
python -m venv venv source venv/bin/activate # On Windows: venv\Scripts\activate
-
Install Dependencies:
pip install -r requirements.txt
-
Set Up Environment Variables (Crucial for API Key):
- Copy
.env.example
to a new file named.env
in the project root:cp .env.example .env
- Edit the
.env
file and add yourGEMINI_API_KEY
:You can obtain your API key from Google AI Studio.GEMINI_API_KEY="YOUR_ACTUAL_GEMINI_API_KEY_HERE"
- The system will prioritize the key from the
.env
file. If it's not found, it will use a non-functional placeholder fromconfig/settings.py
and print a warning. Ensure your.env
file is correctly set up.
- Copy
-
Review Configuration (Optional):
- Open
config/settings.py
. Here you can:- Change the default LLM models used for generation (
GEMINI_PRO_MODEL_NAME
) and evaluation (GEMINI_EVALUATION_MODEL
). - Adjust evolutionary parameters like
POPULATION_SIZE
andGENERATIONS
. - Modify API retry settings or logging levels.
- Change the default LLM models used for generation (
- Open
-
Run OpenAlpha_Evolve! The
main.py
file is configured with an example task (Dijkstra's algorithm). To run it:python -m main
Watch the logs in your terminal to see the evolutionary process unfold! Log files are also saved to
alpha_evolve.log
(by default).
Want to challenge OpenAlpha_Evolve with a new problem? It's easy:
-
Open
main.py
. -
Modify the
TaskDefinition
object:id
: A unique string identifier for your task (e.g., "sort_list_task").description
: A clear, detailed natural language description of the problem. This is crucial for the LLM to understand what to do. Be specific about function names, expected behavior, and constraints.function_name_to_evolve
: The name of the Python function the agent should try to create/evolve (e.g., "custom_sort").input_output_examples
: A list of dictionaries, each containing aninput
(arguments for your function) and the corresponding expectedoutput
. These are vital for evaluation.- Inputs should be provided as a list if the function takes multiple positional arguments, or as a single value if it takes one.
- Use
float('inf')
orfloat('-inf')
directly in your Python code defining these examples if needed by your problem (the evaluation harness handles JSON serialization/deserialization of these).
allowed_imports
: Specify a list of Python standard libraries that the generated code is allowed to import (e.g.,["heapq", "math", "sys"]
). This helps guide the LLM and can be important for the execution sandbox.- (Optional)
evaluation_criteria
: Define how success is measured (currently primarily driven by correctness based on test cases). - (Optional)
initial_code_prompt
: Override the default initial prompt if you need more specific instructions for the first code generation attempt.
-
Run the agent as before:
python -m main
.
The quality of your description
and the comprehensiveness of your input_output_examples
significantly impact the agent's success!
OpenAlpha_Evolve is a living project. Here are some directions we're excited to explore (and invite contributions for!):
- Advanced Evaluation Sandboxing: Implementing more robust, secure sandboxing (e.g., using Docker or
nsjail
) for code execution to handle potentially unsafe code and complex dependencies. - Sophisticated Fitness Metrics: Beyond correctness and basic runtime, incorporating checks for code complexity (e.g., cyclomatic complexity), style (linting), resource usage (memory), and custom domain-specific metrics.
- Reinforcement Learning for Prompt Strategy: Implementing the
RLFineTunerAgent
to dynamically optimize prompt engineering strategies based on performance feedback. - Enhanced Monitoring & Visualization: Developing tools (via
MonitoringAgent
) to visualize the evolutionary process, track fitness landscapes, and understand agent behavior (e.g., using a simple web dashboard or plots). - Broader LLM Support: Adding easy integrations for other powerful LLMs (e.g., OpenAI models, Anthropic Claude).
- Self-Correction & Reflection: Enabling the agent to analyze its own failures more deeply (e.g., analyze error messages, identify patterns in failed tests) and refine its problem-solving approach.
- Diverse Task Domains: Applying OpenAlpha_Evolve to a wider range of problems in science, engineering, data analysis, and creative coding.
- Community-Driven Task Library: Building a collection of interesting and challenging tasks contributed by the community.
- Improved Diff Application: Making the diff application more robust or exploring alternative ways for the LLM to suggest modifications.
- Crossover Implementation: Adding a genetic crossover mechanism as an alternative or supplement to LLM-driven mutation.
This is an open invitation to collaborate! Whether you're an AI researcher, a Python developer, or simply an enthusiast, your contributions are welcome.
- Report Bugs: Find an issue? Please create an issue on GitHub!
- Suggest Features: Have an idea to make OpenAlpha_Evolve better? Open an issue to discuss it!
- Submit Pull Requests:
- Fork the repository.
- Create a new branch for your feature or bugfix (
git checkout -b feature/your-feature-name
). - Write clean, well-documented code.
- Add tests for your changes if applicable.
- Ensure your changes don't break existing functionality.
- Submit a pull request with a clear description of your changes!
Let's evolve this agent together!
This project is licensed under the MIT License. See the LICENSE.md
file for details.
OpenAlpha_Evolve is proudly inspired by the pioneering work of the Google DeepMind team on AlphaEvolve and other related research in LLM-driven code generation and automated discovery. This project aims to make the core concepts more accessible for broader experimentation and learning. We stand on the shoulders of giants.
Disclaimer: This is an experimental project. Generated code may not always be optimal, correct, or secure. Always review and test code thoroughly, especially before using it in production environments.