@@ -5,11 +5,10 @@ import { ChatGPTAPI, ChatGPTUnofficialProxyAPI } from 'chatgpt'
55import { SocksProxyAgent } from 'socks-proxy-agent'
66import httpsProxyAgent from 'https-proxy-agent'
77import fetch from 'node-fetch'
8- import axios from 'axios'
98import { sendResponse } from '../utils'
109import { isNotEmptyString } from '../utils/is'
1110import type { ApiModel , ChatContext , ChatGPTUnofficialProxyAPIOptions , ModelConfig } from '../types'
12- import type { RequestOptions } from './types'
11+ import type { BalanceResponse , RequestOptions } from './types'
1312
1413const { HttpsProxyAgent} = httpsProxyAgent
1514
@@ -126,6 +125,8 @@ async function chatReplyProcess(options: RequestOptions) {
126125}
127126
128127async function fetchBalance ( ) {
128+ // 计算起始日期和结束日期
129+
129130const OPENAI_API_KEY = process . env . OPENAI_API_KEY
130131const OPENAI_API_BASE_URL = process . env . OPENAI_API_BASE_URL
131132
@@ -136,17 +137,38 @@ async function fetchBalance() {
136137 ?OPENAI_API_BASE_URL
137138 :'https://api.openai.com'
138139
140+ const [ startDate , endDate ] = formatDate ( )
141+
142+ // 每月使用量
143+ const urlUsage = `${ API_BASE_URL } ${ startDate } ${ endDate }
144+
145+ const headers = {
146+ 'Authorization' :`Bearer${ OPENAI_API_KEY } ,
147+ 'Content-Type' :'application/json' ,
148+ }
149+
139150try {
140- const headers = { 'Content-Type' :'application/json' , 'Authorization' :`Bearer${ OPENAI_API_KEY } }
141- const response = await axios . get ( `${ API_BASE_URL } , { headers} )
142- const balance = response . data . total_available ?? 0
143- return Promise . resolve ( balance . toFixed ( 3 ) )
151+ // 获取已使用量
152+ const useResponse = await fetch ( urlUsage , { headers} )
153+ const usageData = await useResponse . json ( ) as BalanceResponse
154+ const usage = Math . round ( usageData . total_usage ) / 100
155+ return Promise . resolve ( usage ?`$${ usage } :'-' )
144156}
145157catch {
146158return Promise . resolve ( '-' )
147159}
148160}
149161
162+ function formatDate ( ) :string [ ] {
163+ const today = new Date ( )
164+ const year = today . getFullYear ( )
165+ const month = today . getMonth ( ) + 1
166+ const lastDay = new Date ( year , month , 0 )
167+ const formattedFirstDay = `${ year } ${ month . toString ( ) . padStart ( 2 , '0' ) }
168+ const formattedLastDay = `${ year } ${ month . toString ( ) . padStart ( 2 , '0' ) } ${ lastDay . getDate ( ) . toString ( ) . padStart ( 2 , '0' ) }
169+ return [ formattedFirstDay , formattedLastDay ]
170+ }
171+
150172async function chatConfig ( ) {
151173const balance = await fetchBalance ( )
152174const reverseProxy = process . env . API_REVERSE_PROXY ?? '-'