one-api/providers/aigc2d/balance.go

package aigc2d

import (
	"errors"
	"one-api/providers/base"
)

func (p *Aigc2dProvider) 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 base.BalanceResponse
	_, 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
}