honeymoose
/
chatgpt-web
mirror of https://github.com/Chanzhaoyu/chatgpt-web.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: 查询余额使用 API_BASE_URL 参数

This commit is contained in:
ChenZhaoYu 2023-03-17 13:19:03 +08:00
parent 8fad91ade7
commit 08b5772420
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
service/src/chatgpt/index.ts
View File

@ -102,15 +102,18 @@ async function chatReplyProcess(
async function fetchBalance() { async function fetchBalance() {
const OPENAI_API_KEY = process.env.OPENAI_API_KEY const OPENAI_API_KEY = process.env.OPENAI_API_KEY
const OPENAI_API_BASE_URL = process.env.OPENAI_API_BASE_URL
if (!isNotEmptyString(OPENAI_API_KEY)) if (!isNotEmptyString(OPENAI_API_KEY))
return Promise.resolve('-') return Promise.resolve('-')
const API_BASE_URL = isNotEmptyString(OPENAI_API_BASE_URL)
? OPENAI_API_BASE_URL
: 'https://api.openai.com'
try { try {
const headers = { const headers = { 'Content-Type': 'application/json', 'Authorization': `Bearer ${OPENAI_API_KEY}` }
'Content-Type': 'application/json', const response = await axios.get(`${API_BASE_URL}/dashboard/billing/credit_grants`, { headers })
'Authorization': `Bearer ${OPENAI_API_KEY}`,
}
const response = await axios.get('https://api.openai.com/dashboard/billing/credit_grants', { headers })
const balance = response.data.total_available ?? 0 const balance = response.data.total_available ?? 0
return Promise.resolve(balance.toFixed(3)) return Promise.resolve(balance.toFixed(3))
} }