feat: Update Activation Call to Include Billing Limit

ee/api/billing.py

@@ -101,6 +101,7 @@ def activation(self, request: Request, *args: Any, **kwargs: Any) -> HttpRespons
 
         plan = request.GET.get("plan", None)
         product_keys = request.GET.get("products", None)
+        custom_limits = request.GET.get("custom_limits_usd", None)
         if not plan and not product_keys:
             # If no plan or product keys are specified, we default to the standard plan
             # This is to support the old activation flow
@@ -110,6 +111,8 @@ def activation(self, request: Request, *args: Any, **kwargs: Any) -> HttpRespons
             url = f"{url}&plan={plan}"
         if product_keys:
             url = f"{url}&products={product_keys}"
+        if custom_limits:
+            url = f"{url}&custom_limits_usd={custom_limits}"
 
         if license:
             billing_service_token = build_billing_token(license, organization)

frontend/src/scenes/billing/billing-utils.ts

@@ -175,6 +175,10 @@ export const getUpgradeProductLink = (
     }
     // remove the trailing comma that will be at the end of the url
     url = url.slice(0, -1)
+    if (product.type === 'data_warehouse') {
+        // TODO: remove once data_warehouse is out of beta
+        url += `&custom_limits_usd=${product.type}:500`
+    }
     if (redirectPath) {
         url += `&redirect_path=${redirectPath}`
     }

frontend/src/scenes/onboarding/onboardingLogic.tsx

@@ -258,8 +258,6 @@ export const onboardingLogic = kea<onboardingLogicType>([
             }
             if (step) {
                 actions.setStepKey(step)
-            } else {
-                actions.resetStepKey()
             }
         },
     })),