跳过价格为 0 的 (#287)

ecodeclub · Dec 1, 2024 · 60da318 · 60da318
2 parents b9b620f + 76b0faf
commit 60da318
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/ai/internal/service/llm/handler/credit/builder.go b/internal/ai/internal/service/llm/handler/credit/builder.go
@@ -38,6 +38,10 @@ func NewHandlerBuilder(creSvc credit.Service, repo repository.LLMCreditLogRepo)
 
 func (h *HandlerBuilder) Next(next handler.Handler) handler.Handler {
 	return handler.HandleFunc(func(ctx context.Context, req domain.LLMRequest) (domain.LLMResponse, error) {
+		// 不需要扣除积分
+		if req.Config.Price == 0 {
+			return next.Handle(ctx, req)
+		}
 		cre, err := h.creditSvc.GetCreditsByUID(ctx, req.Uid)
 		if err != nil {
 			return domain.LLMResponse{}, err