def add_run_arguments(parser: argparse.ArgumentParser):
from rasa.core.cli.run import add_run_arguments
from rasa.core.cli.arguments import add_logging_option_arguments
add_run_arguments(parser)
add_model_param(parser)
add_logging_option_arguments(parser)
def add_run_arguments(parser: argparse.ArgumentParser):
from rasa.core.cli.run import add_run_arguments
add_run_arguments(parser)
add_model_param(parser)
parser.add_argument(
"--credentials",
type=str,
default="credentials.yml",
help="Authentication credentials for the connector as a yml file")
def add_subparser(
subparsers: argparse._SubParsersAction, parents: List[argparse.ArgumentParser]
):
run_parser = subparsers.add_parser(
"run",
parents=parents,
conflict_handler="resolve",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
help="Start a Rasa server which loads a trained model",
)
add_run_arguments(run_parser)
run_parser.set_defaults(func=run)
run_subparsers = run_parser.add_subparsers()
run_core_parser = run_subparsers.add_parser(
"core",
parents=parents,
conflict_handler="resolve",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
help="Run a trained Core model",
)
add_run_arguments(run_core_parser)
run_core_parser.set_defaults(func=run)
nlu_subparser = run_subparsers.add_parser(
"nlu",
parents=parents,
conflict_handler="resolve",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
help="Run a trained NLU model",
)
_add_nlu_arguments(nlu_subparser)
nlu_subparser.set_defaults(func=run_nlu)
sdk_subparser = run_subparsers.add_parser(
"actions",
parents=parents,
conflict_handler="resolve",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
help="Run the action server",
)
_adk_sdk_arguments(sdk_subparser)
sdk_subparser.set_defaults(func=run_actions)
