Add check for limits exceeded in PlaygroundConsumer

Author: ajhai

llmstack/server/consumers.py

@@ -10,6 +10,7 @@
 from django.core.exceptions import PermissionDenied
 from django.http import HttpRequest, QueryDict
 from django_ratelimit.exceptions import Ratelimited
+from flags.state import flag_enabled
 
 from llmstack.connections.actors import ConnectionActivationActor
 from llmstack.connections.models import (
@@ -30,6 +31,19 @@ class UsageLimitReached(PermissionDenied):
     pass
 
 
+class OutOfCredits(PermissionDenied):
+    pass
+
+
+@database_sync_to_async
+def _usage_limit_exceeded(request, user):
+    return flag_enabled(
+        "HAS_EXCEEDED_MONTHLY_PROCESSOR_RUN_QUOTA",
+        request=request,
+        user=request.user,
+    )
+
+
 @database_sync_to_async
 def _build_request_from_input(post_data, scope):
     session = dict(scope["session"])
@@ -253,6 +267,11 @@ async def connect(self):
     def _run_app(self, request_uuid, request, **kwargs):
         from llmstack.apps.apis import AppViewSet
 
+        if _usage_limit_exceeded(request, self._run_app):
+            raise OutOfCredits(
+                "You have exceeded your usage credits. Please add credits to your account from settings to continue using the platform.",
+            )
+
         return AppViewSet().run_playground_internal_async(
             session_id=self._session_id,
             request_uuid=request_uuid,