dca4d45d4e
把 21 个内置 agent 的 system_prompt 从 specs.py 的 Python 常量外置为 prompts/<spec.name>.md，import 期由 load_prompt 确定性加载；建立 SPECS 名册 + SCHEMA_CATALOG（Pydantic 类型留 Python）+ 统一只读解析入口 SpecResolver。 纯重构、零功能/schema 变更，缓存断点前块字节级不变（不变量 #9）。 @llm packages/agents（步骤1-3） - spec_model.py：抽出 AgentSpec（frozen，字段不变） - prompt_loader.py：load_prompt = utf-8-sig 去BOM → LF 归一 → NFC → rstrip尾LF， 内存缓存 + fail-fast（PromptNotFoundError），import 期确定性 - schema_catalog.py：SCHEMA_CATALOG[name]→output type 唯一真相源（refiner=None） - prompts/*.md ×21：取常量「运行时值」程序化外迁（反斜杠折行已塌缩， 物理换行≡运行时换行）；文件名按 spec.name 连字符（style.md/character-gen.md 等） - specs.py：删 21 常量 + AgentSpec 类；system_prompt=load_prompt(name)、 output_schema=SCHEMA_CATALOG[name]；建 SPECS + REVIEW_RESERVED_NAMES； *_spec 兼容期保留且 SPECS[name] is *_spec（同一实例）。804→337 行 - __init__.py：显式 __all__ 重导出（避 F401） @backend packages/skills（步骤4-5） - SpecResolver：内置 SPECS（纯内存、零 DB）+ 用户 SkillRegistry 统一 get； 内置 name 永不触发 DB；output_schema_for 精确匹配 - skill_registry：保留命名空间守卫前移至入库校验，拒同名内置 → VALIDATION - toolbox_registry：GeneratorTool.spec 改走 SPECS[...]，删 12 个 *_spec 直接 import @devops repo-root - .gitattributes：prompts/*.md text eol=lf（修正：须用完整嵌套路径才匹配） - packages/agents/pyproject：hatchling artifacts 纳入 prompts/*.md 随 wheel/sdist 分发 - ci.yml：新增 build wheel → 裸装 → import ww_agents.SPECS 冒烟 TDD 全程 mock 网关；门禁绿：ruff/format clean · mypy 209 files · pytest 744 passed （含金标准 sha256 回归 / md↔spec↔catalog 一一对应 / fail-fast / BOM+NFC / 内置守卫 / 同一实例 / 编排器无回归 / apps/api import-smoke / 打包冒烟）
108 lines
3.7 KiB
Python
108 lines
3.7 KiB
Python
"""T6.2 创作工具箱注册表（`TOOLBOX`）单测。
|
||
|
||
校验：15 条齐全（legacy 3 + 新 8 + 竞品快赢 4）；legacy 携 legacy_route 且 spec=None；
|
||
新工具携 spec + output_schema（与 spec.output_schema 一致）；ingest 工具的 table 在白名单内；
|
||
`get_tool` 解析。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from ww_agents import SPECS
|
||
from ww_skills import TOOLBOX, get_tool
|
||
|
||
_LEGACY_KEYS = {"worldbuilding", "character", "outline"}
|
||
_NEW_KEYS = {
|
||
"brainstorm",
|
||
"book-title",
|
||
"blurb",
|
||
"name",
|
||
"golden-finger",
|
||
"glossary",
|
||
"opening",
|
||
"fine-outline",
|
||
# 竞品快赢（Scope B）：续写/扩写/降AI率 preview-only；拆书 F1 可落库 rules。
|
||
"continue",
|
||
"expand",
|
||
"de-ai",
|
||
"teardown",
|
||
}
|
||
_INGEST_KEYS = {"golden-finger", "glossary", "fine-outline", "teardown"}
|
||
|
||
|
||
def test_toolbox_has_all_tools() -> None:
|
||
# Arrange / Act
|
||
keys = set(TOOLBOX.keys())
|
||
|
||
# Assert：legacy 3 + 新 8 + 竞品快赢 4 = 15。
|
||
assert keys == _LEGACY_KEYS | _NEW_KEYS
|
||
assert len(TOOLBOX) == 15
|
||
|
||
|
||
def test_legacy_tools_have_route_and_no_spec() -> None:
|
||
for key in _LEGACY_KEYS:
|
||
tool = TOOLBOX[key]
|
||
assert tool.spec is None, key
|
||
assert tool.output_schema is None, key
|
||
assert tool.legacy_route is not None, key
|
||
assert "{id}" in tool.legacy_route, key
|
||
|
||
|
||
def test_new_tools_carry_spec_and_matching_schema() -> None:
|
||
for key in _NEW_KEYS:
|
||
tool = TOOLBOX[key]
|
||
assert tool.spec is not None, key
|
||
assert tool.output_schema is not None, key
|
||
# 描述符的 output_schema 必须与其 spec 的 output_schema 一致（同一真类）。
|
||
assert tool.output_schema is tool.spec.output_schema, key
|
||
assert tool.legacy_route is None, key
|
||
# 每个新工具的 key 与 spec.name 一致（路由稳定）。
|
||
assert tool.key == tool.spec.name, key
|
||
|
||
|
||
def test_ingest_tools_declare_known_table() -> None:
|
||
for key, tool in TOOLBOX.items():
|
||
if key in _INGEST_KEYS:
|
||
assert tool.ingest is not None, key
|
||
else:
|
||
assert tool.ingest is None, key
|
||
assert TOOLBOX["golden-finger"].ingest is not None
|
||
assert TOOLBOX["golden-finger"].ingest.table == "world_entities"
|
||
assert TOOLBOX["glossary"].ingest is not None
|
||
assert TOOLBOX["glossary"].ingest.table == "world_entities"
|
||
assert TOOLBOX["fine-outline"].ingest is not None
|
||
assert TOOLBOX["fine-outline"].ingest.table == "outline"
|
||
# F1：拆书结论落库为项目 rules。
|
||
assert TOOLBOX["teardown"].ingest is not None
|
||
assert TOOLBOX["teardown"].ingest.table == "rules"
|
||
|
||
|
||
def test_chapter_tools_have_chapter_no_field() -> None:
|
||
for key in ("opening", "fine-outline"):
|
||
field_names = {f.name for f in TOOLBOX[key].input_fields}
|
||
assert "chapter_no" in field_names, key
|
||
|
||
|
||
def test_every_tool_has_brief_field_where_applicable() -> None:
|
||
# 新工具均带 brief textarea（声明驱动表单的统一入口）。
|
||
for key in _NEW_KEYS:
|
||
field_names = {f.name for f in TOOLBOX[key].input_fields}
|
||
assert "brief" in field_names, key
|
||
|
||
|
||
def test_new_tools_spec_is_registry_instance() -> None:
|
||
# 桥接后：新工具 spec 经 SPECS[key] 解析，是注册表同一实例（非直接 import）。
|
||
for key in _NEW_KEYS:
|
||
tool = TOOLBOX[key]
|
||
assert tool.spec is SPECS[key], key
|
||
|
||
|
||
def test_legacy_tools_spec_is_none_unchanged() -> None:
|
||
for key in _LEGACY_KEYS:
|
||
assert TOOLBOX[key].spec is None, key
|
||
|
||
|
||
def test_get_tool_resolves_known_and_unknown() -> None:
|
||
assert get_tool("brainstorm") is TOOLBOX["brainstorm"]
|
||
assert get_tool("worldbuilding") is TOOLBOX["worldbuilding"]
|
||
assert get_tool("does-not-exist") is None
|