mirror of
https://github.com/langgenius/dify.git
synced 2026-01-14 06:07:33 +08:00
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
Block a user