Feature-flag pricing tier availability for alpha/demo modes

backend/services/startup_health.py

@@ -15,7 +15,6 @@ from services.database import (
     init_database,
     default_engine,
 )
-from services.user_api_key_context import get_user_api_keys
 
 _REQUIRED_SCHEMA: Dict[str, List[str]] = {
     "onboarding_sessions": ["id", "user_id", "updated_at"],
@@ -145,62 +144,6 @@ def _check_db_access(checks: List[Dict[str, Any]], errors: List[str], warnings:
     return candidate_user
 
 
-def _check_production_api_key_loading(
-    checks: List[Dict[str, Any]],
-    errors: List[str],
-    warnings: List[str],
-) -> None:
-    deploy_env = os.getenv("DEPLOY_ENV", "local").strip().lower()
-    if deploy_env == "local":
-        _record_check(checks, "production_api_key_loading", True, "skipped in local deploy mode")
-        return
-
-    test_tenant_id = os.getenv("ALWRITY_STARTUP_TEST_TENANT_ID", "").strip()
-    if not test_tenant_id:
-        message = (
-            "Missing ALWRITY_STARTUP_TEST_TENANT_ID for production API key startup check."
-        )
-        errors.append(message)
-        _record_check(checks, "production_api_key_loading", False, message)
-        return
-
-    try:
-        keys = get_user_api_keys(test_tenant_id)
-    except Exception as exc:
-        errors.append(
-            f"Failed to load API keys for startup test tenant '{test_tenant_id}': {exc}"
-        )
-        _record_check(checks, "production_api_key_loading", False, str(exc))
-        return
-
-    if not isinstance(keys, dict):
-        errors.append(
-            f"API key loader returned invalid payload type for startup test tenant '{test_tenant_id}'."
-        )
-        _record_check(checks, "production_api_key_loading", False, "invalid payload type")
-        return
-
-    non_empty_keys = [provider for provider, value in keys.items() if value]
-    if not non_empty_keys:
-        errors.append(
-            f"No API keys could be loaded for startup test tenant '{test_tenant_id}'."
-        )
-        _record_check(checks, "production_api_key_loading", False, "no non-empty keys loaded")
-        return
-
-    warning = None
-    if len(non_empty_keys) < len(keys):
-        warning = (
-            f"Startup test tenant '{test_tenant_id}' has {len(non_empty_keys)}/{len(keys)} non-empty API keys."
-        )
-        warnings.append(warning)
-
-    detail = f"loaded {len(non_empty_keys)} non-empty keys for tenant {test_tenant_id}"
-    if warning:
-        detail = f"{detail}; {warning}"
-    _record_check(checks, "production_api_key_loading", True, detail)
-
-
 def run_startup_health_routine() -> Dict[str, Any]:
     checks: List[Dict[str, Any]] = []
     errors: List[str] = []
@@ -209,8 +152,6 @@ def run_startup_health_routine() -> Dict[str, Any]:
     _check_workspace_root(checks, errors)
     if not errors:
         _check_db_access(checks, errors, warnings)
-    if not errors:
-        _check_production_api_key_loading(checks, errors, warnings)
 
     status = "healthy" if not errors else "failed"
     report = {

backend/services/user_api_key_context.py

@@ -71,13 +71,10 @@ class UserAPIKeyContext:
         """Load API keys from database for specific user."""
         try:
             from api.content_planning.services.content_strategy.onboarding import OnboardingDataIntegrationService
-            from services.database import get_session_for_user
+            from services.database import SessionLocal
             
             integration_service = OnboardingDataIntegrationService()
-            db = get_session_for_user(user_id)
-            if not db:
-                logger.error(f"Failed to create DB session for user {user_id}")
-                return {}
+            db = SessionLocal()
             try:
                 integrated_data = integration_service.get_integrated_data_sync(user_id, db)
                 keys = integrated_data.get('api_keys_data', {})
@@ -156,3 +153,4 @@ def get_tavily_key(user_id: Optional[str] = None) -> Optional[str]:
 def get_copilotkit_key(user_id: Optional[str] = None) -> Optional[str]:
     """Get CopilotKit API key for user."""
     return UserAPIKeyContext.get_user_key(user_id, 'copilotkit')
+

frontend/src/components/Pricing/PricingPage.tsx

@@ -52,6 +52,27 @@ export interface SubscriptionPlan {
 }
 
 const PricingPage: React.FC = () => {
+  const pricingMode = (process.env.REACT_APP_PRICING_MODE || 'alpha').toLowerCase();
+  const isAlphaMode = pricingMode === 'alpha';
+  const tierPolicyByMode: Record<string, { visible: string[]; selectable: string[]; unavailableLabels: Record<string, string> }> = {
+    alpha: {
+      visible: ['free', 'basic', 'pro', 'enterprise'],
+      selectable: ['free', 'basic'],
+      unavailableLabels: { pro: 'Coming Soon', enterprise: 'Contact Sales' },
+    },
+    demo: {
+      visible: ['free', 'basic', 'pro', 'enterprise'],
+      selectable: ['free', 'basic', 'pro'],
+      unavailableLabels: { enterprise: 'Contact Sales' },
+    },
+    production: {
+      visible: ['free', 'basic', 'pro', 'enterprise'],
+      selectable: ['free', 'basic', 'pro'],
+      unavailableLabels: { enterprise: 'Contact Sales' },
+    },
+  };
+  const activeTierPolicy = tierPolicyByMode[pricingMode] || tierPolicyByMode.alpha;
+
   const navigate = useNavigate();
   const [plans, setPlans] = useState<SubscriptionPlan[]>([]);
   const [loading, setLoading] = useState(true);
@@ -76,9 +97,11 @@ const PricingPage: React.FC = () => {
     try {
       setLoading(true);
       const response = await apiClient.get('/api/subscription/plans');
-      // Filter out any alpha plans and ensure we only show the 4 main tiers
+      // Filter out legacy alpha-named plans and enforce tier visibility policy.
       const filteredPlans = response.data.data.plans.filter(
-        (plan: SubscriptionPlan) => !plan.name.toLowerCase().includes('alpha')
+        (plan: SubscriptionPlan) =>
+          !plan.name.toLowerCase().includes('alpha') &&
+          activeTierPolicy.visible.includes(plan.tier)
       );
       setPlans(filteredPlans);
     } catch (err) {
@@ -111,10 +134,13 @@ const PricingPage: React.FC = () => {
       return;
     }
 
-    // For alpha testing, only allow Free and Basic plans (Pro features not ready)
-    if (plan.tier !== 'free' && plan.tier !== 'basic') {
+    if (!activeTierPolicy.selectable.includes(plan.tier)) {
       console.error('[PricingPage] Plan tier not available:', plan.tier);
-      setError('This plan is not available for alpha testing');
+      setError(
+        isAlphaMode
+          ? 'This plan is not available during alpha testing'
+          : 'This plan is currently not available for self-serve checkout'
+      );
       return;
     }
 
@@ -351,6 +377,8 @@ const PricingPage: React.FC = () => {
               yearlyBilling={yearlyBilling}
               selectedPlanId={selectedPlan}
               subscribing={subscribing}
+              canSelectPlan={activeTierPolicy.selectable.includes(plan.tier)}
+              unavailableLabel={activeTierPolicy.unavailableLabels[plan.tier]}
               onSelectPlan={setSelectedPlan}
               onSubscribe={handleSubscribe}
               openKnowMoreModal={openKnowMoreModal}
@@ -392,42 +420,48 @@ const PricingPage: React.FC = () => {
           }}>
             <Typography variant="h6" component="h2" gutterBottom sx={{ display: 'flex', alignItems: 'center', gap: 1 }}>
               <Warning sx={{ color: 'warning.main' }} />
-              Alpha Testing Subscription
+              {isAlphaMode ? 'Alpha Testing Subscription' : 'Confirm Subscription'}
             </Typography>
-            
-            {/* Alpha Testing Notice */}
-            <Alert severity="warning" sx={{ mb: 2 }}>
-              <Typography variant="body2" sx={{ fontWeight: 600, mb: 0.5 }}>
-                ⚠️ Alpha Testing Mode - No Payment Required
+
+            {isAlphaMode ? (
+              <>
+                <Alert severity="warning" sx={{ mb: 2 }}>
+                  <Typography variant="body2" sx={{ fontWeight: 600, mb: 0.5 }}>
+                    ⚠️ Alpha Testing Mode - No Payment Required
+                  </Typography>
+                  <Typography variant="caption" sx={{ display: 'block' }}>
+                    Payment integration is coming soon. For now, subscriptions are activated without charge.
+                  </Typography>
+                </Alert>
+
+                <Typography variant="body1" sx={{ mb: 2 }}>
+                  Thank you for participating in our alpha testing! We&apos;re crediting this plan to your account.
+                </Typography>
+
+                <Box sx={{
+                  p: 2,
+                  mb: 3,
+                  bgcolor: 'info.lighter',
+                  borderRadius: 1,
+                  border: '1px solid',
+                  borderColor: 'info.light'
+                }}>
+                  <Typography variant="body2" color="info.dark">
+                    <strong>Coming in Production:</strong>
+                  </Typography>
+                  <Typography variant="caption" color="info.dark" sx={{ display: 'block', mt: 0.5 }}>
+                    • Secure Stripe/PayPal payment processing<br />
+                    • Automatic renewal management<br />
+                    • Payment verification & receipts<br />
+                    • Upgrade/downgrade options
+                  </Typography>
+                </Box>
+              </>
+            ) : (
+              <Typography variant="body1" sx={{ mb: 3 }}>
+                Please confirm to continue with your selected subscription plan.
               </Typography>
-              <Typography variant="caption" sx={{ display: 'block' }}>
-                Payment integration is coming soon. For now, subscriptions are activated without charge.
-              </Typography>
-            </Alert>
-            
-            <Typography variant="body1" sx={{ mb: 2 }}>
-              Thank you for participating in our alpha testing! We're crediting the Basic plan ($29 value) to your account.
-            </Typography>
-            
-            {/* TODO: Payment Integration Notice */}
-            <Box sx={{ 
-              p: 2, 
-              mb: 3, 
-              bgcolor: 'info.lighter', 
-              borderRadius: 1, 
-              border: '1px solid',
-              borderColor: 'info.light'
-            }}>
-              <Typography variant="body2" color="info.dark">
-                <strong>Coming in Production:</strong>
-              </Typography>
-              <Typography variant="caption" color="info.dark" sx={{ display: 'block', mt: 0.5 }}>
-                • Secure Stripe/PayPal payment processing<br />
-                • Automatic renewal management<br />
-                • Payment verification & receipts<br />
-                • Upgrade/downgrade options
-              </Typography>
-            </Box>
+            )}
             
             {/* Note: Current behavior allows renewal without payment verification */}
             {/* This is intentional for alpha testing but will be secured in production */}

frontend/src/components/Pricing/PricingPage/PlanCard.tsx

@@ -69,6 +69,8 @@ interface PlanCardProps {
   yearlyBilling: boolean;
   selectedPlanId: number | null;
   subscribing: boolean;
+  canSelectPlan: boolean;
+  unavailableLabel?: string;
   onSelectPlan: (planId: number) => void;
   onSubscribe: (planId: number) => void;
   openKnowMoreModal: (title: string, content: React.ReactNode) => void;
@@ -79,6 +81,8 @@ const PlanCard: React.FC<PlanCardProps> = ({
   yearlyBilling,
   selectedPlanId,
   subscribing,
+  canSelectPlan,
+  unavailableLabel,
   onSelectPlan,
   onSubscribe,
   openKnowMoreModal,
@@ -909,13 +913,9 @@ const PlanCard: React.FC<PlanCardProps> = ({
       </CardContent>
 
       <CardActions sx={{ justifyContent: 'center', pb: 3, flexDirection: 'column', gap: 1 }}>
-        {plan.tier === 'pro' ? (
+        {!canSelectPlan ? (
           <Button variant="outlined" size="large" fullWidth disabled sx={{ mb: 1 }}>
-            Coming Soon
-          </Button>
-        ) : plan.tier === 'enterprise' ? (
-          <Button variant="outlined" size="large" fullWidth disabled sx={{ mb: 1 }}>
-            Contact Sales
+            {unavailableLabel || 'Unavailable'}
           </Button>
         ) : (
           <>
@@ -951,4 +951,3 @@ const PlanCard: React.FC<PlanCardProps> = ({
 };
 
 export default PlanCard;
-