diff --git a/api/services/credit_pool_service.py b/api/services/credit_pool_service.py index 6214900c61..479d54f68d 100644 --- a/api/services/credit_pool_service.py +++ b/api/services/credit_pool_service.py @@ -1,5 +1,7 @@ import logging +from typing import cast + from sqlalchemy import CursorResult, update from sqlalchemy.orm import Session @@ -61,7 +63,7 @@ class CreditPoolService: TenantCreditPool.quota_used + credits_required <= TenantCreditPool.quota_limit, ] stmt = update(TenantCreditPool).where(*where_conditions).values(**update_values) - result: CursorResult = session.execute(stmt) + result = cast(CursorResult, session.execute(stmt)) session.commit() if result.rowcount == 0: raise QuotaExceededError(