U
    ·ÜŸjP  ã                   @  s�   d dl mZ d dlZd dlZddlmZmZ ddlmZ ddl	m
Z
 G dd„ dƒZd	d
œdd„Zdd
œdd„Zdddœdd„Zdd
œdd„ZdS )é    )ÚannotationsNé   )ÚOpenAIÚ_load_client)Ú
model_json)Ú	BaseModelc                   @  s,   e Zd ZdZdZdZdZdZdZdZ	dZ
d	S )
ÚColorsz[95mz[94mz[92mz[93mz[91mz[0mz[1mz[4mN)Ú__name__Ú
__module__Ú__qualname__ÚHEADERZOKBLUEZOKGREENÚWARNINGÚFAILZENDCZBOLDZ	UNDERLINE© r   r   ú5/tmp/pip-unpacked-wheel-2qj3ofc_/openai/cli/_utils.pyr      s   r   r   )Úreturnc                   C  s   t ƒ S )N)r   r   r   r   r   Ú
get_client   s    r   Ústrc                  C  s   t j} | d k	rd | ¡S dS )Nz[organization={}] Ú )ÚopenaiÚorganizationÚformat)r   r   r   r   Úorganization_info   s    
r   r   ÚNone)Úmodelr   c                 C  s   t j t| dd�d ¡ d S )Nr   )ÚindentÚ
)ÚsysÚstdoutÚwriter   )r   r   r   r   Úprint_model#   s    r    Úboolc                  C  s(   zdd l } W n tk
r"   Y dS X dS )Nr   FT)Úh2ÚImportError)r"   r   r   r   Úcan_use_http2'   s
    r$   )Ú
__future__r   r   r   r   r   r   Ú_compatr   Z_modelsr   r   r   r   r    r$   r   r   r   r   Ú<module>   s   