Improve demo seed
This commit is contained in:
@@ -70,6 +70,11 @@ class AuthMiddleware(BaseHTTPMiddleware):
|
||||
# But we still need to inject context headers for downstream services
|
||||
user_context = request.state.user
|
||||
tenant_id = user_context.get("tenant_id") or getattr(request.state, "tenant_id", None)
|
||||
|
||||
# Inject subscription tier for demo sessions - always enterprise tier for full feature access
|
||||
user_context["subscription_tier"] = "enterprise"
|
||||
logger.debug(f"Demo session subscription tier set to enterprise", tenant_id=tenant_id)
|
||||
|
||||
self._inject_context_headers(request, user_context, tenant_id)
|
||||
return await call_next(request)
|
||||
|
||||
|
||||
@@ -87,6 +87,11 @@ class SubscriptionMiddleware(BaseHTTPMiddleware):
|
||||
async def dispatch(self, request: Request, call_next):
|
||||
"""Process the request and check subscription requirements"""
|
||||
|
||||
# Skip subscription check for demo sessions - they get enterprise tier
|
||||
if hasattr(request.state, "is_demo_session") and request.state.is_demo_session:
|
||||
logger.debug("Skipping subscription check for demo session", path=request.url.path)
|
||||
return await call_next(request)
|
||||
|
||||
# Skip subscription check for certain routes
|
||||
if self._should_skip_subscription_check(request):
|
||||
return await call_next(request)
|
||||
|
||||
Reference in New Issue
Block a user