o Ďi6B@s8dZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZmZddlZddlZddlmZmZddlmZmZmZm Z m!Z!m"Z"e!#Z$dZ%erndd l&m'Z'd]d ee(d d fd dZ)dZ*d]de+e(dee(d e(fddZ,eede(fZ-dee(dej.dej/d dfddZ0Gdddej1j2Z3d]de-dee+e(d e4ej1j5fddZ6Gdd d ej7Z8d]d!e(dee(d d fd"d#Z9Gd$d%d%e(e Z:ee(ej;d&d'fZeee(ej=d)d'fZ?eee@ej=d*d'fZAeee(ej=d+d'fZBeeCej=d,d'fZDeee(ej=d-d'fZEeee+e(ej=d.d'fZFeee(ej=d/d'fZGGd0d1d1e(e ZHeeHej=d2d'fZIee@ej=d3d4d5d'fZJd6e(d e(fd7d8ZKd9ed e(fd:d;ZLe%fd9edZMd?eeNe(efd@e+e(dAeeNe(efge+e(fd dfdBdCZO D  d^d?eeNe(efdEeHdFe@dGe(d@ee+e(dAeeeNe(efge+e(fd dfdHdIZPdJeQd eQfdKdLZRdMed eNe(effdNdOZSdPe+e(fdQdRZTdSedTd dfdUdVZUdSedTd dfdWdXZVd e(fdYdZZWd e(fd[d\ZXdS)_z*Contains CLI utilities (styling, helpers).N)Enum)Path) TYPE_CHECKING AnnotatedAnyCallableLiteralOptionalSequenceUnioncast) __version__ constants)ANSI get_sessionhf_raise_for_statusinstallation_methodloggingtabulate#HfApitokenreturnrcCsddlm}||dtdS)Nrrzhuggingface-cli)r library_namelibrary_version)huggingface_hub.hf_apirr )rrrZ/home/jeff/fluffinator/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 Bsz"generate_epilog..z Examples zk Learn more Use `hf --help` for more information about a command. Read the documentation at )CLI_REFERENCE_URLjoin)r r!docs_url examples_strrrrgenerate_epilog7s r*)mainhelpepilogctx formattercCs.|r||dD] }||q dSdS)z%Write the epilog without indentation.r"N)write_paragraphsplit write_text)r-r.r/linerrr_format_epilog_no_indentPs  r4c@s\eZdZdZdejdejddfddZdejdejddfdd Zdejde e fd d Z dS) HFCliTyperGroupz Typer Group that: - lists commands alphabetically within sections. - separates commands by topic (main, help, etc.). - formats epilog without extra indentation. r.r/rNc Csi}||D](}|||}|dus|jrq|j|jd}t|dd}||g||fq|d| |dWdn1sGwYt | D](}|dkrYqR|| d| ||Wdn1suwYqRdS)N)limittopicr+z Main commandsz commands) list_commands get_commandhiddenget_short_help_strwidthgetattr setdefaultappendsectionwrite_dlsortedkeys capitalize)selfr.r/topicsnamecmd help_textr7rrrformat_commands`s&   zHFCliTyperGroup.format_commandscCs~g}||D]}|||}|dus|jrqt|dg}||q|r1t|}t|||dS|jr=t|j||dSdS)Nr )r8r9r:r=extendr*r4r-)rEr.r/ all_examplesrGrH cmd_examplesr-rrr format_epilogss   zHFCliTyperGroup.format_epilogcCst|jSN)rBcommandsrC)rEr.rrrr8szHFCliTyperGroup.list_commands) __name__ __module__ __qualname____doc__clickContext HelpFormatterrJrNliststrr8rrrrr5Xs r5r7cCsHdtjdtjdtjddfdd}td|tjjf||pg|dS) NrEr.r/rcSst|j||dSrO)r4r-)rEr.r/rrrrNz#HFCliCommand..format_epilog TyperCommandr7r rN) rUCommandrVrWtyperDtypercorer[r\rrr HFCliCommands  rac"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) r7r context_settingsr,r- short_helpoptions_metavaradd_help_optionno_args_is_helpr: deprecatedrich_help_panelrGr7r rcr,r-rdrerfrgr:rhrir.c sZdur r tdtdtfdtdtff fdd }|S)Nfunc.rcs2tt jt   d |S)N) clsrcr,r-rdrerfrgr:rhri)superrbcommandra)rj) __class__rfrcrhr-r r,r:rGrgrerirErdr7rr_inners  z HFCliApp.command.._inner)r*rr)rErGr7r rcr,r-rdrerfrgr:rhrirorn)rfrcrhr-r r,r:rGrgrerirErdr7rrms BzHFCliApp.commandrO)rQrRrSrTr rYTOPIC_TrXdictrboolrrm __classcell__rrrprrbsZ      rbr,c Cst||ddtdddddid S)zCreate a Typer app with consistent settings. Args: help: Help text for the app. epilog: Optional epilog text (use `generate_epilog` to create one). Returns: A configured Typer app. TNFmax_content_widthx) r,r-add_completionrgrkrich_markup_moderipretty_exceptions_enablerc)rbr5)r,r-rrr typer_factorys rzc@eZdZdZdZdZdS)RepoTypemodeldatasetspaceN)rQrRrSr}r~rrrrrr|sr|z/The ID of the repo (e.g. `username/repo-name`).)r,z2The 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.c@r{) OutputFormatz$Output format for CLI list commands.tablejsonN)rQrRrSrTrrrrrrrsrzOutput format (table or json).z-qz--quietzPrint only IDs (one per line).rGcCstdd|}|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)rGsrrr _to_header1srvaluecCs|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.u✔z%Y-%m-%dz^\d{4}-\d{2}-\d{2}TN , css|]}t|VqdSrO) _format_value)r#vrrrr%Bz _format_value..rG) isinstancersdatetimestrftimerYrmatchrXr'rrrdumps)rrrrr7s         rmax_lencCs,t|}t||kr|d|dd}|S)z/Format a value + truncate it for table display.Nz...)rlen)rrcellrrr _format_cellKs ritemsheadersrow_fncsV|stddSttttttffdd|D}tt|dd|DddS)a Print 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. zNo results found.Ncsg|]}|qSrrr#itemrrr bz"print_as_table..cSg|]}t|qSr)r)r#hrrrrcrr)printr rXr rYintr)rrrrowsrrrprint_as_tableSs  (ridformatquietid_keycs|rD]}t||qdS|tjkr!ttjtdddSdur;r/tdn|g}fdd|D|durPdtttfdttffd d }t |d dS) aPrint 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. N)indentrcs&g|]tfddDrqS)c3s|] }t|VqdSrOrgetrcolrrr%sz/print_list_output...)any)r#)rrrrs&z%print_list_output..rrcsfddDS)Ncsg|] }t|qSrr)r#rrrrrsz5print_list_output..row_fn..rrrrrrrZz!print_list_output..row_fn)rr) rrrrrXrCrrrYrr)rrrrrrr all_columnsr)rrrprint_list_outputfs "rrcCsLt|tjr |St|trdd|DSt|tr$dd|DS|S)z4Recursively serialize a value to be JSON-compatible.cS"i|] \}}|dur|t|qSrO_serialize_value)r#keyvalrrr "z$_serialize_value..cSrrrrrrrrrz$_serialize_value..)rr isoformatrrrrX)rrrrrs   rinfocCsddt|DS)z9Convert repo info dataclasses to json-serializable dicts.cSrrOr)r#krrrrrrz&api_object_to_dict..) dataclassesasdictr)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: r)r1r_ BadParameterr')r propertiesproprrr_parse_expand_propertiessz?make_expand_properties_parser.._parse_expand_properties)r rYrX)rrrrrmake_expand_properties_parsers" rlibrary)huggingface_hub 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   rcstj|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|]}|vVqdSrOr)r#tagcurrent_versionrrr%rz$_check_cli_update..)rcdeviQT)parentsexist_okzhttps://pypi.org/pypi/z/jsonr)timeoutrversionrzA new version of z (z&) is available! You are using version z. To update, run: r") importlibmetadatarrospathexistsrCHECK_FOR_UPDATE_DONE_PATHgetmtimetimerparentmkdirtouchrrrr#_get_huggingface_hub_update_command _get_transformers_update_commandrUechoryellowbold)rmtimeresponsedatalatest_versionupdate_commandrrrrs6  rcCs8t}|dkr dS|dkrtjdkrdS|dkrdSdS)z-Return the command to update huggingface_hub.brewzbrew upgrade huggingface-cli 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_hubrrrGmethodrrrrsrcCs,t}|dkrtjdkrdS|dkrdSdS)z*Return the command to update transformers.rrz`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 transformersrrrrrrs rrO)rNN)YrTrrimportlib.metadatarrrrrenumrpathlibrtypingrrrrrr r r r rUr_rr rhuggingface_hub.utilsrrrrrr get_loggerr_MAX_CELL_LENGTHrrrYrr&rXr*rqrVrWr4r` TyperGroupr5r^r[raTyperrbrzr|Argument RepoIdArgOption RepoTypeOptTokenOptrs PrivateOpt RevisionOptrLimitOpt AuthorOpt FilterOpt SearchOptr FormatOptQuietOptrrrrrrrobjectrrrrrrrrrrrs   ,   "(0 +         ' '