From b465fd1598b1a923346743c922bff12886a1d6e0 Mon Sep 17 00:00:00 2001
From: yinpeng <2291314224@qq.com>
Date: Tue, 4 Feb 2025 21:51:11 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E5=B0=86MAX=5FFAILURES=E9=85=8D?=
 =?UTF-8?q?=E7=BD=AE=E7=A7=BB=E8=87=B3settings=E5=B9=B6=E6=9B=B4=E6=96=B0K?=
 =?UTF-8?q?eyManager=E4=BB=A5=E4=BD=BF=E7=94=A8=E8=AF=A5=E9=85=8D=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/core/config.py          | 1 +
 app/services/key_manager.py | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/app/core/config.py b/app/core/config.py
index 790f841..ea989b4 100644
--- a/app/core/config.py
+++ b/app/core/config.py
@@ -11,6 +11,7 @@ class Settings(BaseSettings):
     SHOW_SEARCH_LINK: bool = True
     SHOW_THINKING_PROCESS: bool = True
     AUTH_TOKEN: str = ""
+    MAX_FAILURES: int = 3
 
     def __init__(self):
         super().__init__()
diff --git a/app/services/key_manager.py b/app/services/key_manager.py
index c62b539..0048591 100644
--- a/app/services/key_manager.py
+++ b/app/services/key_manager.py
@@ -2,6 +2,7 @@ import asyncio
 from itertools import cycle
 from typing import Dict
 from app.core.logger import get_key_manager_logger
+from app.core.config import settings
 
 logger = get_key_manager_logger()
 
@@ -13,7 +14,7 @@ class KeyManager:
         self.key_cycle_lock = asyncio.Lock()
         self.failure_count_lock = asyncio.Lock()
         self.key_failure_counts: Dict[str, int] = {key: 0 for key in api_keys}
-        self.MAX_FAILURES = 10
+        self.MAX_FAILURES = settings.MAX_FAILURES
 
     async def get_next_key(self) -> str:
         """获取下一个API key"""