jprdonnelly/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-09 18:00:30 -04:00
Code Issues Packages Projects Releases Wiki Activity

rebuild quota service

Browse Source
This commit is contained in:
Yansong Zhang
2026-03-31 14:59:45 +08:00
parent 6e745f9e9b
commit 422bf3506e
11 changed files with 222 additions and 241 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/tasks/trigger_processing_tasks.py
View File

@@ -42,7 +42,7 @@ from models.workflow import Workflow, WorkflowAppLog, WorkflowAppLogCreatedFrom,
from services.async_workflow_service import AsyncWorkflowService
from services.end_user_service import EndUserService
from services.errors.app import QuotaExceededError
from services.quota_service import unlimited
from services.quota_service import QuotaService, unlimited
from services.trigger.app_trigger_service import AppTriggerService
from services.trigger.trigger_provider_service import TriggerProviderService
from services.trigger.trigger_request_service import TriggerHttpRequestCachingService
@@ -302,7 +302,7 @@ def dispatch_triggered_workflow(
# reserve quota before invoking trigger
quota_charge = unlimited()
try:
quota_charge = QuotaType.TRIGGER.reserve(subscription.tenant_id)
quota_charge = QuotaService.reserve(QuotaType.TRIGGER, subscription.tenant_id)
except QuotaExceededError:
AppTriggerService.mark_tenant_triggers_rate_limited(subscription.tenant_id)
logger.info(