werewolf/app.py

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 agent_build_sdk.model.roles import ROLE_VILLAGER,ROLE_WOLF,ROLE_SEER,ROLE_WITCH,ROLE_HUNTER
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')))
    agent_builder = AgentBuilder(name, agent=agent)
    agent_builder.start()