feat: M4 文风 + M5 生成/多provider/Skill + Kimi Code 订阅接入 + 本地联调修复
M4(文风): style-auditor 双轨(提取指纹/漂移第四审)+ jobs 长任务框架(zombie reaper) + 回炉 refine + GET /style read-back。 M5(生成+扩展): worldbuilder/character-gen(入库 continuity 409 gate + partition_writes 白名单 + schema→JSONB 形变); 网关多 provider 回退链/熔断/能力降级(Anthropic/Gemini 适配器);Skill registry + 表权限沙箱 + 规则; 前端 角色生成器/世界观/Codex/规则页/技能库/⌘K 命令面板。 K1(Kimi Code 订阅接入): OAuth device-flow(kimi-code)+ 静态 Console key(kimi-code-key)两路径; coding 端点 KimiCLI 伪造头(实测 UA allow-list 门禁,缺则 403)+ JSON 模式结构化(thinking ⊥ tool_choice)。 本地联调修复: CORS 中间件;assemble 注入 premise+「写第N章」指令(修空 prompt 400); GET /outline·/draft read-back + 大纲/工作台/审稿页重载;写页 client/server 常量边界 + notFound 健壮化; 字数 toLocaleString locale 水合;审稿页终稿从已存草稿 seed(修 accept 422)。 门禁: backend ruff/mypy(157)/alembic 无漂移/pytest 451 · frontend lint/tsc/vitest/build。 Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
183
packages/agents/tests/test_generation_specs.py
Normal file
183
packages/agents/tests/test_generation_specs.py
Normal file
@@ -0,0 +1,183 @@
|
||||
"""T5.1 worldbuilder + character-gen 契约测试:spec + schema(C6 扩 / ARCH §5.4 §6.5)。
|
||||
|
||||
契约测试——构造符合 schema 的 mock 响应,校验字段、tier、读写权限。
|
||||
worldbuilder:`{entities:[{type,name,rules}]}` 硬规则显式(写手档,reads=projects)。
|
||||
character-gen:`{cards:[{name,role,traits,backstory,arc,speech_tics,tags,relations}]}`
|
||||
(写手档,reads=world_entities+characters,writes=characters)。不联网、无 DB。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import pytest
|
||||
from pydantic import ValidationError
|
||||
from ww_agents import (
|
||||
AgentSpec,
|
||||
CharacterCard,
|
||||
CharacterGenResult,
|
||||
CharacterRelation,
|
||||
WorldEntityCard,
|
||||
WorldGenResult,
|
||||
character_gen_spec,
|
||||
worldbuilder_spec,
|
||||
)
|
||||
|
||||
# ---- WorldGenResult schema(worldbuilder)----
|
||||
|
||||
|
||||
def test_world_gen_parses_mock_response() -> None:
|
||||
# Arrange:模拟网关 instructor 校验后的结构化世界观产出(硬规则显式)
|
||||
mock = {
|
||||
"entities": [
|
||||
{
|
||||
"type": "力量体系",
|
||||
"name": "九转炼气",
|
||||
"rules": ["只能逐境突破,不可越级", "突破必有反噬代价"],
|
||||
},
|
||||
{"type": "地理", "name": "无雨城", "rules": ["终年无雨"]},
|
||||
]
|
||||
}
|
||||
|
||||
# Act
|
||||
result = WorldGenResult.model_validate(mock)
|
||||
|
||||
# Assert
|
||||
assert len(result.entities) == 2
|
||||
assert result.entities[0].type == "力量体系"
|
||||
assert result.entities[0].name == "九转炼气"
|
||||
assert result.entities[0].rules == ["只能逐境突破,不可越级", "突破必有反噬代价"]
|
||||
assert result.entities[1].rules == ["终年无雨"]
|
||||
|
||||
|
||||
def test_world_gen_defaults_to_empty_entities() -> None:
|
||||
assert WorldGenResult().entities == []
|
||||
|
||||
|
||||
def test_world_entity_rules_default_empty() -> None:
|
||||
card = WorldEntityCard.model_validate({"type": "势力", "name": "天剑宗"})
|
||||
assert card.rules == []
|
||||
|
||||
|
||||
def test_world_entity_requires_type_and_name() -> None:
|
||||
with pytest.raises(ValidationError):
|
||||
WorldEntityCard.model_validate({"type": "势力"})
|
||||
|
||||
|
||||
# ---- CharacterGenResult schema(character-gen)----
|
||||
|
||||
|
||||
def test_character_gen_parses_mock_response() -> None:
|
||||
mock = {
|
||||
"cards": [
|
||||
{
|
||||
"name": "苏黎",
|
||||
"role": "对手",
|
||||
"traits": ["表面冷漠", "内心挣扎"],
|
||||
"backstory": "出身敌对势力,幼年遭灭门。",
|
||||
"arc": "复仇者 → 动摇 → 救赎",
|
||||
"speech_tics": ["惯用反问", "句末带『罢了』"],
|
||||
"tags": ["亦正亦邪", "宿命纠葛"],
|
||||
"relations": [{"name": "主角", "kind": "宿敌", "note": "灭门之仇"}],
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
result = CharacterGenResult.model_validate(mock)
|
||||
|
||||
assert len(result.cards) == 1
|
||||
card = result.cards[0]
|
||||
assert card.name == "苏黎"
|
||||
assert card.role == "对手"
|
||||
assert card.traits == ["表面冷漠", "内心挣扎"]
|
||||
assert card.speech_tics == ["惯用反问", "句末带『罢了』"]
|
||||
assert card.relations[0].name == "主角"
|
||||
assert card.relations[0].kind == "宿敌"
|
||||
assert card.relations[0].note == "灭门之仇"
|
||||
|
||||
|
||||
def test_character_gen_defaults_to_empty_cards() -> None:
|
||||
assert CharacterGenResult().cards == []
|
||||
|
||||
|
||||
def test_character_card_optional_collections_default_empty() -> None:
|
||||
card = CharacterCard.model_validate(
|
||||
{"name": "甲", "role": "工具人", "backstory": "无名小卒", "arc": "始终如一"}
|
||||
)
|
||||
assert card.traits == []
|
||||
assert card.speech_tics == []
|
||||
assert card.tags == []
|
||||
assert card.relations == []
|
||||
|
||||
|
||||
def test_character_card_requires_core_fields() -> None:
|
||||
# name/role/backstory/arc 为必填
|
||||
with pytest.raises(ValidationError):
|
||||
CharacterCard.model_validate({"name": "甲", "role": "对手"})
|
||||
|
||||
|
||||
def test_character_relation_note_optional() -> None:
|
||||
rel = CharacterRelation.model_validate({"name": "师父", "kind": "师徒"})
|
||||
assert rel.note is None
|
||||
|
||||
|
||||
# ---- worldbuilder_spec 声明 ----
|
||||
|
||||
|
||||
def test_worldbuilder_spec_is_writer_tier() -> None:
|
||||
# 不变量 #2:世界观创意用写手档,只声明 tier
|
||||
assert worldbuilder_spec.tier == "writer"
|
||||
assert worldbuilder_spec.name == "worldbuilder"
|
||||
|
||||
|
||||
def test_worldbuilder_spec_reads_projects_writes_world_entities() -> None:
|
||||
assert worldbuilder_spec.reads == ["projects"]
|
||||
assert worldbuilder_spec.writes == ["world_entities"]
|
||||
|
||||
|
||||
def test_worldbuilder_spec_output_schema() -> None:
|
||||
assert worldbuilder_spec.output_schema is WorldGenResult
|
||||
|
||||
|
||||
def test_worldbuilder_spec_prompt_documents_hard_rules() -> None:
|
||||
# 硬规则须显式(供 continuity 校验引用)——system_prompt 必须强调
|
||||
assert "硬规则" in worldbuilder_spec.system_prompt
|
||||
|
||||
|
||||
# ---- character_gen_spec 声明 ----
|
||||
|
||||
|
||||
def test_character_gen_spec_is_writer_tier() -> None:
|
||||
assert character_gen_spec.tier == "writer"
|
||||
assert character_gen_spec.name == "character-gen"
|
||||
|
||||
|
||||
def test_character_gen_spec_reads_world_and_characters() -> None:
|
||||
# reads=world_entities+characters(约束 + 防雷同对照),writes=characters
|
||||
assert character_gen_spec.reads == ["world_entities", "characters"]
|
||||
assert character_gen_spec.writes == ["characters"]
|
||||
|
||||
|
||||
def test_character_gen_spec_output_schema() -> None:
|
||||
assert character_gen_spec.output_schema is CharacterGenResult
|
||||
|
||||
|
||||
def test_character_gen_spec_prompt_documents_anti_duplication() -> None:
|
||||
# M5-a 群像防雷同:prompt 必须注入「已生成卡 + 已有角色」差异化要求
|
||||
prompt = character_gen_spec.system_prompt
|
||||
assert "已有角色" in prompt
|
||||
assert "已生成卡" in prompt
|
||||
assert "防雷同" in prompt
|
||||
|
||||
|
||||
# ---- 共性:immutable + 是 AgentSpec ----
|
||||
|
||||
|
||||
def test_generation_specs_are_agent_specs() -> None:
|
||||
assert isinstance(worldbuilder_spec, AgentSpec)
|
||||
assert isinstance(character_gen_spec, AgentSpec)
|
||||
|
||||
|
||||
def test_generation_specs_are_immutable() -> None:
|
||||
with pytest.raises(ValidationError):
|
||||
worldbuilder_spec.tier = "analyst"
|
||||
with pytest.raises(ValidationError):
|
||||
character_gen_spec.tier = "analyst"
|
||||
191
packages/agents/tests/test_style_specs.py
Normal file
191
packages/agents/tests/test_style_specs.py
Normal file
@@ -0,0 +1,191 @@
|
||||
"""T4.2 style-auditor 双轨契约测试:style_extract / style_drift / refiner spec + schema(C6 扩)。
|
||||
|
||||
契约测试——构造符合 schema 的 mock 响应,校验字段、tier、双轨读写权限。
|
||||
提取轨 `StyleFingerprintResult`(16 维带证据);第四审 `StyleDriftReview`(段级相似度);
|
||||
回炉 `refiner_spec`(writer 纯文本,output_schema=None)。不联网、无 DB。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import pytest
|
||||
from pydantic import ValidationError
|
||||
from ww_agents import (
|
||||
AgentSpec,
|
||||
StyleDimension,
|
||||
StyleDriftReview,
|
||||
StyleDriftSegment,
|
||||
StyleFingerprintResult,
|
||||
refiner_spec,
|
||||
style_drift_spec,
|
||||
style_extract_spec,
|
||||
)
|
||||
|
||||
# ---- StyleFingerprintResult schema(提取轨)----
|
||||
|
||||
|
||||
def test_style_fingerprint_parses_mock_response() -> None:
|
||||
# Arrange:模拟网关 instructor 校验后的结构化指纹产出(每维带原文证据)
|
||||
mock = {
|
||||
"dimensions": [
|
||||
{
|
||||
"name": "句长节奏",
|
||||
"value": "短句为主、节奏明快",
|
||||
"evidence": ["他一刀劈下。血溅三尺。", "风停了。"],
|
||||
},
|
||||
{
|
||||
"name": "叙事人称",
|
||||
"value": "第三人称限知",
|
||||
"evidence": ["他不知道门后有什么。"],
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
# Act
|
||||
fp = StyleFingerprintResult.model_validate(mock)
|
||||
|
||||
# Assert
|
||||
assert len(fp.dimensions) == 2
|
||||
assert fp.dimensions[0].name == "句长节奏"
|
||||
assert fp.dimensions[0].evidence == ["他一刀劈下。血溅三尺。", "风停了。"]
|
||||
assert fp.dimensions[1].value == "第三人称限知"
|
||||
|
||||
|
||||
def test_style_fingerprint_defaults_to_empty_dimensions() -> None:
|
||||
assert StyleFingerprintResult().dimensions == []
|
||||
|
||||
|
||||
def test_style_dimension_evidence_defaults_empty() -> None:
|
||||
dim = StyleDimension.model_validate({"name": "用词", "value": "古雅"})
|
||||
assert dim.evidence == []
|
||||
|
||||
|
||||
def test_style_dimension_requires_name_and_value() -> None:
|
||||
with pytest.raises(ValidationError):
|
||||
StyleDimension.model_validate({"name": "用词"})
|
||||
|
||||
|
||||
# ---- StyleDriftReview schema(第四审)----
|
||||
|
||||
|
||||
def test_style_drift_parses_mock_response() -> None:
|
||||
mock = {
|
||||
"score": 72,
|
||||
"segments": [
|
||||
{"idx": 3, "score": 40, "label": "机翻腔"},
|
||||
{"idx": 7, "score": 55, "label": None},
|
||||
],
|
||||
}
|
||||
|
||||
review = StyleDriftReview.model_validate(mock)
|
||||
|
||||
assert review.score == 72
|
||||
assert len(review.segments) == 2
|
||||
assert review.segments[0].idx == 3
|
||||
assert review.segments[0].label == "机翻腔"
|
||||
assert review.segments[1].label is None
|
||||
|
||||
|
||||
def test_style_drift_defaults_score_100_empty_segments() -> None:
|
||||
# 无指纹优雅降级:默认整体相似度 100、无漂移段
|
||||
review = StyleDriftReview()
|
||||
assert review.score == 100
|
||||
assert review.segments == []
|
||||
|
||||
|
||||
def test_style_drift_segment_label_optional() -> None:
|
||||
seg = StyleDriftSegment.model_validate({"idx": 1, "score": 30})
|
||||
assert seg.label is None
|
||||
|
||||
|
||||
def test_style_drift_segment_requires_idx_and_score() -> None:
|
||||
with pytest.raises(ValidationError):
|
||||
StyleDriftSegment.model_validate({"idx": 1})
|
||||
|
||||
|
||||
# ---- style_extract_spec 声明 ----
|
||||
|
||||
|
||||
def test_style_extract_spec_is_analyst_tier() -> None:
|
||||
assert style_extract_spec.tier == "analyst"
|
||||
assert style_extract_spec.name == "style_extract"
|
||||
|
||||
|
||||
def test_style_extract_spec_reads_and_writes_fingerprint() -> None:
|
||||
assert style_extract_spec.reads == ["style_fingerprint"]
|
||||
assert style_extract_spec.writes == ["style_fingerprint"]
|
||||
|
||||
|
||||
def test_style_extract_spec_output_schema() -> None:
|
||||
assert style_extract_spec.output_schema is StyleFingerprintResult
|
||||
|
||||
|
||||
def test_style_extract_spec_has_nonempty_system_prompt() -> None:
|
||||
assert style_extract_spec.system_prompt.strip()
|
||||
|
||||
|
||||
# ---- style_drift_spec 声明(第四审,name="style")----
|
||||
|
||||
|
||||
def test_style_drift_spec_is_light_tier_named_style() -> None:
|
||||
# 不变量 #2:第四审用轻量档,只声明 tier;列名/section 名为 "style"
|
||||
assert style_drift_spec.tier == "light"
|
||||
assert style_drift_spec.name == "style"
|
||||
|
||||
|
||||
def test_style_drift_spec_is_read_only() -> None:
|
||||
# 不变量 #3:四审只读
|
||||
assert style_drift_spec.writes == []
|
||||
|
||||
|
||||
def test_style_drift_spec_reads_fingerprint() -> None:
|
||||
assert style_drift_spec.reads == ["style_fingerprint"]
|
||||
|
||||
|
||||
def test_style_drift_spec_output_schema() -> None:
|
||||
assert style_drift_spec.output_schema is StyleDriftReview
|
||||
|
||||
|
||||
def test_style_drift_spec_prompt_documents_graceful_degrade() -> None:
|
||||
# 无指纹→返回 score=100/空段(system_prompt 必须含降级指令,否则 LLM 不知如何兜底)
|
||||
assert "100" in style_drift_spec.system_prompt
|
||||
|
||||
|
||||
# ---- refiner_spec 声明(writer 纯文本回炉)----
|
||||
|
||||
|
||||
def test_refiner_spec_is_writer_tier() -> None:
|
||||
assert refiner_spec.tier == "writer"
|
||||
assert refiner_spec.name == "refiner"
|
||||
|
||||
|
||||
def test_refiner_spec_output_schema_is_none() -> None:
|
||||
# 回炉产纯文本(重写段),无结构化 schema
|
||||
assert refiner_spec.output_schema is None
|
||||
|
||||
|
||||
def test_refiner_spec_is_read_only_and_no_writes() -> None:
|
||||
# 回炉非持久(不变量 #3):不写库,作者采纳经既有 draft 自动保存合入
|
||||
assert refiner_spec.reads == []
|
||||
assert refiner_spec.writes == []
|
||||
|
||||
|
||||
def test_refiner_spec_has_nonempty_system_prompt() -> None:
|
||||
assert refiner_spec.system_prompt.strip()
|
||||
|
||||
|
||||
# ---- 共性:immutable + 是 AgentSpec ----
|
||||
|
||||
|
||||
def test_style_specs_are_agent_specs() -> None:
|
||||
assert isinstance(style_extract_spec, AgentSpec)
|
||||
assert isinstance(style_drift_spec, AgentSpec)
|
||||
assert isinstance(refiner_spec, AgentSpec)
|
||||
|
||||
|
||||
def test_style_specs_are_immutable() -> None:
|
||||
with pytest.raises(ValidationError):
|
||||
style_extract_spec.tier = "writer"
|
||||
with pytest.raises(ValidationError):
|
||||
style_drift_spec.tier = "writer"
|
||||
with pytest.raises(ValidationError):
|
||||
refiner_spec.tier = "analyst"
|
||||
Reference in New Issue
Block a user