add rowcount check

This commit is contained in:
Yansong Zhang 2026-01-08 11:07:09 +08:00
parent c3d0c80ca1
commit dbc8ffccbd
1 changed files with 3 additions and 1 deletions

View File

@ -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(