o di @s8ddlZddlmZddlmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZe ZeegZgZdd eeefd d ZdedeeedeeeeffddZdeee eeefeeeffddZde eeefd efddZddZedkredSdS)N)Path)DictListOptionalTupleTypeUnion)load_subpackages)logging)BaseOptimumCLICommand CommandInfoRootOptimumCLICommand)EnvironmentCommand) ExportCommandparent_commandcs2durttstddfdd}|S)a4 A decorator to declare optimum-cli subcommands. The declaration of an optimum-cli subcommand looks like this: ``` @optimum_cli_subcommand() class MySubcommand(BaseOptimumCLICommand): ``` or ``` @optimum_cli_subcommand(ExportCommand) class MySubcommand(BaseOptimumCLICommand): ``` Args: parent_command: (`Optional[Type[BaseOptimumCLICommand]]`): The class of the parent command or None if this is a top-level command. Defaults to None. NzThe parent command , must be a subclass of BaseOptimumCLICommandcs,t|ts td|dt|fdS)NzThe subcommand r) issubclassr ValueError_OPTIMUM_CLI_SUBCOMMANDSappend) subcommandrj/lsinfo/ai/hellotax_ai/llm_service/venv_embed/lib/python3.10/site-packages/optimum/commands/optimum_cli.pywrapperAs z'optimum_cli_subcommand..wrapper)rr r)rrrrroptimum_cli_subcommand$s rrootcommandsreturncCs|g}t|}i}|r*|d}|j|vr ||j|||j<|s#n||j7}|s |rAdd|D}td|dd|d|S)a Retrieves command instances in the root CLI command from a list of command classes. Args: root (`RootOptimumCLICommand`): The root CLI command instance. commands (`List[Type[BaseOptimumCLICommand]]`): The list of command classes to retrieve the instances of in root. Returns: `Dict[Type[BaseOptimumCLICommand], BaseOptimumCLICommand]`: A dictionary mapping a command class to a command instance in the root CLI. rcss|]}|jVqdSN)__name__).0commandrrr esz6resolve_command_to_command_instance..z@Could not find an instance of the following commands in the CLI z: z, .)setpop __class__removeregistered_subcommands RuntimeErrorjoin)rrto_visitremaining_commandscommand2command_instancecurrent_command_instance class_namesrrr#resolve_command_to_command_instanceIs$     r3c Csg}ttjd}|D]e}|s|jdkr'|jdvr&td|dq d|j }t j |dd}t |d g}t |D]3\}}t|trL|\}} n|}d } t|tsjt|tsjtd |d |d t|d||| fq>q |S)a Loads a list of command classes to register to the CLI from the `optimum/commands/register/` directory. It will look in any python file if there is a `REGISTER_COMMANDS` list, and load the commands to register accordingly. At this point, nothing is actually registered in the root CLI. Returns: `List[Tuple[Union[Type[BaseOptimumCLICommand], CommandInfo], Optional[Type[BaseOptimumCLICommand]]]]`: A list of tuples with two elements. The first element corresponds to the command to register, it can either be a subclass of `BaseOptimumCLICommand` or a `CommandInfo`. The second element corresponds to the parent command, where `None` means that the parent command is the root CLI command. registerz.py) __pycache__z README.mdz Skipping zM because only python files are allowed when registering commands dynamically.z .register.zoptimum.commands)packageREGISTER_COMMANDSNzThe command at index z in z is not of the right type: r&)r__file__parentiterdiris_dirsuffixnameloggerwarningstem importlib import_modulegetattr enumerate isinstancetuplerrr rtyper) commands_to_registerregister_dir_pathfilename module_namemodulecommands_to_register_in_file command_idxr$command_or_command_infoparent_command_clsrrr!dynamic_load_commands_in_registerls4        rQrOcCs<t|tst|jj|jj|d}n|}|||dS)a/ Registers a command as being a subcommand of `parent_command`. Args: command_or_command_info (`Union[Type[BaseOptimumCLICommand], CommandInfo]`): The command to register. parent_command (`BaseOptimumCLICommand`): The instance of the parent command. )helpsubcommand_classN)rErCOMMANDr=rRis_subcommand_info_or_raiseregister_subcommand)rOr command_inforrrregister_optimum_cli_subcommands rXc Cstddd}|j}tD]}t||dq ttt}t|dd|D}|D]\}}|dur3|}n||}t||dq(|}t |dsO| t d| |} | dS) NzOptimum CLI toolz optimum-cli)usagercSsg|] \}}|dur|qSr!r)r#_rPrrr szmain..funcr )rparserOPTIMUM_CLI_ROOT_SUBCOMMANDSrXr rrQr3 parse_argshasattr print_helpexitr\run) rr]subcommand_clsrHr0rOrparent_command_instanceargsservicerrrmains(      rh__main__r!) rApathlibrtypingrrrrrr subpackagesr utilsr baser rrenvrexportr get_loggerr>r^rrr3rQrXrhr"rrrrs>      %  #" , "