Dwild/ChuanhuChatGPT
0
1a
2 u+,dK � @ sx d dl Z d dlZd dlmZmZmZmZmZmZm Z ddl3m Z d dl m
Z
d dlZd dlZdd� Zdd � Zd4d � ZdS ) � N)� timeout_all�
USAGE_API_URL�BALANCE_API_URL�standard_error_msg�connection_timeout_prompt�error_retrieve_prompt�read_timeout_prompt� )�shared)�retrieve_proxyc C s| dd| � �d�}t }t� � tj|||d�}W d � n1 s@0 Y |jdkr`|�� }|S td|j� d|j� ���d S )Nzapplication/jsonzBearer )z Content-Type�
Authorization)�headers�timeout�� z$API request failed with status code z: )r r �requests�get� status_code�json� Exception�text)�openai_api_keyZ billing_urlr
r �response�data� r �L/Users/treediagram/Documents/Developer/ChuanhuChatGPT/modules/openai_func.py�get_billing_data s ��$5r c 6 C s< z�t j �� }t|��d�}|jdd��d�}tjj� d|� d|� �}zt| |�}W n< t y� } z$t7� dt |� � dW Y d }~W S d }~0 0 d�
|d d8 �}d |� �W S tjjy� tt t }| Y S tjj� y� tt t }| Y S t �y6 } z&t9� dt |� � tt W Y d }~S d }~0 0 d S ) Nz%Y-%m-%dr ��dayz ?start_date=z10&end_date=u 获取API使用情况失败:u **获取API使用情况失败**z{:.5f}Z total_usage�d u **本月使用金额** $)�datetime�now�get_last_day_of_month�strftime�replacer11 �state�
usage_api_urlr r �logging�error�str�formatr �12exceptions�ConnectTimeoutr r r � ReadTimeoutr ) r Z curr_timeZlast_day_of_monthZfirst_day_of_monthZ usage_urlZ13usage_data�eZ
rounded_usage� status_textr r r � get_usage& s* 14 r/ c C s* | j dd�tjdd� }|tj|jd� S )N� r � )�days)r# r � timedeltar )Zany_dayZ15next_monthr r r r! = s r! )r r& �modules.presetsr r r r r r r � r16 �modules.configr �osr r r/ r! r r r r �<module> s $17