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