update prompt

.idea/workspace.xml

@@ -3,6 +3,8 @@
   <component name="ChangeListManager">
     <list default="true" id="be0bdb17-6f52-4720-b934-341fa7978d29" name="Changes" comment="">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/werewolf/witch/prompt.py" beforeDir="false" afterPath="$PROJECT_DIR$/werewolf/witch/prompt.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/werewolf/witch/witch_agent.py" beforeDir="false" afterPath="$PROJECT_DIR$/werewolf/witch/witch_agent.py" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -70,7 +72,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1745132074997</updated>
-      <workItem from="1745132078229" duration="7018000" />
+      <workItem from="1745132078229" duration="10057000" />
     </task>
     <servers />
   </component>

werewolf/witch/prompt.py

@@ -29,11 +29,10 @@ VOTE_PROMPT = """{history}
 
 SKILL_PROMPT = """{history}
 你是{name}，作为女巫，你现在可以使用你的技能。
-今晚{tonight_killed}被狼人杀死了。
+今晚信息：{tonight_killed}。
 
 你目前拥有的药水：
-- 解药：{'有' if has_antidote else '已用完'}
-- 毒药：{'有' if has_poison else '已用完'}
+{skill_info}
 
 你可以：
 1. 使用解药救活{tonight_killed}（如果你还有解药）

werewolf/witch/witch_agent.py

@@ -83,17 +83,17 @@ class WitchAgent(BasicRoleAgent):
         elif req.status == STATUS_SKILL:
             has_poison = self.memory.load_variable("has_poison")
             has_antidote = self.memory.load_variable("has_antidote")
-            tonight_killed = req.name
+            tonight_killed = req.message
             
             choices = [name for name in req.message.split(",") if name != self.memory.load_variable("name")]  # 排除自己
             self.memory.set_variable("choices", choices)
-            
+
+            skill_info = "女巫有{}瓶毒药和{}瓶解药".format("有" if has_poison else "没有", "有" if has_antidote else "没有")
             prompt = format_prompt(SKILL_PROMPT, {
                 "name": self.memory.load_variable("name"),
                 "choices": choices,
                 "tonight_killed": tonight_killed,
-                "has_poison": has_poison,
-                "has_antidote": has_antidote,
+                "skill_info": skill_info,
                 "history": "\n".join(self.memory.load_history())
             })