20 lines
483 B
Go
20 lines
483 B
Go
package util
|
|
|
|
import (
|
|
"context"
|
|
"github.com/songquanpeng/one-api/common/logger"
|
|
"github.com/songquanpeng/one-api/model"
|
|
)
|
|
|
|
func ReturnPreConsumedQuota(ctx context.Context, preConsumedQuota int64, tokenId int) {
|
|
if preConsumedQuota != 0 {
|
|
go func(ctx context.Context) {
|
|
// return pre-consumed quota
|
|
err := model.PostConsumeTokenQuota(tokenId, -preConsumedQuota)
|
|
if err != nil {
|
|
logger.Error(ctx, "error return pre-consumed quota: "+err.Error())
|
|
}
|
|
}(ctx)
|
|
}
|
|
}
|