iP@ rUdZddlZddlmZmZddlmZddlmZddl m Z m Z ddl m Z d ejDZeeefed <ejd ZeejZeeed <ed GddZed GddZdedefdZdedefdZdededeeedzedzffdZdededeej effdZ!dedej dedefdZ"dedej dedefdZ#dS) a|Centralized parser for Hugging Face Hub URIs ('hf://...') and mount specifications. A HF URI is a URI-like string that identifies a location on the Hugging Face Hub: a model/dataset/space/kernel repository, a bucket, optionally a revision, and optionally a path inside the repo or bucket. Canonical syntax: ``` hf://[/][@][/] ``` A HF mount wraps a HF URI with a local mount path and an optional ':ro'/':rw' flag (used by Spaces and Jobs volumes): ``` hf://[/][@][/]:[:ro|:rw] ``` See 'docs/source/en/package_reference/hf_uris.md' for the full grammar and examples. N) dataclassfield)unquote) constants) HfUriErrorHFValidationError)validate_repo_idci|]\}}|| Sr ).0kvs _/usr/local/lib/hermes-agent/venv/lib/python3.11/site-packages/huggingface_hub/utils/_hf_uris.py r0s"["["[DAq1a"["["[_TYPE_TO_PREFIXz ^refs/(?:convert/[\w.-]+|pr/\d+)_VALID_URI_TYPEST)frozenceZdZUdZejed<eed<dZedzed<dZ eed<e dddd Z edzed <dd Z e d efd Ze d efdZd efdZdS)HfUriaParsed representation of a Hugging Face Hub URI ('hf://...'). Attributes: type (`str`): One of 'model', 'dataset', 'space', 'kernel' or 'bucket'. id (`str`): The repository id ('namespace/name', e.g. 'my-org/my-model') for repo URIs, or the bucket id ('namespace/name') for bucket URIs. revision (`str`, *optional*): The revision specified after '@' in the URI, URL-decoded. 'None' if no revision was specified, or for bucket URIs (which never carry a revision). Special refs like 'refs/pr/10' and 'refs/convert/parquet' are preserved as-is. path_in_repo (`str`): The path inside the repo or bucket. Empty string if the URI points at the root. typeidNrevision path_in_repoFreprhashcomparedefault_rawreturnc |jpd}|jtvr/t|d|jdt td|jr|jddkrt|d|jd |jd krG t|jn1#t$r$}t|t||d}~wwxYw|j |j st|d |jd kr|j t|d |j r=|j ds d |j vrt|d|j ddSdS)NrzInvalid type 'z'. Must be one of .urimsg/r z"Id must be 'namespace/name', got ''.bucketz%Revision must not be an empty string.z&Bucket URIs do not support a revision.z//z+Path must not contain empty segments (got 'z').) r"rrrsortedrcountr rstrrr startswith)selfr'es r __post_init__zHfUri.__post_init__Ssio2 9, , ,*s49*s*sX^_oXpXp*s*s*sttt tw \$'--,,11*Ztw*Z*Z*Z[[[ [ 9  = ))))$ = = = Sc!ff5551< = = $T] $*QRRR R 9 T]%>*RSSS S   t ++C00 tDD/][@][/]:[:ro|:rw] ``` Attributes: source ([`HfUri`]): The parsed HF URI identifying the Hub resource to mount. mount_path (`str`): The local mount path (always starts with '/'). read_only (`bool`, *optional*): True if the mount ends with ':ro', False if it ends with ':rw', 'None' if no flag was provided. source mount_pathN read_onlyFrr"r#c|jpd}|jdr |jdkrt|d|jddS)Nrr)zEMount path must be a non-empty absolute path starting with '/', got 'r*r&)r"rPr/r)r0raws rr2zHfMount.__post_init__sfio2))#.. $/S2H2Ho\`\kooo 3I2Hrc|jd|jg}|j||jrdndd|S)zqRender the mount as a canonical 'hf://' string. Example: 'hf://models/my-org/my-model:/data:ro' :N:ro:rwr)rOrCrPrQr@rA)r0rBs rrCzHfMount.to_urisU ##%%sDO< > % LL$.;e < < <wwu~~rrD) rErFrGrHrrJr.rQrLrr"r2rCr rrrNrNs" MMMOOO!Itd{!!!u%eUDQQQD#*QQQrrNr'r#c|tjs+t|dtjdtjd|}|t tjd}|st|dtjdt ||\}}|dkrt |||St|||S) aParse a Hugging Face Hub URI ('hf://...'). A HF URI is a URI-like string identifying a location on the Hugging Face Hub. The full grammar is: ``` hf://[/][@][/] ``` See 'docs/source/en/package_reference/hf_uris.md' for the full specification. Args: uri (`str`): The URI to parse. Must start with 'hf://'. Returns: [`HfUri`]: the parsed URI. Raises: [`HfUriError`]: If the URI is malformed (missing prefix, invalid type, missing id, etc.). Examples: ```py >>> from huggingface_hub.utils import parse_hf_uri >>> parse_hf_uri("hf://my-org/my-model") HfUri(type='model', id='my-org/my-model', revision=None, path_in_repo='') >>> parse_hf_uri("hf://datasets/my-org/my-dataset@refs/pr/3/train.json") HfUri(type='dataset', id='my-org/my-dataset', revision='refs/pr/3', path_in_repo='train.json') ``` Must start with 'z'. Expected format: z#[/][@][/]NEmpty body after 'r*rSr+)r/rr<rlen _split_type_parse_bucket_body_parse_repo_body)r'rSbodytype_locations r parse_hf_urircs> >>)/ 0 0   [ 5 [ [ ) 5 [ [ [   C s9())++ ,D NL93HLLLMMM!$C000OE8 !(Es;;;; He 5 5 55r mount_strc|tjst|dtjd|}|t tjd}|st|dtjdt ||\}}}|t|dtj|z} t |}n)#t$r}t||j|d}~wwxYwt||||S) aParse a HF mount specification ('hf://...:[:ro|:rw]'). A mount specification is a HF URI followed by a local mount path and an optional read-only/read-write flag. The full grammar is: ``` hf://[/][@][/]:[:ro|:rw] ``` See 'docs/source/en/package_reference/hf_uris.md' for the full specification. Args: mount_str (`str`): The mount string to parse. Must start with 'hf://' and contain a ':' segment. Returns: [`HfMount`]: the parsed mount. Raises: [`HfUriError`]: If the mount string is malformed (missing mount path, invalid URI, etc.). Examples: ```py >>> from huggingface_hub.utils import parse_hf_mount >>> parse_hf_mount("hf://my-org/my-model:/data:ro") HfMount(source=HfUri(type='model', id='my-org/my-model', revision=None, path_in_repo=''), mount_path='/data', read_only=True) >>> parse_hf_mount("hf://buckets/my-org/my-bucket/sub/dir:/mnt:rw") HfMount(source=HfUri(type='bucket', id='my-org/my-bucket', revision=None, path_in_repo='sub/dir'), mount_path='/mnt', read_only=False) ``` rYr*r&NrZr[zKMissing mount path. Expected ':' (e.g. 'hf://org/model:/data').)rOrPrQr") r/rr<rr\ _split_mountrcr(rN) rdrSr`rbrPrQuri_strrOr1s rparse_hf_mountrhs<@    5 6 6 =I$9===    C S.//11 2D VS&T9;P&T&T&TUUUU&24S&A&A&A#Hj)S&stttt#h.G4g&& 444Sae,,,!34 &Z9SV W W WWs;C C1C,,C1r`rSc~|drd|d}}n/|drd|d}}nd}|d}|dkr|t|d |ddfS|d|}||d zd}|st|d |||fS) zSplit the ':[:ro|:rw]' suffix from 'body'. Returns '(location, mount_path, read_only)' where 'mount_path' is 'None' if no mount segment is present. rVTrWFNz:/zb':ro'/':rw' suffix is only valid when a mount path is provided (e.g. 'hf://...:/:ro').r&r z#Missing location before mount path.)endswith removesuffixrfindr)r`rSrQidxrbrPs rrfrf"s  }}U 1 1% 8 84 u  !2!25!9!94   **T  C byy  x T4DSDzHcAgiiJ MS&KLLLL Z **rrbc |d}|dkrs|tjvr$t|d|dtj|dt |x}!t|dtj|d d |fS|d|}||d zd}|tjvrtj||fSt |x}t|d |d |dd |fS)zDetect the (optional) type prefix and return '(type, remaining_location)'. A missing type prefix defaults to 'model'. Singular forms ('model/', 'dataset/', etc.) are explicitly rejected with a helpful error. r)rjzMissing identifier after 'z '. Expected 'z/'.r&Nz*Type prefix must be plural. Did you mean 'z/...'?modelr z!Type prefix must be plural, got 'z/'. Did you mean 'z/'?)findrHF_URI_TYPE_PREFIXESrr<rget)rbrS slash_idxsingular_pluralfirstrests rr]r]@sV  c""IB y5 5 5ppp H]p_gppp  /228<< [@][/]'.r)zMissing repository id.r&r;rjNryz-Repository id must be 'namespace/name', got 'z'. rr rz!Missing repository id before '@'.r*zEmpty revision after '@'.r{) r|rrqr}r\r-r=matchgroup removeprefixrr) rbrarSat_idxrrBrepo_idr rev_and_pathrrts rr_r_ws%~~c""H @S&>????]]3  F ||sA&& u::>>*gZb*g*g*ghhh h1X**a**#&u::>>uQxxr 7F7# -  O*MNNN N ==   " "*eZa*e*e*efff f-22<@@  {{}}H'H 8EEcJJLL$))#..IB'! '  3+IMOO< 8$$ G*EFFF F   !     r)$rHre dataclassesrr urllib.parserhuggingface_hubrhuggingface_hub.errorsrr _validatorsr rritemsrdictr.rJcompiler= frozensetvaluesrrrNrcrhtuplerLrfrIr]r^r_r rrrs, (((((((( %%%%%%@@@@@@@@)))))) #\"[I4R4X4X4Z4Z"["["[c3h[[[ *rz*MNN$-9Y-K-R-R-T-T#U#U)C.UUU $JJJJJJJJZ $''''''''T/6c/6e/6/6/6/6d7Xc7Xg7X7X7X7Xt+s+C+E#sTz4$;2N,O++++<#suY5H#5M/N>     066  6 6  666666r