From dbc8ffccbd3a00723c3e98efa030459425221a70 Mon Sep 17 00:00:00 2001 From: Yansong Zhang <916125788@qq.com> Date: Thu, 8 Jan 2026 11:07:09 +0800 Subject: [PATCH] add rowcount check --- api/services/credit_pool_service.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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(