mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-30 07:06:38 +08:00
27 lines
633 B
Go
27 lines
633 B
Go
package closeai
|
|
|
|
import (
|
|
"errors"
|
|
)
|
|
|
|
func (p *CloseaiProxyProvider) Balance() (float64, error) {
|
|
fullRequestURL := p.GetFullRequestURL("/dashboard/billing/credit_grants", "")
|
|
headers := p.GetRequestHeaders()
|
|
|
|
req, err := p.Requester.NewRequest("GET", fullRequestURL, p.Requester.WithHeader(headers))
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
// 发送请求
|
|
var response OpenAICreditGrants
|
|
_, errWithCode := p.Requester.SendRequest(req, &response, false)
|
|
if errWithCode != nil {
|
|
return 0, errors.New(errWithCode.OpenAIError.Message)
|
|
}
|
|
|
|
p.Channel.UpdateBalance(response.TotalAvailable)
|
|
|
|
return response.TotalAvailable, nil
|
|
}
|