From c418e6c95de3eca21a1551baa3e852dcb44491da Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Wed, 19 Mar 2025 15:41:46 +0800 Subject: [PATCH] fix --- api/services/app_generate_service.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/api/services/app_generate_service.py b/api/services/app_generate_service.py index 47fb75b49f..80de862e6c 100644 --- a/api/services/app_generate_service.py +++ b/api/services/app_generate_service.py @@ -14,6 +14,7 @@ from core.app.features.rate_limiting import RateLimit from libs.helper import RateLimiter from models.model import Account, App, AppMode, EndUser from models.workflow import Workflow +from services.billing_service import BillingService from services.errors.llm import InvokeRateLimitError from services.workflow_service import WorkflowService @@ -40,14 +41,14 @@ class AppGenerateService: :return: """ # system level rate limiter - if not dify_config.BILLING_ENABLED: + if dify_config.BILLING_ENABLED: # 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 cls.system_rate_limiter.is_rate_limited(app_model.tenant_id): 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)