Spaces:
Sleeping
Sleeping
File size: 2,644 Bytes
ff3263f 0137c6b ff3263f 5817e2c b3cdf66 ff3263f b3cdf66 ff3263f b3cdf66 ff3263f b3cdf66 ff3263f 5817e2c b3cdf66 5817e2c b3cdf66 ff3263f b3cdf66 ff3263f b3cdf66 ff3263f 9310dbb 0137c6b b3cdf66 0137c6b b3cdf66 0137c6b b3cdf66 ff3263f 5817e2c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
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
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}"
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
|