Ziu+ <dZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z m Z ddl m Z ddlmZddlmZmZddlZddlmZdd lmZmZd d lmZd Ze d ZdZdZeeZe GddZ e!dddg dCdej"dee#ej$dfdee%ej&ddfddfdZ'e!d d!dd!d"d#d$g%dej"d&ee#ej$d'fddfd(Z(e!d)d*gdDd+Z)e!d,d-gd&ee#ej$d.fddfd/Z*d0e+e#d1e,e#dee-fd2Z.d3e d4e d5e ddfd6Z/de fd7Z0d8e#de fd9Z1d8e#de#fd:Z2d8e#de fd;Z3ej4d<Z5d8e#d=e#de#fd>Z6de#de7e#e#e#ffd?Z8d&e#de#fd@Z9dAe d0e+e#de-fdBZ:dS)Ez0Contains helper utilities for hf CLI extensions.N)asdict dataclass)datetimetimezone)Path)TemporaryDirectory) AnnotatedOptional)CLIError) get_sessiontabulate) typer_factory huggingfacez~/.local/share/hf/extensionsz manifest.jsonzyManage hf CLI extensions. Security Warning: extensions are third-party executables. Install only from sources you trust.)helpcjeZdZUeed<eed<eed<eed<eed<eed<eed<eed<eed <d S) ExtensionManifestownerreporepo_id short_nameexecutable_nameexecutable_pathtype installed_atsourceN)__name__ __module__ __qualname__str__annotations__Y/root/projects/butler/venv/lib/python3.11/site-packages/huggingface_hub/cli/extensions.pyrr/sj JJJ III LLLOOO III KKKKKr#rinstallzhf extensions install hf-claudez-hf extensions install hanouticelina/hf-claude)examplesFctxrz:GitHub extension repository in `[OWNER/]hf-` format.forcez--forcezOverwrite if already installed.returnct|\}}}|}tt|jdi}||vrt d|d|dt|}|r)|st d|dtj |t|} d|d|d | } t | d } | n.#t$r!} t d | d |d|d| | d} ~ wwxYwt!5} t#| | z }|| jt(jdkrt)j|dt/|||d||| t1t3|dt5jt8jd|d| }t?|||dddn #1swxYwYtAd|d|dtAd|tAd|dS)zInstall an extension from a public GitHub repository. Security warning: this installs a third-party executable. 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.z"https://raw.githubusercontent.com//z/refs/heads/main/T)follow_redirectszFailed to download 'z' from 'z': Nntbinaryzhttps://github.com/) rrrrrrrrr) extension_dirsource_executablemanifestzInstalled extension ''.zRun it with: hf zOr with: hf extensions exec )!_normalize_repo_id find_rootsetgetattrcommandkeysr _get_extension_direxistsshutilrmtree_get_executable_namer getraise_for_status Exceptionrr write_bytescontentosnamechmodrr _get_extension_executable_pathrnowrutc isoformat_persist_installed_extensionprint)r'rr(r repo_namerroot_ctxreserved_commandsr2rraw_urlresponseetmp_dirtmp_executabler4s r$extension_installrW<s%&$6g#>#> E9j}}HGH$4j"EEJJLLMM&&& w w wak w w w   'z22M% hffffgg g m$$$*:66Oh5hh9hhWfhhGi==$$Wt$DD!!#### iii`o``u``y``]^``aaghhi    g8""8#3444 7d?? H^U + + +$**y**!+ >z J JKK!hl33==??<<<<<     %',    #               . 7% 7 7) 7 7 7888 )Z ) )*** 5 5 566666s+7D E D<<ECH--H14H1execT)allow_extra_argsallow_interspersed_argsignore_unknown_optionsz#hf extensions exec claude -- --helpz/hf extensions exec claude --model zai-org/GLM-5)context_settingsr&rGz.Extension name (with or without `hf-` prefix).ct|}t|}|std|dt |t |j}tj|)zExecute an installed extension.r,' is not installed.rargs)code) _normalize_extension_namerIis_filer _execute_extension_binarylistr`typerExit)r'rGrr exit_codes r$extension_execris +400J4Z@@O  " " $ $FDZDDDEEE)/PTUXU]P^P^___I *) $ $ $$r#rezhf extensions listct}|stddSg}t|D]}|r|jds1|jdd}|tz }d}d}|rntj | d}t| dd}t| d d}|d |||g|stddStt|gd  dS) z"List installed extension commands.zNo extensions installed.Nhf-utf-8encodingrrzhf )COMMAND REPOSITORY INSTALLED_AT)headers)_get_extensions_rootis_dirrNsortediterdirrG startswithMANIFEST_FILENAMErcjsonloads read_textr rAappendr )root_dirrowsr2r manifest_path repositoryrdatas r$extension_listrs$%%H ??   ())) D 0 0 2 233DD ##%% ]-?-J-J5-Q-Q  "'+ %(99    " " =:m55w5GGHHDTXXi4455Jtxx;;<r?rN)rGrr2s r$extension_removers}+400J&z22M    ! !FDZDDDEEE M-    . . . ./////r#r`known_commandsc4|sdS|d}|drdS||vrdS|dr |ddn|}|sdSt|}|sdSt|t |ddS)Nr-rkrlrr_)ryrIrcrdre)r`r command_namerrs r$%_dispatch_unknown_top_level_extensionrs t7Ls##t~%%t%1%<%> Z Z ZZr#r2r3r4c||jz }|tz } |ddtj||t jdkrt j|d|tj t|dddzd dS#t$rtj |d wxYw) NTFparentsexist_okr/r0)indent sort_keys rnro) ignore_errors)rrzmkdirr>copy2rFrGrH write_textr{dumpsrrCr?)r2r3r4rrs r$rMrMs#h&>>O!$55MD5999 &888 7d?? H_e , , ,  F8,<,!!@*@@@AAA Mr#c:d|}tjdkr|dz }|S)Nrkr/z.exe)rFrG)rrGs r$r@r@s(   D w$  Kr#c@t|t|z SN)r<r@)rs r$rIrI s j ) ),@,L,L LLr#z^[A-Za-z0-9][A-Za-z0-9._-]*$rc>|stdtfddDrtd|ddvsdvrtd|dtstd|dS) NExtension name cannot be empty.c3 K|]}|vV dSrr").0seprGs r$ z1_validate_extension_short_name..s' . .33$; . . . . . .r#)r-\rr5z..:z9'. Allowed characters: letters, digits, '.', '_' and '-'.)stripr any_ALLOWED_EXTENSION_NAME fullmatch)rrrGs @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 d}|std t || |||fS)Nz://zEOnly GitHub repositories in `[OWNER/]hf-` format are supported.r-rrrz%Expected `[OWNER/]REPO` format, got 'r5rkz6Extension repository name must start with 'hf-', got 'rlz(Invalid extension repository name 'hf-'.r)r splitlenDEFAULT_EXTENSION_OWNERallryr)rpartsrrOrs r$r6r6 s ^___ MM#  E 5zzQ'!H UqSZZ yyJwJJJKKK    & &_]PY]]]^^^122J CABBB":gFFFF )Z ''r#c|}|std|dr |ddn|}t||S)Nrrkrlr)rr ryr)rG candidate normalizeds r$rbrb8s] I :8999"+"6"6u"="=L1229J )*T J J JJr#rc tjt|g|zS#t$rV}tjdks|jt jkrtjdt|g|zcYd}~Sd}~wwxYw)Nr/sh) subprocesscallr OSErrorrFrGerrnoENOEXEC)rr`rTs r$rdrd@sDO 4 45<=== DDD 7d??ag66 c/&:&:;dBCCCCCCCCDs$' BA B<BB)F)r)N);__doc__rr{rFrer>r dataclassesrrrrpathlibrtempfilertypingr r rfhuggingface_hub.errorsr huggingface_hub.utilsr r _cli_utilsrrrrzEXTENSIONS_HELPextensions_clirr:Contextr ArgumentboolOptionrWrirrrer8intrrMrur<r@rIcompilerrtupler6rbrdr"r#r$rs"76 ))))))))''''''''''''''&&&&&&&& ++++++77777777%%%%%%($566#+ O444           )7_d ;7;7 ;7  XYYY [;7 T<5< 8YZZZZ [ ;7 ;7;7;7;7| *.5lpqq-9% %  LMMM O %  %%%%$*>)?@@MMMA@M>,I+JKK0  VWWW Y 0  000LK0([S [3s8[X`adXe[[[[*   Yj os    d 34SSMsMtMMMM%"*%DEE s s s    ((c3m(<((((0KCKCKKKKDtD49DDDDDDDr#