新增自定义日志模块,替换全局logging实现;优化日志输出格式与颜色支持

This commit is contained in:
yinpeng
2024-12-16 16:20:53 +08:00
parent 829c1b02ec
commit 56a069dbac
10 changed files with 131 additions and 24 deletions
+3 -3
View File
@@ -1,9 +1,9 @@
import asyncio
from itertools import cycle
import logging
from typing import Dict
from app.core.logger import get_key_manager_logger
logger = logging.getLogger(__name__)
logger = get_key_manager_logger()
class KeyManager:
@@ -42,7 +42,7 @@ class KeyManager:
current_key = await self.get_next_key()
if current_key == initial_key:
await self.reset_failure_counts()
# await self.reset_failure_counts() 取消重置
return current_key
async def handle_api_failure(self, api_key: str) -> str: