&`i,%ddlZddlZddlmZGddZejddeDZedZ e e_ dZ e e_ dS) N)urlparseceZdZdZedZedZedZedZedZ edZ ede d e d e fd Z d S) ProtocolsProviderz~Note that these must be preinstalled on all nodes in the Ray cluster; it is not sufficient to install them in the runtime_env.c hdS)N> gss3uvgcspipfileabfssazurecondahttpsclss u/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/runtime_env/protocol.py get_protocolszProtocolsProvider.get_protocols s    c hdS)N>rrr r rrrrs rget_remote_protocolsz&ProtocolsProvider.get_remote_protocols&s>>>>rcJ ddl}ddlm}n%#t$rtd|jzwxYw|}|r|d}n-ddlm }ddl m }|d|| }d |i}||fS) zSet up S3 protocol handling. Returns: tuple: (open_file function, transport_params) Raises: ImportError: If required dependencies are not installed. rNopenzBYou must `pip install smart_open[s3]` to fetch URIs in s3 bucket. r)UNSIGNED)Config)signature_version)configclient) boto3 smart_openr ImportError_MISSING_DEPENDENCIES_WARNINGSessionget_credentialsr botocorerbotocore.configr)rr! open_filesession s3_clientrrtransport_paramss r_handle_s3_protocolz%ProtocolsProvider._handle_s3_protocol*s  LLL 4 4 4 4 4 4 4   /141RS  --//  " " $ $ Vt,,II * ) ) ) ) ) . . . . . . T&&82T2T2T UUI$i0***s "/cp ddlm}ddlm}n%#t$rt d|jzwxYw|dfS)zSet up Google Cloud Storage protocol handling. Returns: tuple: (open_file function, transport_params) Raises: ImportError: If required dependencies are not installed. r)storagerzTYou must `pip install smart_open[gcs]` to fetch URIs in Google Cloud Storage bucket.N) google.cloudr/r"rr#r$)rr/r)s r_handle_gs_protocolz%ProtocolsProvider._handle_gs_protocolMsx  , , , , , , 4 4 4 4 4 4 4   @34  $s "1c ddlm}ddlm}ddlm}n%#t $rt d|jzwxYwtj d}|stdd|d }d ||| i}||fS) aSet up Azure blob storage protocol handling. Returns: tuple: (open_file function, transport_params) Raises: ImportError: If required dependencies are not installed. ValueError: If required environment variables are not set. rDefaultAzureCredential)BlobServiceClientrzpYou must `pip install azure-storage-blob azure-identity smart_open[azure]` to fetch URIs in Azure Blob Storage. AZURE_STORAGE_ACCOUNTz`Azure Blob Storage authentication requires AZURE_STORAGE_ACCOUNT environment variable to be set.zhttps://z.blob.core.windows.net/r ) account_url credential) azure.identityr4azure.storage.blobr5r"rr#r$osgetenv ValueError)rr4r5r)azure_storage_account_namer7r,s r_handle_azure_protocolz(ProtocolsProvider._handle_azure_protocolcs  = = = = = = < < < < < < 4 4 4 4 4 4 4   834  &(Y/F%G%G") H  U!;TTT '''4J4J4L4L  ***s"7c ddlddlmn%#t$rtd|jzwxYwddfd }|dfS)a&Set up Azure Blob File System Secure (ABFSS) protocol handling. Returns: tuple: (open_file function, transport_params) Raises: ImportError: If required dependencies are not installed. ValueError: If the ABFSS URI format is invalid. rNr3z\You must `pip install adlfs azure-identity` to fetch URIs in Azure Blob File System Secure. r,ct|}|jr d|jvrtd||jdd\}}|std||r|dstd||dd}|std | | }|||S) N@z6Invalid ABFSS URI format - missing container@account: z1Invalid ABFSS URI format - empty container name: z.dfs.core.windows.netzSInvalid ABFSS URI format - invalid hostname (must end with .dfs.core.windows.net): .rz/Invalid ABFSS URI format - empty account name: ) account_namer8)rnetlocr=splitendswithAzureBlobFileSystemr) urimoder,parsedcontainer_part hostname_partr> filesystemr4adlfss rr)z;ProtocolsProvider._handle_abfss_protocol..open_filesGc]]F= Cv}$<$< RSRR-3M,?,?Q,G,G )NM"  MMM !  (>(>?V(W(W  ojmoo *7)<).open_filesC&rrrrr rrz+You must `pip install smart_open` to fetch z URIs. rbwb) rlenr-r1r?rRr"rr#upperr$writeread)rrSrTrUtpr)finfouts rdownload_remote_uriz%ProtocolsProvider.download_remote_uris`333555555  v  #C NN$4$45J9= ' ' ' ' ' '  3355MIrr   3355MIrr  6688MIrr  6688MIrr 8888888   !: ( 0 0:::78 Yz4" = = = 'i&& '$ 388::&&& ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 'sB:C8C9 E'(E E'E E'E E''E+.E+N) __name__ __module__ __qualname__r$ classmethodrrr-r1r?rRstrrbrrrrrs 9"   [ 0??[? + +[ +D[*%+%+[%+N::[:x*'3*'C*'C*'*'*'[*'*'*'rrProtocolc8i|]}||Srr\).0rSs r rls$RRRHX^^xRRRrcNfdtDS)NcDg|]}|Srrj)rkrSrs r z%_remote_protocols..s5   "*HNN     r)rrrs`r_remote_protocolsrps:    .?.T.T.V.V   rcDt|j||SrX)rrbvalue)selfrTrUs r_download_remote_urirts  0 0Z S SSr) enumr; urllib.parserrEnumrrhrfrpremote_protocolsrtrbrrrrys !!!!!!m'm'm'm'm'm'm'm'` 49RR0A0O0O0Q0QRRR     .TTT 4r