update

.idea/workspace.xml

@@ -4,7 +4,6 @@
     <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" />
@@ -72,7 +71,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1745132074997</updated>
-      <workItem from="1745132078229" duration="10057000" />
+      <workItem from="1745132078229" duration="11140000" />
     </task>
     <servers />
   </component>

werewolf/witch/prompt.py

@@ -1,6 +1,7 @@
 DESC_PROMPT = """{history}
 你是{name}，你是一名女巫。你的目标是找出狼人，保护村庄。你拥有两瓶药：一瓶解药可以救活一名被狼人杀死的玩家，一瓶毒药可以毒死一名玩家。每种药只能使用一次。
-目前你的药水状态：解药（{'有' if has_antidote else '已用完'}），毒药（{'有' if has_poison else '已用完'}）。
+你目前拥有的药水：
+{skill_info}
 
 请根据游戏规则和此前的对话，提供一个自然且合理的描述，确保：
 
@@ -40,7 +41,7 @@ SKILL_PROMPT = """{history}
 3. 不使用任何药水
 
 请仔细分析当前游戏局势，做出最有利于村民阵营的决定。
-如果你决定使用解药，请回复"救{tonight_killed}"
+如果你决定使用解药，请回复"救[玩家名]"
 如果你决定使用毒药，请回复"毒[玩家名]"，从以下玩家中选择：{choices}
 如果你决定不使用任何药水，请回复"不使用"
 

werewolf/witch/witch_agent.py

@@ -56,10 +56,13 @@ class WitchAgent(BasicRoleAgent):
     def interact(self, req=AgentReq) -> AgentResp:
         logger.info("witch interact: {}".format(req))
         if req.status == STATUS_DISCUSS:
+            has_poison = self.memory.load_variable("has_poison")
+            has_antidote = self.memory.load_variable("has_antidote")
+            skill_info = "女巫有{}瓶毒药和{}瓶解药".format("有" if has_poison else "没有", "有" if has_antidote else "没有")
+
             prompt = format_prompt(DESC_PROMPT,
                                   {"name": self.memory.load_variable("name"),
-                                   "has_poison": self.memory.load_variable("has_poison"),
-                                   "has_antidote": self.memory.load_variable("has_antidote"),
+                                   "skill_info": skill_info,
                                    "history": "\n".join(self.memory.load_history())
                                   })
             logger.info("prompt:" + prompt)