Ma/sora2api
1
0
Fork 0
mirror of https://github.com/TheSmallHanCat/sora2api.git synced 2026-02-14 18:14:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 错误计数统计

Browse Source
This commit is contained in:
TheSmallHanCat
2025-12-02 23:39:26 +08:00
parent 83f2ed9468
commit 0ca2a44885
3 changed files with 26 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/services/token_manager.py
View File

@@ -913,12 +913,12 @@ class TokenManager:
async def record_error(self, token_id: int):
"""Record token error"""
await self.db.increment_error_count(token_id)
# Check if should ban
stats = await self.db.get_token_stats(token_id)
admin_config = await self.db.get_admin_config()
if stats and stats.error_count >= admin_config.error_ban_threshold:
if stats and stats.consecutive_error_count >= admin_config.error_ban_threshold:
await self.db.update_token_status(token_id, False)
async def record_success(self, token_id: int, is_video: bool = False):