ZjA UdZddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z ddl mZe jj!Dcic]\}}|| c}}Zeeefed <ej*d Zee jj1Zeeed <ed GddZed GddZdedefdZej<dedefdZdedefdZ dedede!eedzedzffdZ"dedede!e jFeffdZ$dede jFdedefdZ%dede jFdedefdZ&ycc}}w) 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_id_TYPE_TO_PREFIXz ^refs/(?:convert/[\w.-]+|pr/\d+)_VALID_URI_TYPEST)frozenceZdZUdZej ed<eed<dZedzed<dZ eed<e dddd Z edzed <dd Z e d efd Ze d efdZd efdZy)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 |jxsd}|jtvr+t|d|jdt td|j r|j j ddk7rt|d|j d |jd k7r t|j |j|js t|d |jd k(r|j t|d |jrE|jjdsd |jvrt|d|jdyy#t$r}t|t||d}~wwxYw)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').) rrr rsortedrcountr rstrrr startswith)selfres ]/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/huggingface_hub/utils/_hf_uris.py __post_init__zHfUri.__post_init__Ts]iio2 99, ,N499+EWX^_oXpWqqr*st tww$''--,1,NtwwiWY*Z[ [ 99  = ) == $T]]*QR R 99 T]]%>*RS S     ++C0DDs     GHcDjL#%eUDQD#*Qt6%4%%%%%r1rcveZdZUdZeed<eed<dZedzed<e ddddZ edzed<d d Z d efd Z y) HfMountaFA HF URI paired with a local mount path and optional read-only flag. Used by Spaces and Jobs to describe volume mounts. The full syntax is: ``` hf://[/][@][/]:[: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_onlyFrrrc|jxsd}|jjdr|jdk(rt|d|jdy)Nrr!zEMount path must be a non-empty absolute path starting with '/', got 'r"r)rrKr'r)r(raws r*r+zHfMount.__post_init__sWiio2))#.$//S2H[\`\k\k[llno 3Ir1c|jjd|jg}|j|j |jrdnddj |S)zqRender the mount as a canonical 'hf://' string. Example: 'hf://models/my-org/my-model:/data:ro' ::ro:rwr)rJr=rKrLr:r;)r(r<s r*r=zHfMount.to_urisM ##%sDOO< >> % LL$..e <wwu~r1r>) r?r@rArBrrDr&rLrFrrr+r=rGr1r*rIrIsK" MO!Itd{!%eUDQD#*Qr1rIrrc: t|y#t$rYywxYw)z'Check if a string is a valid hf:// URI.TF) parse_hf_urir)rs r* is_hf_urirUs$S s  c||jtjs/t|dtjdtjd|}|t tjd}|st|dtjdt ||\}}|dk(rt |||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"rNr#)r'rr6rlen _split_type_parse_bucket_body_parse_repo_body)rrNbodytype_locations r*rTrTs@ >>)// 0  5 567 ) 5 566Y [  C s9(()+ ,D  293H3H2ILMM!$C0OE8 !(Es;; He 55r1 mount_strc|jtjst|dtjd|}|t tjd}|st|dtjdt ||\}}}| t|dtj|z} t |}t||||S#t$r}t||j|d}~wwxYw) 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) ``` rWr"rNrXrYzKMissing mount path. Expected ':' (e.g. 'hf://org/model:/data').)rJrKrLr) r'rr6rrZ _split_mountrTr rI) rarNr^r`rKrLuri_strrJr)s r*parse_hf_mountres@    5 5 6#I$9$9#:"=  C S../1 2D S(:9;P;P:QQS&TUU&24S&A#Hj)S&stt##h.G4g& &Z9SV WW 4Saee,!34s2 C C2C--C2r^rNc0|jdrd|jd}}n'|jdrd|jd}}nd}|jd}|dk(r| t|d |ddfS|d|}||d zd}|s t|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. rQTrRFNz:/zb':ro'/':rw' suffix is only valid when a mount path is provided (e.g. 'hf://...:/:ro').rr z#Missing location before mount path.)endswith removesuffixrfindr)r^rNrLidxr`rKs r*rcrc-s  }}U 1 1% 84 u !2!25!94   **T C by  x T4DSzHcAgiJ S&KLL Z **r1r`c |jd}|dk(rr|tjvr$t|d|dtj|dt j |x}!t|dtj|d d |fS|d|}||d zd}|tjvrtj||fSt j |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!rgzMissing identifier after 'z '. Expected 'z/'.rNz*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_PREFIXESrr6r get)r`rN slash_idxsingular_pluralfirstrests r*r[r[Ks/  c"IB y55 50 - H]H]G^_g^hhop  /228< [@][/]'.r!zMissing repository id.rr5rgNr rvz-Repository id must be 'namespace/name', got 'z'. rrz!Missing repository id before '@'.r"zEmpty revision after '@'.rx) ryrrnr%rzrZr7matchgroup removeprefixrr) r`r_rNat_idxrr<repo_idr rev_and_pathr~rqs r*r]r]s~~c"H S&>?? ]]3 F|x(..s3a7sA& u:>,YZbYccf*gh h1XJaaz*#&u:>uQxr 7F# - *MN N ==  ",YZaYbbd*ef f-22<@  {{}H'H 8EEcJL$))#.IB'! ' 3+IMO< 8$*EF F   !   r1)'rB functoolsre dataclassesrr urllib.parserhuggingface_hubrhuggingface_hub.errorsrr _validatorsr roitemsr dictr&rDcompiler7 frozensetvaluesr rrIrFrU lru_cacherTretuplercrCr[r\r])kvs00r*rs, ( %@) 5>4R4R4X4X4Z"[DAq1a4"[c3h[ *rzz*MN$-Y-K-K-R-R-T#U)C.U $JJJZ $'''T34 /6c/6e/6/6d7Xc7Xg7Xt+s+C+E#sTz4$;2N,O+<#suY5H5H#5M/N>      088   8 8  8c #\s E