feat: add wallet settlement audit flow

2026-05-11 22:59:26 +08:00
parent da1e19d0a9
commit c992f1de60
22 changed files with 1452 additions and 44 deletions
@@ -67,6 +67,19 @@ func (s *Service) execute(ctx context.Context, task store.GatewayTask, user *aut
 		}
 		return Result{Task: failed, Output: failed.Result}, err
 	}
+	if len(candidates) > 0 {
+		estimatedBillings := s.estimatedBillings(ctx, user, task.Kind, body, candidates[0])
+		if err := s.ensureWalletBalance(ctx, user, estimatedBillings); err != nil {
+			if errors.Is(err, store.ErrInsufficientWalletBalance) {
+				failed, finishErr := s.failTask(ctx, task.ID, "insufficient_balance", err.Error(), task.RunMode == "simulation", err)
+				if finishErr != nil {
+					return Result{}, finishErr
+				}
+				return Result{Task: failed, Output: failed.Result}, err
+			}
+			return Result{}, err
+		}
+	}
 	if err := s.store.MarkTaskRunning(ctx, task.ID, modelType, body); err != nil {
 		return Result{}, err
 	}