mirror of https://github.com/langgenius/dify.git
add rowcount check
This commit is contained in:
parent
8a38489e53
commit
75ccba6e52
|
|
@ -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(
|
||||
|
|
|
|||
Loading…
Reference in New Issue