Skip to content

Commit 256decc

Browse files
committed
Add package data for default prompts and clean up templates
Added MANIFEST.in and updated pyproject.toml to include default prompt files in package data. Removed hardcoded minimal template fallbacks from TemplateManager, relying on external files for critical templates and fragments.
1 parent 5a2f6f1 commit 256decc

File tree

3 files changed

+5
-20
lines changed

3 files changed

+5
-20
lines changed

MANIFEST.in

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
include openevolve/prompts/defaults/*.txt
2+
include openevolve/prompts/defaults/*.json

openevolve/prompt/templates.py

Lines changed: 0 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -187,9 +187,6 @@ def __init__(self, custom_template_dir: Optional[str] = None):
187187
# 2. Override with custom templates (if provided)
188188
if self.custom_dir and self.custom_dir.exists():
189189
self._load_from_directory(self.custom_dir)
190-
191-
# 3. Minimal hardcoded fallbacks (for safety/compatibility)
192-
self._ensure_minimal_templates()
193190

194191
def _load_from_directory(self, directory: Path) -> None:
195192
"""Load all templates and fragments from a directory"""
@@ -208,23 +205,6 @@ def _load_from_directory(self, directory: Path) -> None:
208205
with open(fragments_file, 'r') as f:
209206
loaded_fragments = json.load(f)
210207
self.fragments.update(loaded_fragments)
211-
212-
def _ensure_minimal_templates(self) -> None:
213-
"""Ensure critical templates exist (backward compatibility)"""
214-
if "system_message" not in self.templates:
215-
self.templates["system_message"] = "You are an AI assistant helping with code evolution."
216-
if "diff_user" not in self.templates:
217-
self.templates["diff_user"] = "# Task\nImprove the program:\n```\n{current_program}\n```"
218-
if "full_rewrite_user" not in self.templates:
219-
self.templates["full_rewrite_user"] = "# Task\nRewrite the program:\n```\n{current_program}\n```"
220-
221-
# Ensure critical fragments exist
222-
if "fitness_improved" not in self.fragments:
223-
self.fragments["fitness_improved"] = "Fitness improved: {prev} → {current}"
224-
if "fitness_declined" not in self.fragments:
225-
self.fragments["fitness_declined"] = "Fitness declined: {prev} → {current}"
226-
if "no_specific_guidance" not in self.fragments:
227-
self.fragments["no_specific_guidance"] = "Focus on improving fitness while maintaining diversity"
228208

229209
def get_template(self, name: str) -> str:
230210
"""Get a template by name"""

pyproject.toml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,5 +50,8 @@ openevolve-run = "openevolve.cli:main"
5050
[tool.setuptools.packages.find]
5151
include = ["openevolve*"]
5252

53+
[tool.setuptools.package-data]
54+
openevolve = ["prompts/defaults/*.txt", "prompts/defaults/*.json"]
55+
5356
[tool.setuptools.dynamic]
5457
version = {attr = "openevolve._version.__version__"}

0 commit comments

Comments
 (0)