mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
|
"""compose_prompt includes the Fable doctrine layer when the flag is on."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from roboco.agents.factories._base import compose_prompt
|
||
|
|
from roboco.models import AgentRole, Team
|
||
|
|
|
||
|
|
|
||
|
|
def test_doctrine_included_when_flag_enabled() -> None:
|
||
|
|
with patch("roboco.config.settings.fable_mode_enabled", True):
|
||
|
|
prompt = compose_prompt(AgentRole.DEVELOPER, Team.BACKEND, "be-dev-1")
|
||
|
|
assert "# Fable Doctrine" in prompt
|
||
|
|
assert "Turn discipline" in prompt
|
||
|
|
|
||
|
|
|
||
|
|
def test_doctrine_absent_when_flag_disabled() -> None:
|
||
|
|
with patch("roboco.config.settings.fable_mode_enabled", False):
|
||
|
|
prompt = compose_prompt(AgentRole.DEVELOPER, Team.BACKEND, "be-dev-1")
|
||
|
|
assert "# Fable Doctrine" not in prompt
|
||
|
|
|
||
|
|
|
||
|
|
def test_doctrine_frontmatter_not_leaked() -> None:
|
||
|
|
with patch("roboco.config.settings.fable_mode_enabled", True):
|
||
|
|
prompt = compose_prompt(AgentRole.DEVELOPER, Team.BACKEND, "be-dev-1")
|
||
|
|
assert "keep-coding-instructions" not in prompt
|
||
|
|
|
||
|
|
|
||
|
|
def test_doctrine_applies_to_every_role() -> None:
|
||
|
|
with patch("roboco.config.settings.fable_mode_enabled", True):
|
||
|
|
for role in AgentRole:
|
||
|
|
prompt = compose_prompt(role, None, f"probe-{role.value}")
|
||
|
|
assert "# Fable Doctrine" in prompt, f"missing for role={role.value}"
|