cathub/LocalChatGPT
0
1a
2 /� d� � @ sX d dl Z d dlZd dlmZmZmZmZmZmZ d dl m3Z4 d dl Z dd� Z dd� Z
dS )� N)� timeout_all�BALANCE_API_URL�standard_error_msg�connection_timeout_prompt�error_retrieve_prompt�read_timeout_prompt)�sharedc C s� dd| � �d�}t }tj�d�p*tj�d�}tj�d�pBtj�d�}i }|rdt�d|� �� ||d < |r�t�d5|� �� ||d < |r�tjt|||d �}ntjt||d
�}|S )Nzapplication/jsonzBearer )z Content-Type�
Authorization�6HTTP_PROXY�7http_proxy� HTTPS_PROXY� https_proxyu 使用 HTTP 代理: �httpu 使用 HTTPS 代理: �https)�headers�timeout�proxies)r r )r �os�environ�get�logging�info�requestsr )�openai_api_keyr r r r
r �response� r �L/Users/treediagram/Documents/Developer/ChuanhuChatGPT/modules/openai_func.py�get_usage_response s< ��8���r c 9 C s z�t | d�}t�|�� � zD|�� �d�r8|�� �d�nd}|�� �d�rX|�� �d�nd}W n> ty� } z&t�dt|� � d}d}W Y d }~n10d }~0 0 d|� d|� d�W S tj j11y� t t t
}| Y S tj j� y� t t t
}| Y S 0 d S ) N)r Ztotal_availabler �12total_usedu API使用情况解析失败:u) **API使用情况**（已用/余额） z$ / �$)r r �debug�jsonr � Exception�error�strr �13exceptions�ConnectTimeoutr r r � ReadTimeoutr )r r Zbalancer �e� status_textr r r � get_usage3 s2 14���� r* )r r �modules.presetsr r r r r r �modulesr r r r* r r r r �<module> s +