Xjr dZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z ddl mZddlmZddlmZddlmZmZddlZddlZdd lmZmZmZdd lmZmZd d lm Z d d l!m"Z"m#Z#d dl$m%Z%dZ&edZ'dZ(dZ)e e)Z*dZ+dZ,dZ-ej.e/Z0GddeZ1Gdde2eZ3e GddZ4e*5dgd dld!ej6d"ee2e"d#fd$ee7e#d%d&fd'dfd(Z8e*5d)gd* dmd+ee2dze"d,fd'dfd-Z9e*5d.d/d d/d0d1d2g3d!ej6d+ee2e"d4fd'dfd5Z:e*5d6d7gdnd8Z;e*5d9d:gdnd;Zfd'dfd?Z=d'e>e4fd@Z?d'e>e@e2e2ffdAZAdBe>e2dCeBe2d'eCdzfdDZDdEe2d'edzfdFZEd+e2d'e4fdGZFdHe4d'e3fdIZGddJdKe2dLe2dEe2dMe2dzd'e4f dNZHdKe2dLe2d'e2fdOZIdKe2dLe2dEe2d'eJfdPZKdQedEe2dReJd'efdSZLdQedKe2dLe2dEe2d'ef dTZMejNdUZOdKe2dLe2d'e2dzfdVZPdEe2d'efdWZQdddXdYe2dZe2d[eRdzd\eRdzd'ejSf d]ZTd\ejUd'e2fd^ZVdKe2dLe2d'e7fd_ZWdEe2d'e2fd`ZXdEe2d'efdaZYdbedce2d'efddZZejNdeZ[dEe2dfe2d'e2fdgZ\d"e2d'e@e2e2e2ffdhZ]d+e2d'e2fdiZ^djedBe>e2d'eCfdkZ_dS)oz0Contains helper utilities for hf CLI extensions.N)asdict dataclass)datetimetimezone)Enum)unescape)Path) AnnotatedLiteral)CLIErrorCLIExtensionInstallErrorConfirmationError) get_sessionlogging) typer_factory)ArgumentOption)out huggingfacez~/.local/share/hf/extensionsz manifest.jsonzManage hf CLI extensions. Security Warning: extensions are third-party executables or Python packages. Install only from sources you trust.)helpz hf-extension i,ceZdZdZdS)_GitHubRateLimitErroraThe GitHub API refused the request because a rate limit is exhausted. Distinguished from other `CLIError`s because it says nothing about the extension being handled: it is not worth retrying on the next one, and it must not be mistaken for an install failure. N)__name__ __module__ __qualname____doc__OC:\Projects\Historical\venv\Lib\site-packages\huggingface_hub/cli/extensions.pyrr9sr rceZdZdZdZdZdS)_ExtensionUpdateStatusupdated up_to_dateskippedN)rrrUPDATED UP_TO_DATESKIPPEDrr r!r#r#AsGJGGGr r#ceZdZUeed<eed<eed<eed<eed<eded<eed<d Zed zed <d Zed zed <e d e d dfdZ d e d d fdZ d S)ExtensionManifestownerreporepo_id short_nameexecutable_path)binarypythontype installed_atN description commit_shapathreturncD|tz }|std|dtj|}fd|D}tj|d|d<tdi|S)NzManifest file not found at z". Your extension may be corrupted.c.i|]\}}|jv||Sr)__dataclass_fields__).0keyvalueclss r! z*ExtensionManifest.load..Zs,]]]zsESCD\=\=\U=\=\=\r r4r) MANIFEST_FILENAMEis_filer jsonloads read_textitemsr fromisoformatr+)r?r7 manifest_pathdatas` r!loadzExtensionManifest.loadSs00 $$&& ljjjjkk kz-113344]]]]TZZ\\]]]'5d>6JKK^ ((4(((r c|tz }|jddt|}|j|d<|tj|dddS)NTparentsexist_okr4)indent sort_keys) rAparentmkdirrr4 isoformat write_textrCdumps)selfr7rHrIs r!savezExtensionManifest.save^sx00 ""4$"???d||#0::<<^  Dd!K!K!KLLLLLr ) rrrstr__annotations__r rr5r6 classmethodr rJrXrr r!r+r+Gs JJJ III LLLOOO $ %%%%"Kt"""!Jd !!!))!4)))[)MM$MMMMMMr r+install)zhf extensions install hf-claudez-hf extensions install hanouticelina/hf-claudez(hf extensions install alvarobartt/hf-mem)examplesFctxr.z:GitHub extension repository in `[OWNER/]hf-` format.forcez--forcezOverwrite if already installed.r8ct|\}}}|}tt|jdi}||vrt d|d|dt|}|r|st d|dt||st d|d |d |rtj |t||| } | j } tj| d |d |d |tjd|dS)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_namezGitHub repository '/z)' not found. It must exist and be public.r,rdr/z extension installedhf )sourcecommandzRun it with: hf N)_normalize_repo_id find_rootsetgetattrrikeysr _get_extension_direxists_github_repo_existsshutilrmtree_install_extensionr3 capitalizerresulthint) r^r.r_r,rdr/root_ctxreserved_commands extension_dirmanifestext_types r!extension_installr}fs*$6g#>#> E9j}}HGH$4j"EEJJLLMM&&& w w wak w w w   'z22MdedbZbbbccc Ui @ @ @kiUiiYiiijjj% m$$$!zZZZH}''))HJ )))%%)%%"j"" H , , ,-----r update)zhf extensions updatezhf extensions update hf-claudez'hf extensions update alvarobartt/hf-memnamezExtension to update (with or without `hf-` prefix, optionally as `OWNER/hf-`). If omitted, all installed extensions are checked and the outdated ones are updated.c |t|}t|}|xtjkr#t jd|j|jnFxtjkr&t jd|jd|jntj kr dSt}|s*t j dt j ddSg}g}|D]}t j d |jd |jd  t|}nI#t$rt$r2}t j d |jd |jd |dYd}~yd}~wwxYw|xtjkr||jxtjkr||jtj krt jd|rd|nd|rd|nddS)z6Update installed extension(s) to their latest version.NzExtension updated)rrhrbz' is already up to daterzNo extensions installed.z1Install one with: hf extensions install z Checking '' (z)...zCould not update '): . Skipping.zExtensions update completez, )r$r%)$_load_installed_extension_for_update_update_installed_extensionr#r'rrvr/r.r(r)_list_installed_extensionswarningrwlogr Exceptionappendjoin)rr{ update_status manifestsr$r%errors r!extension_updatersv( 7==3H== /'//// .X5HQYQabbbbb2'2222 U)<UUU\d\oppppp'///*,,I  ./// DEEEGJ KX0KKX5EKKKLLL 7AAMM$        KlX-@llXEUllZ_lll m m m HHHH /'////x233332'2222!!("56666'///0J$&-7 '"""4,6@499Z(((DsDE)(EEexecT)allow_extra_argsallow_interspersed_argsignore_unknown_optionsz#hf extensions exec claude -- --helpz/hf extensions exec claude --model zai-org/GLM-5)context_settingsr]z.Extension name (with or without `hf-` prefix).ct|}t|}|std|dt |t |j}tj |)zExecute an installed extension.rb' is not installed.r0args)code) _normalize_extension_name"_resolve_installed_executable_pathrBr _execute_extension_binarylistrclick exceptionsExit)r^rr/r0 exit_codes r!extension_execrs +400J8DDO  " " $ $FDZDDDEEE)/PTUXU]P^P^___I    Y  / //r z list | lszhf extensions listcbdtD}tj|ddS)z"List installed extension commands.cg|]S}d|jt|jt|j|jd|jdTS)rgz%Y-%m-%d)rirhr3 installedr5)r/rYr.r3r4strftimer5)r<r{s r! z"extension_list..st     3X022(*++ &&!.77 CC#/      r riid_keyN)rrtable)rowss r!extension_listrsE  344   DId9%%%%%%r searchzhf extensions searchc tdddtdddd}|}d tD}g}|d gD]m}|d d }|||d |dd|dpd||vrdnddntj|ddS)zISearch extensions available on GitHub (tagged with 'hf-extension' topic).GETz*https://api.github.com/search/repositoriesztopic:starsdescd)qsortorderper_page)paramsch|] }|j Srr/)r<ms r! z#extension_search.. sDDD!DDDr rFrhf- full_namestargazers_countrr5yes)rr-rr5rr-rN) _github_request_EXTENSIONS_GITHUB_TOPICrCrget removeprefixrrr)responserIrrr-r/s r!extension_searchrs 48688'TZhkllH ==??DDD'A'C'CDDDI D"%%   &\..u55  "[)"4a88#xx 66<"&0I&=&=UU2       Id6""""""r z remove | rmzhf extensions remove claudez8Extension name to remove (with or without `hf-` prefix).ct|}t|}|std|dt j|t jd|dS)zRemove an installed extension.rbrzExtension removedrN)rrois_dirr rrrsrrv)rr/rzs r!extension_removersy+400J&z22M    ! !FDZDDDEEE M-   J"444444r ct}|sgSg}t|D]}|r|jds1 |t |`#t$r/}t d|jd|Yd}~d}~wwxYw|S)zPReturn manifests for all validly-installed extensions, sorted by directory name.rz'Failed to load manifest for extension '': N) EXTENSIONS_ROOT expanduserrsortediterdirr startswithrr+rJrloggerdebug)root_dirrrzes r!rr2s))++H ??   I 0 0 2 233 ##%% ]-?-J-J5-Q-Q      .33MBB C C C C    LL]=CU]]Z[]] ^ ^ ^ HHHH  s-B33 C,=%C''C,cg}tD]>}d|jd}|j |jd|n|}||j|f?|S)Nz [extension ] )rr.r5rr/)entriesr{tag help_texts r!"list_installed_extensions_for_helprCsuG.0099/H,///7?7K7Wx+33c333]` +Y78888 Nr rknown_commandsc|sdS|d}|drdSd|D}||vrdS t|d|}n#t$rYdSwxYw t |}n#t $rt |}YnwxYw||sdSt|t|ddS)Nr-cfh|].}|dD]}|/S)|)splitstrip)r<cmdas r!rz7dispatch_unknown_top_level_extension..Ss7MMMsciinnMMMMMMr roriginal_inputrr) r_validate_extension_short_namerr rr _auto_install_official_extensionrBrr)rr command_name all_knownr/r0s r!$dispatch_unknown_top_level_extensionrLs) t7Ls##tMMnMMMIy  t3L4M4Me4T4Teqrrr ttG> Z Z ZZs#$A A*)A*.A>>BBr/c ~td|}}t|rdS t||sdSn#t$rYdSwxYw t jd|d|d|dn#t$rYdSwxYwt|||}t|j  S) zKTry to auto-install huggingface/hf-. Returns executable path or None.rNrc'z)' is an official Hugging Face extension (rez). Install it?rf) DEFAULT_EXTENSION_OWNERrorprqrrconfirmrrtr r0r)r/r,rdr{s r!rrhs.0Bj0B0B9E*%%,,..t")DDD 4  tt n nnUnnU^nnnoooo tt!zZZZH ( ) ) 4 4 6 66s#A AAA66 BBc6t|}t|}|sN|d\}}}|r|d|nd|}t d|d|t |S)Nrez/hf-rrbzA' is not installed. Install it first with: hf extensions install )rrorr rpartitionr r+rJ)rr/rzr,_install_targets r!rr~s*400J&z22M    ! ! jjll--c22 q!7<TE33z333BT BTBT w* w wgu w w     ! !- 0 00r r{c Z|j|j|j}}} t||}nN#t$rt $r7}t jd|d|d|d|d tj cYd}~Sd}~wwxYw||j kr tj St||||tj S)NrczCould not check updates for 'rrerr)r,rdr/r6)r,r-r/_fetch_latest_commit_sharrrrr#r)r6r(rtr')r{r,rdr/ latest_shars r!rrs#+>8=(BUj9E.-EYOOO  ... kJkk5kk9kkY^kkklll%-------.X(((%00UiJ[effff ! ))s)A4,A/)A4/A4)r6r,rdr6c Vt|}| }d} t|||}n#t$rd}YnwxYw|9tjdkrt d|d|dt|||}nt|||| }|G t|| }n4#t$r'} tj d |d|d | Yd} ~ nd} ~ wwxYwt|||d||t||d ndtjt jt%|| | } | |d}| |s|rt)j|dSSS#t $rt,j$r$} t1dt2d|d|d| d} ~ wt,j$r$} t1d|d|d| jd| d} ~ wt$r} t1d|d|d| | d} ~ wwxYw#|s|rt)j|dwwwxYw)aFetch and install an extension (binary or Python package), then persist its manifest. Installs in place: an existing install is overwritten without being removed first, so a failed update keeps the previous version working. A failed fresh install is cleaned up entirely. FrfNntrrezu' is a shell-script extension, which is not supported on Windows. Only Python extensions can be installed on Windows.)rzr/r1)rzr,rdr/rcz Could not record a version for 'z', installing anyway: r1r2) r,r-r.r/r0r3r4r5r6T) ignore_errorszPip 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 install extension from 'r)rorp_fetch_remote_binaryrosrr _install_binary_extension_install_python_extensionrrrr+rYrnowrutc_try_fetch_remote_descriptionrXrrrs subprocessTimeoutExpiredr _EXTENSIONS_PIP_INSTALL_TIMEOUTCalledProcessError returncode) r,rdr/r6rz fresh_installrr1r0rr{rs r!rtrts'z22M%,,...MI;= )WabbbFF   FFF   w$JJJ JJJ8+ 6OO8+5IZdO   q5EYWWW  q q q ouooyoohmoopppppppp q%**y**!00#/X!hl335EYWWW!      m$$$   =] = M-t < < < < < = =   $& 0+J 0 0SX 0 0[d 0 0 0     (& 05 0 09 0 0STS_ 0 0 0    nnn&'eE'e'eT]'e'ebc'e'efflmmn =] = M-t < < < < < = =s<E3 A E3 A  AE3B10E31 C";CE3C""A4E33H F++H=G H)HHH H(cjtdd|d|dddi}|jS)aFetch the latest commit SHA on the default branch, used to detect available updates. The only REST API call left in the install and update paths. Raises on failure: callers decide whether a missing SHA is fatal, and report the reason themselves. rzhttps://api.github.com/repos/rez /commits/HEADAcceptzapplication/vnd.github.sha)headers)rtextr)r,rdrs r!rrsP  HHH HHH78H =    r cFd|d|d|}td|}|jS)N"https://raw.githubusercontent.com/rez /HEAD/hf-r)rcontent)r,rdr/raw_urlrs r!rrs7[5[[9[[z[[Gug..H  r rzr1c|dd|t|z }||tjdkrtj|d|S)NTrLri)rS_get_executable_name write_bytesrrchmod)rzr/r1r0s r!rrset444#&::&F&FFO''' w$ %((( r c d|d|d}|dz }t|tjdkrdnd}tjd}t j}|s|d || d d |r'tj |dt|gd n5tjd t||d||d||r|ddddt||g} nt|dddddd|g} tj | d t"|d|t%|} t|| } | st'd|d|d| d| S)Nhttps://github.com/rez/archive/HEAD.zipvenvrz python.exer2uvz Creating virtual environment in TrL)check)with_pipzVirtual environment created in zInstalling package from pipr\z --reinstallz--pythonz-mz--disable-pip-version-checkz --no-inputz--force-reinstall)rtimeoutzPackage installed from zInstalled package from 'z/' does not expose the required console script ''.)_get_venv_bin_pathrrrrwhichrstatusrBr~rSrrunrYr EnvBuildercreatedonerrr resolve) rzr,rdr/ source_urlvenv_dir venv_pythonuv_pathr install_cmdexecutable_namevenv_executables r!rr s;KuKKyKKKJv%H$Xrw$||T\]]Kl4  G Z\\F    B CCCDDDD4888  A NGVS]];4 H H H H H OT * * * 1 1#h-- @ @ @ @h@@AAA MM9Z99:::  y-SQ\M]M]_ij        )     N;d4STTTT KK6*66777*:66O(?CCO  " " $ $  $u $ $y $ $ $ $ $     " " $ $$r z*matchrs r!rr8s I H H H H HD "5T*G*G4E*G*GHHmmoo))-88 k3 ' '           "5T*B*B*BCCM,,.. 2 2D::<AA$$ A10A15BDD DD!A;F F+*F+cbt||td|z S)Nrr)rrrrs r!rorofs7#:jIIII  % % ' '*< *<*< </ ////;;233 3% @ @-c%jjX\JJJUUWWH ????? ?GGw 3    D  Ns?BB B c  tdd|d|n`#tj$rN}t|tjr|jjdkrYd}~dStd|d|d||d}~wwxYwd S) aEWhether a public GitHub repo exists. Resolved against `github.com` rather than `GET /repos/{owner}/{repo}` to keep the check off the metered REST API (see `_github_request`). Only a 404 answers `False`: any other failure is reported as such, so callers never present an unreachable GitHub as a missing repo. HEADrreiNFz)Could not reach GitHub to check whether 'z ' exists: T)rhttpx2 HTTPErrorr.HTTPStatusErrorrr@r )r,rdrs r!rqrqst Ie I Ii I IJJJJ  ttt eV3 4 4 9SWZ9Z9Z55555g5gg9gg`egghhnsst 4sA6*A1A11A6c:d|}tjdkr|dz }|S)Nrrz.exerr)r/rs r!rrs(   D w$  Kr ct|}t|}t|jSN)ror+rJr r0r)r/rzr{s r!rrs?&z22M %%m44H ( ) ) 4 4 6 66r r&r*cBtjdkr|dz |z S|dz |z S)NrScriptsbinrV)r&r*s r!rrs- w$)#o55 e o --r z^[A-Za-z0-9][A-Za-z0-9._-]*$rc>|stdtfddDrtd|ddvsdvrtd|dtstd|dS) NzExtension name cannot be empty.c3 K|]}|vV dSrXr)r<seprs r! z1_validate_extension_short_name..s' . .33$; . . . . . .r )re\zInvalid extension name 'rz..:z9'. Allowed characters: letters, digits, '.', '_' and '-'.)rr any_ALLOWED_EXTENSION_NAME fullmatch)r/rrs @r!rrs     D :8999 . . . .+ . . ...FD.DDDEEE t||sd{{D.DDDEEE " , ,T 2 2  p~ p p p    Kr cd|vrtd|d}t|dkrt}|d}n;t|dkrt |r|\}}ntd|d|d std |d|d }|std t|| |||fS) Nz://zEOnly GitHub repositories in `[OWNER/]hf-` format are supported.rerrrOz%Expected `[OWNER/]REPO` format, got 'rrz6Extension repository name must start with 'hf-', got 'z(Invalid extension repository name 'hf-'.r)r rlenrallrrr)r.partsr,rdr/s r!rjrjs ^___ MM#  E 5zzQ'!H UqSZZ yyJwJJJKKK    & &_]PY]]]^^^''..J CABBB":gFFFF )Z ''r c|ddd}t|d|S)Nrerrr)rrsplitrr)rrds r!rrsF ##C++B/I ))*@*@*G*GX\ ] ] ]]r r0c tjt|g|zS#t$rV}tjdks|jt jkrtjdt|g|zcYd}~Sd}~wwxYw)Nrsh)rcallrYrJrrerrnoENOEXEC)r0rrs r!rrsDO 4 45<=== DDD 7d??ag66 c/&:&:;dBCCCCCCCCDs$' BA B<BB)FrX)r8N)`rrorCrrerrrr dataclassesrrrrenumrhtmlrpathlibr typingr r rrRhuggingface_hub.errorsr r rhuggingface_hub.utilsrr _cli_utilsr _frameworkrr_outputrrrrAEXTENSIONS_HELPextensions_clirr?r get_loggerrrrrYr#r+riContextboolr}rrrrrrrtuplerrlrGrrrrrtrbytesrrrcompiler1rrodictResponserHeadersrArqrrrrcrrjrrrr r!rs\ 76  ))))))))''''''''%%%%%%%% XXXXXXXXXX66666666%%%%%%((((((((($566#+ O444)!"%  H % %HS$  MMMMMMM M< Y^ %.%. %.  RSSS U%. T66)2STTTT U %. %.%.%.%.P " 66  d f     6 6666r *.5lpqq-90 0  FGGG I 0  0000$ /C.DEE & & &FE &,B+CDD###ED#6 1N0OPP5  PQQQ S 5  555QP5(D):$;"DsCx,A[tCy[#c([WZ]aWa[[[[8777777, 1s 17H 1 1 1 1**;*@V****0" J=J=J= J=J= J= d J=  J=J=J=J=Z !s !s !s ! ! ! !33CE #uY])%)%S)%S)%^a)%fj)%)%)%)%X2:KLL+++t++++\=3=4====59QU &*TkCG$; _>C4 # # $     SS73747777 ....... %"*%DEE s s s    ((c3m(<((((0^C^C^^^^ DtD49DDDDDDDr