a
    _g~                     @  st   d dl mZ d dlZd dlmZ d dlmZ G dd deZG dd dej	Z
d	d
ddddZdddddZdS )    )annotationsN)Callable)overridec                      s:   e Zd Zddd fddZeddddZeZ  ZS )	CancelledErrorstrNone)msgreturnc                   s   || _ t | d S N)r   super__init__)selfr   	__class__ b/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/openai/cli/_progress.pyr   	   s    zCancelledError.__init__r	   c                 C  s   | j S r
   )r   r   r   r   r   __str__   s    zCancelledError.__str__)__name__
__module____qualname__r   r   r   __repr____classcell__r   r   r   r   r      s   r   c                      sL   e Zd Zddddd fddZd	d
ddZeddddddZ  ZS )BufferReader    Nbytes
str | Noner   )bufdescr	   c                   s2   t  | t|| _d| _tt||d| _d S )Nr   )r   )r   r   len_len	_progressprogress	_callback)r   r   r   r   r   r   r      s    
zBufferReader.__init__intr   c                 C  s   | j S r
   )r!   r   r   r   r   __len__   s    zBufferReader.__len__z
int | None)nr	   c              
   C  sj   t j| |}|  jt|7  _z| | j W n4 tyd } ztd||W Y d }~n
d }~0 0 |S )NzThe upload was cancelled: {})	ioBytesIOreadr"   r    r$   	Exceptionr   format)r   r(   chunker   r   r   r+      s    &zBufferReader.read)r   N)r'   )r   r   r   r   r&   r   r+   r   r   r   r   r   r      s   r   floatr   zCallable[[float], None])totalr   r	   c                   s2   dd l }|j d|d ddd fdd}|S )	Nr   T)r1   Z
unit_scaler   r0   r   )r#   r	   c                   s$   |  _ | kr   n   d S r
   )r(   closerefresh)r#   Zmeterr1   r   r   incr0   s    
zprogress.<locals>.incr)tqdm)r1   r   r6   r5   r   r4   r   r#   +   s    r#   r%   )ir	   c                 C  s   t | d S )Ni   )r%   )r7   r   r   r   MB:   s    r8   )
__future__r   r)   typingr   Ztyping_extensionsr   r,   r   r*   r   r#   r8   r   r   r   r   <module>   s   