GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-09-07 16:56:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 修复 OpenAI 和 Gemini API 调用重试逻辑及日志记录

Browse Source
This commit is contained in:
yinpeng
2024-12-21 02:27:59 +08:00
parent 80bcaf5cd4
commit 33a5f9d89f
9 changed files with 192 additions and 168 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/api/openai_routes.py
+5 -5
View File
@@ -48,9 +48,9 @@ async def chat_completion(
api_key = await key_manager.get_next_working_key()
logger.info(f"Using API key: {api_key}")
retries = 0
MAX_RETRIES = 3
max_retries = 3
while retries < MAX_RETRIES:
while retries < max_retries:
try:
response = await chat_service.create_chat_completion(
request=request,
@@ -64,13 +64,13 @@ async def chat_completion(
except Exception as e:
logger.warning(
f"API call failed with error: {str(e)}. Attempt {retries + 1} of {MAX_RETRIES}"
f"API call failed with error: {str(e)}. Attempt {retries + 1} of {max_retries}"
)
api_key = await key_manager.handle_api_failure(api_key)
logger.info(f"Switched to new API key: {api_key}")
retries += 1
if retries >= MAX_RETRIES:
logger.error(f"Max retries ({MAX_RETRIES}) reached. Raising error")
if retries >= max_retries:
logger.error(f"Max retries ({max_retries}) reached. Raising error")
raise