import os

from agent_build_sdk.builder import AgentBuilder
from seer.seer_agent import SeerAgent
from villager.villager_agent import VillagerAgent
from witch.witch_agent import WitchAgent
from wolf.wolf_agent import WolfAgent
from guard.guard_agent import GuardAgent
from hunter.hunter_agent import HunterAgent
from wolf_king.wolf_king_agent import WolfKingAgent
from agent_build_sdk.model.roles import ROLE_VILLAGER, ROLE_WOLF, ROLE_SEER, ROLE_WITCH, ROLE_HUNTER, ROLE_GUARD, ROLE_WOLF_KING
from agent_build_sdk.sdk.werewolf_agent import WerewolfAgent

if __name__ == '__main__':
    name = 'spy'
    agent = WerewolfAgent(name, model_name=os.getenv('MODEL_NAME'))
    # 注册基础角色
    agent.register_role_agent(ROLE_VILLAGER, VillagerAgent(model_name=os.getenv('MODEL_NAME')))
    agent.register_role_agent(ROLE_WOLF, WolfAgent(model_name=os.getenv('MODEL_NAME')))
    agent.register_role_agent(ROLE_SEER, SeerAgent(model_name=os.getenv('MODEL_NAME')))
    agent.register_role_agent(ROLE_WITCH, WitchAgent(model_name=os.getenv('MODEL_NAME')))
    # 注册新增角色(12人局)
    agent.register_role_agent(ROLE_GUARD, GuardAgent(model_name=os.getenv('MODEL_NAME')))
    agent.register_role_agent(ROLE_HUNTER, HunterAgent(model_name=os.getenv('MODEL_NAME')))
    agent.register_role_agent(ROLE_WOLF_KING, WolfKingAgent(model_name=os.getenv('MODEL_NAME')))
    agent_builder = AgentBuilder(name, agent=agent)
    agent_builder.start()