def setup_args():
"""
Set up args.
"""
parser = ParlaiParser(False, False)
parser.add_parlai_data_path()
parser.add_messenger_args()
return parser.parse_args()
def main():
argparser = ParlaiParser(False, False)
argparser.add_parlai_data_path()
argparser.add_messenger_args()
opt = argparser.parse_args()
opt['task'] = os.path.basename(os.path.dirname(os.path.abspath(__file__)))
# Initialize a SQuAD teacher agent, which we will get context from
module_name = 'parlai.tasks.squad.agents'
class_name = 'DefaultTeacher'
my_module = importlib.import_module(module_name)
task_class = getattr(my_module, class_name)
task_opt = {}
task_opt['datatype'] = 'train'
task_opt['datapath'] = opt['datapath']
messenger_manager = MessengerManager(opt=opt)
messenger_manager.setup_server()
messenger_manager.init_new_state()
def get_overworld(agent):
return MessengerOverworld(None, agent)
def assign_agent_role(agent):
agent[0].disp_id = 'Agent'
def run_conversation(manager, opt, agents, task_id):
task = task_class(task_opt)
agent = agents[0]
world = QADataCollectionWorld(
opt=opt,
task=task,
agent=agent
)
while not world.episode_done():
world.parley()
world.shutdown()
# World with no onboarding
messenger_manager.set_onboard_functions({'default': None})
task_functions = {'default': run_conversation}
assign_agent_roles = {'default': assign_agent_role}
messenger_manager.set_agents_required({'default': 1})
messenger_manager.set_overworld_func(get_overworld)
messenger_manager.setup_socket()
try:
messenger_manager.start_new_run()
messenger_manager.start_task(
assign_role_functions=assign_agent_roles,
task_functions=task_functions,
)
except BaseException:
raise
finally:
messenger_manager.shutdown()
def main():
argparser = ParlaiParser(True, True)
argparser.add_messenger_args()
opt = argparser.parse_args()
print(opt)
if opt['model'] == None and opt['model_file'] == None:
print("Model must be specified")
return
bot = create_agent(opt)
shared_bot_params = bot.share()
messenger_manager = MessengerManager(opt=opt)
messenger_manager.setup_server()
messenger_manager.init_new_state()
def get_overworld(agent):
return MessengerOverworld(None, agent)
def assign_agent_role(agent):
agent[0].disp_id = 'Agent'
def run_conversation(manager, opt, agents, task_id):
agent = agents[0]
this_bot = create_agent_from_shared(shared_bot_params)
world = MessengerBotChatTaskWorld(opt=opt, agent=agent, bot=this_bot)
while not world.episode_done():
world.parley()
world.shutdown()
# World with no onboarding
messenger_manager.set_onboard_functions({'default': None})
task_functions = {'default': run_conversation}
assign_agent_roles = {'default': assign_agent_role}
messenger_manager.set_agents_required({'default': 1})
messenger_manager.set_overworld_func(get_overworld)
messenger_manager.setup_socket()
try:
messenger_manager.start_new_run()
messenger_manager.start_task(
assign_role_functions=assign_agent_roles,
task_functions=task_functions,
)
except BaseException:
raise
finally:
messenger_manager.shutdown()
def main():
argparser = ParlaiParser(False, False)
argparser.add_parlai_data_path()
argparser.add_messenger_args()
opt = argparser.parse_args()
opt['task'] = os.path.basename(os.path.dirname(os.path.abspath(__file__)))
opt['password'] = '******' # If password is none anyone can chat
messenger_manager = MessengerManager(opt=opt)
messenger_manager.setup_server()
messenger_manager.init_new_state()
def get_overworld(agent):
return MessengerOverworld(None, agent)
onboard_functions = {
name: worlds[0].run
for (name, worlds) in MessengerOverworld.DEMOS.items()
}
messenger_manager.set_onboard_functions(onboard_functions)
task_functions = {
name: worlds[1].run
for (name, worlds) in MessengerOverworld.DEMOS.items()
}
assign_agent_roles = {
name: worlds[1].assign_roles
for (name, worlds) in MessengerOverworld.DEMOS.items()
}
agents_required = {
name: worlds[1].MAX_AGENTS
for (name, worlds) in MessengerOverworld.DEMOS.items()
}
messenger_manager.set_agents_required(agents_required)
messenger_manager.set_overworld_func(get_overworld)
messenger_manager.setup_socket()
try:
messenger_manager.start_new_run()
messenger_manager.start_task(
assign_role_functions=assign_agent_roles,
task_functions=task_functions,
)
except BaseException:
raise
finally:
messenger_manager.shutdown()
def _get_args(self):
parser = ParlaiParser(False, False)
parser.add_parlai_data_path()
parser.add_messenger_args()
return parser.parse_args([])
