o 0iP @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)rrLbranchrNz/Binary found, installing as binary extension...)rrLr extension_dirrz-Binary extension installed successfully from /.z:Binary not found, trying to install as Python extension...)rrLrrNrMz-Python extension installed successfully from )rrLrMcandidate_descriptionzRun it with: hf )_normalize_repo_idZ find_rootsetgetattrcommandkeysr _get_extension_direxists_resolve_github_repo_infoshutilrmtree_fetch_remote_binary Exceptionprint_install_binary_extension_install_python_extension_try_fetch_remote_descriptionr$r?)rHrrIrrLrZroot_ctxZreserved_commandsrNZextension_existsrMr$rmanifestr'r'r1extension_installSsX   rcexecT)Zallow_extra_argsallow_interspersed_argsZignore_unknown_optionsz#hf extensions exec claude -- --helpz/hf extensions exec claude --model zai-org/GLM-5)Zcontext_settingsrGnamez.Extension name (with or without `hf-` prefix).cCsFt|}t|}|std|dt|t|jd}tj|d)zExecute an installed extension.rK' is not installed.rargs)code) _normalize_extension_name"_resolve_installed_executable_pathr)r _execute_extension_binarylistrityperZExit)rHrfrr exit_coder'r'r1extension_execs  rqz 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)rUr#r! installedr$)rrCrr!r"strftimer$).0rbr'r'r1 s  z"extension_list..rU)rrrsid_keyN)_list_installed_extensionsr)rrrsrowsr'r'r1extension_lists r{searchzhf 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:starsZdescd)qsortorderZper_pageT)paramsfollow_redirectstimeoutcSsh|]}|jqSr')r)rvmr'r'r1 sz#extension_search..itemsrfhf-NZ full_nameZstargazers_countrr$yes)rfrr}r$rtrright)rrrsrxZ alignments) r get_EXTENSIONS_GITHUB_TOPIC_EXTENSIONS_DOWNLOAD_TIMEOUTraise_for_statusr*ry startswithappendr) rrrsresponser0rtrzrrLrr'r'r1extension_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.rKrgzRemoved extension ''.N)rkrWis_dirr rZr[r^)rfrrNr'r'r1extension_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 expanduserrsortediterdirrfrrrr2r]loggerdebug)root_dirZ manifestsrNer'r'r1rys rycCsPg}tD] }d|jd}|jdur|jd|n|}||j|fq|S)Nz [extension ] )ryrr$rr)entriesrbtag help_textr'r'r1"list_installed_extensions_for_helps  rriknown_commandscCs|sdS|d}|drdS||vrdS|dr |ddn|}|s&dSzt|}Wn ty6YdSw|s=dSt|t|dddS)Nr-rrrrh)rrlr]r)rmrn)rirZ command_namerrr'r'r1$dispatch_unknown_top_level_extensions$   rrrLrMrcCsDt|}d|d|d|d|}tj|dtd}||jS)N"https://raw.githubusercontent.com/rO /refs/heads/Trr)_get_executable_namer rrrcontent)rrLrMrrZraw_urlrr'r'r1r\1s r\rNrc Cszt|}|jddd||}||tjdkrt|dt|||d|||t|dt t j d|d|d S) NTFr3ntirOrhttps://github.com/ rrrrrrr!r"r#) rr: write_bytesosrfchmodrrCrnowrutc)rrLrrNrrrr'r'r1r_9s"     r_c 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)%NrrOz/archive/refs/heads/z.zipvenvFz Creating virtual environment in T) ignore_errorsr3Zuv)checkzVirtual environment created in zInstalling package from piprFz--python)rr)Zwith_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 updaterXrZr[r:which_get_venv_python_path subprocessrunrCdone_EXTENSIONS_PIP_INSTALL_TIMEOUTrZ EnvBuildercreater#_get_venv_extension_executable_pathr)r rresolverrrrTimeoutExpiredr CalledProcessError returncoder])rrLrrNrM source_urlvenv_dirrtstatusZuv_pathZ venv_pythonrZvenv_executablerbrr'r'r1r`Ts         r`rQc Csz+tjd|d|d|dtdd}||}|d}t|tr*|WSWn ty4Ynwzsz1_validate_extension_short_name..)rO\rrz..:z9'. Allowed characters: letters, digits, '.', '_' and '-'.)rr any_ALLOWED_EXTENSION_NAME fullmatch)rrr'rr1rs  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.rOrrr6z%Expected `[OWNER/]REPO` format, got 'rrz6Extension repository name must start with 'hf-', got 'rz(Invalid extension repository name 'hf-'.r)r splitlenDEFAULT_EXTENSION_OWNERallrr)rpartsrrLrr'r'r1rRs         rRcCs:|}|s td|dr|ddn|}t||dS)Nrrrr)rr rr)rf candidate normalizedr'r'r1rk7s  rkrc Csjz tt|g|WSty4}ztjdks|jtjkrtdt|g|WYd}~Sd}~ww)Nrsh)rcallrCOSErrorrrferrnoZENOEXEC)rrirr'r'r1rm?s"rm)F)S__doc__rr*rrerZrr dataclassesrrrrpathlibrtypingrrr roZhuggingface_hub.errorsr r Zhuggingface_hub.utilsr r rZ _cli_utilsrrrrrrrr(ZEXTENSIONS_HELPZextensions_clirrrrZ get_loggerr@rrrUContextrCZArgumentboolOptionrcrqtabler{rrrnrytuplerrSintrbytesr\r_r`rarrWrYrrlrrcompilerrrRrkrmr'r'r'r1s      E       "   V ,"