30 lines
695 B
Go
30 lines
695 B
Go
|
package api2d
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"one-api/common"
|
||
|
|
"one-api/model"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (p *Api2dProvider) Balance(channel *model.Channel) (float64, error) {
|
||
|
|
fullRequestURL := "https://api.aigc2d.com/dashboard/billing/credit_grants"
|
||
|
|
headers := p.GetRequestHeaders()
|
||
|
|
|
||
|
|
client := common.NewClient()
|
||
|
|
req, err := client.NewRequest("GET", fullRequestURL, common.WithHeader(headers))
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// 发送请求
|
||
|
|
var response APGC2DGPTUsageResponse
|
||
|
|
_, errWithCode := common.SendRequest(req, &response, false)
|
||
|
|
if errWithCode != nil {
|
||
|
|
return 0, errors.New(errWithCode.OpenAIError.Message)
|
||
|
|
}
|
||
|
|
|
||
|
|
channel.UpdateBalance(response.TotalAvailable)
|
||
|
|
|
||
|
|
return response.TotalAvailable, nil
|
||
|
|
}
|