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

Two complete working examples are available in the examples/ directory:

Both 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.1.0.tar.gz (94.3 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.1.0-py3-none-any.whl (109.6 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: gepa_adk-0.1.0.tar.gz
  • Upload date:
  • Size: 94.3 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.1.0.tar.gz
Algorithm Hash digest
SHA256 4f044735de0991481a4de0c05a827a4149f4fd8538af070702e0bd5fa7370549
MD5 a5a90f3f0359c26a1587f8edd94ebe47
BLAKE2b-256 8f64106c7fa460fa41b109c01f8dbc99bed528c76a81e47a5289fadfc343bdc5

See more details on using hashes here.

Provenance

The following attestation bundles were made for gepa_adk-0.1.0.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.1.0-py3-none-any.whl.

File metadata

  • Download URL: gepa_adk-0.1.0-py3-none-any.whl
  • Upload date:
  • Size: 109.6 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.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 68ed45cc07aeab24248a14f12610fe034adf79c19cb83611de0d70a2dcab50cd
MD5 422e756b4f996b2e96a46a7c046bc84d
BLAKE2b-256 9cfeef9ecfcd03c2d64cd5c3875873dd1d7b944b8ad582a6fc17043723eee54a

See more details on using hashes here.

Provenance

The following attestation bundles were made for gepa_adk-0.1.0-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