outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Yeuoly 2025-03-19 15:41:46 +08:00
parent af4c47bfab
commit c418e6c95d
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api/services/app_generate_service.py
View File

@ -14,6 +14,7 @@ from core.app.features.rate_limiting import RateLimit
from libs.helper import RateLimiter from libs.helper import RateLimiter
from models.model import Account, App, AppMode, EndUser from models.model import Account, App, AppMode, EndUser
from models.workflow import Workflow from models.workflow import Workflow
from services.billing_service import BillingService
from services.errors.llm import InvokeRateLimitError from services.errors.llm import InvokeRateLimitError
from services.workflow_service import WorkflowService from services.workflow_service import WorkflowService
@ -40,14 +41,14 @@ class AppGenerateService:
:return: :return:
""" """
# system level rate limiter # system level rate limiter
if not dify_config.BILLING_ENABLED: if dify_config.BILLING_ENABLED:
# check if it's free plan # check if it's free plan
limit_info = {"subscription": {"plan": "sandbox"}} limit_info = BillingService.get_info(app_model.tenant_id)
# limit_info = BillingService.get_info(app_model.tenant_id)
if limit_info["subscription"]["plan"] == "sandbox": if limit_info["subscription"]["plan"] == "sandbox":
if cls.system_rate_limiter.is_rate_limited(app_model.tenant_id): if cls.system_rate_limiter.is_rate_limited(app_model.tenant_id):
raise InvokeRateLimitError( raise InvokeRateLimitError(
f"Rate limit exceeded, please upgrade your plan, or your RPD was {dify_config.APP_DAILY_RATE_LIMIT} requests/day" "Rate limit exceeded, please upgrade your plan"
f"or your RPD was {dify_config.APP_DAILY_RATE_LIMIT} requests/day"
) )
cls.system_rate_limiter.increment_rate_limit(app_model.tenant_id) cls.system_rate_limiter.increment_rate_limit(app_model.tenant_id)