U
    ·ÜŸjG  ã                   @   sŒ   U d dl Z d dlZd dlmZ ddlmZ e d¡Zeje	d< e d¡Z
eje	d< d	d
hZddœdd„Zddœdd„ZG dd„ dejƒZdS )é    N)Úoverrideé   )Úis_dictZopenaiÚloggerZhttpxÚhttpx_loggerzapi-keyÚauthorization)Úreturnc                   C   s   t jddd� d S )Nz?[%(asctime)s - %(name)s:%(lineno)d - %(levelname)s] %(message)sz%Y-%m-%d %H:%M:%S)ÚformatÚdatefmt)ÚloggingÚbasicConfig© r   r   ú7/tmp/pip-unpacked-wheel-2qj3ofc_/openai/_utils/_logs.pyÚ_basic_config   s    þr   c                  C   s^   t j d¡} | dkr4tƒ  t tj¡ t tj¡ n&| dkrZtƒ  t tj	¡ t tj	¡ d S )NZ
OPENAI_LOGÚdebugÚinfo)
ÚosÚenvironÚgetr   r   ÚsetLevelr   ÚDEBUGr   ÚINFO)Úenvr   r   r   Úsetup_logging   s    r   c                   @   s"   e Zd Zeejedœdd„ƒZdS )ÚSensitiveHeadersFilter)Úrecordr   c                 C   s^   t |jƒrZd|jkrZt |jd ƒrZ|jd – }|jd< |D ]}t|ƒ ¡ tkr<d||< q<dS )NÚheadersz
<redacted>T)r   ÚargsÚstrÚlowerÚSENSITIVE_HEADERS)Úselfr   r   Úheaderr   r   r   Úfilter#   s    "
zSensitiveHeadersFilter.filterN)Ú__name__Ú
__module__Ú__qualname__r   r   Ú	LogRecordÚboolr#   r   r   r   r   r   "   s   r   )r   r   Útyping_extensionsr   Ú_utilsr   Ú	getLoggerr   ÚLoggerÚ__annotations__r   r    r   r   ÚFilterr   r   r   r   r   Ú<module>   s    