GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-05-26 10:49:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 将 Gemini API 连接超时时间统一调整为 300 秒

Browse Source
This commit is contained in:
yinpeng
2024-12-21 23:16:47 +08:00
parent 33a5f9d89f
commit 98ba46f779
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
app/services/chat_service.py
View File

@@ -260,8 +260,8 @@ class ChatService:
while retries < max_retries:
try:
timeout = httpx.Timeout(
60.0, read=60.0
) # 连接超时60秒，读取超时60秒
300.0, read=300.0
) # 连接超时300秒，读取超时300秒
async with httpx.AsyncClient(timeout=timeout) as async_client:
stream_url = f"https://generativelanguage.googleapis.com/v1beta/models/{gemini_model}:streamGenerateContent?alt=sse&key={current_api_key}"
async with async_client.stream(
@@ -344,7 +344,7 @@ class ChatService:
return generate()
else:
try:
timeout = httpx.Timeout(60.0, read=60.0) # 连接超时60秒，读取超时60秒
timeout = httpx.Timeout(300.0, read=300.0) # 连接超时300秒，读取超时300秒
async with httpx.AsyncClient(timeout=timeout) as client:
url = f"https://generativelanguage.googleapis.com/v1beta/models/{gemini_model}:generateContent?key={api_key}"
response = await client.post(url, json=payload)
@@ -375,7 +375,7 @@ class ChatService:
"""调用Gemini API生成内容"""
url = f"{self.base_url}/models/{model_name}:generateContent?key={api_key}"
timeout = httpx.Timeout(60.0, read=60.0) # 连接超时30秒，读取超时60秒
timeout = httpx.Timeout(300.0, read=300.0) # 连接超时300秒，读取超时300秒
async with httpx.AsyncClient(timeout=timeout) as client:
try:
response = await client.post(url, json=request.model_dump())
@@ -403,7 +403,7 @@ class ChatService:
while retries < MAX_RETRIES:
try:
url = f"{self.base_url}/models/{model_name}:streamGenerateContent?alt=sse&key={current_api_key}"
timeout = httpx.Timeout(60.0, read=60.0)
timeout = httpx.Timeout(300.0, read=300.0)
async with httpx.AsyncClient(timeout=timeout) as client:
async with client.stream(