compatible balance query with third-party services

This commit is contained in:
josc146
2023-06-12 16:32:52 +08:00
parent 4dba9d579e
commit b0c8bb0621
+12 -3
View File
@@ -81,9 +81,18 @@ export function GeneralPart({ config, updateConfig }) {
const [balance, setBalance] = useState(null)
const getBalance = async () => {
const billing = await checkBilling(config.apiKey, config.customOpenAiApiUrl)
if (billing && billing.length > 2 && billing[2]) setBalance(`${billing[2].toFixed(2)}`)
else openUrl('https://platform.openai.com/account/usage')
const response = await fetch(`${config.customOpenAiApiUrl}/dashboard/billing/credit_grants`, {
headers: {
'Content-Type': 'application/json',
Authorization: `Bearer ${config.apiKey}`,
},
})
if (response.ok) setBalance((await response.json()).total_available.toFixed(2))
else {
const billing = await checkBilling(config.apiKey, config.customOpenAiApiUrl)
if (billing && billing.length > 2 && billing[2]) setBalance(`${billing[2].toFixed(2)}`)
else openUrl('https://platform.openai.com/account/usage')
}
}
return (