"""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_has_appearance_motive_defaults() -> None: # ⑧ 穿线：appearance/motive 有默认值守解析韧性（LLM 漏产字段不致整卡解析失败）。 card = CharacterCard.model_validate( {"name": "甲", "role": "工具人", "backstory": "无名小卒", "arc": "始终如一"} ) assert card.appearance == "" assert card.motive == "" def test_character_card_parses_appearance_and_motive() -> None: card = CharacterCard.model_validate( { "name": "苏璃", "role": "CP", "backstory": "遗孤", "arc": "由敌转盟", "appearance": "银发赤瞳，惯着玄色劲装", "motive": "替亡兄查清族灭真相", } ) assert card.appearance == "银发赤瞳，惯着玄色劲装" assert card.motive == "替亡兄查清族灭真相" 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 def test_character_gen_prompt_declares_appearance_and_motive_fields() -> None: # ⑧ 穿线：产出契约须显式声明 appearance/motive 两个新字段。 prompt = character_gen_spec.system_prompt assert "appearance" in prompt assert "motive" in prompt def test_motive_not_double_written_in_traits() -> None: # D6 重切回归：动机唯一落 `motive` 字段，`traits` 只留性格三层—— # traits 契约行须把动机/欲望/恐惧**改导向 motive**，绝不再指示折进 traits # （否则双写→漂移→喂 ③ 人物塑造审查假信号）。 prompt = character_gen_spec.system_prompt lines = prompt.splitlines() traits_line = next(line for line in lines if line.lstrip().startswith("- **traits**")) motive_line = next(line for line in lines if line.lstrip().startswith("- **motive**")) # traits 契约行显式把动机重导向 motive 字段（而非折进 traits）。 assert "motive" in traits_line # 旧的「写明核心动机」折入指示已被移除。 assert "写明核心动机" not in prompt assert "并写明" not in traits_line # motive 契约行承接内驱三要素。 assert "动机" in motive_line assert "恐惧" in motive_line # ---- 共性：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"