74 lines
2.2 KiB
Python
74 lines
2.2 KiB
Python
"""StoryRouteAgent：LLM JSON 决策与非法 target 回退。"""
|
||
|
||
import unittest
|
||
from types import SimpleNamespace
|
||
from unittest.mock import Mock
|
||
|
||
from app.agents.memoir.story_route_agent import StoryRouteAgent
|
||
|
||
|
||
def _story_stub(sid: str, title: str = "T"):
|
||
return SimpleNamespace(
|
||
id=sid,
|
||
title=title,
|
||
canonical_markdown="预览正文",
|
||
chapter_links=[],
|
||
)
|
||
|
||
|
||
class StoryRouteAgentTest(unittest.TestCase):
|
||
def test_no_llm_returns_new_story(self):
|
||
agent = StoryRouteAgent()
|
||
out = agent.decide(
|
||
chapter_category="childhood",
|
||
chapter_title="童年",
|
||
batch_transcript="hello",
|
||
candidate_stories=[_story_stub("s1")],
|
||
llm=None,
|
||
valid_story_ids={"s1"},
|
||
)
|
||
self.assertEqual(out.decision, "new_story")
|
||
self.assertIsNone(out.new_story_title)
|
||
|
||
def test_append_invalid_id_falls_back_to_new_story(self):
|
||
agent = StoryRouteAgent()
|
||
llm = Mock()
|
||
bound = Mock()
|
||
llm.bind.return_value = bound
|
||
bound.invoke.return_value = SimpleNamespace(
|
||
content='{"decision":"append_story","target_story_id":"unknown"}'
|
||
)
|
||
out = agent.decide(
|
||
chapter_category="childhood",
|
||
chapter_title="童年",
|
||
batch_transcript="hello",
|
||
candidate_stories=[_story_stub("s1")],
|
||
llm=llm,
|
||
valid_story_ids={"s1"},
|
||
)
|
||
self.assertEqual(out.decision, "new_story")
|
||
self.assertEqual(out.reason, "invalid_target")
|
||
|
||
def test_append_valid_target(self):
|
||
agent = StoryRouteAgent()
|
||
llm = Mock()
|
||
bound = Mock()
|
||
llm.bind.return_value = bound
|
||
bound.invoke.return_value = SimpleNamespace(
|
||
content='{"decision":"append_story","target_story_id":"s1"}'
|
||
)
|
||
out = agent.decide(
|
||
chapter_category="childhood",
|
||
chapter_title="童年",
|
||
batch_transcript="more text",
|
||
candidate_stories=[_story_stub("s1")],
|
||
llm=llm,
|
||
valid_story_ids={"s1"},
|
||
)
|
||
self.assertEqual(out.decision, "append_story")
|
||
self.assertEqual(out.target_story_id, "s1")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|