tj'ddlmZddlZddlZddlmZddlmZmZddl m Z m Z m Z ddl mZejeZGddeZ dd dZ d!d"dZ dd#dZd$dZdS)%) annotationsN)Path)hf_hub_downloadsnapshot_download)EntryNotFoundErrorHFValidationErrorLocalEntryNotFoundError)tqdmc.eZdZdZfdZdfd ZxZS) disabled_tqdmz Class to override `disable` argument in case progress bars are globally disabled. Taken from https://github.com/tqdm/tqdm/issues/619#issuecomment-619639324. cDd|d<tj|i|dS)NTdisable)super__init__)selfargskwargs __class__s `/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/sentence_transformers/util/file_io.pyrzdisabled_tqdm.__init__s/ y$)&)))))attrstrreturnNonec| t|dS#t$r |dkrYdSwxYw)zBFix for https://github.com/huggingface/huggingface_hub/issues/1603_lockN)r __delattr__AttributeError)rrrs rrzdisabled_tqdm.__delattr__sV  GG   % % % % %   w s !&;;)rrrr)__name__ __module__ __qualname____doc__rr __classcell__)rs@rr r s` *****rr Fmodel_name_or_pathrtokenbool | str | None cache_folder str | Nonerevisionlocal_files_onlyboolrc Ftt|d||||S)a Checks if the given model name or path corresponds to a SentenceTransformer model. Args: model_name_or_path (str): The name or path of the model. token (Optional[Union[bool, str]]): The token to be used for authentication. Defaults to None. cache_folder (Optional[str]): The folder to cache the model files. Defaults to None. revision (Optional[str]): The revision of the model. Defaults to None. local_files_only (bool): Whether to only use local files for the model. Defaults to False. Raises: Exception: Propagates errors from the Hub that are not "file not found" (e.g. authentication, rate-limit, or network errors). The probe deliberately surfaces these instead of returning ``False``, so callers don't mistake a flaky Hub for "this isn't a SentenceTransformer". Returns: bool: True if the model is a SentenceTransformer model, False if it exists but does not contain a ``modules.json``. z modules.json)r%r'r)r*)r+load_file_path)r$r%r'r)r*s ris_sentence_transformer_modelr."s:6   %-       rfilename str | Path subfolderc t|||}|rt|St|rdSt||} t ||j|j|d|||S#tttf$r.}t d|d|d|Yd}~dSd}~wwxYw)a Loads a file from a local or remote location. Args: model_name_or_path (str): The model name or path. filename (str): The name of the file to load. subfolder (str): The subfolder within the model subfolder (if applicable). token (Optional[Union[bool, str]]): The token to access the remote file (if applicable). cache_folder (Optional[str]): The folder to cache the downloaded file (if applicable). revision (Optional[str], optional): The revision of the file (if applicable). Defaults to None. local_files_only (bool, optional): Whether to only consider local files. Defaults to False. Raises: Exception: Errors that are not unambiguously "this file is not on the Hub" propagate to the caller (e.g. authentication, rate-limit, network). Only ``EntryNotFoundError``, ``HFValidationError``, and ``LocalEntryNotFoundError`` are converted to a ``None`` return. Returns: Optional[str]: The path to the loaded file, or ``None`` if the file is not present locally and the Hub confirms it is not in the repo (or no valid repo id was given). Nsentence-transformers)r0r2r) library_namer% cache_dirr*zCould not load  from : ) rexistsris_dirrnameparentas_posixrrr loggerdebug) r$r0r2r%r'r)r* file_pathexcs rr-r-Is@'H==I9~~ &&((tY))I ^&//110"-      13J K  VxVV9KVVQTVVWWWttttt s)2BC!3#CC!c xt|tr|}t||}|rt |St|rdS|||dvr|dndd|||t d} td i|}nx#ttf$r.} t d|d|d| Yd} ~ dSd} ~ wt$r.} d |d < td i|}n#t$r| dwxYwYd} ~ nd} ~ wwxYwt t||S) ar Loads the subfolder path for a given model name or path. Args: model_name_or_path (str): The name or path of the model. subfolder (str): The subfolder to load. token (Optional[Union[bool, str]]): The token for authentication. cache_folder (Optional[str]): The folder to cache the downloaded files. revision (Optional[str], optional): The revision of the model. Defaults to None. local_files_only (bool, optional): Whether to only use local files. Defaults to False. Raises: Exception: Errors that are not unambiguously "this is not on the Hub" propagate to the caller. ``HFValidationError`` and ``LocalEntryNotFoundError`` are converted to a ``None`` return. Other failures (auth, rate-limit, network) trigger a single cache-only retry; if the cache also lacks the file, the original exception is re-raised (a cache miss would mask the real cause). Returns: Optional[str]: The subfolder path, or ``None`` if the parent is a local directory without the subfolder, or no valid repo id was given. N)r/.z/**r4)repo_idr)allow_patternsr5r%r6r* tqdm_classzCould not load subfolder r7r8Tr*) isinstancerr=r9rr:r rrr r>r? Exception) r$r2r%r'r)r*dir_pathdownload_kwargs repo_pathrA first_errors r load_dir_pathrNs<)T"")&&(( & 22H8}} &&((t&/8 /I/IY++++t/!,#  O(%8888 6 7 aaaDVaa\_aabbbttttt  ( ( (.2*+ ()<_part"`` file, which is atomically moved to ``path`` once the download has completed successfully. Parent directories of ``path`` are created automatically if they do not exist. Args: url (str): The HTTP(S) URL to download. path (str): Destination file path on the local filesystem. Raises: ImportError: If the optional ``httpx`` dependency is not installed. httpx.HTTPStatusError: If the HTTP request returns a non-success status code. OSError: If the file cannot be written to ``path``. Returns: None rNzRhttpx is required to use this function. Please install it via `pip install httpx`.r/T)exist_ok_partGET)follow_redirectszContent-LengthBFz Downloading )unittotal unit_scaleleavedescwbi) chunk_size)httpx ImportErrorosrPdirnamemakedirsstreamraise_for_statusheadersgetintr basenameopen iter_bytesupdatelenwritereplacerIr9removeclose) rOrPr^download_filepathresponsecontent_lengthrXprogress file_binarychunks rhttp_getrws(p pppnooop wt"" BGOOD))D9999w eS4 8 8H!!###!)--.>??'5'AN###tEd%FmUWU\UeUefjUkUkFmFm    '.. 1+%00D0AA11E1 E 333#))%0001 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 J($ / / / /   w~~/00 - +,,,   0 NN    HNN     )sj!A6HFAE>2 F>F FF FG7?GG7"H7H  HH"%H")NNNF) r$rr%r&r'r(r)r(r*r+rr+)r/NNNF)r$rr0r1r2rr%r&r'r(r)r(r*r+rr()r$rr2rr%r&r'r(r)r(r*r+rr()rOrrPrrr) __future__rloggingr`pathlibrhuggingface_hubrrhuggingface_hub.utilsrrr tqdm.autonotebookr getLoggerrr>r r.r-rNrwrGrrrs]"""""" >>>>>>>>``````````""""""  8 $ $D, $#" $$$$$T##"99999~ $#" E+E+E+E+E+P111111r