From a1a7b38c64f30940d65ebe32bc3561d196123b09 Mon Sep 17 00:00:00 2001 From: "Haihui.Wang" Date: Fri, 8 May 2026 17:34:08 +0800 Subject: [PATCH] Auto create builtin apikey quota --- user/component/access_token.go | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/user/component/access_token.go b/user/component/access_token.go index ea0ed018b..490420f08 100644 --- a/user/component/access_token.go +++ b/user/component/access_token.go @@ -698,8 +698,20 @@ func (c *accessTokenComponentImpl) checkOrCreateBuiltinAPIKey(ctx context.Contex UserID: user.ID, TokenType: types.AccessTokenTypeBuiltIn, } - - err = c.ts.Create(ctx, newToken, nil) + quotaReq := &types.CreateUserTokenRequest{ + QuotaType: types.AccountingQuotaTypeUnlimited, + ValueType: types.AccountingQuotaValueTypeFee, + Quota: 0, + } + quota, err := c.buildNewAccessTokenQuota(ctx, newToken, quotaReq) + if err != nil { + return nil, fmt.Errorf("failed to build API key quota, error: %w", err) + } + var quotas []database.AccountAccessTokenQuota + if quota != nil { + quotas = []database.AccountAccessTokenQuota{*quota} + } + err = c.ts.Create(ctx, newToken, quotas) if err != nil { return nil, fmt.Errorf("failed to create builtin api key, error: %w", err) }