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