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 / 打包冒烟）
83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
"""T6 脑洞生成器契约测试：spec + schema（创作工具箱通用框架首个最简生成器）。
|
||
|
||
契约测试——构造符合 schema 的 mock 响应，校验字段、tier、读写权限。
|
||
brainstorm：`{ideas:[{premise,hook,genre_fit}]}`（轻量档，reads=projects，writes=[] 纯预览）。
|
||
不联网、无 DB。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import pytest
|
||
from pydantic import ValidationError
|
||
from ww_agents import (
|
||
AgentSpec,
|
||
Idea,
|
||
IdeaListResult,
|
||
brainstorm_spec,
|
||
)
|
||
|
||
# ---- IdeaListResult schema ----
|
||
|
||
|
||
def test_idea_list_parses_mock_response() -> None:
|
||
# Arrange：模拟网关 instructor 校验后的结构化脑洞产出
|
||
mock = {
|
||
"ideas": [
|
||
{
|
||
"premise": "废柴觉醒可吞噬规则的瞳术",
|
||
"hook": "每次升级都要付出一段记忆为代价",
|
||
"genre_fit": "玄幻·热血",
|
||
},
|
||
{"premise": "末世里唯一能种粮的农场主", "hook": "粮食成了最硬的通货"},
|
||
]
|
||
}
|
||
|
||
# Act
|
||
result = IdeaListResult.model_validate(mock)
|
||
|
||
# Assert
|
||
assert len(result.ideas) == 2
|
||
assert result.ideas[0].premise == "废柴觉醒可吞噬规则的瞳术"
|
||
assert result.ideas[0].hook == "每次升级都要付出一段记忆为代价"
|
||
assert result.ideas[0].genre_fit == "玄幻·热血"
|
||
assert result.ideas[1].genre_fit is None # 可缺
|
||
|
||
|
||
def test_idea_list_defaults_to_empty() -> None:
|
||
assert IdeaListResult().ideas == []
|
||
|
||
|
||
def test_idea_requires_premise_and_hook() -> None:
|
||
with pytest.raises(ValidationError):
|
||
Idea.model_validate({"premise": "只有前提没有钩子"})
|
||
|
||
|
||
# ---- brainstorm_spec 声明 ----
|
||
|
||
|
||
def test_brainstorm_spec_is_light_tier() -> None:
|
||
# 不变量 #2：脑洞发散用轻量档，只声明 tier
|
||
assert brainstorm_spec.tier == "light"
|
||
assert brainstorm_spec.name == "brainstorm"
|
||
|
||
|
||
def test_brainstorm_spec_reads_projects_writes_nothing() -> None:
|
||
# 纯预览：读作品设定、不写任何业务表（不变量 #3）
|
||
assert brainstorm_spec.reads == ("projects",)
|
||
assert brainstorm_spec.writes == ()
|
||
|
||
|
||
def test_brainstorm_spec_output_schema() -> None:
|
||
assert brainstorm_spec.output_schema is IdeaListResult
|
||
|
||
|
||
def test_brainstorm_spec_prompt_documents_diversification() -> None:
|
||
# 一次多条须逐条差异化——system_prompt 必须强调
|
||
assert "差异化" in brainstorm_spec.system_prompt
|
||
|
||
|
||
def test_brainstorm_spec_is_agent_spec_and_immutable() -> None:
|
||
assert isinstance(brainstorm_spec, AgentSpec)
|
||
with pytest.raises(ValidationError):
|
||
brainstorm_spec.tier = "writer"
|