o 0io@sdZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZmZddlZddlZddlmZmZddlmZmZmZm Z m!Z!m"Z"ddl#m$Z$e!%Z&d Z'ertdd l(m)Z)dyd ee*d d fddZ+dZ,dyde-e*dee*d e*fddZ.eede*fZ/ee-e*e0e*gee1fZ2dee*dej3dej4d dfddZ5e6dZ7Gdddej8j9Z: dyde2deege-e;e*e*ffd efd#d$Z?Gd%d&d&ej@ZA dzd'e*dee*d(eed.fZQeee*ejFd?d@d.fZReee*ejFdAd.fZSd eTe*e*ffdBdCZU  dzdDee-e*dEee*d eTe*ee*ffdFdGZVdHeTe*ee*fd ee-eTe*e*ffdIdJZWGdKdLdLe*e ZXeeXejFdMd.fZYeeIejFdNdOdPd.fZZdQe*d e*fdRdSZ[dTed e*fdUdVZ\e'fdTedWe1d e*fdXdYZ] dydZeeTe*efd[e-e*d\eeTe*efge-e*fd]eeTe*e*fd df d^d_Z^ `   d{dZeeTe*efdaeXdbeIdce*d[ee-e*d\eeeTe*efge-e*fd]eeTe*e*fd dfdddeZ_dfe`d e`fdgdhZadied eTe*effdjdkZbdle-e*fdmdnZcdoedpd dfdqdrZddoedpd dfdsdtZed e*fdudvZfd e*fdwdxZgdS)|z*Contains CLI utilities (styling, helpers).N)Enum)Path) TYPE_CHECKING AnnotatedAnyCallableLiteralOptionalSequenceUnioncast) __version__ constants)ANSI get_sessionhf_raise_for_statusinstallation_methodloggingtabulate) load_dotenv#HfApitokenreturnrcCsddlm}||dtdS)Nrrzhuggingface-cli)rZ library_nameZlibrary_version)huggingface_hub.hf_apirr )rrrX/home/kim/smarthome/.venv/lib/python3.10/site-packages/huggingface_hub/cli/_cli_utils.py get_hf_api,s rz9https://huggingface.co/docs/huggingface_hub/en/guides/cliexamples docs_anchorcCs:|rt|nt}ddd|D}d|d|dS)a Generate an epilog with examples and a Learn More section. Args: examples: List of example commands (without the `$ ` prefix). docs_anchor: Optional anchor for the docs URL (e.g., "#hf-download"). Returns: Formatted epilog string.  css|]}d|VqdS)z $ Nr).0exrrr Csz"generate_epilog..z Examples zk Learn more Use `hf --help` for more information about a command. Read the documentation at )CLI_REFERENCE_URLjoin)rr Zdocs_urlZ examples_strrrrgenerate_epilog8s r')mainhelpepilogctx formattercCs.|r||dD] }||q dSdS)z%Write the epilog without indentation.r!N)Zwrite_paragraphsplit write_text)r*r+r,linerrr_format_epilog_no_indentRs  r0z\s*\|\s*cseZdZdZdejdeedeffdd Z e dej deeddfd d Z e dej deeddfd d Z dejd edeej ffdd ZdeeeeffddZdejdejddfddZdejdejddfddZdejdeefddZZS)HFCliTyperGroupa Typer Group that: - lists commands alphabetically within sections. - separates commands by topic (main, help, etc.). - formats epilog without extra indentation. - supports aliases via pipe-separated names (e.g. ``name="list | ls"``). - rewrites ``--json`` to ``--format json`` for commands that accept ``--format``. - rewrites ``spaces/user/repo`` to ``user/repo --type space`` for commands that accept ``--type``. r+argsrcs`|r |dds |dnd}|r|||nd}|dur)||||||t||S)Nr-) startswith get_command_rewrite_json_shorthand_rewrite_repo_type_prefixsuperresolve_command)selfr+r2cmd_namecmd __class__rrr9hs   zHFCliTyperGroup.resolve_commandr<NcCsfd|vrdStdd|jD}|r1tdd|Dr td|d}ddg|||d <dSdS) a2Rewrite hidden ``--json`` shorthand to ``--format json``. Only applies to commands that accept ``--format``. This avoids rewriting ``--json`` for commands that pass args through to external binaries (e.g. ``hf extensions exec``) or that simply don't support ``--format``. z--jsonNcs&|]}t|tjod|jvVqdS)--formatN isinstanceclickOptionoptsr"paramrrrr$|$z:HFCliTyperGroup._rewrite_json_shorthand..css"|] }|dkp |dVqdS)r@z --format=Nr4r"argrrrr$~s z/'--json' and '--format' are mutually exclusive.r@json)anyparamsrC UsageErrorindex)r<r2Zhas_format_optionidxrrrr6rs  z'HFCliTyperGroup._rewrite_json_shorthandcCs2tdd|jD}|sdSt}d}|jD]}t|tjr,|jdvr(|||d7}q|s1dSt}|jD]}t|tjrS|j sSg|j |j RD]}||qKq7d}g} d} | t |kr|| } | dkrin.| drd | vsv| |vr{| d7} n| d 7} n||vr| | |d7}| d7} | t |ks`| sdSd} d} g}| D]I}||d d }t |d ks|dtjvrq|d}tj|}| dur|| krtd | d|d|} |} |||dd |d fq|sdStdd|Drtd| d|D] \}}|||<q|d| gdS)uRewrite prefixed repo IDs (e.g. ``spaces/user/repo``) to ``user/repo --type space``. Only applies to commands that have a ``--type`` / ``--repo-type`` option and at least one repo-ID positional argument (any ``click.Argument`` whose name ends with ``_id``, e.g. ``repo_id``, ``from_id``, ``to_id``). When the token that maps to such an argument matches ``{prefix}/org/repo`` (where *prefix* is one of ``spaces``, ``datasets``, or ``models``), the prefix is stripped and an implicit ``--type {type}`` is appended. An error is raised if ``--type`` is also provided explicitly or if multiple prefixed arguments disagree on the repo type. Only repo-ID positional slots are inspected so that other positional arguments (filenames, local paths, patterns …) are never misinterpreted as prefixed repo IDs. csr?)--typeNrArFrrrr$rHz.Nr)Zrepo_idZfrom_idZto_idrMz--r3=/z!Conflicting repo type prefixes: 'z/' and 'z/'.css4|]}|dkp|dp|dkp|dVqdS)rSz--type= --repo-typez --repo-type=NrIrJrrrr$s " z!Ambiguous repo type: got prefix 'z8/' in repo ID and explicit --type. Use one or the other.rS)rNrOsetrBrCArgumentnameaddrDZis_flagrEZsecondary_optslenr4appendr-rZREPO_TYPES_MAPPINGrPextend)r<r2Zhas_type_optionZrepo_id_positionsZarg_idxrGZ value_optionsoptZpositional_countZrepo_id_arg_indicesirKZ inferred_typeZ first_prefixZrewritesZ arg_indexpartsprefixZ mapped_type new_valuerrrr7s|             "    z)HFCliTyperGroup._rewrite_repo_type_prefixr;csLt||}|dur |S|jD]\}}t|}||vr#|SqdSN)r8r5commandsitems _ALIAS_SPLITr-)r:r+r;r<registered_nameZregistered_cmdaliasesr=rrr5s zHFCliTyperGroup.get_commandcCs6i}|jD]}t|}|d}|dd||<q|S)zBBuild a mapping from primary command name to its aliases (if any).rrMN)rfrhr-)r:resultrirbprimaryrrr _alias_maps   zHFCliTyperGroup._alias_mapr,c Cs(i}|}||D];}|||}|dus|jrq |j|jd}||g}|r5|dd|d}t|dd} | | g ||fq | d| |dWdn1s^wYt |D](} | dkrpqi| | d| || Wdn1swYqidS) N)limitz [alias: , ]topicr(z Main commandsz commands)rm list_commandsr5hiddenZget_short_help_strwidthgetr&getattr setdefaultr^sectionwrite_dlsortedkeys capitalize) r:r+r,ZtopicsZ alias_mapr[r< help_textrjrqrrrformat_commandss.    zHFCliTyperGroup.format_commandscCsg}||D]}|||}|dus|jrqt|dg}|r&||dq|r5t|}t|||dS|jrAt|j||dSdS)Nrr)rrr5rsrvr^r'r0r*)r:r+r,Z all_examplesr[r<Z cmd_examplesr*rrr format_epilogs  zHFCliTyperGroup.format_epilogcCs0g}|jD]}t|d}||qt|S)Nr)rfrhr-r^rz)r:r+Z primary_namesr[rlrrrrr#s   zHFCliTyperGroup.list_commands)__name__ __module__ __qualname____doc__rCContextliststrtupler9 staticmethodCommandr6r7r r5dictrm HelpFormatterr~rrr __classcell__rrr=rr1]s  "`   r1fallback_handlerextra_commands_providercsGfdddt}|S)zRReturn a Typer group class that runs a fallback handler before command resolution.csReZdZdejdeedeffdd Zdejdej ddffdd Z Z S) z8fallback_typer_group_factory..FallbackTyperGroupr+r2rcs2|t|j}|durt|t||Sre)rYrfr{ SystemExitr8r9)r:r+r2Zfallback_exit_code)r>rrrr93szHfallback_typer_group_factory..FallbackTyperGroup.resolve_commandr,Ncsbt||dur-}|r/|d||WddS1s&wYdSdSdS)NzExtension commands)r8r~rxry)r:r+r,entries)r>rrrr~9s  "zHfallback_typer_group_factory..FallbackTyperGroup.format_commands) rrrrCrrrrr9rr~rrrrr=rFallbackTyperGroup2s"(r)r1)rrrrrrfallback_typer_group_factory,srrqcCsHdtjdtjdtjddfdd}td|tjjf||pg|dS) Nr:r+r,rcSst|j||dSre)r0r*)r:r+r,rrrrEz#HFCliCommand..format_epilog TyperCommandrqrr) rCrrrtyper|typercorerrrrr HFCliCommandDs  rc"seZdZdZ dddddddddddddd deed ed eeed eeee fd eed eedeedede de de de deede e de fge de ffffddZ Z S)HFCliAppz&Custom Typer app for Hugging Face CLI.Nr(z [OPTIONS]TF) rqrcontext_settingsr)r* short_helpoptions_metavaradd_help_optionno_args_is_helprs deprecatedrich_help_panelr[rqrrr)r*rrrrrsrrr.c sZdur r tdtdtfdtdtff fdd }|S)Nfunc.rcs2tt jt   d |S)N) clsrr)r*rrrrrsrr)r8rcommandr)r)r>rrrr*rr)rsr[rrrr:rrqrr_innergs  z HFCliApp.command.._inner)r'rr)r:r[rqrrr)r*rrrrrsrrrr=)rrrr*rr)rsr[rrrr:rrqrrRs BzHFCliApp.commandre)rrrrr rTOPIC_Trrrboolrrrrrr=rrOsZ      rr)rc Cs0|durt}t||dd|ddddddgdd S) a+Create a Typer app with consistent settings. Args: help: Help text for the app. epilog: Optional epilog text (use `generate_epilog` to create one). cls: Optional Click group class to use (defaults to `HFCliTyperGroup`). Returns: A configured Typer app. NTFxz-hz--help)Zmax_content_widthZhelp_option_names) r)r*Zadd_completionrrZrich_markup_moderZpretty_exceptions_enabler)r1r)r)r*rrrr typer_factoryzs rc@eZdZdZdZdZdS)RepoTypemodeldatasetspaceN)rrrrrrrrrrrsrzNThe ID of the repo (e.g. `username/repo-name` or `spaces/username/repo-name`).)r)rSrXz2The type of repository (model, dataset, or space).zJA User Access Token generated from https://huggingface.co/settings/tokens.zfWhether to create a private repo if repo doesn't exist on the Hub. Ignored if the repo already exists.zDGit revision id which can be a branch name, a tag, or a commit hash.zLimit the number of results.z!Filter by author or organization.zHFilter by tags (e.g. 'text-classification'). Can be used multiple times.z Search query.z-ez--envz/Set environment variables. E.g. --env ENV=valuez-sz --secretszvSet secret environment variables. E.g. --secrets SECRET=value or `--secrets HF_TOKEN` to pass your Hugging Face token.z --env-filez(Read in a file of environment variables.z/Read in a file of secret environment variables.cCs0ddlm}tj}|}dur||d<|S)zQReturn a copy of ``os.environ`` with the user's HF token injected (if available).r) get_tokenNZHF_TOKEN)huggingface_hubrosenvironcopy)rextended_environrrrr_get_extended_environs  renvenv_filecCsLt}i}|r|tt||d|pgD] }|t||dq|S)zParse ``-e``/``--env``/``-s``/``--secrets`` and ``--env-file``/``--secrets-file`` CLI args into a dict. Uses an extended environment that includes the user's HF token so that bare ``--secrets HF_TOKEN`` resolves correctly. )r)rupdaterr read_text)rrrenv_mapZ env_valuerrr parse_env_maps  rrcCs|sdSdd|DS)z_Convert an env/secrets dict to the ``[{"key": ..., "value": ...}]`` format used by the Hub API.NcSsg|] \}}||p ddqS))keyvaluerr"kvrrr $sz-env_map_to_key_value_list..rg)rrrrenv_map_to_key_value_list src@r) OutputFormatz$Output format for CLI list commands.tablerLN)rrrrrrLrrrrr'srzOutput format (table or json).z-qz--quietzPrint only IDs (one per line).r[cCstdd|}|S)z\Convert a camelCase or PascalCase string to SCREAMING_SNAKE_CASE to be used as table header.z([a-z])([A-Z])z\1_\2)resubupper)r[srrr _to_header?srrcCs|sdSt|tr|r dSdSt|tjr|dSt|tr+td|r+|ddSt|tr:ddd |DSt|t rNd |vrIt|d St |St|S) z/Convert a value to string for terminal display.ru✔z%Y-%m-%dz^\d{4}-\d{2}-\d{2}TN rocss|]}t|VqdSre) _format_value)r"rrrrr$Pz _format_value..r[) rBrdatetimestrftimerrmatchrr&rrLdumps)rrrrrEs         rmax_lencCs,t|}t||kr|d|dd}|S)z/Format a value + truncate it for table display.NrWz...)rr])rrcellrrr _format_cellYs rrgheadersrow_fn alignmentscsr|stddSttttttffdd|D}dd|D}dd|p)iD}tt|||ddS) aiPrint items as a formatted table. Args: items: Sequence of dictionaries representing the items to display. headers: List of column headers. row_fn: Function that takes an item dict and returns a list of string values for each column. alignments: Optional mapping of header name to "left" or "right". Defaults to "left". zNo results found.Ncsg|]}|qSrrr"itemrrrrrz"print_as_table..cSg|]}t|qSrr)r"hrrrrsrcSsi|] \}}t||qSrrrrrr uz"print_as_table..)rr)printr rr rintrgr)rgrrrrowsZscreaming_headersZscreaming_alignmentsrrrprint_as_tableas(ridformatquietid_keyc s|rD]}t||qdS|tjkr"ttjtdtddSdur<r0tdn|g}fdd|D|durQdtttfdttffd d }t ||d dS) a`Print list command output in the specified format. Args: items: Sequence of dictionaries representing the items to display. format: Output format (table or json). quiet: If True, print only IDs (one per line). id_key: Key to use for extracting IDs in quiet mode. headers: Optional list of column names for headers. If not provided, auto-detected from keys. row_fn: Optional function to extract row values. If not provided, uses _format_cell on each column. alignments: Optional mapping of header name to "left" or "right". Defaults to "left". NrU)indentdefaultrcs&g|]tfddDrqS)c3s|] }t|VqdSrerrurcolrrr$sz/print_list_output...)rN)r"rrrrs&z%print_list_output..rrcsfddDS)Ncsg|] }t|qSrr)r"rrrrrrz5print_list_output..row_fn..rr)rrrrrz!print_list_output..row_fn)rrr) rrrLrrrr{rrr) rgrrrrrrrZ all_columnsr)rrgrprint_list_outputys "rrcCsLt|tjr |St|trdd|DSt|tr$dd|DS|S)z4Recursively serialize a value to be JSON-compatible.cS"i|] \}}|dur|t|qSre_serialize_value)r"rvalrrrr"z$_serialize_value..cSrrrrrrrrrz$_serialize_value..)rBr isoformatrrgr)rrrrrs   rinfocCsddt|DS)z9Convert repo info dataclasses to json-serializable dicts.cSrrerrrrrrrz&api_object_to_dict..) dataclassesasdictrg)rrrrapi_object_to_dictsrvalid_propertiescs&dttdtttffdd }|S)zJCreate a callback to parse and validate comma-separated expand properties.rrc sR|durdSdd|dD}|D]}|vr&td|ddq|S)NcSsg|]}|qSr)strip)r"prrrrrzSmake_expand_properties_parser.._parse_expand_properties..,zInvalid expand property: 'z'. Valid values are: ro)r-rZ BadParameterr&)r propertiesproprrr_parse_expand_propertiessz?make_expand_properties_parser.._parse_expand_properties)r rr)rrrrrmake_expand_properties_parsers" rlibrary)rZ transformerscCs2zt|WdStytjdddYdSw)aB Check whether a newer version of a library is available on PyPI. If a newer version is found, notify the user and suggest updating. If current version is a pre-release (e.g. `1.0.0.rc1`), or a dev version (e.g. `1.0.0.dev1`), no check is performed. This function is called at the entry point of the CLI. It only performs the check once every 24 hours, and any error during the check is caught and logged, to avoid breaking the CLI. Args: library: The library to check for updates. Currently supports "huggingface_hub" and "transformers". z$Error while checking for CLI update.T)exc_infoN)_check_cli_update Exceptionloggerdebug)rrrrcheck_cli_updates   r cstj|tfdddDrdStjtjr+tj tj}t |dkr+dSt tjj j dddt tjtjd|dd d }t||}|d d }|kr|d krct}nt}ttd|d|ddt|d dSdS)Nc3s|]}|vVqdSrer)r"tagcurrent_versionrrr$rz$_check_cli_update..)rcdeviQT)parentsexist_okzhttps://pypi.org/pypi/z/jsonrU)timeoutrversionrzA new version of z (z&) is available! You are using version z. To update, run: r!) importlibmetadatarrNrpathexistsrZCHECK_FOR_UPDATE_DONE_PATHgetmtimetimerparentmkdirtouchrrurrL#_get_huggingface_hub_update_command _get_transformers_update_commandrCZechoryellowbold)rmtimeresponsedataZlatest_versionZupdate_commandrr rrs6  rcCs8t}|dkr dS|dkrtjdkrdS|dkrdSdS)z-Return the command to update huggingface_hub.Zbrewzbrew upgrade hf hf_installerntzNpowershell -NoProfile -Command "iwr -useb https://hf.co/cli/install.ps1 | iex"z0curl -LsSf https://hf.co/cli/install.sh | bash -zpip install -U huggingface_hubrrr[methodrrrrsrcCs,t}|dkrtjdkrdS|dkrdSdS)z*Return the command to update transformers.r%r&z`powershell -NoProfile -Command "iwr -useb https://hf.co/cli/install.ps1 | iex" -WithTransformerszHcurl -LsSf https://hf.co/cli/install.sh | bash -s -- --with-transformerszpip install -U transformersr'r(rrrrs rre)NN)rNNN)hrrrimportlib.metadatarrLrrrenumrpathlibrtypingrrrrrr r r r rCrrr rZhuggingface_hub.utilsrrrrrrZhuggingface_hub.utils._dotenvrZ get_loggerr Z_MAX_CELL_LENGTHrrrrr%rr'rrYrZFallbackHandlerTrrr0compilerhrZ TyperGroupr1rrrrrZTyperrrrrZZ RepoIdArgrDZ RepoTypeOptZTokenOptrZ PrivateOptZ RevisionOptZLimitOptZ AuthorOptZ FilterOptZ SearchOptZEnvOptZ SecretsOptZ EnvFileOptZSecretsFileOptrrrrrZ FormatOptZQuietOptrrrrrobjectrrrr rrrrrrrs   ,    " R ( , !            .     ) '