o di@sdZddlmZddlmZmZddlmZddlm Z m Z m Z m Z e r,ddlm Z mZeddGd d d ZGd d d eZGd ddeZdS)z,Optimum command-line interface base classes.)ABC)ArgumentParserRawTextHelpFormatter) dataclass) TYPE_CHECKINGOptionalTupleType) Namespace_SubParsersActionT)frozenc@sReZdZUeed<eed<dZeeded<eZ eed<e ddZ d d Z dS) CommandInfonamehelpNBaseOptimumCLICommandsubcommand_classformatter_classcCs |jduSN)rselfrc/lsinfo/ai/hellotax_ai/llm_service/venv_embed/lib/python3.10/site-packages/optimum/commands/base.pyis_subcommand_info" zCommandInfo.is_subcommand_infocCs|js td|ddS)NzDThe command info must define a subcommand_class attribute, but got: .)r ValueErrorrrrris_subcommand_info_or_raise&sz'CommandInfo.is_subcommand_info_or_raise) __name__ __module__ __qualname__str__annotations__rrr rrpropertyrrrrrrr s    r c @seZdZUeed<dZeedfed<    ddedd ed d eed ed edf ddZ e ddZ e j dedfddZ e ddZ edddZdefddZddZdS)rCOMMANDr. SUBCOMMANDSNF subparsersr argsr commandfrom_defaults_factoryparserrcs|dur|_|r|durtdd|_|_nE|dur(tdd|jjjjjd_jfdd}jj|dj D]}t |t s[td t |d  |qJ|_dS) a0 Initializes the instance. Args: subparsers (`Optional[_SubParsersAction]`): The parent subparsers this command will create its parser on. args (`Optional[Namespace]`, defaults to `None`): The parsed arguments that are going to be use in self.run(). command (`Optional[CommandInfo]`, defaults to `None`): The command info for this instance. This can be used to set the class attribute `COMMAND`. from_defaults_factory (`bool`, defaults to `False`): When setting the parser defaults, we create a second instance of self. By setting `from_defaults_factory=True`, we do not do unnecessary actions for setting the defaults, such as creating a parser. Nz^The instance of the original parser must be passed when creating a defaults factory, command: rzKA subparsers instance is needed when from_defaults_factory=False, command: )rcsjj|jdjdS)NT)r'r(r)) __class__r%r#r))r&rrrdefaults_factoryYsz8BaseOptimumCLICommand.__init__..defaults_factory)funcz6Subcommands must be instances of CommandInfo, but got z here.)r#rr)r% add_parserrr parse_args set_defaultsr$ isinstancer typeregister_subcommandr&)rr%r&r'r(r)r+ subcommandrrr__init__/s,      zBaseOptimumCLICommand.__init__cCs8t|dd}|dur|jr|j|_|jSd|_|jS)a2 This property handles how subparsers are created, which are only needed when registering a subcommand. If `self` does not have any subcommand, no subparsers should be created or it will mess with the command. This property ensures that we create subparsers only if needed. _subparsersN)getattrr$r)add_subparsersr5rr%rrrr%gs  z BaseOptimumCLICommand.subparserscCs ||_dSr)r5r8rrrr%vrcCst|dsg|_|jS)N_registered_subcommands)hasattrr9rrrrregistered_subcommandszs z,BaseOptimumCLICommand.registered_subcommandscCsdSrr)r)rrrr.sz BaseOptimumCLICommand.parse_args command_infocCs2||j|f|_|j|j|j|ddS)N)r')rr$r;appendrr%)rr<rrrr2sz)BaseOptimumCLICommand.register_subcommandcCs|jdSr)r) print_helprrrrrunszBaseOptimumCLICommand.run)NNFN)r)r)rrrr r!r$rrboolr4r"r%setterr; staticmethodr.r2r?rrrrr+s8  8    rc@s8eZdZedddZd dedeededfd d ZdS) RootOptimumCLICommandrootzoptimum-cli root command)rrNcli_nameusager&r cCs$t||d|_|j|_d|_dS)N)rF)rr)r7r%r&)rrErFr&rrrr4s  zRootOptimumCLICommand.__init__)NN)rrrr r#r rr4rrrrrCs $rCN)__doc__abcrargparserr dataclassesrtypingrrrr r r r rrCrrrrs  b