Skip to main content

Evolutionary optimization for Google ADK agents

Project description

gepa-adk

Evolutionary optimization for Google ADK agents.

What is this?

gepa-adk makes your AI agents better automatically. It takes an agent, runs it against examples, gets feedback, and evolves the agent's instructions until performance improves.

Think of it as natural selection for AI prompts—the best instructions survive and improve.

Who is this for?

Teams building AI agents with Google's Agent Development Kit (ADK) who want to:

  • Improve agent performance without manual prompt tweaking
  • Use structured feedback (not just pass/fail) to guide improvements
  • Evolve multiple agents working together
  • Get 3-5x faster optimization through parallel evaluation

Installation

Prerequisites

Before installing gepa-adk, you need:

  1. Python 3.12+
  2. Ollama with the gpt-oss:20b model:
    # Install Ollama (if not already installed)
    # Visit https://ollama.ai for installation instructions
    
    # Pull the required model
    ollama pull gpt-oss:20b
    
  3. Set environment variable:
    export OLLAMA_API_BASE=http://localhost:11434
    

Why gpt-oss:20b? The evolutionary optimization engine uses this model internally to generate improved agent instructions. Without it, evolution will fail.

Why local models? Evolutionary optimization makes many LLM calls per run (evaluating multiple candidates across iterations). We recommend Ollama with open-source models to avoid API costs and rate limits. However, gepa-adk works with any Google ADK-supported model (including Gemini) - just be aware of potential costs.

Install gepa-adk

pip install gepa-adk

For development or if using uv:

uv add gepa-adk

Quick Start

from pydantic import BaseModel, Field
from google.adk.agents import LlmAgent
from gepa_adk import evolve_sync

class Output(BaseModel):
    answer: str
    score: float = Field(ge=0.0, le=1.0)

agent = LlmAgent(name="assistant", model="gemini-2.0-flash",
                 instruction="You are a helpful assistant.", output_schema=Output)
trainset = [{"input": "What is 2+2?", "expected": "4"}]
result = evolve_sync(agent, trainset)
print(f"Evolved: {result.evolved_instruction}")

Examples

Complete working examples are available in the examples/ directory:

All examples require Ollama with gpt-oss:20b model (see Prerequisites above).

Run an example:

python examples/basic_evolution.py

Documentation

  • Getting Started Guide — Step-by-step walkthrough from installation to first evolution
  • Use Case Guides — Patterns for single-agent, critic agents, multi-agent, and workflows
  • API Reference — Complete documentation for all public functions and classes

Troubleshooting

"Model not found" or "Connection refused" errors

Ensure Ollama is running and the model is pulled:

# Check Ollama is running
curl http://localhost:11434/api/tags

# Pull the model if not present
ollama pull gpt-oss:20b

# Verify the model is available
ollama list | grep gpt-oss

Evolution is slow or uses too many iterations

Adjust the EvolutionConfig parameters:

from gepa_adk import EvolutionConfig

config = EvolutionConfig(
    max_iterations=3,  # Reduce iterations
    patience=2,        # Stop early if no improvement
)

result = evolve_sync(agent, trainset, config=config)

Want to use a different model?

For your agents: You can use any model supported by Google ADK (Gemini, Ollama models, etc.). The examples use ollama_chat/gpt-oss:20b but you can change this to gemini-2.0-flash or other ADK-supported models.

For the evolution engine: Currently, the reflection model is hardcoded to ollama_chat/gpt-oss:20b. Future versions will support custom model configuration. For now, ensure this model is available in your Ollama instance.

Cost warning: Using cloud APIs like Gemini for agents during evolution can result in high costs due to the many evaluation calls required.

Status

In Development — Not yet ready for production use.

See docs/proposals/ for technical design and roadmap.

Credits

This project implements concepts from GEPA (Genetic-Pareto optimization) and integrates with Google ADK.

License

Apache License 2.0 - see LICENSE for details.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

gepa_adk-0.2.2.tar.gz (132.0 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

gepa_adk-0.2.2-py3-none-any.whl (154.3 kB view details)

Uploaded Python 3

File details

Details for the file gepa_adk-0.2.2.tar.gz.

File metadata

  • Download URL: gepa_adk-0.2.2.tar.gz
  • Upload date:
  • Size: 132.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: uv/0.9.26 {"installer":{"name":"uv","version":"0.9.26","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"22.04","id":"jammy","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

File hashes

Hashes for gepa_adk-0.2.2.tar.gz
Algorithm Hash digest
SHA256 b8c66553f90d75810264a6c10a82ab568a2de22f21ae3c6fa2e9003670919503
MD5 80a85d9beb5bbcc6092f2215ad0f6754
BLAKE2b-256 c01fa511d899b3433329b887f127203a8911d7ac1350ffc2f4d793a2c4edcc78

See more details on using hashes here.

Provenance

The following attestation bundles were made for gepa_adk-0.2.2.tar.gz:

Publisher: publish.yml on Alberto-Codes/gepa-adk

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file gepa_adk-0.2.2-py3-none-any.whl.

File metadata

  • Download URL: gepa_adk-0.2.2-py3-none-any.whl
  • Upload date:
  • Size: 154.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: uv/0.9.26 {"installer":{"name":"uv","version":"0.9.26","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"22.04","id":"jammy","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

File hashes

Hashes for gepa_adk-0.2.2-py3-none-any.whl
Algorithm Hash digest
SHA256 ebdd47f0a85f5b94bb2545559b4f62fcdff474698ba242095581f601b2d3f1a3
MD5 72b854b221c6b5a8dea72c19681f5bdc
BLAKE2b-256 981096ee114cc7b5d50deede12089b2d89c1e81b1e0d0d5d86aa906a8c06384c

See more details on using hashes here.

Provenance

The following attestation bundles were made for gepa_adk-0.2.2-py3-none-any.whl:

Publisher: publish.yml on Alberto-Codes/gepa-adk

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page