GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-09-06 16:16:37 +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
app/services/chat_service.py
+5 -5
View File
@@ -260,8 +260,8 @@ class ChatService:
while retries < max_retries: while retries < max_retries:
try: try:
timeout = httpx.Timeout( timeout = httpx.Timeout(
60.0, read=60.0 300.0, read=300.0
) # 连接超时60秒，读取超时60秒 ) # 连接超时300秒，读取超时300秒
async with httpx.AsyncClient(timeout=timeout) as async_client: 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}" stream_url = f"https://generativelanguage.googleapis.com/v1beta/models/{gemini_model}:streamGenerateContent?alt=sse&key={current_api_key}"
async with async_client.stream( async with async_client.stream(
@@ -344,7 +344,7 @@ class ChatService:
return generate() return generate()
else: else:
try: 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: async with httpx.AsyncClient(timeout=timeout) as client:
url = f"https://generativelanguage.googleapis.com/v1beta/models/{gemini_model}:generateContent?key={api_key}" url = f"https://generativelanguage.googleapis.com/v1beta/models/{gemini_model}:generateContent?key={api_key}"
response = await client.post(url, json=payload) response = await client.post(url, json=payload)
@@ -375,7 +375,7 @@ class ChatService:
"""调用Gemini API生成内容""" """调用Gemini API生成内容"""
url = f"{self.base_url}/models/{model_name}:generateContent?key={api_key}" 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: async with httpx.AsyncClient(timeout=timeout) as client:
try: try:
response = await client.post(url, json=request.model_dump()) response = await client.post(url, json=request.model_dump())
@@ -403,7 +403,7 @@ class ChatService:
while retries < MAX_RETRIES: while retries < MAX_RETRIES:
try: try:
url = f"{self.base_url}/models/{model_name}:streamGenerateContent?alt=sse&key={current_api_key}" 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 httpx.AsyncClient(timeout=timeout) as client:
async with client.stream( async with client.stream(