o
    VÛ·iè  ã                   @   sÀ   d Z ddlmZmZmZ ddlmZ dd„ Zejdd„ ƒZ	d	d
„ Z
dd„ Zejdd„ ƒZdd„ Zdd„ Zejd dd„ƒZd dd„Zd dd„Zejd dd„ƒZd dd„Zd dd„Ze eƒ ¡ dS )!z'
Wrapper for _yajl2 C extension module
é    )ÚcommonÚcompatÚutilsé   )Ú_yajl2c                 C   s   |   dd¡S )NÚbuf_sizei   )Úpop)Úkwargs© r
   úL/home/ubuntu/vllm_env/lib/python3.10/site-packages/ijson/backends/yajl2_c.pyÚ<lambda>   s    r   c                 K   ó   t j| jfi |¤ŽS ©N)r   Úbasic_parse_basecoroÚsend©Útargetr	   r
   r
   r   r      ó   r   c                 K   ó&   t  | ¡}t|ƒ}tj||fi |¤ŽS r   )r   Úbytes_readerÚ_get_buf_sizer   Úbasic_parse©Úfiler	   Úfr   r
   r
   r   Úbasic_parse_gen   ó   
r   c                 K   ó   t |ƒ}tj| |fi |¤ŽS r   )r   r   Úbasic_parse_async©r   r	   r   r
   r
   r   r      ó   r   c                 K   r   r   )r   Úparse_basecoror   r   r
   r
   r   r!      r   r!   c                 K   r   r   )r   r   r   r   Úparser   r
   r
   r   Ú	parse_gen#   r   r#   c                 K   r   r   )r   r   Úparse_asyncr   r
   r
   r   r$   (   r    r$   Nc                 K   ó   t j| j||fi |¤ŽS r   )r   Úkvitems_basecoror   ©r   ÚprefixÚmap_typer	   r
   r
   r   r&   ,   ó   r&   c                 K   ó*   t  | ¡}t|ƒ}tj||||fi |¤ŽS r   )r   r   r   r   Úkvitems©r   r(   r)   r	   r   r   r
   r
   r   Úkvitems_gen0   ó   
r.   c                 K   ó    t |ƒ}tj| |||fi |¤ŽS r   )r   r   Úkvitems_async©r   r(   r)   r	   r   r
   r
   r   r1   5   ó   r1   c                 K   r%   r   )r   Úitems_basecoror   r'   r
   r
   r   r4   9   r*   r4   c                 K   r+   r   )r   r   r   r   Úitemsr-   r
   r
   r   Ú	items_gen=   r/   r6   c                 K   r0   r   )r   r   Úitems_asyncr2   r
   r
   r   r7   B   r3   r7   r   )Ú__doc__Úijsonr   r   r   Ú r   r   Ú	coroutiner   r   r   r!   r#   r$   r&   r.   r1   r4   r6   r7   Úenrich_backendÚglobalsr
   r
   r
   r   Ú<module>   s*   





