From 98ba46f7796a2036afa1ba40c338ea3f20b93301 Mon Sep 17 00:00:00 2001
From: yinpeng <2291314224@qq.com>
Date: Sat, 21 Dec 2024 23:16:47 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=B0=86=20Gemini=20API=20=E8=BF=9E?=
 =?UTF-8?q?=E6=8E=A5=E8=B6=85=E6=97=B6=E6=97=B6=E9=97=B4=E7=BB=9F=E4=B8=80?=
 =?UTF-8?q?=E8=B0=83=E6=95=B4=E4=B8=BA=20300=20=E7=A7=92?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/services/chat_service.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/app/services/chat_service.py b/app/services/chat_service.py
index 3c2cd42..14cf39d 100644
--- a/app/services/chat_service.py
+++ b/app/services/chat_service.py
@@ -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(