fix: treat local rate limit queueing separately

apps/api/internal/store/rate_limits.go

@@ -230,6 +230,42 @@ WHERE id = $1::uuid AND released_at IS NULL`, leaseID); err != nil && !errors.Is
 	return nil
 }
 
+func (s *Store) AttachRateLimitResultToAttempt(ctx context.Context, attemptID string, result RateLimitResult) error {
+	if attemptID == "" || (len(result.Reservations) == 0 && len(result.LeaseIDs) == 0) {
+		return nil
+	}
+	tx, err := s.pool.Begin(ctx)
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback(ctx)
+
+	for _, reservation := range result.Reservations {
+		if reservation.ReservationID == "" {
+			continue
+		}
+		if _, err := tx.Exec(ctx, `
+UPDATE gateway_rate_limit_reservations
+SET attempt_id = $2::uuid,
+    updated_at = now()
+WHERE id = $1::uuid`, reservation.ReservationID, attemptID); err != nil {
+			return err
+		}
+	}
+	for _, leaseID := range result.LeaseIDs {
+		if leaseID == "" {
+			continue
+		}
+		if _, err := tx.Exec(ctx, `
+UPDATE gateway_concurrency_leases
+SET attempt_id = $2::uuid
+WHERE id = $1::uuid`, leaseID, attemptID); err != nil {
+			return err
+		}
+	}
+	return tx.Commit(ctx)
+}
+
 func (s *Store) RecoverInterruptedRuntimeState(ctx context.Context) (RuntimeRecoveryResult, error) {
 	tx, err := s.pool.Begin(ctx)
 	if err != nil {