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

fix: 将 Gemini API 连接超时时间统一调整为 60 秒

Browse Source
This commit is contained in:
yinpeng
2024-12-18 20:10:03 +08:00
parent 6ef21b9a8a
commit 914922624e
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/services/chat_service.py
+4 -4
View File
@@ -195,7 +195,7 @@ class ChatService:
while retries < MAX_RETRIES: while retries < MAX_RETRIES:
try: try:
timeout = httpx.Timeout(30.0, read=60.0) # 连接超时30秒，读取超时60秒 timeout = httpx.Timeout(60.0, read=60.0) # 连接超时60秒，读取超时60秒
async with httpx.AsyncClient(timeout=timeout) as client: async with httpx.AsyncClient(timeout=timeout) as 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 client.stream("POST", stream_url, json=payload) as response: async with client.stream("POST", stream_url, json=payload) as response:
@@ -254,7 +254,7 @@ class ChatService:
return generate() return generate()
else: else:
try: try:
timeout = httpx.Timeout(60.0, read=60.0) # 连接超时30秒，读取超时60秒 timeout = httpx.Timeout(60.0, read=60.0) # 连接超时60秒，读取超时60秒
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)
@@ -290,7 +290,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(30.0, read=60.0) # 连接超时30秒，读取超时60秒 timeout = httpx.Timeout(60.0, read=60.0) # 连接超时30秒，读取超时60秒
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())
@@ -314,7 +314,7 @@ class ChatService:
"""调用Gemini API流式生成内容""" """调用Gemini API流式生成内容"""
url = f"{self.base_url}/models/{model_name}:streamGenerateContent?alt=sse&key={api_key}" url = f"{self.base_url}/models/{model_name}:streamGenerateContent?alt=sse&key={api_key}"
timeout = httpx.Timeout(30.0, read=60.0) # 连接超时30秒，读取超时60秒 timeout = httpx.Timeout(60.0, read=60.0) # 连接超时60秒，读取超时60秒
async with httpx.AsyncClient(timeout=timeout) as client: async with httpx.AsyncClient(timeout=timeout) as client:
try: try:
async with client.stream('POST', url, json=request.model_dump()) as response: async with client.stream('POST', url, json=request.model_dump()) as response: