Examples¶

retrAI ships with ready-to-run example projects in the examples/ directory.

01 — SQL Optimization (DuckDB)¶

Optimize a slow analytical query against a DuckDB database.

cd examples/01_sql_duckdb
python seed.py              # generate sample data
retrai run sql-benchmark    # agent optimises the query

Goal: Reduce query execution time below the threshold defined in .retrai.yml.

A Python utility module with deliberately broken tests.

cd examples/02_pytest_fix
retrai run pytest           # agent fixes the source code

Goal: All tests in tests/test_utils.py pass.

Train and optimise a churn prediction model.

cd examples/03_ml_churn
python generate_data.py     # create synthetic dataset
retrai run ml-optimize      # agent improves model accuracy

Goal: Achieve target accuracy on the holdout set.

A benchmark script with a deliberately slow implementation.

cd examples/04_perf_optimize
retrai run perf-check       # agent optimises the code

Goal: bench.py completes under the time threshold.

A Python module with deliberate type errors.

cd examples/05_pyright_typing
retrai run pyright          # agent fixes all type errors

Goal: Zero pyright errors.

A Python script with style issues to be cleaned up.

cd examples/06_shell_linter
retrai run shell-goal       # agent fixes linting issues

Goal: Ruff reports zero violations.

for dir in examples/*/; do
  echo "=== Running $dir ==="
  retrai run --cwd "$dir"
done

Try different models

Each example can be run with any model:

retrai run pytest --cwd examples/02_pytest_fix --model gpt-4o
retrai run pytest --cwd examples/02_pytest_fix --model gemini/gemini-2.0-flash