refactor relationship changes

src/classes/story_teller.py

@@ -8,6 +8,10 @@ if TYPE_CHECKING:
 
 from src.utils.config import CONFIG
 from src.utils.llm import call_llm_with_template, LLMMode
+from src.classes.relations import (
+    process_relation_changes,
+    get_relation_change_context
+)
 
 story_styles = [
     "平淡叙述：语句克制、少修饰、像旁观者记录。",
@@ -31,9 +35,11 @@ story_styles = [
 class StoryTeller:
     """
     故事生成器：基于模板与 LLM，将给定事件扩展为简短的小故事。
+    同时负责处理可能的后天关系变化。
     """
     
-    TEMPLATE_PATH = CONFIG.paths.templates / "story.txt"
+    TEMPLATE_SINGLE_PATH = CONFIG.paths.templates / "story_single.txt"
+    TEMPLATE_DUAL_PATH = CONFIG.paths.templates / "story_dual.txt"
 
     @staticmethod
     def _build_avatar_infos(*actors: "Avatar") -> Dict[str, dict]:
@@ -54,45 +60,82 @@ class StoryTeller:
         return avatar_infos
 
     @staticmethod
-    def _build_template_data(event: str, res: str, avatar_infos: Dict[str, dict], prompt: str) -> dict:
+    def _build_template_data(event: str, res: str, avatar_infos: Dict[str, dict], prompt: str, *actors: "Avatar") -> dict:
         """构建模板渲染所需的数据字典"""
+        
+        # 默认空关系列表
+        possible_new_relations = []
+        possible_cancel_relations = []
+        avatar_name_1 = ""
+        avatar_name_2 = ""
+        
+        # 如果有两个有效角色，计算可能的关系
+        non_null = [a for a in actors if a is not None]
+        if len(non_null) >= 2:
+            # 计算 actors[1] 相对于 actors[0] 的可能关系
+            possible_new_relations, possible_cancel_relations = get_relation_change_context(non_null[0], non_null[1])
+            avatar_name_1 = non_null[0].name
+            avatar_name_2 = non_null[1].name
+
         return {
             "avatar_infos": avatar_infos,
+            "avatar_name_1": avatar_name_1,
+            "avatar_name_2": avatar_name_2,
             "event": event,
             "res": res,
             "style": random.choice(story_styles),
             "story_prompt": prompt,
+            "possible_new_relations": possible_new_relations,
+            "possible_cancel_relations": possible_cancel_relations,
         }
 
     @staticmethod
     def _make_fallback_story(event: str, res: str, style: str) -> str:
         """生成降级文案"""
-        return f"{event}。{res}。{style}"
+        # 不再显示 style，避免出戏
+        return f"{event}。{res}。"
 
     @staticmethod
-    async def tell_story(event: str, res: str, *actors: "Avatar", prompt: str = "") -> str:
+    async def tell_story(event: str, res: str, *actors: "Avatar", prompt: str = "", allow_relation_changes: bool = False) -> str:
         """
         生成小故事（异步版本）。
-        基于 `static/templates/story.txt` 模板，失败时返回降级文案。
+        根据 allow_relation_changes 参数选择模板：
+        - True: 使用 story_dual.txt，支持关系变化（需要至少2个角色）
+        - False: 使用 story_single.txt，仅生成故事（无论角色数量）
         
         Args:
             event: 事件描述
             res: 结果描述
             *actors: 参与的角色（1-2个）
             prompt: 可选的故事提示词
+            allow_relation_changes: 是否允许故事导致关系变化，默认为False（单人模式）
         """
-        avatar_infos = StoryTeller._build_avatar_infos(*actors)
-        infos = StoryTeller._build_template_data(event, res, avatar_infos, prompt)
+        non_null = [a for a in actors if a is not None]
         
-        try:
-            data = await call_llm_with_template(StoryTeller.TEMPLATE_PATH, infos, LLMMode.FAST)
-            story = data.get("story", "").strip()
-            if story:
-                return story
-        except Exception:
-            pass
+        # 只有当允许关系变化且有至少2个角色时，才使用双人模板
+        is_dual = allow_relation_changes and len(non_null) >= 2
+        
+        template_path = StoryTeller.TEMPLATE_DUAL_PATH if is_dual else StoryTeller.TEMPLATE_SINGLE_PATH
+        
+        avatar_infos = StoryTeller._build_avatar_infos(*actors)
+        infos = StoryTeller._build_template_data(event, res, avatar_infos, prompt, *actors)
+        
+        # 移除了 try-except 块，允许异常向上冒泡，以便 Fail Fast
+        data = await call_llm_with_template(template_path, infos, LLMMode.FAST)
+        story = data.get("story", "").strip()
+        
+        # 仅在双人模式下处理关系变化
+        if is_dual:
+            avatar_1 = non_null[0]
+            avatar_2 = non_null[1]
+            # 尝试获取 month_stamp
+            month_stamp = getattr(avatar_1.world, "month_stamp", 0)
+            process_relation_changes(avatar_1, avatar_2, data, month_stamp)
+
+        if story:
+            return story
         
         return StoryTeller._make_fallback_story(event, res, infos["style"])
 
 
-__all__ = ["StoryTeller"]
+__all__ = ["StoryTeller"]