sladro
6f785c9f2c
Co-authored-by: factory-droid[bot] <138933559+factory-droid[bot]@users.noreply.github.com>
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
from bidmaster.utils.prompt_planner import PromptPlanner
|
|
|
|
|
|
def test_prompt_planner_uses_metadata_and_parent_context() -> None:
|
|
parent = {"id": "chapter_1", "title": "1. 总体技术方案", "level": 1}
|
|
child = {
|
|
"id": "chapter_1_1",
|
|
"title": "1.1 技术方案-基本要求 (3.0分)",
|
|
"level": 2,
|
|
"parent_id": "chapter_1",
|
|
}
|
|
|
|
state = {
|
|
"chapter_queue": [parent, child],
|
|
"chapter_metadata": {
|
|
"chapter_1_1": {
|
|
"title": child["title"],
|
|
"score": 3.0,
|
|
"category": "technical_solution",
|
|
"requirements": "方案需覆盖总体架构；提供性能指标；描述安全策略。",
|
|
"rubric_points": ["总体架构完整", "性能指标量化"],
|
|
}
|
|
},
|
|
"chapter_children_map": {"chapter_1": ["chapter_1_1"]},
|
|
"generated_contents": {"chapter_1": "本章节描述总体设计与安全架构要点。"},
|
|
"chapter_configs": {"chapter_1": {"emphasis": "突出安全策略"}},
|
|
"expanded_configs": {"chapter_1": "chapter_1", "chapter_1_1": "chapter_1"},
|
|
}
|
|
|
|
planner = PromptPlanner(state)
|
|
spec = planner.build_prompt_spec(child)
|
|
|
|
assert "总体架构完整" in spec["rubric_points"]
|
|
assert "父章节摘要" in spec["context_summary"]
|
|
assert "突出安全策略" in spec["objectives"]
|
|
|
|
|
|
def test_prompt_planner_fallback_without_metadata() -> None:
|
|
root = {"id": "chapter_root", "title": "1. 服务方案", "level": 1}
|
|
child = {
|
|
"id": "chapter_child",
|
|
"title": "1.1 现场服务体系",
|
|
"level": 2,
|
|
"parent_id": "chapter_root",
|
|
}
|
|
|
|
state = {
|
|
"chapter_queue": [root, child],
|
|
"chapter_children_map": {"chapter_root": ["chapter_child"]},
|
|
}
|
|
|
|
planner = PromptPlanner(state)
|
|
spec = planner.build_prompt_spec(root)
|
|
|
|
assert "1.1 现场服务体系" in spec["requirements_summary"]
|
|
assert "围绕章节主题" in spec["objectives"]
|