o ~riP @sdZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z ddl mZddlmZmZmZddlZddlmZmZddlmZmZmZd d lmZmZmZmZm Z d Z!ed Z"d Z#dZ$e e$dZ%dZ&dZ'dZ(dZ)e*e+Z,e GdddZ-e%j.dgdd dmdej/dee0ej1ddfdee2ej3dddfd dfd!d"Z4e%j.d#d$dd$d%d&d'gd(dej/d)ee0ej1d*dfd dfd+d,Z5e%j.d-d.gdej6dfd/ed0ed dfd1d2Z7e%j.d3d4gdej6dfd/ed0ed dfd5d6Z8e%j.d7d8gdd)ee0ej1d9dfd dfd:d;Z9d e:e-fdd?Z=d@e:e0dAe>e0d ee?fdBdCZ@dDe0dEe0dFe0dGe0d eAf dHdIZBdDe0dEe0dGe0dJedKeAd e-f dLdMZCdDe0dEe0dGe0dJedFe0d e-f dNdOZDdDe0dEe0dFe0dPee0d ee0f dQdRZEd efdSdTZFdGe0d efdUdVZGdDe0dEe0d e` format.forcez--forcezOverwrite if already installed.r'cCsXt|\}}}|}tt|jdi}||vr%td|d|dt|}|} | r9|s9td|dt ||d\} } | rHt |z t ||| |d} Wn t y]d } Ynw| d ur{td t||||| d } td |d |dntdt||||| d} td|d |dt||| | d} | | _| |td|d S)zInstall an extension from a public GitHub repository. Security warning: this installs a third-party executable or Python package. Install only from sources you trust. commandszCannot install extension 'z,' because it conflicts with an existing `hf z ` command. Extension 'z1' is already installed. Use --force to overwrite.)r repo_name)rrMbranchrNz/Binary found, installing as binary extension...)rrMr extension_dirr z-Binary extension installed successfully from /.z:Binary not found, trying to install as Python extension...)rrMrrOrNz-Python extension installed successfully from )rrMrNcandidate_descriptionzRun it with: hf )_normalize_repo_id find_rootsetgetattrcommandkeysr _get_extension_direxists_resolve_github_repo_infoshutilrmtree_fetch_remote_binary Exceptionprint_install_binary_extension_install_python_extension_try_fetch_remote_descriptionr%r@)rIrrJrrMrroot_ctxreserved_commandsrOextension_existsrNr%r manifestr(r(r2extension_installSsX   rhexecT)allow_extra_argsallow_interspersed_argsignore_unknown_optionsz#hf extensions exec claude -- --helpz/hf extensions exec claude --model zai-org/GLM-5)context_settingsrHnamez.Extension name (with or without `hf-` prefix).cCsFt|}t|}|std|dt|t|jd}tj|d)zExecute an installed extension.rL' is not installed.rargs)code) _normalize_extension_name"_resolve_installed_executable_pathr*r _execute_extension_binarylistrqtyperExit)rIrnrr exit_coder(r(r2extension_execs  rzz list | lszhf extensions listformatquietcCs$ddtD}t|||dddS)z"List installed extension commands.cSs:g|]}d|jt|jt|j|jd|jdqS)zhf z%Y-%m-%d)rWr$r" installedr%)rrDrr"r#strftimer%).0rgr(r(r2 s  z"extension_list..rW)r{r|id_keyN)_list_installed_extensionsr)r{r|rowsr(r(r2extension_lists rsearchzhf extensions searchc Cstjddtdddddtd}||}d d tD}g}|d gD]/}|d }|d r;|ddn|}|||d|dd|dpNd||vrTdnddq*t |||dddiddS)zISearch extensions available on GitHub (tagged with 'hf-extension' topic).z*https://api.github.com/search/repositoriesztopic:starsdescd)qsortorderper_pageT)paramsfollow_redirectstimeoutcSsh|]}|jqSr()r)rmr(r(r2 sz#extension_search..itemsrnhf-N full_namestargazers_countrr%yes)rnrrr%r}rright)r{r|r alignments) r get_EXTENSIONS_GITHUB_TOPIC_EXTENSIONS_DOWNLOAD_TIMEOUTraise_for_statusr+r startswithappendr) r{r|responser1r}rrrMrr(r(r2extension_searchs,   rz remove | rmzhf extensions remove claudez8Extension name to remove (with or without `hf-` prefix).cCsFt|}t|}|std|dt|td|ddS)zRemove an installed extension.rLrozRemoved extension ''.N)rsrYis_dirr r\r]r`)rnrrOr(r(r2extension_removes  rc Cst}|s gSg}t|D]5}|r|jdsqz |t |Wqt yG}zt d|jd|WYd}~qd}~ww|S)zPReturn manifests for all validly-installed extensions, sorted by directory name.rz'Failed to load manifest for extension '': N) EXTENSIONS_ROOT expanduserrsortediterdirrnrrrr3r_loggerdebug)root_dir manifestsrOer(r(r2rs rcCsPg}tD] }d|jd}|jdur|jd|n|}||j|fq|S)Nz [extension ] )rrr%rr)entriesrgtag help_textr(r(r2"list_installed_extensions_for_helps  rrqknown_commandscCs|sdS|d}|drdS||vrdS|dr |ddn|}|s&dSzt|}Wn ty6YdSw|s=dSt|t|dddS)Nr-rrrrp)rrtr_r*rurv)rqr command_namerrr(r(r2$dispatch_unknown_top_level_extensions$   rrrMrNrcCsDt|}d|d|d|d|}tj|dtd}||jS)N"https://raw.githubusercontent.com/rP /refs/heads/Trr)_get_executable_namer rrrcontent)rrMrNrrraw_urlrr(r(r2r^1s r^rOr c Cszt|}|jddd||}||tjdkrt|dt|||d|||t|dt t j d|d|d S) NTFr4ntirPr https://github.com/ rrrrrrr"r#r$) rr; write_bytesosrnchmodrrDrnowrutc)rrMrrOr rrr(r(r2ra9s"     rac Csd|d|d|d}|d}d}t}z-z|d||r+tj|dd |jddd td } t|} | rjtj | dt |gdd | d ||d|tj | dddt | |gdt dn,t jddt || d ||d|tj t | ddddd|gdt d| d|t|} t||} | std|d|d| dt|||d||| t | dttjd|d|d } d}| WW|stj|dd SStytjy }ztdt d|d|d|d}~wtjy)}ztd |d|d!|jd"|d}~wtyC}ztd#|d|d$||d}~ww|sOtj|dd ww)%NrrPz/archive/refs/heads/z.zipvenvFz Creating virtual environment in T) ignore_errorsr4uv)checkzVirtual environment created in zInstalling package from piprGz--python)rr)with_pipz-mz--disable-pip-version-checkz --no-inputzPackage installed from zInstalled package from 'z/' does not expose the required console script 'rr!rzPip install timed out after zs for 'z$'. See pip output above for details.z$Failed to install pip package from 'z ' (exit code z$). See pip output above for details.z%Failed to set up pip extension from 'r)r updaterZr\r]r;which_get_venv_python_path subprocessrunrDdone_EXTENSIONS_PIP_INSTALL_TIMEOUTr EnvBuildercreater#_get_venv_extension_executable_pathr*r rresolverrrrTimeoutExpiredr CalledProcessError returncoder_)rrMrrOrN source_urlvenv_dirr}statusuv_path venv_pythonrvenv_executablergrr(r(r2rbTs         rbrRc Csz+tjd|d|d|dtdd}||}|d}t|tr*|WSWn ty4Ynwzsz1_validate_extension_short_name..)rP\rrz..:z9'. Allowed characters: letters, digits, '.', '_' and '-'.)rr any_ALLOWED_EXTENSION_NAME fullmatch)rrr(rr2rs  rcCsd|vrtd|d}t|dkrt}|d}nt|dkr)t|r)|\}}ntd|d|d s>td |d|d d}|sJtd t||d |||fS)Nz://zEOnly GitHub repositories in `[OWNER/]hf-` format are supported.rPrrr7z%Expected `[OWNER/]REPO` format, got 'rrz6Extension repository name must start with 'hf-', got 'rz(Invalid extension repository name 'hf-'.r)r splitlenDEFAULT_EXTENSION_OWNERallrr)rpartsrrMrr(r(r2rSs         rScCs:|}|s td|dr|ddn|}t||dS)Nrrrr)rr rr)rn candidate normalizedr(r(r2rs7s  rsrc Csjz tt|g|WSty4}ztjdks|jtjkrtdt|g|WYd}~Sd}~ww)Nrsh)rcallrDOSErrorrrnerrnoENOEXEC)rrqrr(r(r2ru?s"ru)F)S__doc__r r+rrer\rr dataclassesrrrrpathlibrtypingrrr rwhuggingface_hub.errorsr r huggingface_hub.utilsr r r _cli_utilsrrrrrrrr)EXTENSIONS_HELPextensions_clirrrr get_loggerrArrrWContextrDArgumentboolOptionrhrztablerrrrvrtuplerrUintrbytesr^rarbrcrrYr[rrtrrcompilerrrSrsrur(r(r(r2s      E       "   V ,"