86 lines
3.3 KiB
Python
86 lines
3.3 KiB
Python
from app.config.ai_prompt_registry import (
|
|
PROMPT_REGISTRY_MAP,
|
|
PROMPT_FAMILY_ALLOWED,
|
|
get_prompt_registry,
|
|
get_prompt_registry_entry,
|
|
iter_prompt_registry_keys,
|
|
)
|
|
from app.config.ai_prompts import (
|
|
PROMPT_DECOMPOSE_TASKS,
|
|
PROMPT_FEISHU_PROJECT_SIGNAL_CLASSIFICATION,
|
|
PROMPT_FEISHU_REQUIREMENT_ANALYSIS,
|
|
PROMPT_TASK_BREAKDOWN,
|
|
)
|
|
from app.config.ai_prompts_catalog import __all__ as PROMPT_CATALOG_EXPORTS
|
|
|
|
|
|
def test_prompt_registry_covers_all_catalog_exports() -> None:
|
|
catalog_keys = set(PROMPT_CATALOG_EXPORTS)
|
|
registry_keys = set(iter_prompt_registry_keys())
|
|
|
|
assert registry_keys == catalog_keys
|
|
assert set(PROMPT_REGISTRY_MAP) == catalog_keys
|
|
|
|
for key in catalog_keys:
|
|
entry = get_prompt_registry_entry(key)
|
|
assert entry is not None
|
|
assert entry["prompt_key"] == key
|
|
assert entry["prompt_family"] in PROMPT_FAMILY_ALLOWED
|
|
assert entry["responsibility_scope"]
|
|
assert entry["priority"] in {"high", "medium", "low"}
|
|
assert entry["audience"]
|
|
|
|
|
|
def test_prompt_registry_returns_detached_copies() -> None:
|
|
registry = get_prompt_registry()
|
|
original_key = registry[0]["prompt_key"]
|
|
|
|
registry[0]["prompt_key"] = "BROKEN_KEY"
|
|
|
|
assert PROMPT_REGISTRY_MAP[original_key]["prompt_key"] == original_key
|
|
|
|
|
|
def test_high_frequency_registry_entries_have_complete_metadata() -> None:
|
|
expected_families = {
|
|
"PROMPT_FEISHU_PROJECT_SIGNAL_CLASSIFICATION": "classification",
|
|
"PROMPT_FEISHU_LEVEL_CLASSIFICATION": "classification",
|
|
"PROMPT_FEISHU_SIGNAL_EXTRACTION": "extraction",
|
|
"PROMPT_FEISHU_TASK_ORCHESTRATION": "planning",
|
|
"PROMPT_SIGNAL_PLANNER": "planning",
|
|
"PROMPT_GENERATE_CHARTER": "planning",
|
|
"PROMPT_DECOMPOSE_TASKS": "planning",
|
|
"PROMPT_PROJECT_STATUS_QA": "response",
|
|
"PROMPT_MEMBER_WORKLOAD_QA": "response",
|
|
"PROMPT_AI_PMO_QA": "response",
|
|
"PROMPT_EXTERNAL_TREND_QA": "response",
|
|
"PROMPT_INDUSTRY_TREND_QA": "response",
|
|
}
|
|
|
|
for prompt_key, family in expected_families.items():
|
|
entry = get_prompt_registry_entry(prompt_key)
|
|
assert entry is not None
|
|
assert entry["module"]
|
|
assert entry["business_scenario"]
|
|
assert entry["input_contract"]
|
|
assert entry["output_contract"]
|
|
assert isinstance(entry["call_sites"], tuple)
|
|
assert entry["call_sites"], prompt_key
|
|
assert entry["prompt_family"] == family
|
|
assert entry["responsibility_scope"]
|
|
|
|
signal_planner_entry = get_prompt_registry_entry("PROMPT_SIGNAL_PLANNER")
|
|
assert signal_planner_entry is not None
|
|
assert "charter_json" in signal_planner_entry["input_contract"]
|
|
assert any("mixin_prompt.py" in site for site in signal_planner_entry["call_sites"])
|
|
assert any("mixin_analyze.py" in site for site in signal_planner_entry["call_sites"])
|
|
assert "规划" in signal_planner_entry["responsibility_scope"]
|
|
|
|
|
|
def test_deprecated_prompt_aliases_point_to_canonical_prompts() -> None:
|
|
registry_keys = set(iter_prompt_registry_keys())
|
|
|
|
assert "PROMPT_FEISHU_REQUIREMENT_ANALYSIS" not in registry_keys
|
|
assert "PROMPT_TASK_BREAKDOWN" not in registry_keys
|
|
assert PROMPT_FEISHU_REQUIREMENT_ANALYSIS == PROMPT_FEISHU_PROJECT_SIGNAL_CLASSIFICATION
|
|
assert PROMPT_TASK_BREAKDOWN == PROMPT_DECOMPOSE_TASKS
|