def main(args):
utils.configure_colored_logging(args.loglevel)
pre_load = args.pre_load
_endpoints = read_endpoints(args.endpoints)
router = DataRouter(args.path,
args.max_training_processes,
args.response_log,
args.emulate,
args.storage,
model_server=_endpoints.model,
wait_time_between_pulls=args.wait_time_between_pulls)
if pre_load:
logger.debug('Preloading....')
if 'all' in pre_load:
pre_load = router.project_store.keys()
router._pre_load(pre_load)
rasa = RasaNLU(router,
args.loglevel,
args.write,
args.num_threads,
get_token(cmdline_args.token),
args.cors,
default_config_path=args.config)
logger.info('Started http server on port %s' % args.port)
rasa.app.run('0.0.0.0', args.port)
def main(args):
pre_load = args.pre_load
_endpoints = read_endpoints(args.endpoints)
router = DataRouter(
args.path,
args.max_training_processes,
args.response_log,
args.emulate,
args.storage,
model_server=_endpoints.model,
wait_time_between_pulls=args.wait_time_between_pulls,
)
if pre_load:
logger.debug("Preloading....")
if "all" in pre_load:
pre_load = router.project_store.keys()
router._pre_load(pre_load)
rasa = RasaNLU(
router,
args.loglevel,
args.write,
args.num_threads,
get_token(args.token),
args.cors,
default_config_path=args.config,
)
logger.info("Started http server on port %s" % args.port)
rasa.app.run("0.0.0.0", args.port)
def main(args):
pre_load = args.pre_load
_endpoints = read_endpoints(args.endpoints)
router = DataRouter(
args.path,
args.max_training_processes,
args.response_log,
args.emulate,
args.storage,
model_server=_endpoints.model,
wait_time_between_pulls=args.wait_time_between_pulls,
)
if pre_load:
logger.debug("Preloading....")
if "all" in pre_load:
pre_load = router.project_store.keys()
router._pre_load(pre_load)
rasa = create_app(
router, args.loglevel, args.write, get_token(args.token), args.cors
)
rasa.add_task(configure_logging)
logger.info("Started http server on port %s" % args.port)
rasa.run(
host="0.0.0.0",
port=args.port,
workers=1,
access_log=logger.isEnabledFor(logging.DEBUG),
)
def app_without_model():
_, nlu_log_file = tempfile.mkstemp(suffix="_rasa_nlu_logs.json")
router = DataRouter()
rasa = create_app(router, logfile=nlu_log_file)
return rasa.test_client
def router(component_builder):
if "TRAVIS_BUILD_DIR" in os.environ:
root_dir = os.environ["TRAVIS_BUILD_DIR"]
else:
root_dir = os.getcwd()
train_models(component_builder,
os.path.join(root_dir, "data/examples/rasa/demo-rasa.json"))
router = DataRouter(os.path.join(root_dir, "test_projects"))
return router
def app(tmpdir_factory):
"""Use IResource interface of Klein to mock Rasa HTTP server.
:param component_builder:
:return:
"""
_, nlu_log_file = tempfile.mkstemp(suffix="_rasa_nlu_logs.json")
router = DataRouter(tmpdir_factory.mktemp("projects").strpath)
rasa = RasaNLU(router, logfile=nlu_log_file, testing=True)
return StubTreq(rasa.app.resource())
def app(tmpdir_factory, trained_nlu_model):
"""Use IResource interface of Klein to mock Rasa HTTP server"""
_, nlu_log_file = tempfile.mkstemp(suffix="_rasa_nlu_logs.json")
temp_path = tmpdir_factory.mktemp("projects")
try:
shutil.copytree(
trained_nlu_model,
os.path.join(
temp_path.strpath,
"{}/{}".format(KEYWORD_PROJECT_NAME, KEYWORD_MODEL_NAME),
),
)
except FileExistsError:
pass
router = DataRouter(temp_path.strpath)
rasa = RasaNLU(router, logfile=nlu_log_file, testing=True)
return StubTreq(rasa.app.resource())
def app(component_builder):
"""Use IResource interface of Klein to mock Rasa HTTP server.
:param component_builder:
:return:
"""
if "TRAVIS_BUILD_DIR" in os.environ:
root_dir = os.environ["TRAVIS_BUILD_DIR"]
else:
root_dir = os.getcwd()
_, nlu_log_file = tempfile.mkstemp(suffix="_rasa_nlu_logs.json")
train_models(component_builder,
os.path.join(root_dir, "data/examples/rasa/demo-rasa.json"))
router = DataRouter(os.path.join(root_dir, "test_projects"))
rasa = RasaNLU(router, logfile=nlu_log_file, testing=True)
return StubTreq(rasa.app.resource())