
    QЦiW                     
   S SK r S SKJr  S SKJrJrJrJrJr  S SK	r	SSK
Jr  SSKJr  SSKJr  SS	KJrJr  S
r " S S\5      r " S S\5      r " S S\5      r " S S5      rS\S\S\S\\   S\\   S\\\\\4      4S jrg)    N)deque)IteratorLiteralOptional	TypedDictUnion   )build_hf_headers)hf_raise_for_status   )	SSEClient)ApiGetReloadEventSourceDataApiGetReloadRequesti  c                   0    \ rS rSr% \S   \S'   \\S'   Srg)MultiReplicaStreamEvent   eventkind N)__name__
__module____qualname____firstlineno__r   __annotations__r   __static_attributes__r       a/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/huggingface_hub/_hot_reload/client.pyr   r      s    
'
&&r   r   c                   0    \ rS rSr% \S   \S'   \\S'   Srg)MultiReplicaStreamReplicaHash$   replicaHashr   hashr   N)r   r   r   r   r   r   strr   r   r   r   r   r   $   s    
-
  
Ir   r   c                   &    \ rS rSr% \S   \S'   Srg)MultiReplicaStreamFullMatch)   	fullMatchr   r   N)r   r   r   r   r   r   r   r   r   r   r%   r%   )   s    
+
r   r%   c            	       L    \ rS rSrS\S\S\S\\   4S jrS\S\\   4S	 jr	S
r
g)ReloadClient-   host	subdomainreplica_hashtokenc                    UR                  X" S[         35      nX0l        [        R                  " U SU 3[        US9S9U l        g )Nz--z/--replicas/+)r.   )base_urlheaders)replaceHOT_RELOADING_PORTr-   httpxClientr
   client)selfr+   r,   r-   r.   	base_hosts         r   __init__ReloadClient.__init__.   sK     LLk<N;O,PQ	(ll!{-~>$51
r   	reload_idreturnc              #   b  #    [        US9nU R                  R                  SSUS9 n[        U5        [	        UR                  5       5      R                  5        H7  nUR                  S:X  d  M  [        R                  " UR                  5      v   M9     S S S 5        g ! , (       d  f       g = f7f)N)reloadIdPOSTz/get-reload)jsonmessage)r   r6   streamr   r   
iter_byteseventsr   r@   loadsdata)r7   r;   reqresr   s        r   
get_reloadReloadClient.get_reload=   s|     !95[[C@C$"3>>#34;;=;;)+**UZZ00 > A@@s#   %B/AB/&B	B/
B,(B/)r6   r-   N)r   r   r   r   r#   r   r9   r   r   rI   r   r   r   r   r)   r)   -   sL    
 
 	

 
 }
1C 1H5P,Q 1r   r)   
commit_shar+   r,   replica_hashesr.   r<   c           
   #   "  #    U Vs/ s H  n[        UUUUS9PM     nn0 n[        U5       H  u  p[        U5      S:  a  SU	R                  S.v   Sn
[	        5       n[        U	R                  U 5      5       Hs  u  pUS:X  a  XU'   O7U
=(       a    UR                  U5      U:H  =n
(       a  UR                  U5        MG  U(       a  SUR                  5       S.v   U(       a  M  SUS.v   Mu     US:  d  M  U
(       d  M  S	S
0v   M     g s  snf 7f)N)r+   r,   r-   r.   r   r!   )r   r"   Tr   r   )r   r   r   r'   )	r)   	enumeratelenr-   r   rI   getappendpopleft)rK   r+   r,   rL   r.   r"   clientsfirst_client_eventsclient_indexr6   
full_matchreplayevent_indexr   s                 r   multi_replica_reload_eventsrY   F   s%     # #D 			
 #   CE )' 2w<!(&2E2EFF
5:W"+F,=,=j,I"JKq 38K0)[.A.E.Ek.RV[.[[[e$&1ABB &"U33 #K !

;'' !3s"   DD
CD#D6D?D)r@   collectionsr   typingr   r   r   r   r   r4   utils._headersr
   utils._httpr   
sse_clientr   typesr   r   r3   r   r   r%   r)   r#   listrY   r   r   r   <module>ra      s       @ @  - - ! C  'i '
I 
) 1 12!(!(
!( !( I	!(
 C=!( e+-JLgghi!(r   