def set_x_arguments(parser: argparse.ArgumentParser):
add_model_param(parser, add_positional_arg=False)
add_data_param(parser,
default=DEFAULT_DATA_PATH,
data_type="stories and Rasa NLU ")
parser.add_argument(
"--no-prompt",
action="store_true",
help=
"Automatic yes or default options to prompts and oppressed warnings.",
)
parser.add_argument(
"--production",
action="store_true",
help="Run Rasa X in a production environment.",
)
parser.add_argument(
"--rasa-x-port",
default=DEFAULT_RASA_X_PORT,
type=int,
help="Port to run the Rasa X server at.",
)
add_server_arguments(parser)
def set_interactive_core_arguments(parser: argparse.ArgumentParser) -> None:
add_model_param(parser, model_name="Rasa Core", default=None)
add_stories_param(parser)
_add_common_params(parser)
_add_training_arguments(parser)
add_port_argument(parser)
def set_x_arguments(parser: argparse.ArgumentParser):
add_model_param(parser, add_positional_arg=False)
add_data_param(parser, default=DEFAULT_DATA_PATH, data_type="stories and Rasa NLU ")
parser.add_argument(
"--no-prompt",
action="store_true",
help="Automatic yes or default options to prompts and oppressed warnings.",
)
parser.add_argument(
"--production",
action="store_true",
help="Run Rasa X in a production environment.",
)
parser.add_argument(
"--rasa-x-port",
default=DEFAULT_RASA_X_PORT,
type=int,
help="Port to run the Rasa X server at.",
)
parser.add_argument(
"--config-endpoint",
type=str,
help="Rasa X endpoint URL from which to pull the runtime config. This URL "
"typically contains the Rasa X token for authentication. Example: "
"https://example.com/api/config?token=my_rasa_x_token",
)
add_server_arguments(parser)
def add_run_arguments(parser: argparse.ArgumentParser):
from rasa.cli.arguments.run import add_run_arguments
from rasa.cli.arguments.arguments import add_logging_option_arguments
add_run_arguments(parser)
add_model_param(parser)
add_logging_option_arguments(parser)
def add_subparser(subparsers: argparse._SubParsersAction,
parents: List[argparse.ArgumentParser]):
shell_parser = subparsers.add_parser(
"shell",
parents=parents,
conflict_handler="resolve",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
help="Speak to a trained model on the command line",
)
rasa.cli.run.add_run_arguments(shell_parser)
run_subparsers = shell_parser.add_subparsers()
shell_nlu_subparser = run_subparsers.add_parser(
"nlu",
parents=parents,
conflict_handler="resolve",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
help="Interpret messages on the command line using your NLU model.",
)
shell_nlu_subparser.set_defaults(func=shell_nlu)
add_model_param(shell_nlu_subparser)
rasa.cli.arguments.arguments.add_logging_option_arguments(shell_parser)
rasa.cli.arguments.arguments.add_logging_option_arguments(
shell_nlu_subparser)
shell_parser.set_defaults(func=shell)
def set_test_arguments(parser: argparse.ArgumentParser):
add_model_param(parser, add_positional_arg=False)
core_arguments = parser.add_argument_group("Core Test Arguments")
add_test_core_argument_group(core_arguments)
nlu_arguments = parser.add_argument_group("NLU Test Arguments")
add_test_nlu_argument_group(nlu_arguments)
def set_test_arguments(parser):
add_model_param(parser, add_positional_arg=False)
core_arguments = parser.add_argument_group("Core Test arguments")
add_test_core_arguments(core_arguments)
add_stories_param(core_arguments, "test")
add_url_param(core_arguments)
nlu_arguments = parser.add_argument_group("NLU Test arguments")
add_test_nlu_arguments(nlu_arguments)
def set_interactive_arguments(parser: argparse.ArgumentParser):
add_model_param(parser)
add_config_param(parser)
add_domain_param(parser)
add_data_param(parser)
add_out_param(parser)
add_force_param(parser)
add_skip_visualization_param(parser)
add_server_arguments(parser)
def set_interactive_core_arguments(parser: argparse.ArgumentParser):
add_model_param(parser)
add_config_param(parser)
add_domain_param(parser)
add_stories_param(parser)
add_out_param(parser)
add_augmentation_param(parser)
add_debug_plots_param(parser)
add_dump_stories_param(parser)
add_skip_visualization_param(parser)
add_server_arguments(parser)
def set_interactive_arguments(parser: argparse.ArgumentParser) -> None:
parser.add_argument(
"--e2e",
action="store_true",
help="Save story files in e2e format. In this format user messages will be included in the stories.",
)
add_model_param(parser, default=None)
add_data_param(parser)
_add_common_params(parser)
train_arguments = _add_training_arguments(parser)
add_force_param(train_arguments)
add_persist_nlu_data_param(train_arguments)
def set_test_arguments(parser: argparse.ArgumentParser) -> None:
add_model_param(parser, add_positional_arg=False)
core_arguments = parser.add_argument_group("Core Test Arguments")
add_test_core_argument_group(core_arguments)
nlu_arguments = parser.add_argument_group("NLU Test Arguments")
add_test_nlu_argument_group(nlu_arguments)
add_no_plot_param(parser)
add_errors_success_params(parser)
add_out_param(
parser,
default=DEFAULT_RESULTS_PATH,
help_text="Output path for any files created during the evaluation.",
)
def set_x_arguments(shell_parser: argparse.ArgumentParser):
add_model_param(shell_parser, add_positional_arg=False)
shell_parser.add_argument(
"--no-prompt",
action="store_true",
help="Automatic yes or default options to prompts and oppressed warnings.",
)
shell_parser.add_argument(
"--production",
action="store_true",
help="Run Rasa X in a production environment.",
)
shell_parser.add_argument(
"--nlg",
type=str,
default="http://localhost:5002/api/nlg",
help="Rasa NLG endpoint.",
)
shell_parser.add_argument(
"--model-endpoint-url",
type=str,
default="http://localhost:5002/api/projects/default/models/tags/production",
help="Rasa model endpoint URL.",
)
shell_parser.add_argument(
"--project-path",
type=str,
default=".",
help="Path to the Rasa project directory.",
)
shell_parser.add_argument(
"--data-path",
type=str,
default="data",
help=(
"Path to the directory containing Rasa NLU training data "
"and Rasa Core stories."
),
)
add_server_arguments(shell_parser)
def set_interactive_core_arguments(parser: argparse.ArgumentParser):
add_model_param(parser, model_name="Rasa Core", default=None)
add_stories_param(parser)
add_skip_visualization_param(parser)
add_endpoint_param(
parser,
help_text=
"Configuration file for the model server and the connectors as a yml file.",
)
train_arguments = parser.add_argument_group("Train Arguments")
add_config_param(train_arguments)
add_domain_param(train_arguments)
add_out_param(train_arguments)
add_augmentation_param(train_arguments)
add_debug_plots_param(train_arguments)
add_dump_stories_param(train_arguments)
def set_x_arguments(parser: argparse.ArgumentParser):
add_model_param(parser, add_positional_arg=False)
parser.add_argument(
"--no-prompt",
action="store_true",
help=
"Automatic yes or default options to prompts and oppressed warnings.",
)
parser.add_argument(
"--production",
action="store_true",
help="Run Rasa X in a production environment.",
)
parser.add_argument(
"--nlg",
type=str,
default="http://localhost:5002/api/nlg",
help="Rasa NLG endpoint.",
)
parser.add_argument(
"--model-endpoint-url",
type=str,
default=
"http://localhost:5002/api/projects/default/models/tags/production",
help="Rasa model endpoint URL.",
)
parser.add_argument(
"--project-path",
type=str,
default=".",
help="Path to the Rasa project directory.",
)
add_data_param(parser,
default=DEFAULT_DATA_PATH,
data_type="stories and Rasa NLU ")
add_server_arguments(parser)
def set_interactive_arguments(parser: argparse.ArgumentParser):
add_model_param(parser, default=None)
add_data_param(parser)
add_skip_visualization_param(parser)
add_endpoint_param(
parser,
help_text=
"Configuration file for the model server and the connectors as a yml file.",
)
train_arguments = parser.add_argument_group("Train Arguments")
add_config_param(train_arguments)
add_domain_param(train_arguments)
add_out_param(train_arguments,
help_text="Directory where your models should be stored.")
add_augmentation_param(train_arguments)
add_debug_plots_param(train_arguments)
add_dump_stories_param(train_arguments)
add_force_param(train_arguments)
def set_test_nlu_arguments(parser: argparse.ArgumentParser):
add_model_param(parser, add_positional_arg=False)
add_test_nlu_argument_group(parser)
def set_run_arguments(parser: argparse.ArgumentParser) -> None:
"""Arguments for running Rasa directly using `rasa run`."""
add_model_param(parser)
add_server_arguments(parser)
def set_run_arguments(parser: argparse.ArgumentParser):
add_model_param(parser)
add_server_arguments(parser)
def set_shell_nlu_arguments(parser: argparse.ArgumentParser):
add_model_param(parser)
def set_shell_arguments(parser: argparse.ArgumentParser) -> None:
add_model_param(parser)
add_server_arguments(parser)
