AlanChatGPT / modules /openai_func.py
XuAlan's picture
Duplicate from JohnSmith9982/ChuanhuChatGPT
baeb4b5
raw
history blame
3.14 kB
import requests
import logging
from modules.presets import (
timeout_all,
USAGE_API_URL,
BALANCE_API_URL,
standard_error_msg,
connection_timeout_prompt,
error_retrieve_prompt,
read_timeout_prompt
)
from modules import shared
from modules.utils import get_proxies
import os, datetime
def get_billing_data(openai_api_key, billing_url):
headers = {
"Content-Type": "application/json",
"Authorization": f"Bearer {openai_api_key}"
}
timeout = timeout_all
proxies = get_proxies()
response = requests.get(
billing_url,
headers=headers,
timeout=timeout,
proxies=proxies,
)
if response.status_code == 200:
data = response.json()
return data
else:
raise Exception(f"API request failed with status code {response.status_code}: {response.text}")
def get_usage(openai_api_key):
try:
balance_data=get_billing_data(openai_api_key, BALANCE_API_URL)
logging.debug(balance_data)
try:
balance = balance_data["total_available"] if balance_data["total_available"] else 0
total_used = balance_data["total_used"] if balance_data["total_used"] else 0
usage_percent = round(total_used / (total_used+balance) * 100, 2)
except Exception as e:
logging.error(f"API使用情况解析失败:"+str(e))
balance = 0
total_used=0
return f"**API使用情况解析失败**"
if balance == 0:
last_day_of_month = datetime.datetime.now().strftime("%Y-%m-%d")
first_day_of_month = datetime.datetime.now().replace(day=1).strftime("%Y-%m-%d")
usage_url = f"{USAGE_API_URL}?start_date={first_day_of_month}&end_date={last_day_of_month}"
try:
usage_data = get_billing_data(openai_api_key, usage_url)
except Exception as e:
logging.error(f"获取API使用情况失败:"+str(e))
return f"**获取API使用情况失败**"
return f"**本月使用金额** \u3000 ${usage_data['total_usage'] / 100}"
# return f"**免费额度**(已用/余额)\u3000${total_used} / ${balance}"
return f"""\
<b>免费额度使用情况</b>
<div class="progress-bar">
<div class="progress" style="width: {usage_percent}%;">
<span class="progress-text">{usage_percent}%</span>
</div>
</div>
<div style="display: flex; justify-content: space-between;"><span>已用 ${total_used}</span><span>可用 ${balance}</span></div>
"""
except requests.exceptions.ConnectTimeout:
status_text = standard_error_msg + connection_timeout_prompt + error_retrieve_prompt
return status_text
except requests.exceptions.ReadTimeout:
status_text = standard_error_msg + read_timeout_prompt + error_retrieve_prompt
return status_text
except Exception as e:
logging.error(f"获取API使用情况失败:"+str(e))
return standard_error_msg + error_retrieve_prompt