Skip to content

Comments

Enable more customization for which prompt components gepa can optimize#826

Draft
eligotts wants to merge 3 commits intomainfrom
eli/gepa-for-rlm
Draft

Enable more customization for which prompt components gepa can optimize#826
eligotts wants to merge 3 commits intomainfrom
eli/gepa-for-rlm

Conversation

@eligotts
Copy link
Contributor

@eligotts eligotts commented Feb 5, 2026

Description

Previously, GEPA could only optimized system prompts that were defined in env.system_prompt. Now, we can optimize multiple prompt components, as well as dynamically loaded system prompts (like in rlm_env.py). We can also optimize env_groups as long as each environment exposes the same prompt components and starting prompts for those components. We can also optimize tool defs as well. The user can see which components are optimizable by vf-gepa <env> --see-components and specify which components to optimize with --components

Environments are now responsible for exposing prompt components. Default is exposing self.system_prompt and the tools registered in self.oai_tools.

wordle:
Screenshot 2026-02-04 at 6 05 55 PM

rlm_secrets:
Screenshot_2026-02-04_at_8 18 33_PM

Type of Change

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • Documentation update
  • Test improvement

Testing

  • All existing tests pass when running uv run pytest locally.
  • New tests have been added to cover the changes

Checklist

  • My code follows the style guidelines of this project as outlined in AGENTS.md
  • I have performed a self-review of my own code
  • I have commented my code, particularly in hard-to-understand areas
  • I have made corresponding changes to the documentation
  • My changes generate no new warnings
  • Any dependent changes have been merged and published

Additional Notes

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant