From 3c015349640d4a3456e283b142b06a86b3dbdbd8 Mon Sep 17 00:00:00 2001 From: mannaandpoem <1580466765@qq.com> Date: Mon, 25 Sep 2023 23:22:18 +0800 Subject: [PATCH] Modify moderator --- .../actions/moderator_actions.py | 109 +++++++++++++++--- examples/werewolf_game/roles/moderator.py | 11 +- 2 files changed, 100 insertions(+), 20 deletions(-) diff --git a/examples/werewolf_game/actions/moderator_actions.py b/examples/werewolf_game/actions/moderator_actions.py index 7850ab478..9e51a9106 100644 --- a/examples/werewolf_game/actions/moderator_actions.py +++ b/examples/werewolf_game/actions/moderator_actions.py @@ -4,45 +4,128 @@ STAGE_INSTRUCTIONS = { # 上帝需要介入的全部步骤和对应指令 # The 1-st night 0: {"content": "It’s dark, everyone close your eyes. I will talk with you/your team secretly at night.", - "send_to": "Moderator", # for moderator to continuen speaking + "send_to": "Moderator", # for moderator to continuen speaking "restricted_to": ""}, - 1: {"content": "Werewolves, please open your eyes!", - "send_to": "Moderator", # for moderator to continuen speaking + 1: {"content": "Guard, please open your eyes!", + "send_to": "Moderator", # for moderator to continuen speaking "restricted_to": ""}, - 2: {"content": """Werewolves, I secretly tell you that Player 3 and Player 4 are + 2: {"content": """Guard, now tell me who you protect tonight? + You only choose one from the following living options please: {living_players}.""", + "send_to": "Guard", + "restricted_to": "Guard"}, + 3: {"content": "Guard, close your eyes", + "send_to": "Moderator", + "restricted_to": ""}, + 4: {"content": "Werewolves, please open your eyes!", + "send_to": "Moderator", + "restricted_to": ""}, + 5: {"content": """Werewolves, I secretly tell you that xxx and xxx are all of the 2 werewolves! Keep in mind you are teammates. The rest players are not werewolves. choose one from the following living options please: - [Player 1, Player2]. """, # send to werewolf restrictedly for a response + {living_players}. """, # send to werewolf restrictedly for a response "send_to": "Werewolf", "restricted_to": "Werewolf"}, - 3: {"content": "Werewolves, close your eyes", - "send_to": "Moderator", # for moderator to continuen speaking + 6: {"content": "Werewolves, close your eyes", + "send_to": "Moderator", "restricted_to": ""}, - 4: {"content": """It's daytime. No one dies last night. Now freely talk about roles of other players with each other based on your observation and reflection - with few sentences. Decide whether to reveal your identity based on your reflection.""", - "send_to": "", # send to all to speak in daytime - "restricted_to": ""} + 7: {"content": "Seer, please open your eyes!", + "send_to": "Moderator", + "restricted_to": ""}, + 8: {"content": """Seer, you can check one player's identity. Who are you going to verify its identity tonight? + Choose only one from the following living options:{living_players}.""", + "send_to": "Seer", + "restricted_to": "Seer"}, + 9: {"content": "Seer, close your eyes", + "send_to": "Moderator", + "restricted_to": ""}, + 10: {"content": "Witch, please open your eyes!", + "send_to": "Moderator", + "restricted_to": ""}, + 11: {"content": """Witch, you have a bottle of poison, who are you going to kill tonight? + Choose one from the following living options: {living_players}.""", + "send_to": "Witch", + "restricted_to": "Witch"}, + 12: {"content": """Witch, you have a bottle of antidote and a bottle of poison. + Who are you going to save tonight or kill tonight? Choose one from the following living options: + {living_players}.""", + "send_to": "Witch", + "restricted_to": "Witch"}, + 13: {"content": "Witch, close your eyes", + "send_to": "Moderator", + "restricted_to": ""}, + # The 1-st daytime + 14: {"content": """It's daytime. Everyone woke up except those who had been killed.""", + "send_to": "Moderator", + "restricted_to": ""}, + 15: {"content": "xxxx was killed last night. Or, it was a peaceful night and no one died!", + "send_to": "Moderator", + "restricted_to": ""}, + 16: {"content": """Now freely talk about roles of other players with each other based on your observation and + reflection with few sentences. Decide whether to reveal your identity based on your reflection.""", + "send_to": "", # send to all to speak in daytime + "restricted_to": ""}, + 17: {"content": """Now vote and tell me who you think is the werewolf. Don’t mention your role. + You only choose one from the following living options please: + {living_players}. Or you can pass. For example: I vote to kill xxxx""", + "send_to": "Moderator", + "restricted_to": ""}, + 18: {"content": """xxxx was eliminated.""", + "send_to": "Moderator", + "restricted_to": ""}, } +INSTRUCT_SPEAK_TEMPLATE = """ +## BACKGROUND +It's a Werewolf game, you are moderator. + +## STAGE +The current stage of the game is: +{} + +## CONTEXT +Here's the current context: +{} + +What would you like to instruct? +""" class InstructSpeak(Action): def __init__(self, name="InstructSpeak", context=None, llm=None): super().__init__(name, context, llm) - async def run(self, context, stage_idx): - return STAGE_INSTRUCTIONS[stage_idx] + async def run(self, context,living_players, stage_idx): + + instruction_info = STAGE_INSTRUCTIONS.get(stage_idx, "Unknown instruction.") + # 利用上下文信息context和所处阶段的信息 + if "{living_players}" in instruction_info["content"]: + content = instruction_info["content"].format(living_players) + else: + content = instruction_info["content"] + + prompt = INSTRUCT_SPEAK_TEMPLATE.format(content, context) + + rsp = await self._aask(prompt) + + send_to = instruction_info["send_to"] + restricted_to = instruction_info["restricted_to"] + + return rsp, send_to, restricted_to + class ParseSpeak(Action): async def run(self): return "" + class SummarizeNight(Action): """consider all events at night, conclude which player dies (can be a peaceful night)""" pass + class SummarizeDay(Action): """consider all votes at day, conclude which player dies""" pass + class AnnounceGameResult(Action): async def run(self, winner: str): diff --git a/examples/werewolf_game/roles/moderator.py b/examples/werewolf_game/roles/moderator.py index e5a61e6c1..9b59300ef 100644 --- a/examples/werewolf_game/roles/moderator.py +++ b/examples/werewolf_game/roles/moderator.py @@ -24,15 +24,12 @@ class Moderator(Role): self._watch([UserRequirement, InstructSpeak, ParseSpeak]) self._init_actions([InstructSpeak, ParseSpeak, AnnounceGameResult]) self.stage_idx = 0 + self.living_players = "[Player 1, Player 2, Player 3, Player 4]" - async def _instruct_speak(self): + async def _instruct_speak(self,context): stage_idx = self.stage_idx % len(STAGE_INSTRUCTIONS) - - stage_info = await InstructSpeak().run(context="", stage_idx=stage_idx) - self.stage_idx += 1 - - return stage_info["content"], stage_info["send_to"], stage_info["restricted_to"] + return await InstructSpeak().run(context=context, stage_idx=stage_idx, living_players=self.living_players) async def _parse_speak(self): # 解析玩家消息并返回结果 @@ -69,7 +66,7 @@ class Moderator(Role): # 根据_think的结果，执行InstructSpeak还是ParseSpeak, 并将结果返回 if isinstance(todo, InstructSpeak): - msg_content, msg_to_send_to, msg_restriced_to = await self._instruct_speak() + msg_content, msg_to_send_to, msg_restriced_to = await self._instruct_speak(memories) msg = Message(content=msg_content, role=self.profile, sent_from=self.name, cause_by=InstructSpeak, send_to=msg_to_send_to, restricted_to=msg_restriced_to)