def test_create_single_input_channels_by_class_wo_credentials():
channels = run.create_http_input_channels(
"rasa.core.channels.rest.RestInput", credentials_file=None
)
assert len(channels) == 1
assert channels[0].name() == "rest"
async def serve_application(core_model: Text = None,
nlu_model: Optional[Text] = None,
tracker_dump: Optional[Text] = None,
port=constants.DEFAULT_SERVER_PORT,
enable_api=True,
endpoints=None
):
input_channels = run.create_http_input_channels("cmdline", None)
app = run.configure_app(input_channels, enable_api=enable_api, port=port)
logger.info("Starting Rasa Core server on "
"{}".format(constants.DEFAULT_SERVER_FORMAT.format(port)))
# noinspection PyShadowingNames
async def load_agent_and_tracker(app, loop):
agent = await run.load_agent_on_start(core_model, endpoints,
nlu_model, app, loop)
tracker = load_tracker_from_json(tracker_dump,
agent.domain)
await replay_events(tracker, agent)
app.register_listener(load_agent_and_tracker, 'before_server_start')
app.run(host='0.0.0.0', port=port,
access_log=logger.isEnabledFor(logging.DEBUG))
def test_create_http_input_channels():
channels = run.create_http_input_channels(None, CREDENTIALS_FILE)
assert len(channels) == 7
# ensure correct order
assert {c.name()
for c in channels} == {
"twilio", "slack", "telegram", "mattermost", "facebook",
"webexteams", "rocketchat"
}
def test_create_single_input_channels_by_class():
channels = run.create_http_input_channels(
"rasa.core.channels.channel.RestInput", CREDENTIALS_FILE)
assert len(channels) == 1
assert channels[0].name() == "rest"
def test_create_single_input_channels():
channels = run.create_http_input_channels("facebook", CREDENTIALS_FILE)
assert len(channels) == 1
assert channels[0].name() == "facebook"
def serve_application(
model_path: Optional[Text] = None,
channel: Optional[Text] = None,
port: int = constants.DEFAULT_SERVER_PORT,
credentials: Optional[Text] = None,
cors: Optional[Union[Text, List[Text]]] = None,
auth_token: Optional[Text] = None,
enable_api: bool = True,
response_timeout: int = constants.DEFAULT_RESPONSE_TIMEOUT,
jwt_secret: Optional[Text] = None,
jwt_method: Optional[Text] = None,
endpoints: Optional[AvailableEndpoints] = None,
remote_storage: Optional[Text] = None,
log_file: Optional[Text] = None,
ssl_certificate: Optional[Text] = None,
ssl_keyfile: Optional[Text] = None,
ssl_ca_file: Optional[Text] = None,
ssl_password: Optional[Text] = None,
conversation_id: Optional[Text] = uuid.uuid4().hex,
):
"""Run the API entrypoint."""
from rasa import server
if not channel and not credentials:
channel = "cmdline"
input_channels = create_http_input_channels(channel, credentials)
app = configure_app(
input_channels,
cors,
auth_token,
enable_api,
response_timeout,
jwt_secret,
jwt_method,
port=port,
endpoints=endpoints,
log_file=log_file,
conversation_id=conversation_id,
)
ssl_context = server.create_ssl_context(ssl_certificate, ssl_keyfile,
ssl_ca_file, ssl_password)
protocol = "https" if ssl_context else "http"
logger.info("Starting Rasa server on "
"{}".format(
constants.DEFAULT_SERVER_FORMAT.format(protocol, port)))
app.register_listener(
partial(load_agent_on_start, model_path, endpoints, remote_storage),
"before_server_start",
)
# noinspection PyUnresolvedReferences
async def clear_model_files(_app: Sanic, _loop: Text) -> None:
if app.agent.model_directory:
shutil.rmtree(_app.agent.model_directory)
app.register_listener(clear_model_files, "after_server_stop")
rasa.utils.common.update_sanic_log_level(log_file)
# 注册自己的蓝图
app.register_blueprint(api)
app.run(
host="0.0.0.0",
port=port,
ssl=ssl_context,
backlog=int(os.environ.get(ENV_SANIC_BACKLOG, "100")),
workers=rasa.core.utils.number_of_sanic_workers(
endpoints.lock_store if endpoints else None),
)