# 具体参见https://github.com/modelscope/agentscope/blob/main/examples/game_werewolf/configs/agent_configs.json
agent_configs = [
    {
        "class": "DictDialogAgent",
        "args": {
            "name": "Player1",
            # 包含狼人杀的具体规则
            "sys_prompt": "Act as a player in a werewolf game. You are Player1 ...",
            "model": "gpt-4",
            "use_memory": True
        }
    },
    ...
    
    {
        "class": "DictDialogAgent",
        "args": {
            "name": "Player6",
            # 包含狼人杀的具体规则
            "sys_prompt": "Act as a player in a werewolf game. You are Player6 ...",
            "model": "gpt-4",
            "use_memory": True
        }
    }
]

# 主持人
HostMsg = partial(Msg, name="Moderator", echo=True)
# 女巫在开始的时候有解药和毒药
healing, poison = True, True
# 狼人夜间最多讨论回合数
MAX_WEREWOLF_DISCUSSION_ROUND = 3
# 游戏进行的最多天数（防止大模型陷入投票得不到结果或者其他原因导致游戏无法结束的情况）
MAX_GAME_ROUND = 6
# 初始化agent
survivors = agentscope.init(
        model_configs="./configs/model_configs.json",
        agent_configs="./configs/agent_configs.json",
)
# 6个角色
roles = ["werewolf", "werewolf", "villager", "villager", "seer", "witch"]
# 给每个agent按照顺序分配角色
wolves, witch, seer = survivors[:2], survivors[-1], survivors[-2]

for i in range(1, MAX_GAME_ROUND + 1):
    # 夜晚来临，狼人首先讨论
    hint = HostMsg(content=Prompts.to_wolves.format(n2s(wolves)))
    with msghub(wolves, announcement=hint) as hub:
        for _ in range(MAX_WEREWOLF_DISCUSSION_ROUND):
            x = sequentialpipeline(wolves)
            if x.get("agreement", False):
                break
    
        # 狼人讨论结束，进行投票
        hint = HostMsg(content=Prompts.to_wolves_vote)
        votes = [extract_name_and_id(wolf(hint).content)[0] for wolf in wolves]
        # 主持人对狼人公布投票结果
        dead_player = [majority_vote(votes)]
        hub.broadcast(
            HostMsg(content=Prompts.to_wolves_res.format(dead_player[0])),
        )
    # 女巫阶段，是否使用解药和毒药（一个晚上最多使用一瓶）
    healing_used_tonight = False
    if witch in survivors:
        if healing:
            hint = HostMsg(
                content=Prompts.to_witch_resurrect.format_map(
                    {"witch_name": witch.name, "dead_name": dead_player[0]},
                ),
            )
            if witch(hint).get("resurrect", False):
                healing_used_tonight = True
                dead_player.pop()
                healing = False
    
        if poison and not healing_used_tonight:
            x = witch(HostMsg(content=Prompts.to_witch_poison))
            if x.get("eliminate", True):
                dead_player.append(extract_name_and_id(x.content)[0])
                poison = False
    # 预言家验人环节
    if seer in survivors:
        hint = HostMsg(
            content=Prompts.to_seer.format(seer.name, n2s(survivors)),
        )
        x = seer(hint)
    
        player, idx = extract_name_and_id(x.content)
        role = "werewolf" if roles[idx] == "werewolf" else "villager"
        hint = HostMsg(content=Prompts.to_seer_result.format(player, role))
        # 预言家获得验人结果
        seer.observe(hint)
    # 根据夜晚发生的情况，更新目前的存活情况并判断游戏是否结束
    survivors, wolves = update_alive_players(survivors, wolves, dead_player)
    if check_winning(survivors, wolves, "Moderator"):
        break
    
    # 根据是否平安夜决定主持人白天的开场白
    content = (
        Prompts.to_all_danger.format(n2s(dead_player))
        if dead_player
        else Prompts.to_all_peace
    )
    hints = [
        HostMsg(content=content),
        HostMsg(content=Prompts.to_all_discuss.format(n2s(survivors))),
    ]
    with msghub(survivors, announcement=hints) as hub:
        # 白天讨论环节
        x = sequentialpipeline(survivors)
        # 白天投票环节
        hint = HostMsg(content=Prompts.to_all_vote.format(n2s(survivors)))
        votes = [extract_name_and_id(_(hint).content)[0] for _ in survivors]
        vote_res = majority_vote(votes)
        # 公布投票结果
        result = HostMsg(content=Prompts.to_all_res.format(vote_res))
        hub.broadcast(result)
        # 更新存活情况并判断游戏是否结束
        survivors, wolves = update_alive_players(survivors, wolves, vote_res)
        if check_winning(survivors, wolves, "Moderator"):
            break
        # 有序继续，新的夜晚来临
        hub.broadcast(HostMsg(content=Prompts.to_all_continue))

def parse_dict(response: ModelResponse) -> ModelResponse:
    """Parse function for DictDialogAgent"""
    try:
        response_dict = json.loads(response.text)
    except json.decoder.JSONDecodeError:
        response_dict = json.loads(response.text.replace("'", '"'))
    return ModelResponse(raw=response_dict)
def default_response(response: ModelResponse) -> ModelResponse:
    """The default response of fault_handler"""
    return ModelResponse(raw={"speak": response.text})
class DictDialogAgent(AgentBase):
    
    def reply(self, x: dict = None) -> dict:
        # 省略具体代码...
        # call llm
        response = self.model(
            prompt,
            parse_func=self.parse_func,
            fault_handler=self.fault_handler,
            max_retries=self.max_retries,
        ).raw
        # 省略代码...
        return msg