mirror of https://github.com/aliasrobotics/cai.git
46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
"""Ensure packaged prompt markdown loads through `load_prompt_template` (Mako render succeeds)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from cai.util import load_prompt_template
|
|
from cai.util.prompts import _MICRO_PROFILE_PATHS
|
|
|
|
# Primary agent system templates (short list for CI time).
|
|
_SYSTEM_PROMPT_RELPATHS = (
|
|
"prompts/system_orchestration_agent.md",
|
|
"prompts/system_selection_agent.md",
|
|
"prompts/system_red_team_agent.md",
|
|
"prompts/system_blue_team_agent.md",
|
|
"prompts/system_bug_bounter.md",
|
|
"prompts/system_web_pentester.md",
|
|
"prompts/system_ctf_agent.md",
|
|
"prompts/system_compliance_agent.md",
|
|
"prompts/system_use_cases.md",
|
|
"prompts/memory_analysis_agent.md",
|
|
"prompts/reverse_engineering_agent.md",
|
|
"prompts/subghz_agent.md",
|
|
"prompts/wifi_security_agent.md",
|
|
"prompts/system_dfir_agent.md",
|
|
"prompts/system_network_analyzer.md",
|
|
"prompts/system_replay_attack_agent.md",
|
|
)
|
|
|
|
|
|
def test_system_templates_load(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.delenv("CAI_CYBER_PROFILE", raising=False)
|
|
for rel in _SYSTEM_PROMPT_RELPATHS:
|
|
text = load_prompt_template(rel)
|
|
assert isinstance(text, str)
|
|
assert len(text) > 80, f"prompt unexpectedly short: {rel}"
|
|
|
|
|
|
def test_micro_profile_templates_load(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.delenv("CAI_CYBER_PROFILE", raising=False)
|
|
for rel in sorted(set(_MICRO_PROFILE_PATHS.values())):
|
|
text = load_prompt_template(rel)
|
|
assert isinstance(text, str)
|
|
assert len(text) > 40, f"micro prompt unexpectedly short: {rel}"
|
|
assert "MICRO-PROFILE" in text.upper(), rel
|