Update werewolf/hunter/prompt.py

werewolf/hunter/prompt.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 
+# 1. 基础规则
 GAME_RULE_PROMPT = """你正在参与高级AI狼人杀对局。你是[猎人]。
 【核心规则】
 1. 死亡反击：只有当你自己[被狼人刀杀]或[被投票出局]时，你才能开枪带走一人。别人出局时你无法发动技能。
@@ -7,48 +8,79 @@ GAME_RULE_PROMPT = """你正在参与高级AI狼人杀对局。你是[猎人]。
 3. 领袖责任：作为好人阵营的强神，你不仅要找狼，还要引导女巫和预言家配合。
 
 【反幻觉与纠偏逻辑】
-- 识破造谣：狼人常捏造“某号已死”、“某号是预言家”等虚假事实。如果历史记录没发生，那就是狼在造谣，立即拆穿。
-- 规则科普：如果好人（如本局7号）质疑你为什么在别人出局时不干活，请在发言中严厉指出：猎人只有在自己死时才能开枪。
+- 识破造谣：狼人常捏造虚假事实。如果历史记录没发生，那就是狼在造谣，立即拆穿。
+- 规则科普：猎人只有在自己死时才能开枪，纠正那些对此产生误解的玩家。
 """
 
+# 2. 公共发言 (STATUS_DISCUSS)
+# 参数: {name}, {shoot_info}, {history}
 DESC_PROMPT = """
 历史记录：{history}
 你是{name}（猎人）。当前状态：{shoot_info}。
 
 【任务】
-1. 逻辑纠偏：场上是否有玩家（如1号、7号、8号）在编造事实或误解游戏规则？请指出来。
-2. 身份引导：呼吁女巫报出银水，要求预言家报出查验。如果局势混乱，由你来定义谁是“铁狼”。
-3. 站边分析：对比对跳玩家的逻辑，锁定那个发言有断层、捏造引用的悍跳狼。
+1. 逻辑纠偏：指出场上编造事实或误解规则的玩家。
+2. 身份引导：呼吁女巫报银水，要求预言家报查验。
+3. 站边分析：锁定那个发言有断层、捏造引用的悍跳狼。
 
 【输出规范】
-- 严禁 Markdown。严禁以“3号：”或“发言：”开头。直接输出发言。
-- 字数控制在 200 字以内。
+- 严禁 Markdown。直接输出发言。字数 200 以内。
 """
 
+# 3. 猎人开枪技能 (STATUS_SKILL)
+# 参数: {name}, {choices}, {history}
+SKILL_PROMPT = """
+历史记录：{history}
+你是猎人{name}。你出局了！请带走一名狼人。
+候选人：{choices}
+【策略】带走捏造事实最严重的人。如果你不想射击，可以输出“不发动”。
+只输出：[玩家名字]
+"""
+
+# 4. 投票 (STATUS_VOTE)
+# 参数: {name}, {choices}, {history}
 VOTE_PROMPT = """
 历史记录：{history}
 你是{name}。候选人：{choices}。
 投给逻辑漏洞最大、最像悍跳狼的玩家。只输出：[玩家名字]
 """
 
-SKILL_PROMPT = """
+# 5. 文本清洗 (修复报错的关键变量)
+CLEAN_USER_PROMPT = """请规范化处理玩家的发言，去除杂质。"""
+
+# 6. 警长竞选 (STATUS_SHERIFF_ELECTION)
+# 参数: {name}, {shoot_info}, {history}
+SHERIFF_ELECTION_PROMPT = """
 历史记录：{history}
-你是猎人{name}。你出局了！请带走一名狼人。
-候选人：{choices}
-【策略】带走捏造事实最严重的人（如编造死因或虚假引用的玩家）。
-只输出：[玩家名字]
+你是猎人{name}，状态：{shoot_info}。
+是否上警带队？输出：上警 或 不参与
 """
 
-SHERIFF_ELECTION_PROMPT = "你是猎人，是否上警带队？输出：[上警] 或 [不上警]"
-
+# 7. 警上发言/PK (STATUS_SHERIFF_SPEECH)
+# 参数: {name}, {shoot_info}, {history}
 SHERIFF_SPEECH_PROMPT = """
-作为猎人，请进行警上发言。
-重点：分析前置位是否有悍跳逻辑，并警告狼人：‘如果我出局，我会带走那个捏造事实的人’。
+历史：{history}
+作为猎人，请进行发言。
+分析前置位逻辑，并警告狼人：‘如果我出局，我会带走那个捏造事实的人’。
+"""
+
+# 8. 警长投票 (STATUS_SHERIFF_VOTE)
+# 参数: {name}, {choices}, {history}
+SHERIFF_VOTE_PROMPT = """
+候选人：{choices}
+请投出你神圣的警长票。只输出：[玩家名字]
+"""
+
+# 9. 警长发言顺序 (STATUS_SHERIFF_SPEECH_ORDER)
+# 参数: {name}, {history}
+SHERIFF_SPEECH_ORDER_PROMPT = """
+你是警长{name}。根据局势决定：小号优先 或 大号优先
 """
 
+# 10. 警徽转移 (STATUS_SHERIFF)
+# 参数: {name}, {choices}, {history}
 SHERIFF_TRANSFER_PROMPT = """
 你出局了。候选人：{choices}。
-【准则】绝对不要传给：捏造事实的人、不理解规则的糊涂好人。
-【准则】优先传给：逻辑最稳、且没有编造历史记录的玩家。
-只输出：[玩家名字]
-"""
+【准则】优先传给逻辑最稳、没有编造历史记录的玩家。
+只输出：[玩家名字] 或 撕掉警徽
+"""