diff --git a/api/services/credit_pool_service.py b/api/services/credit_pool_service.py index 479d54f68d..ada7f41ec2 100644 --- a/api/services/credit_pool_service.py +++ b/api/services/credit_pool_service.py @@ -1,8 +1,6 @@ import logging -from typing import cast - -from sqlalchemy import CursorResult, update +from sqlalchemy import update from sqlalchemy.orm import Session from configs import dify_config @@ -63,12 +61,8 @@ class CreditPoolService: TenantCreditPool.quota_used + credits_required <= TenantCreditPool.quota_limit, ] stmt = update(TenantCreditPool).where(*where_conditions).values(**update_values) - result = cast(CursorResult, session.execute(stmt)) + session.execute(stmt) session.commit() - if result.rowcount == 0: - raise QuotaExceededError( - f"Insufficient credits. Required: {credits_required}, Available: {pool.remaining_credits}" - ) except Exception: logger.exception("Failed to deduct credits for tenant %s", tenant_id) raise QuotaExceededError("Failed to deduct credits")