feat(engine): 实现阵营投票系统

- 新增 Votes 模型,包含 optimists/fearful 计数和 voted_users 去重列表
- 扩展 Event 模型,添加可选 faction 字段支持投票事件
- 新增 voting.py 模块处理投票逻辑
- 投票规则:每用户每 tick 限投一次,下一 tick 生效
- 投票累加到 factions.power,达到 threshold 触发阵营技能
- 添加 5 个投票系统测试用例,全部 26 个测试通过

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
empty
2025-12-30 13:12:22 +08:00
parent 87007d9b43
commit 75e84f2ba3
4 changed files with 163 additions and 0 deletions

View File

@@ -10,6 +10,7 @@ from .social_influence import apply_social_influence
from .factions import (
update_factions, apply_faction_influence, check_and_trigger_faction_event
)
from .voting import process_votes, apply_votes_to_factions
MAX_EVENTS = 20
MAX_MEMORY_PER_AGENT = 3
@@ -43,6 +44,9 @@ def process_events(state: WorldState, events: List[Event]) -> WorldState:
# tick 递增
state.tick += 1
# ★ 在 tick 开始时:将上一轮投票累加到阵营能量
apply_votes_to_factions(state)
# cooldown 递减
if state.global_meter.cooldown > 0:
state.global_meter.cooldown -= 1
@@ -50,6 +54,9 @@ def process_events(state: WorldState, events: List[Event]) -> WorldState:
# 更新世界效果(持续影响)
update_world_effects(state)
# ★ 处理投票事件(记录到 votes下一 tick 生效)
process_votes(state, events)
for event in events:
text = event.text

View File

@@ -58,6 +58,14 @@ class Factions(BaseModel):
fearful: FactionData = Field(default_factory=FactionData)
class Votes(BaseModel):
"""投票统计"""
optimists: int = 0
fearful: int = 0
# 记录本 tick 已投票的用户(用于去重)
voted_users: List[str] = Field(default_factory=list)
class FactionEventResult(BaseModel):
"""阵营事件触发结果"""
type: Optional[str] = None # "festival" | "panic" | None
@@ -104,6 +112,7 @@ class WorldState(BaseModel):
world_effects: List[WorldEffect] = Field(default_factory=list)
factions: Factions = Field(default_factory=Factions)
story_arcs: Dict[str, StoryArc] = Field(default_factory=dict)
votes: Votes = Field(default_factory=Votes)
class Event(BaseModel):
@@ -111,6 +120,7 @@ class Event(BaseModel):
text: str
user: str
ts: float
faction: Optional[str] = None # 用于投票事件: "optimists" | "fearful"
class StepRequest(BaseModel):

View File

@@ -0,0 +1,46 @@
"""阵营投票系统 - 允许用户通过投票影响阵营能量"""
from typing import List
from .models import WorldState, Event, Votes
def process_votes(state: WorldState, events: List[Event]) -> None:
"""处理投票事件
规则:
1. 每个用户在一个 tick 内只能投 1 次票(按 user 去重)
2. 每票增加对应 faction 的 votes += 1
"""
for event in events:
if event.type != "vote":
continue
# 验证 faction 有效性
if event.faction not in ("optimists", "fearful"):
continue
# 检查用户是否已投票(去重)
if event.user in state.votes.voted_users:
continue
# 记录投票
state.votes.voted_users.append(event.user)
if event.faction == "optimists":
state.votes.optimists += 1
elif event.faction == "fearful":
state.votes.fearful += 1
def apply_votes_to_factions(state: WorldState) -> None:
"""将投票累加到阵营能量,然后清空投票
在每个 tick 开始时调用:
1. 将 votes 累加进 factions.power
2. 清空 votes包括 voted_users
"""
# 累加投票到阵营能量
state.factions.optimists.power += state.votes.optimists
state.factions.fearful.power += state.votes.fearful
# 清空投票
state.votes = Votes()