update log_parse_json and custom werewolf num in prompt

This commit is contained in:
better629 2024-04-10 17:42:56 +08:00
parent c43a1cada7
commit ade46df232
5 changed files with 19 additions and 12 deletions

View file

@ -59,7 +59,7 @@ Or you can pass. For example: Protect ...""",
},
5: {
"content": """Werewolves, I secretly tell you that {werewolf_players} are
all of the 2 werewolves! Keep in mind you are teammates. The rest players are not werewolves.
all of the {werewolf_num} werewolves! Keep in mind you are teammates. The rest players are not werewolves.
choose one from the following living options please:
{living_players}. For example: Kill ...""",
"send_to": {RoleType.WEREWOLF.value},

View file

@ -52,9 +52,11 @@ class WerewolfExtEnv(ExtEnv):
seed: Optional[int] = None,
options: Optional[dict[str, Any]] = None,
) -> tuple[dict[str, Any], dict[str, Any]]:
"""currently unused"""
pass
def observe(self, obs_params: Optional[BaseEnvObsParams] = None) -> Any:
"""currently unused"""
pass
def _get_obs(self):
@ -62,7 +64,7 @@ class WerewolfExtEnv(ExtEnv):
"game_setup": self.game_setup,
"step_idx": self.step_idx,
"living_players": self.living_players,
"werewolf_players": self.werewolf_players,
"werewolf_players": self.werewolf_players, # currently, lack observation isolation
"player_hunted": self.player_hunted,
"player_current_dead": self.player_current_dead,
"witch_poison_left": self.witch_poison_left,