#hLddlmZddlmZddlmZmZddlZddlm Z ddlm Z m Z m Z Gd d e eZed Gd dZed GddZdej"deefdZdee e fdeefdZe dddedee e fdee defdZe dddddede de dee dee e fdee deee e fdefd Ze d#d!e dee e fdeee e fdefd"Zy)$) dataclass)Enum)DictOptionalN) constants) get_sessionhf_raise_for_statusvalidate_hf_hub_argsceZdZdZdZy) XetTokenTypereadwriteN)__name__ __module__ __qualname__READWRITEW/var/www/html/sandstorm/venv/lib/python3.12/site-packages/huggingface_hub/utils/_xet.pyrr s D ErrT)frozenc"eZdZUeed<eed<y) XetFileData file_hash refresh_routeN)rrrstr__annotations__rrrrrs Nrrc,eZdZUeed<eed<eed<y)XetConnectionInfo access_tokenexpiration_unix_epochendpointN)rrrrrintrrrr!r!sMrr!responsereturnc0|y |jtj}tj|jvr!|jtjd}n|jtj }t||S#t $rYywxYw)aa 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 (`requests.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_ROUTEKeyErrorr)r&rrs r!parse_xet_file_data_from_responser0s$$Y%L%LM  9 9X^^ K$NN9+Y+YZ[`aM$,,Y-]-]^M #  sA7B BBr*c |tj}|tj}t|tj}t|||S#t t tf$rYywxYw)a Parse XET connection info from the HTTP headers or return None if not found. Args: headers (`Dict`): HTTP headers to extract the XET metadata from. Returns: `XetConnectionInfo` or `None`: The information needed to connect to the XET storage service. Returns `None` if the headers do not contain the XET connection info. N)r$r"r#) r!HUGGINGFACE_HEADER_X_XET_ENDPOINT%HUGGINGFACE_HEADER_X_XET_ACCESS_TOKENr%#HUGGINGFACE_HEADER_X_XET_EXPIRATIONr/ ValueError TypeErrorr!)r*r$r"r#s r&parse_xet_connection_info_from_headersr7>sp9FFGyNNO #GI,Y,Y$Z [ !3  j) ,sAAA)(A))r$ file_datar$c|j td||ntj}|j}|j dr||}t ||S)a Utilizes the information in the parsed metadata to request the Hub xet connection information. This includes the access token, expiration, and XET service URL. Args: file_data: (`XetFileData`): The file data needed to refresh the xet connection information. headers (`Dict[str, str]`): Headers to use for the request, including authorization headers and user agent. endpoint (`str`, `optional`): The endpoint to use for the request. Defaults to the Hub endpoint. Returns: `XetConnectionInfo`: The connection information needed to make the request to the xet storage service. Raises: [`~utils.HfHubHTTPError`] If the Hub API returned an error. [`ValueError`](https://docs.python.org/3/library/exceptions.html#ValueError) If the Hub API response is improperly formatted. z>The provided xet metadata does not contain a refresh endpoint./)rr5rENDPOINT startswith#_fetch_xet_connection_info_with_url)r8r*r$r)s rrefresh_xet_connection_infor>Wsb4&YZZ#/xY5G5GH  ! !C ~~c 3%  .sG <rFr=rrrrQs!!DD3  $  $ 0A0Ah{F[BDcNxPaGb2 # #=#=#s(^#=sm #=  #=#=L #"'+&E&E&E &E sm &E #s(^ &Esm&E T#s(^ $&E&E&ER(,  #s(^ T#s(^ $ r