Spaces:
Sleeping
Sleeping
| 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() |