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.1.tar.gz (116.8 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.1-py3-none-any.whl (135.9 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: gepa_adk-0.2.1.tar.gz
  • Upload date:
  • Size: 116.8 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.1.tar.gz
Algorithm Hash digest
SHA256 20741e8c4cd049a5a1e918ea11c29b6de33ada2c3cdacf2844f362accf09203b
MD5 6fda0a127b885b5ee6e89cc6db4ec135
BLAKE2b-256 ec24fc1eefc12fa82a7bea32c34e96af73e56682320748dae188044d5efbe726

See more details on using hashes here.

Provenance

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

File metadata

  • Download URL: gepa_adk-0.2.1-py3-none-any.whl
  • Upload date:
  • Size: 135.9 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.1-py3-none-any.whl
Algorithm Hash digest
SHA256 6a0d6cb49e5857c98f954e9d7686bcbdd62d55b15701fa816dddfc6dfaa7034f
MD5 e000fe50326a128f49595458bca174d9
BLAKE2b-256 c0cc3e83fc3e80d9282429178c7b94d653b77cae09e6e196139bbaba2d95bf3a

See more details on using hashes here.

Provenance

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