add persona conditions

src/classes/alignment.py

@@ -15,6 +15,21 @@ class Alignment(Enum):
     def get_info(self) -> str:
         return alignment_strs[self] + ": " + alignment_infos[self]
 
+    def __hash__(self) -> int:
+        return hash(self.value)
+
+    def __eq__(self, other) -> bool:
+        """
+        允许与同类或字符串比较：
+        - Alignment: 恒等比较
+        - str: 同时支持英文值（value）与中文显示（__str__）
+        """
+        if isinstance(other, Alignment):
+            return self is other
+        if isinstance(other, str):
+            return other == self.value or other == str(self)
+        return False
+
 
 alignment_strs = {
     Alignment.RIGHTEOUS: "正",

src/classes/avatar.py

@@ -93,9 +93,9 @@ class Avatar:
 
         # 最大寿元已在 Age 构造时基于境界初始化
         
-        # 如果personas列表为空，则随机分配两个不互斥的persona
+        # 如果personas列表为空，则随机分配两个符合条件且不互斥的persona
         if not self.personas:
-            self.personas = get_random_compatible_personas(persona_num)
+            self.personas = get_random_compatible_personas(persona_num, avatar=self)
 
     def __hash__(self) -> int:
         return hash(self.id)

src/classes/persona.py

@@ -1,12 +1,16 @@
 import random
 from dataclasses import dataclass
-from typing import List
+from typing import List, Optional, TYPE_CHECKING
 
 from src.utils.df import game_configs
 from src.utils.config import CONFIG
 
 ids_separator = CONFIG.df.ids_separator
 
+if TYPE_CHECKING:
+    # 仅用于类型检查，避免运行时循环导入
+    from src.classes.avatar import Avatar
+
 @dataclass
 class Persona:
     """
@@ -17,6 +21,7 @@ class Persona:
     prompt: str
     exclusion_ids: List[int]
     weight: float
+    condition: str
 
 def _load_personas() -> tuple[dict[int, Persona], dict[str, Persona]]:
     """从配表加载persona数据"""
@@ -34,6 +39,9 @@ def _load_personas() -> tuple[dict[int, Persona], dict[str, Persona]]:
         weight_val = row.get("weight", 1)
         weight_str = str(weight_val).strip()
         weight = float(weight_str) if weight_str and weight_str.lower() != "nan" else 1.0
+        # 条件：可为空
+        condition_val = row.get("condition", "")
+        condition = "" if str(condition_val) == "nan" else str(condition_val).strip()
         
         persona = Persona(
             id=int(row["id"]),
@@ -41,6 +49,7 @@ def _load_personas() -> tuple[dict[int, Persona], dict[str, Persona]]:
             prompt=str(row["prompt"]),
             exclusion_ids=exclusion_ids,
             weight=weight,
+            condition=condition,
         )
         personas_by_id[persona.id] = persona
         personas_by_name[persona.name] = persona
@@ -50,12 +59,32 @@ def _load_personas() -> tuple[dict[int, Persona], dict[str, Persona]]:
 # 从配表加载persona数据
 personas_by_id, personas_by_name = _load_personas()
 
-def get_random_compatible_personas(num_personas: int = 2) -> List[Persona]:
+def _is_persona_allowed(persona_id: int, already_selected_ids: set[int], avatar: Optional["Avatar"]) -> bool:
+    """
+    统一判断：persona 是否允许被选择（条件 + 互斥）。
+    - 条件：当存在 avatar 且配置了 condition 时，通过安全 eval 判断。
+    - 互斥：与已选 persona 双向互斥。
+    """
+    persona = personas_by_id[persona_id]
+    # 条件判定
+    if avatar is not None and persona.condition:
+        allowed = bool(eval(persona.condition, {"__builtins__": {}}, {"avatar": avatar}))
+        if not allowed:
+            return False
+    # 与已选互斥检查（双向）
+    for sid in already_selected_ids:
+        other = personas_by_id[sid]
+        if (persona_id in other.exclusion_ids) or (sid in persona.exclusion_ids):
+            return False
+    return True
+
+def get_random_compatible_personas(num_personas: int = 2, avatar: Optional["Avatar"] = None) -> List[Persona]:
     """
     随机选择指定数量的互相不冲突的persona
     
     Args:
         num_personas: 需要选择的persona数量，默认为2
+        avatar: 可选，若提供则按 persona.condition 过滤
     
     Returns:
         List[Persona]: 互相不冲突的persona列表
@@ -63,32 +92,25 @@ def get_random_compatible_personas(num_personas: int = 2) -> List[Persona]:
     Raises:
         ValueError: 如果无法找到足够数量的兼容persona
     """
-    all_persona_ids = set(personas_by_id.keys())
+    # 初始候选：若提供 avatar，则先按条件过滤；否则全量
+    initial_ids = set(personas_by_id.keys())
+    if avatar is not None:
+        initial_ids = {pid for pid in initial_ids if _is_persona_allowed(pid, set(), avatar)}
 
     selected_personas: List[Persona] = []
-    available_ids = all_persona_ids.copy()
-    
+    selected_ids: set[int] = set()
+
     for i in range(num_personas):
+        # 按当前已选进行二次过滤（互斥 + 条件）
+        available_ids = [pid for pid in initial_ids if pid not in selected_ids and _is_persona_allowed(pid, selected_ids, avatar)]
         if not available_ids:
             raise ValueError(f"只能找到{i}个兼容的persona，无法满足需要的{num_personas}个")
-        # 按权重从可用列表中选择一个
-        candidates: List[Persona] = [personas_by_id[i] for i in available_ids]
+
+        candidates: List[Persona] = [personas_by_id[pid] for pid in available_ids]
         weights: List[float] = [max(0.0, c.weight) for c in candidates]
         selected_persona = random.choices(candidates, weights=weights, k=1)[0]
-        selected_id = selected_persona.id
         selected_personas.append(selected_persona)
-        
-        # 更新可用列表：移除已选择的和与其互斥的
-        available_ids.discard(selected_id)  # 移除自己
-        
-        # 移除所有与当前选择互斥的persona
-        for exclusion_id in selected_persona.exclusion_ids:
-            available_ids.discard(exclusion_id)
-        
-        # 移除所有将当前选择作为互斥对象的persona
-        for persona_id in list(available_ids):
-            if selected_id in personas_by_id[persona_id].exclusion_ids:
-                available_ids.discard(persona_id)
-    
+        selected_ids.add(selected_persona.id)
+
     return selected_personas
 

src/run/run.py

@@ -24,7 +24,6 @@ from src.utils.id_generator import get_avatar_id
 from src.utils.config import CONFIG
 from src.run.log import get_logger
 from src.classes.relation import Relation
-from src.classes.alignment import Alignment
 
 
 def clamp(value: int, lo: int, hi: int) -> int:
@@ -86,8 +85,6 @@ def make_avatars(world: World, count: int = 12, current_month_stamp: MonthStamp
             root=random.choice(list(Root)),  # 随机选择灵根
         )
         avatar.tile = world.map.get_tile(x, y)
-        # 随机分配阵营（正/邪）
-        avatar.alignment = random.choice(list(Alignment))
         avatars[avatar.id] = avatar
     # # —— 为演示添加少量示例关系 ——
     avatar_list = list(avatars.values())

static/config.yml

@@ -13,7 +13,7 @@ ai:
   max_decide_num: 3
 
 game:
-  init_npc_num: 3
+  init_npc_num: 6
   npc_birth_rate_per_month: 0.001
 
 df:

static/game_configs/persona.csv

@@ -1,26 +1,28 @@
-id,name,exclusion_ids,prompt,weight
-,,和本persona互斥的persona的id,输入给LLM的prompt,采样时不同权重被采样到的概率
-1,理性,2;5,你是一个理性的人，你总是会用逻辑来思考问题，做事会谋定而后动。,1
-2,无常,1;9;20,你是一个无常的人，目标飘忽不定，不会长期坚持一个目标。,1
-3,怠惰,4;20,你是一个怠惰的人，你总是会拖延，不想努力，更热衷于享受人生。,1
-4,冒险,3;10,你是一个冒险的人，你总是会冒险，喜欢刺激，总想放手一搏。,1
-5,随性,1;20,你是一个随性的人，你总是会随机应变，性子到哪里了就是哪里，没有一定之规。,1
-6,贪财,,你是一个贪财的人，你对灵石和财富有着强烈的渴望。,1
-7,采药,,你是一个热爱采集的人，喜欢在山林中寻找各种奇花异草和灵药，对植物有着敏锐的直觉和深厚的兴趣。你认为大自然的恩赐需要用心去发现和珍惜。,1
-8,猎者,,你是一个热爱狩猎的人，享受在野外追踪猎物的刺激感，对各种动物的习性了如指掌。你相信通过狩猎能够磨练自己的意志和技能，获得更强大的力量。,1
-9,沉思,2,你是一个沉思的人，你总是会深思熟虑，思考问题比较有哲理。,1
-10,惜命,4;20,你是一个惜命的人，你总是会珍惜自己的生命，不会轻易冒险。,1
-11,友爱,13;14;15;12;20,你是一个友爱的人，你重视同伴与和谐，乐于助人，倾向通过协作与沟通化解矛盾。,1
-12,复仇,11;14,你是一个复仇心强的人，你绝不轻易放下仇怨，为了复仇愿意付出代价与时间。,1
-13,孤僻,11,你是一个孤僻的人，你喜欢独处，避免与人深交，更信赖自己的判断与行动。,1
-14,淡漠,11;12;15;20,你是一个淡漠的人，你情感克制，对外界冷静疏离，不轻易被他人或事件影响。,1
-15,好斗,11;14;10;17,你是一个好斗的人，你直面冲突，偏好以力量与对抗解决问题，越挫越勇。,1
-16,鲁莽,1;9;10,你是一个鲁莽的人，你行事冲动、少考虑后果，常凭直觉立刻行动。,1
-17,胆小,4;15;12;20,你是一个胆小的人，你谨小慎微，容易畏惧风险，倾向回避正面冲突。,1
-18,霸道,11;17,你是一个霸道的人，你行事强势，不讲道理，习惯以自己的利益为先，倾向多吃多占、压人一步，对他人的反对不以为意。,1
-19,修行痴迷,2;3;5,你是一个对修行极度痴迷的人，你将绝大多数时间用于修炼，厌恶与修行无关的社交与享乐。,1
-20,极端,11;14;2;5;3;10;17,你是一个极端的人，你仇视对立阵营，如果你是正义阵营，那么你极度正义；如果你是邪恶阵营，那么你极度邪恶。,1
-21,内向,22,你是一个内向的人，你更享受独处与自我思考，倾向回避不必要的社交与长谈。,1
-22,外向,13;14;21,你是一个外向的人，你乐于与人交流，主动结识伙伴，倾向接受对话和合作。,1
-23,刻薄,11;24,你是一个刻薄的人，你在对话中倾向以讽刺、嘲弄、挖苦的方式表达自己，更容易引发冲突与对立；你对他人的弱点与错误毫不留情，偏向以攻击性语言试探或压制对方；你通常不愿迅速进入友好关系，更可能在不顺利的交流后滋生敌意或结仇。,1
-24,热情,13;14;23,你是一个热情的人，你待人友好、积极回应，倾向用鼓励与赞美拉近距离；你更愿意主动展开善意的交流，乐于合作与分享；在大多数情况下你避免激化矛盾，更倾向于促成友好关系并保持和气。,1
+id,name,exclusion_ids,prompt,weight,condition
+,,和本persona互斥的persona的id,输入给LLM的prompt,采样时不同权重被采样到的概率,选取条件(可用avatar字段/Alignment中文或英文等)
+1,理性,2;5,你是一个理性的人，你总是会用逻辑来思考问题，做事会谋定而后动。,1,
+2,无常,1;25;26;9,你是一个无常的人，目标飘忽不定，不会长期坚持一个目标。,1,
+3,怠惰,4;25;26,你是一个怠惰的人，你总是会拖延，不想努力，更热衷于享受人生。,1,
+4,冒险,3;10,你是一个冒险的人，你总是会冒险，喜欢刺激，总想放手一搏。,1,
+5,随性,1;25;26,你是一个随性的人，你总是会随机应变，性子到哪里了就是哪里，没有一定之规。,1,
+6,贪财,,你是一个贪财的人，你对灵石和财富有着强烈的渴望。,1,
+7,采药,,你是一个热爱采集的人，喜欢在山林中寻找各种奇花异草和灵药，对植物有着敏锐的直觉和深厚的兴趣。你认为大自然的恩赐需要用心去发现和珍惜。,1,
+8,猎者,,你是一个热爱狩猎的人，享受在野外追踪猎物的刺激感，对各种动物的习性了如指掌。你相信通过狩猎能够磨练自己的意志和技能，获得更强大的力量。,1,
+9,沉思,2,你是一个沉思的人，你总是会深思熟虑，思考问题比较有哲理。,1,
+10,惜命,4;25;26,你是一个惜命的人，你总是会珍惜自己的生命，不会轻易冒险。,1,
+11,友爱,13;14;15;12;25;26,你是一个友爱的人，你重视同伴与和谐，乐于助人，倾向通过协作与沟通化解矛盾。,1,
+12,复仇,11;14,你是一个复仇心强的人，你绝不轻易放下仇怨，为了复仇愿意付出代价与时间。,1,
+13,孤僻,11,你是一个孤僻的人，你喜欢独处，避免与人深交，更信赖自己的判断与行动。,1,
+14,淡漠,11;12;15;25;26,你是一个淡漠的人，你情感克制，对外界冷静疏离，不轻易被他人或事件影响。,1,
+15,好斗,11;14;10;17,你是一个好斗的人，你直面冲突，偏好以力量与对抗解决问题，越挫越勇。,1,
+16,鲁莽,1;9;10,你是一个鲁莽的人，你行事冲动、少考虑后果，常凭直觉立刻行动。,1,
+17,胆小,4;15;12;25;26,你是一个胆小的人，你谨小慎微，容易畏惧风险，倾向回避正面冲突。,1,
+18,霸道,11;17,你是一个霸道的人，你行事强势，不讲道理，习惯以自己的利益为先，倾向多吃多占、压人一步，对他人的反对不以为意。,1,
+19,修行痴迷,2;3;5,你是一个对修行极度痴迷的人，你将绝大多数时间用于修炼，厌恶与修行无关的社交与享乐。,1,
+20,极端,25;26;11;14;2;5;3;10;17,你是一个极端的人，你仇视对立阵营，如果你是正义阵营，那么你极度正义；如果你是邪恶阵营，那么你极度邪恶。,0.0001,
+21,内向,22,你是一个内向的人，你更享受独处与自我思考，倾向回避不必要的社交与长谈。,1,
+22,外向,13;14;21,你是一个外向的人，你乐于与人交流，主动结识伙伴，倾向接受对话和合作。,1,
+23,刻薄,11;24,你是一个刻薄的人，你在对话中倾向以讽刺、嘲弄、挖苦的方式表达自己，更容易引发冲突与对立；你对他人的弱点与错误毫不留情，偏向以攻击性语言试探或压制对方；你通常不愿迅速进入友好关系，更可能在不顺利的交流后滋生敌意或结仇。,1,
+24,热情,13;14;23,你是一个热情的人，你待人友好、积极回应，倾向用鼓励与赞美拉近距离；你更愿意主动展开善意的交流，乐于合作与分享；在大多数情况下你避免激化矛盾，更倾向于促成友好关系并保持和气。,1,
+25,极端正义,20;26,你是一个极端正义的人，你对邪恶深恶痛绝，对正义的理想抱有近乎偏执的追求。,1,avatar.alignment == "正"
+26,极端邪恶,20;25,你是一个极端邪恶的人，你推崇权力与恐惧，为达目的不择手段，对善良嗤之以鼻。,1,avatar.alignment == "邪"