}tjddlZddlZddlZddlmZddlmZddlZddlm Z ddlm Z ej dZ d e d efd ZGd d e eZedGddZd!dejde dzd edzfdZe ddddede de de dzde dzd e f dZGddZeZdZdee e fd ee e ffdZd ZdS)"N) dataclass)Enum) constants)validate_hf_hub_argsz^[0-9a-fA-F]{64}$xet_hashreturncPtt|S)zLReturn whether `xet_hash` looks like a real Xet content hash (64 hex chars).)bool_REGEX_XET_HASH fullmatch)r s X/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/huggingface_hub/utils/_xet.pyis_valid_xet_hashrs ))(33 4 44ceZdZdZdZdS) XetTokenTypereadwriteN)__name__ __module__ __qualname__READWRITErrrrs D EEErrT)frozenc$eZdZUeed<eed<dS) XetFileData file_hash refresh_routeN)rrrstr__annotations__rrrrrs'NNNrrresponseendpointc|dS |jtj}tj|jvr|jtjd}n|jtj}n#t $rYdSwxYw||n tj}|tj rF| tj d| d}t||S)a^ Parse XET file metadata from an HTTP response. This function extracts XET file metadata from the HTTP headers or HTTP links of a given response object. If the required metadata is not found, it returns `None`. Args: response (`httpx.Response`): The HTTP response object containing headers dict and links dict to extract the XET metadata from. Returns: `Optional[XetFileData]`: An instance of `XetFileData` containing the file hash and refresh route if the metadata is found. Returns `None` if the required metadata is missing. Nurl/)rr ) headersrHUGGINGFACE_HEADER_X_XET_HASH$HUGGINGFACE_HEADER_LINK_XET_AUTH_KEYlinks&HUGGINGFACE_HEADER_X_XET_REFRESH_ROUTEKeyErrorENDPOINT startswithHUGGINGFACE_CO_URL_HOMEreplacerstripr)r#r$rr s r!parse_xet_file_data_from_responser3 st$Y%LM  9X^ K K$N9+YZ[`aMM$,Y-]^M tt#/xxY5GH ABBs%--i.O.V.VWZ.[.[]e]l]lmp]q]qrr #   sAA&& A43A4)revisionr$ token_typerepo_id repo_typer4cl||n tj}|d|d|d|jd}|dks||d|z }|S)a Build the URL used to fetch or refresh a Xet access token for a given repo. Args: token_type (`XetTokenType`): Type of the token to request: `"read"` or `"write"`. repo_id (`str`): A namespace (user or an organization) and a repo name separated by a `/`. repo_type (`str`): Type of the repo (e.g. `"model"`, `"dataset"`, `"space"`, `"bucket"`). revision (`str`, `optional`): The revision of the repo to get the token for. endpoint (`str`, `optional`): The endpoint to use for the request. Defaults to the Hub endpoint. Returns: `str`: The fully-qualified URL of the token refresh endpoint. Nz/api/zs/z/xet-z-tokenbucketr')rr.value)r5r6r7r4r$r&s rxet_connection_info_refresh_urlr;Cse4$/xxY5GH  O OI O O O Oz7G O O OCH 4 ~8~~ Jrc$eZdZdZdZdZdZdS)XetSessionHoldera7Holds an optional XetSession; supports safe re-creation after sigint_abort or fork. Thread-safe: a ``threading.Lock`` guards all state mutations, which matters for free-threaded Python (3.14t) where multiple threads can race on ``get()`` or ``sigint_abort()`` without the GIL serialising them. cRtj|_d|_d|_dSN) threadingLock_lock_session _session_pidselfs r__init__zXetSessionHolder.__init__ps&^%%  (,rc|j5tj}|j|j|krd|_|jddlm}||_||_|jcdddS#1swxYwYdS)aReturn the current session, creating one if needed. Fork-safe: if the current process PID differs from the PID that created the session (i.e. we are in a forked child), the old session is discarded and a fresh session is created for this process. Nr) XetSession)rBosgetpidrCrDhf_xetrI)rF current_pidrIs rgetzXetSessionHolder.getusZ ! !)++K}(T->+-M-M!% }$------ *  $/!= ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !sAA++A/2A/c|j5|j9 |jn#t$rYnwxYwd|_d|_ddddS#1swxYwYdS)zMAbort the current session and clear it so the next get() creates a fresh one.N)rBrC sigint_abort ExceptionrDrEs rrPzXetSessionHolder.sigint_aborts Z ) )}(M..0000 D $ $(! ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s,A+A 8A8AAAN)rrr__doc__rGrNrPrrrr=r=hsK--- !!!0 ) ) ) ) )rr=c4tS)a*Return the global :class:`hf_xet.XetSession`, creating it on first call. The session is shared across all calls within a process, just as the HTTP client returned by :func:`~huggingface_hub.utils._http.get_session` is shared. It is created lazily and is fork-safe and thread-safe. )_GLOBAL_XET_HOLDERrNrrrget_xet_sessionrUs  ! ! # ##rr(c>d|DS)zReturn a copy of headers with the authorization header removed. Xet storage requests use a short-lived xet access token for auth, so the Hub authorization header must not be forwarded to xet storage endpoints. cJi|] \}}|dk||!S) authorization)lower).0keyr:s r z,xet_headers_without_auth..s0 [ [ [:3CIIKK?rise !!!!!! """""""*122555555 3  $   #PT* `knr`r    F   !!!!! ! Dj ! Dj ! !!!!H.).).).).).).).)b&%''$$$\d38n\c3h\\\\&&&&&r