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()