From 75ccba6e52c09b7c9c74445bff2c0675bbe231ce Mon Sep 17 00:00:00 2001
From: Yansong Zhang <916125788@qq.com>
Date: Thu, 8 Jan 2026 11:01:23 +0800
Subject: [PATCH] add rowcount check

---
 api/services/credit_pool_service.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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