
    g*                        d Z ddlZddlZddlmZmZmZmZmZm	Z	m
Z
 ddlZ ed      Z G d dee         Z G d dee         Z G d	 d
      Zg dZ	 ddej"                  deegef   dee   de	edf   fdZ	 ddej"                  deegef   dee   deeddf   fdZ	 ddedeegef   dee   de
ee   ef   fdZdededefdZy)zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.    N)CallableGenericTypeVarOptional	GeneratorAsyncGeneratorTupleTc                       e Zd ZU ej                  ed<   eeddf   ed<   	 ddej                  dee	gef   de
e	   fdZd Zd Zd	 Zd
 Zy)EventStreamresponseN	generatordecodersentinelc                 6    || _         t        |||      | _        y N)r   stream_eventsr   selfr   r   r   s       Y/var/www/openai/venv/lib/python3.12/site-packages/mistralai_azure/utils/eventstreaming.py__init__zEventStream.__init__   s     !&x(C    c                     | S r    r   s    r   __iter__zEventStream.__iter__        r   c                 ,    t        | j                        S r   )nextr   r   s    r   __next__zEventStream.__next__#   s    DNN##r   c                     | S r   r   r   s    r   	__enter__zEventStream.__enter__&   r   r   c                 8    | j                   j                          y r   )r   closer   exc_typeexc_valexc_tbs       r   __exit__zEventStream.__exit__)   s    r   r   )__name__
__module____qualname__httpxResponse__annotations__r   r
   r   strr   r   r   r    r"   r)   r   r   r   r   r      sn    nnD$'' #'	D..D 3%(#D 3-	D$r   r   c                       e Zd ZU ej                  ed<   eedf   ed<   	 ddej                  dee	gef   de
e	   fdZd Zd Zd	 Zd
 Zy)EventStreamAsyncr   Nr   r   r   c                 6    || _         t        |||      | _        y r   )r   stream_events_asyncr   r   s       r   r   zEventStreamAsync.__init__1   s     !,XwIr   c                     | S r   r   r   s    r   	__aiter__zEventStreamAsync.__aiter__:   r   r   c                 R   K   | j                   j                          d {   S 7 wr   )r   	__anext__r   s    r   r8   zEventStreamAsync.__anext__=   s     ^^--////s   '%'c                    K   | S wr   r   r   s    r   
__aenter__zEventStreamAsync.__aenter__@   s     s   c                 T   K   | j                   j                          d {    y 7 wr   )r   acloser%   s       r   	__aexit__zEventStreamAsync.__aexit__C   s     mm""$$$s   (&(r   )r*   r+   r,   r-   r.   r/   r   r
   r   r0   r   r   r6   r8   r:   r=   r   r   r   r2   r2   -   sl    nnag&& #'	J..J 3%(#J 3-	J0%r   r2   c                   ^    e Zd ZU dZee   ed<   dZee   ed<   dZee   ed<   dZ	ee
   ed<   y)ServerEventNideventdataretry)r*   r+   r,   r@   r   r0   r/   rA   rB   rC   intr   r   r   r?   r?   G   s:    BE8C=D(3-E8C=r   r?   )s   

s   

s   r   r   r   returnc                  K   t               }d}d}| j                         2 3 d {   }|r||z  }t        |t        |            D ]Z  }|||dz    }d }	|dv rt        D ]  }
t        |||
      }	|	 n |	0||| }|t        |	      z   }t        |||      \  }}|V| \ |dkD  s||d  }d}7 6 t        |||      \  }}|| y y wNr   F   )      
)	bytearrayaiter_bytesrangelenMESSAGE_BOUNDARIES_peek_sequence_parse_eventr   r   r   bufferpositiondiscardchunkicharseqboundaryblockrA   s                r   r4   r4   U   s    
 [FHG++- e %xV-A!a!e$D#'C~% 2H(FH=C !3 {8A&E3s8|H)%(CNE7  ." a<HI&FH7-: "&'8<NE7 s8   CB4B2B4AC1+CC*C2B44Cc              #     K   t               }d}d}| j                         D ]  }|r||z  }t        |t        |            D ]Y  }|||dz    }d }	|dv rt        D ]  }
t        |||
      }	|	 n |	0||| }|t        |	      z   }t        |||      \  }}|V| [ |dkD  s||d  }d} t        |||      \  }}|| y y wrG   )rK   
iter_bytesrM   rN   rO   rP   rQ   rR   s                r   r   r      s    
 [FHG$$& %xV-A!a!e$D#'C~% 2H(FH=C !3 {8A&E3s8|H)%(CNE7  ." a<HI&FH7 ': "&'8<NE7 s   A(C++CC#!Crawc                 l   | j                         }t        j                  d|      }d}t               }d}|D ]  }|s|j	                  d      }	|	dk  r|d|	 }
|	t        |      dz
  k  r||	dz   d  nd}t        |      r|d   dk(  r|dd  }|
dk(  r
||_        d	}d|
d
k(  r||dz   z  }d	}t|
dk(  r
||_        d	}|
dk(  s|j                         rt        |      nd |_
        d	} |r	|| dk(  ry|r|d d }||_        |j                         xs |dk(  xs |dk(  xs |dk(  }|j                  d      xs$ |j                  d      xs |j                  d      }|s|r	 t        j                  |      |_        d }|r% |t        j"                  |j$                              }|dfS # t         $ r Y 8w xY w)Nz\r?\n|\rF :r   rH    rA   TrB   
r@   rC   )NTtruefalsenull{[")decoderesplitr?   findrN   rA   r@   isdigitrD   rC   rB   	isnumeric
startswithjsonloads	Exceptiondumps__dict__)r^   r   r   r[   linespublishrA   rB   linedelimfieldvaluedata_is_primitivedata_is_jsonouts                  r   rQ   rQ      s    JJLEHH[%(EGMED		#A:Qu%*SY]%:UQY[!u:%(c/!"IEGEKGf_EDL DGd]EHGg(-#e*TEKG1 4 DxjO+CRy
 NNSS$'/STV^ 	 OOC PDOOC$8PDOOC<P 	 !ZZ-
 Cdjj01:  s    F' '	F32F3rT   rS   sequencec                 |    t        |      t        |      | z
  kD  ry t        |      D ]  \  }}|| |z      |k7  s y  |S r   )rN   	enumerate)rT   rS   r   rW   rY   s        r   rP   rP      sH    
8}Fh./H%3(Q,3& & Or   r   )__doc__rl   rr   typingr   r   r   r   r   r   r	   r-   r
   r   r2   r?   rO   r.   r0   r4   r   rK   boolrQ   rD   bytesrP   r   r   r   <module>r      sY   G 	    CL'!* 4%wqz %4     #'nn'seQh' sm' AtG	'Z #'nn'seQh' sm' q$}	'V LP:	:%seQh/:;CC=:
8A;:zS ) u r   