
    Pi-                     ,   d dl mZmZmZ d dlmZ ddlmZ ddlm	Z	 ddl
mZ ddlmZ  ee          Z G d	 d
e	          Zdededeeeef                  deeeef                  fdZdededeeeef                  deeeeef         f         fdZdS )    )AnyOptionalUnion)get_session   )config)DatasetsError   )"get_authentication_headers_for_url)
get_loggerc                       e Zd ZdZdS )DatasetViewerErrorzDataset viewer error.

    Raised when trying to use the dataset viewer HTTP API and when trying to access:
    - a missing dataset, or
    - a private/gated dataset and the user is not authenticated.
    - unavailable /parquet or /info responses
    N)__name__
__module____qualname____doc__     r/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/datasets/utils/_dataset_viewer.pyr   r      s           r   r   datasetcommit_hashtokenreturnc           
      L   t           j                            dd          dz   }	 t                                          || z   t          t           j        d|  z   |          d          }|                                 d|j        v r|j        d         |k    s||                                }|                    d
          du r8|                    dd          s"|                    dd          sd|v r|d         S t          
                    d|  d           n,t          
                    d|  d|j        d          d           nM# t          $ r@}t          
                    d|  dt          |          j         d| d           Y d	}~nd	}~ww xY wt          d          )zn
    Get the dataset exported parquet files
    Docs: https://huggingface.co/docs/datasets-server/parquet
    ://://datasets-server.z/parquet?dataset=	datasets/r         Y@urlheaderstimeout
X-RevisionNpartialFpendingTfailedparquet_fileszParquet export for  is not completely ready yet.) is available but outdated (commit_hash='')zNo parquet export for  available (: )z$No exported Parquet files available.r   HF_ENDPOINTreplacer   getr   raise_for_statusr"   jsonloggerdebug	Exceptiontyper   r   )r   r   r   dataset_viewer_parquet_urlparquet_data_files_response parquet_data_files_response_jsones          r   get_exported_parquet_filesr=      s    "(!3!;!;ECX!Y!Y\o!o]&1mm&7&7*W46v7ILaX_LaLa7ainooo '8 '
 '
#
 	$446666>>>*2<@KOOS^Sf3N3S3S3U3U0488CCuLL<@@DQQ M<@@4PP M (+KKK;OLLLL!]w!]!]!]^^^^ R'  R  R\w\  AM  ]N  R  R  R    ] ] ][g[[477CS[[WX[[[\\\\\\\\]
C
D
DD   CE
 >AE
 

F6FFc           
      L   t           j                            dd          dz   }	 t                                          || z   t          t           j        d|  z   |          d          }|                                 d|j        v r|j        d         |k    s||                                }|                    d
          du r8|                    dd          s"|                    dd          sd|v r|d         S t          
                    d|  d           n,t          
                    d|  d|j        d          d           nM# t          $ r@}t          
                    d|  dt          |          j         d| d           Y d	}~nd	}~ww xY wt          d          )z
    Get the dataset information, can be useful to get e.g. the dataset features.
    Docs: https://huggingface.co/docs/datasets-server/info
    r   r   z/info?dataset=r   r   r   r    r$   Nr%   Fr&   Tr'   dataset_infozDataset info for r)   r*   r+   zNo dataset info for r,   r-   r.   z$No exported dataset infos available.r/   )r   r   r   dataset_viewer_info_urlinfo_responser<   s         r   get_exported_dataset_infosrC   >   s    %088@UVVYii[#))''16v7ILaX_LaLa7ainooo * 
 

 	&&(((=000$\2kAA[EX - 2 2 4 4!%%i00E99)--i>> :)--h== : '-77(88LL![W![![![\\\\ B  B  BZgZop|Z}  B  B  B    [ [ [YGYYaAQYYUVYYYZZZZZZZZ[
C
D
DDr>   N)typingr   r   r   huggingface_hub.utilsr    r   
exceptionsr	   
file_utilsr   loggingr   r   r5   r   strboollistdictr=   rC   r   r   r   <module>rN      s   ' ' ' ' ' ' ' ' ' ' - - - - - -       & & & & & &             
H		       !E!E"!E+3E#t)4D+E!E	$sCx.!E !E !E !EH!E!E"!E+3E#t)4D+E!E	#tCH~
!E !E !E !E !E !Er   