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="4105000" />
+      <workItem from="1745132078229" duration="4676000" />
     </task>
     <servers />
   </component>

werewolf/wolf/prompt.py

@@ -28,7 +28,7 @@ VOTE_PROMPT = """{history}
 """
 
 WOLF_SPEECH_PROMPT = """{history}
-你是{name}，现在是狼人之间的交流时间。你可以与你的狼人队友{teammates}讨论今晚要击杀谁。
+你是{name}，现在是狼人之间的交流时间。你可以与你的狼人队友{teammates}讨论今晚要击杀谁。你只有一次的交流机会，请根据游戏局势做出你的选择。
 
 请考虑以下因素：
 1. 哪个玩家对狼人阵营威胁最大（如预言家、女巫）

werewolf/wolf/wolf_agent.py

@@ -22,10 +22,6 @@ 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:
@@ -90,6 +86,10 @@ 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"),