| 994 | |
| 995 | |
| 996 | def parse_models(subparsers) -> None: |
| 997 | parser_models = subparsers.add_parser( |
| 998 | "models", formatter_class=argparse.RawTextHelpFormatter, help=Help.group_models, aliases=["m"] |
| 999 | ) |
| 1000 | |
| 1001 | subparsers_models = parser_models.add_subparsers(title="commands", dest="command") |
| 1002 | subparsers_models.required = True |
| 1003 | subparsers_models.choices = Help.models_choices |
| 1004 | |
| 1005 | # Models Instances. |
| 1006 | parse_model_instances(subparsers_models) |
| 1007 | |
| 1008 | # Models get |
| 1009 | parser_models_get = subparsers_models.add_parser( |
| 1010 | "get", formatter_class=argparse.RawTextHelpFormatter, help=Help.command_models_get |
| 1011 | ) |
| 1012 | parser_models_get_optional = parser_models_get._action_groups.pop() |
| 1013 | parser_models_get_optional.add_argument("model", help=Help.param_model) |
| 1014 | parser_models_get_optional.add_argument( |
| 1015 | "-p", "--path", dest="folder", required=False, help=Help.param_model_downfile |
| 1016 | ) |
| 1017 | parser_models_get._action_groups.append(parser_models_get_optional) |
| 1018 | parser_models_get.set_defaults(func=api.model_get_cli) |
| 1019 | |
| 1020 | # Models list |
| 1021 | parser_models_list = subparsers_models.add_parser( |
| 1022 | "list", formatter_class=argparse.RawTextHelpFormatter, help=Help.command_models_list |
| 1023 | ) |
| 1024 | parser_models_list_optional = parser_models_list._action_groups.pop() |
| 1025 | parser_models_list.add_argument("--sort-by", dest="sort_by", required=False, help=Help.param_model_sort_by) |
| 1026 | parser_models_list.add_argument("-s", "--search", dest="search", required=False, help=Help.param_search) |
| 1027 | parser_models_list.add_argument("--owner", dest="owner", required=False, help=Help.param_model_owner) |
| 1028 | parser_models_list.add_argument("--page-size", dest="page_size", default=20, type=int, help=Help.param_page_size) |
| 1029 | parser_models_list.add_argument("--page-token", dest="page_token", required=False, help=Help.param_page_token) |
| 1030 | _add_output_format_args(parser_models_list) |
| 1031 | parser_models_list._action_groups.append(parser_models_list_optional) |
| 1032 | parser_models_list.set_defaults(func=api.model_list_cli) |
| 1033 | |
| 1034 | # Models init |
| 1035 | parser_models_init = subparsers_models.add_parser( |
| 1036 | "init", formatter_class=argparse.RawTextHelpFormatter, help=Help.command_models_init |
| 1037 | ) |
| 1038 | parser_models_init_optional = parser_models_init._action_groups.pop() |
| 1039 | parser_models_init_optional.add_argument( |
| 1040 | "-p", "--path", dest="folder", required=False, help=Help.param_model_upfile |
| 1041 | ) |
| 1042 | parser_models_init._action_groups.append(parser_models_init_optional) |
| 1043 | parser_models_init.set_defaults(func=api.model_initialize_cli) |
| 1044 | |
| 1045 | # Models create |
| 1046 | parser_models_create = subparsers_models.add_parser( |
| 1047 | "create", formatter_class=argparse.RawTextHelpFormatter, help=Help.command_models_new |
| 1048 | ) |
| 1049 | parser_models_create_optional = parser_models_create._action_groups.pop() |
| 1050 | parser_models_create_optional.add_argument( |
| 1051 | "-p", "--path", dest="folder", required=False, help=Help.param_model_upfile |
| 1052 | ) |
| 1053 | parser_models_create._action_groups.append(parser_models_create_optional) |