update prompt

werewolf/seer/seer_agent.py

@@ -41,7 +41,7 @@ class SeerAgent(BasicRoleAgent):
                 self.memory.append_history('主持人: 现在进入第{}天。'.format(str(req.round)))
                 self.memory.append_history('主持人: 每个玩家描述自己的信息。')
         elif req.status == STATUS_VOTE:  # 投票环节
-            self.memory.append_history(req.name + ': ' + req.message)
+            self.memory.append_history(f'第{req.round}天的投票环节,{req.name} 投了 {req.message}')
         elif req.status == STATUS_VOTE_RESULT:  # 投票环节
             out_player = req.name if req.name else req.message
             if out_player:

werewolf/villager/villager_agent.py

@@ -33,7 +33,7 @@ class VillagerAgent(BasicRoleAgent):
                 self.memory.append_history('主持人: 现在进入第{}天。'.format(str(req.round)))
                 self.memory.append_history('主持人: 每个玩家描述自己的信息。')
         elif req.status == STATUS_VOTE:  # 投票环节
-            self.memory.append_history(req.name + ': ' + req.message)
+            self.memory.append_history(f'第{req.round}天的投票环节,{req.name} 投了 {req.message}')
         elif req.status == STATUS_VOTE_RESULT:  # 投票环节
             out_player = req.name if req.name else req.message
             if out_player:

werewolf/witch/witch_agent.py

@@ -41,8 +41,8 @@ class WitchAgent(BasicRoleAgent):
                 self.memory.append_history('主持人: 现在进入第{}天。'.format(str(req.round)))
                 self.memory.append_history('主持人: 每个玩家描述自己的信息。')
         elif req.status == STATUS_VOTE:  # 投票环节
-            self.memory.append_history(req.name + ': ' + req.message)
-        elif req.status == STATUS_VOTE_RESULT:  # 投票环节
+            self.memory.append_history(f'第{req.round}天的投票环节,{req.name} 投了 {req.message}')
+        elif req.status == STATUS_VOTE_RESULT:  # 投票结果
             out_player = req.name if req.name else req.message
             if out_player:
                 self.memory.append_history('主持人: 投票结果是：{}。'.format(out_player))
@@ -63,10 +63,10 @@ class WitchAgent(BasicRoleAgent):
             skill_info = "女巫有{}瓶毒药和{}瓶解药".format("1" if has_poison else "0", "1" if has_antidote else "0")
 
             prompt = format_prompt(DESC_PROMPT,
-                                  {"name": self.memory.load_variable("name"),
-                                   "skill_info": skill_info,
-                                   "history": "\n".join(self.memory.load_history())
-                                  })
+                                   {"name": self.memory.load_variable("name"),
+                                    "skill_info": skill_info,
+                                    "history": "\n".join(self.memory.load_history())
+                                    })
             logger.info("prompt:" + prompt)
             result = self.llm_caller(prompt)
             logger.info("witch interact result: {}".format(result))
@@ -77,20 +77,19 @@ class WitchAgent(BasicRoleAgent):
             choices = [name for name in req.message.split(",") if name != self.memory.load_variable("name")]  # 排除自己
             self.memory.set_variable("choices", choices)
             prompt = format_prompt(VOTE_PROMPT, {"name": self.memory.load_variable("name"),
-                                               "choices": choices,
-                                               "history": "\n".join(self.memory.load_history())
-                                              })
+                                                 "choices": choices,
+                                                 "history": "\n".join(self.memory.load_history())
+                                                 })
             logger.info("prompt:" + prompt)
             result = self.llm_caller(prompt)
             logger.info("witch interact result: {}".format(result))
             return AgentResp(success=True, result=result, errMsg=None)
-            
+
         elif req.status == STATUS_SKILL:
             has_poison = self.memory.load_variable("has_poison")
             has_antidote = self.memory.load_variable("has_antidote")
             tonight_killed = req.message
 
-
             skill_info = "女巫有{}瓶毒药和{}瓶解药".format("1" if has_poison else "0", "1" if has_antidote else "0")
             prompt = format_prompt(SKILL_PROMPT, {
                 "name": self.memory.load_variable("name"),
@@ -98,7 +97,7 @@ class WitchAgent(BasicRoleAgent):
                 "skill_info": skill_info,
                 "history": "\n".join(self.memory.load_history())
             })
-            
+
             logger.info("prompt:" + prompt)
             result = self.llm_caller(prompt)
             logger.info("witch skill result: {}".format(result))
@@ -116,4 +115,4 @@ class WitchAgent(BasicRoleAgent):
 
             return AgentResp(success=True, result=result, skillTargetPlayer=skill_target_person, errMsg=None)
         else:
-            raise NotImplementedError
+            raise NotImplementedError

werewolf/wolf/wolf_agent.py

@@ -47,7 +47,7 @@ class WolfAgent(BasicRoleAgent):
                 self.memory.append_history('主持人: 现在进入第{}天。'.format(str(req.round)))
                 self.memory.append_history('主持人: 每个玩家描述自己的信息。')
         elif req.status == STATUS_VOTE:  # 投票环节
-            self.memory.append_history(req.name + ': ' + req.message)
+            self.memory.append_history(f'第{req.round}天的投票环节,{req.name} 投了 {req.message}')
         elif req.status == STATUS_VOTE_RESULT:  # 投票环节
             out_player = req.name if req.name else req.message
             if out_player: