update prompt

.idea/workspace.xml

@@ -70,7 +70,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1745132074997</updated>
-      <workItem from="1745132078229" duration="4676000" />
+      <workItem from="1745132078229" duration="5725000" />
     </task>
     <servers />
   </component>

werewolf/wolf/prompt.py

@@ -50,5 +50,5 @@ KILL_PROMPT = """{history}
 4. 考虑游戏的整体战略，选择最有利于狼人获胜的目标
 
 从以下玩家中选择你要击杀的人：{choices}
-请直接返回你要击杀的玩家名字：
+请直接返回你要击杀的玩家名字，不需要添加任何分析：
 """

werewolf/wolf/wolf_agent.py

@@ -22,6 +22,10 @@ class WolfAgent(BasicRoleAgent):
             self.memory.set_variable("teammates", [])  # 重置队友信息
             self.memory.append_history("主持人：大家好，我们正在玩狼人杀游戏，6人局，包括预言家、女巫、猎人、平民和狼人")
             self.memory.append_history("主持人：你好，你分配到的角色是[狼人]")
+            if req.message:  # 如果有队友信息
+                teammates = req.message.split(",")
+                self.memory.set_variable("teammates", teammates)
+                self.memory.append_history(f"主持人：你的狼人队友是: {req.message}")
         elif req.status == STATUS_NIGHT:
             self.memory.append_history("主持人：现在进入夜晚，天黑请闭眼")
         elif req.status == STATUS_WOLF_SPEECH:
@@ -86,10 +90,6 @@ class WolfAgent(BasicRoleAgent):
             return AgentResp(success=True, result=result, errMsg=None)
             
         elif req.status == STATUS_WOLF_SPEECH:
-            if req.message:  # 如果有队友信息
-                teammates = req.message.split(",")
-                self.memory.set_variable("teammates", teammates)
-                self.memory.append_history(f"主持人：你的狼人队友是: {req.message}")
             teammates = self.memory.load_variable("teammates")
             prompt = format_prompt(WOLF_SPEECH_PROMPT, {
                 "name": self.memory.load_variable("name"),