fix: 在密钥验证成功时重置失败计数

This commit is contained in:
snaily
2025-07-19 10:49:09 +08:00
parent 9ec33ce320
commit 69399c291e
+5 -1
View File
@@ -339,7 +339,9 @@ async def verify_key(api_key: str, chat_service: GeminiChatService = Depends(get
) )
if response: if response:
return JSONResponse({"status": "valid"}) # 如果密钥验证成功,则重置其失败计数
await key_manager.reset_key_failure_count(api_key)
return JSONResponse({"status": "valid"})
except Exception as e: except Exception as e:
logger.error(f"Key verification failed: {str(e)}") logger.error(f"Key verification failed: {str(e)}")
@@ -382,6 +384,8 @@ async def verify_selected_keys(
api_key api_key
) )
successful_keys.append(api_key) successful_keys.append(api_key)
# 如果密钥验证成功,则重置其失败计数
await key_manager.reset_key_failure_count(api_key)
return api_key, "valid", None return api_key, "valid", None
except Exception as e: except Exception as e:
error_message = str(e) error_message = str(e)